HONEYCOMB-9: Remove references to VPP from translation layer

Change-Id: I281db366a112edc08203e8cb392a212708d4552a
Signed-off-by: Maros Marsalek <mmarsale@cisco.com>

2 files changed, 328 insertions, 0 deletions

diff --git a/v3po/translate-utils/src/test/java/io/fd/honeycomb/v3po/translate/impl/write/util/DelegatingWriterRegistryTest.java b/v3po/translate-utils/src/test/java/io/fd/honeycomb/v3po/translate/impl/write/util/DelegatingWriterRegistryTest.java
new file mode 100644
index 000000000..e201890b5
--- /dev/null
+++ b/v3po/translate-utils/src/test/java/io/fd/honeycomb/v3po/translate/impl/write/util/DelegatingWriterRegistryTest.java
@@ -0,0 +1,189 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.honeycomb.v3po.translate.impl.write.util;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+
+import io.fd.honeycomb.v3po.translate.TranslationException;
+import io.fd.honeycomb.v3po.translate.util.write.DelegatingWriterRegistry;
+import io.fd.honeycomb.v3po.translate.write.WriteContext;
+import io.fd.honeycomb.v3po.translate.write.Writer;
+import io.fd.honeycomb.v3po.translate.write.WriterRegistry;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mockito;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.Interfaces;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.Vpp;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.VppState;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+public class DelegatingWriterRegistryTest {
+
+    private final InstanceIdentifier<Vpp> vppId;
+    private final InstanceIdentifier<VppState> vppStateId;
+    private final InstanceIdentifier<Interfaces> interfaceId;
+
+    private WriteContext ctx;
+    private Writer<Vpp> writer;
+    private Writer<VppState> vppStateWriter;
+    private Writer<Interfaces> interfacesWriter;
+
+    private DelegatingWriterRegistry registry;
+
+    public DelegatingWriterRegistryTest() {
+        vppId = InstanceIdentifier.create(Vpp.class);
+        vppStateId = InstanceIdentifier.create(VppState.class);
+        interfaceId = InstanceIdentifier.create(Interfaces.class);
+    }
+
+    @SuppressWarnings("unchecked")
+    private <D extends DataObject> Writer<D> mockWriter(Class<D> clazz) {
+        final Writer<D> mock = (Writer<D>) Mockito.mock(Writer.class);
+        doReturn(InstanceIdentifier.create(clazz)).when(mock).getManagedDataObjectType();
+        return mock;
+    }
+
+    private DataObject mockDataObject(final String name, final Class<? extends DataObject> classToMock) {
+        final DataObject dataBefore = mock(classToMock, name);
+        doReturn(classToMock).when(dataBefore).getImplementedInterface();
+        return dataBefore;
+    }
+
+    @SuppressWarnings("unchecked")
+    private static Map<InstanceIdentifier<?>, DataObject> asMap(DataObject... objects) {
+        final Map<InstanceIdentifier<?>, DataObject> map = new HashMap<>();
+        for (DataObject object : objects) {
+            final Class<? extends DataObject> implementedInterface =
+                    (Class<? extends DataObject>) object.getImplementedInterface();
+            final InstanceIdentifier<?> id = InstanceIdentifier.create(implementedInterface);
+            map.put(id, object);
+        }
+        return map;
+    }
+
+    @Before
+    public void setUp() {
+        ctx = mock(WriteContext.class);
+        writer = mockWriter(Vpp.class);
+        vppStateWriter = mockWriter(VppState.class);
+        interfacesWriter = mockWriter(Interfaces.class);
+
+        final List<Writer<? extends DataObject>> writers = new ArrayList<>();
+        writers.add(writer);
+        writers.add(vppStateWriter);
+        writers.add(interfacesWriter);
+
+        registry = new DelegatingWriterRegistry(writers);
+    }
+
+    @Test(expected = UnsupportedOperationException.class)
+    public void testGetManagedDataObjectType() {
+        registry.getManagedDataObjectType();
+    }
+
+    @Test
+    public void testBulkUpdateRevert() throws Exception {
+        // Prepare data changes:
+        final DataObject dataBefore1 = mockDataObject("Vpp before", Vpp.class);
+        final DataObject dataAfter1 = mockDataObject("Vpp after", Vpp.class);
+
+        final DataObject dataBefore2 = mockDataObject("VppState before", VppState.class);
+        final DataObject dataAfter2 = mockDataObject("VppState after", VppState.class);
+
+        // Fail on update
+        Mockito.doThrow(new TranslationException("vpp failed")).when(vppStateWriter)
+                .update(vppStateId, dataBefore2, dataAfter2, ctx);
+
+        // Run the test
+        try {
+            registry.update(asMap(dataBefore1, dataBefore2), asMap(dataAfter1, dataAfter2), ctx);
+        } catch (WriterRegistry.BulkUpdateException e) {
+            // Check second update failed
+            assertEquals(vppStateId, e.getFailedId());
+            verify(writer).update(vppId, dataBefore1, dataAfter1, ctx);
+            verify(vppStateWriter).update(vppStateId, dataBefore2, dataAfter2, ctx);
+
+            // Try to revert changes
+            e.revertChanges();
+
+            // Check revert was successful
+            verify(writer).update(vppId, dataAfter1, dataBefore1, ctx);
+            verify(vppStateWriter, never()).update(vppStateId, dataAfter2, dataBefore2, ctx);
+
+            return;
+        }
+        fail("BulkUpdateException expected");
+    }
+
+    @Test
+    public void testBulkUpdateRevertFail() throws Exception {
+        // Prepare data changes:
+        final DataObject dataBefore1 = mockDataObject("Vpp before", Vpp.class);
+        final DataObject dataAfter1 = mockDataObject("Vpp after", Vpp.class);
+
+        final DataObject dataBefore2 = mockDataObject("VppState before", VppState.class);
+        final DataObject dataAfter2 = mockDataObject("VppState after", VppState.class);
+
+        final DataObject dataBefore3 = mockDataObject("Interfaces before", Interfaces.class);
+        final DataObject dataAfter3 = mockDataObject("Interfaces after", Interfaces.class);
+
+        // Fail on the third update
+        doThrow(new TranslationException("vpp failed")).when(interfacesWriter)
+                .update(interfaceId, dataBefore3, dataAfter3, ctx);
+
+        // Fail on the second revert
+        doThrow(new TranslationException("vpp failed again")).when(writer)
+                .update(vppId, dataAfter1, dataBefore1, ctx);
+
+        // Run the test
+        try {
+            registry.update(asMap(dataBefore1, dataBefore2, dataBefore3), asMap(dataAfter1, dataAfter2, dataAfter3), ctx);
+        } catch (WriterRegistry.BulkUpdateException e) {
+            // Check third update failed
+            assertEquals(interfaceId, e.getFailedId());
+            verify(writer).update(vppId, dataBefore1, dataAfter1, ctx);
+            verify(vppStateWriter).update(vppStateId, dataBefore2, dataAfter2, ctx);
+            verify(interfacesWriter).update(interfaceId, dataBefore3, dataAfter3, ctx);
+
+            // Try to revert changes
+            try {
+                e.revertChanges();
+            } catch (WriterRegistry.Reverter.RevertFailedException e2) {
+                // Check second revert failed
+                assertEquals(Collections.singletonList(vppId), e2.getNotRevertedChanges());
+                verify(writer).update(vppId, dataAfter1, dataBefore1, ctx);
+                verify(vppStateWriter).update(vppStateId, dataAfter2, dataBefore2, ctx);
+                verify(interfacesWriter, never()).update(interfaceId, dataAfter3, dataBefore3, ctx);
+                return;
+            }
+            fail("WriterRegistry.Revert.RevertFailedException expected");
+        }
+        fail("BulkUpdateException expected");
+    }
+}
\ No newline at end of file
diff --git a/v3po/translate-utils/src/test/java/io/fd/honeycomb/v3po/translate/impl/write/util/TransactionWriteContextTest.java b/v3po/translate-utils/src/test/java/io/fd/honeycomb/v3po/translate/impl/write/util/TransactionWriteContextTest.java
new file mode 100644
index 000000000..5a8740b2d
--- /dev/null
+++ b/v3po/translate-utils/src/test/java/io/fd/honeycomb/v3po/translate/impl/write/util/TransactionWriteContextTest.java
@@ -0,0 +1,139 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.honeycomb.v3po.translate.impl.write.util;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static org.mockito.MockitoAnnotations.initMocks;
+
+import com.google.common.base.Optional;
+import com.google.common.util.concurrent.CheckedFuture;
+import io.fd.honeycomb.v3po.translate.Context;
+import io.fd.honeycomb.v3po.translate.util.write.TransactionWriteContext;
+import java.util.Map;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
+import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.Vpp;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.VppState;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.vpp.BridgeDomains;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.vpp.bridge.domains.BridgeDomain;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+
+public class TransactionWriteContextTest {
+
+    @Mock
+    private BindingNormalizedNodeSerializer serializer;
+    @Mock
+    private DOMDataReadOnlyTransaction beforeTx;
+    @Mock
+    private DOMDataReadOnlyTransaction afterTx;
+    @Mock
+    private CheckedFuture<Optional<NormalizedNode<?, ?>>, ReadFailedException> future;
+    @Mock
+    private Optional<org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode<?, ?>> optional;
+    @Mock
+    private Map.Entry entry;
+
+    private TransactionWriteContext transactionWriteContext;
+
+    @Before
+    public void setUp() {
+        initMocks(this);
+        transactionWriteContext = new TransactionWriteContext(serializer, beforeTx, afterTx);
+    }
+
+    @Test
+    public void testReadBeforeNoData() throws Exception {
+        when(beforeTx.read(eq(LogicalDatastoreType.CONFIGURATION), any(YangInstanceIdentifier.class))).thenReturn(future);
+        when(future.checkedGet()).thenReturn(optional);
+        when(optional.isPresent()).thenReturn(false);
+
+        final InstanceIdentifier<BridgeDomain> instanceId =
+                InstanceIdentifier.create(Vpp.class).child(BridgeDomains.class).child(BridgeDomain.class);
+
+        final Optional<DataObject> dataObjects = transactionWriteContext.readBefore(instanceId);
+        assertNotNull(dataObjects);
+        assertFalse(dataObjects.isPresent());
+
+        verify(serializer).toYangInstanceIdentifier(instanceId);
+        verify(serializer, never()).fromNormalizedNode(any(YangInstanceIdentifier.class), any(NormalizedNode.class));
+    }
+
+
+    @Test
+    public void testReadBefore() throws Exception {
+        when(beforeTx.read(eq(LogicalDatastoreType.CONFIGURATION), any(YangInstanceIdentifier.class))).thenReturn(future);
+        when(future.checkedGet()).thenReturn(optional);
+        when(optional.isPresent()).thenReturn(true);
+
+        final InstanceIdentifier<BridgeDomain> instanceId =
+                InstanceIdentifier.create(Vpp.class).child(BridgeDomains.class).child(BridgeDomain.class);
+        final YangInstanceIdentifier yangId = YangInstanceIdentifier.builder().node(VppState.QNAME).node(
+                BridgeDomains.QNAME).node(BridgeDomain.QNAME).build();
+        when(serializer.toYangInstanceIdentifier(any(InstanceIdentifier.class))).thenReturn(yangId);
+        when(serializer.fromNormalizedNode(eq(yangId), any(NormalizedNode.class))).thenReturn(entry);
+        when(entry.getValue()).thenReturn(mock(DataObject.class));
+
+        final Optional<DataObject> dataObjects = transactionWriteContext.readBefore(instanceId);
+        assertNotNull(dataObjects);
+        assertTrue(dataObjects.isPresent());
+
+        verify(serializer).toYangInstanceIdentifier(instanceId);
+        verify(serializer).fromNormalizedNode(eq(yangId), any(NormalizedNode.class));
+    }
+
+    @Test(expected = IllegalStateException.class)
+    public void testReadBeforeFailed() throws Exception {
+        when(beforeTx.read(eq(LogicalDatastoreType.CONFIGURATION), any(YangInstanceIdentifier.class))).thenReturn(future);
+        when(future.checkedGet()).thenThrow(ReadFailedException.class);
+        transactionWriteContext.readBefore(mock(InstanceIdentifier.class));
+    }
+
+    @Test(expected = IllegalStateException.class)
+    public void testReadAfterFailed() throws Exception {
+        when(afterTx.read(eq(LogicalDatastoreType.CONFIGURATION), any(YangInstanceIdentifier.class))).thenReturn(future);
+        when(future.checkedGet()).thenThrow(ReadFailedException.class);
+        transactionWriteContext.readAfter(mock(InstanceIdentifier.class));
+    }
+
+    @Test
+    public void testGetContext() throws Exception {
+        assertNotNull(transactionWriteContext.getContext());
+    }
+
+    @Test
+    public void testClose() throws Exception {
+        final Context context = transactionWriteContext.getContext();
+        transactionWriteContext.close();
+        // TODO verify context was closed
+    }
+}
\ No newline at end of file