1 files changed, 107 insertions, 0 deletions
diff --git a/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/read/VrfSubtableCustomizerTest.java b/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/read/VrfSubtableCustomizerTest.java
new file mode 100644
index 000000000..f18f349d8
--- /dev/null
+++ b/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/read/VrfSubtableCustomizerTest.java
@@ -0,0 +1,107 @@
+package io.fd.honeycomb.lisp.translate.read;
+
+
+import static io.fd.honeycomb.lisp.translate.read.dump.executor.params.SubtableDumpParams.MapLevel.L3;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import io.fd.honeycomb.lisp.translate.read.trait.SubtableReaderTestCase;
+import io.fd.honeycomb.translate.read.ReadFailedException;
+import io.fd.honeycomb.translate.spi.read.ReaderCustomizer;
+import io.fd.honeycomb.translate.util.read.cache.exceptions.execution.i.DumpCallFailedException;
+import io.fd.vpp.jvpp.VppCallbackException;
+import org.junit.Before;
+import org.junit.Test;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.EidTable;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.eid.table.VniTable;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.eid.table.VniTableBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.eid.table.VniTableKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.eid.table.vni.table.VrfSubtable;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.eid.table.vni.table.VrfSubtableBuilder;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+public class VrfSubtableCustomizerTest extends SubtableReaderTestCase<VrfSubtable, VrfSubtableBuilder> {
+
+    private InstanceIdentifier<VrfSubtable> validId;
+
+    public VrfSubtableCustomizerTest() {
+        super(VrfSubtable.class, VrfSubtableBuilder.class);
+    }
+
+    @Before
+    public void init() {
+        validId = InstanceIdentifier.create(EidTable.class).child(VniTable.class, new VniTableKey(expectedVni))
+                .child(VrfSubtable.class);
+    }
+
+    @Test
+    public void testReadCurrentSuccessfull() throws ReadFailedException {
+        doReturnValidNonEmptyDataOnDump();
+        VrfSubtableBuilder builder = new VrfSubtableBuilder();
+        customizer.readCurrentAttributes(validId, builder, ctx);
+
+        verifyLispEidTableMapDumpCalled(L3);
+
+        final VrfSubtable subtable = builder.build();
+        assertNotNull(subtable);
+        assertEquals(expectedTableId, subtable.getTableId().longValue());
+    }
+
+    @Test
+    public void testReadCurrentEmptyDump() throws ReadFailedException {
+        doReturnEmptyDataOnDump();
+        VrfSubtableBuilder builder = new VrfSubtableBuilder();
+        customizer.readCurrentAttributes(validId, builder, ctx);
+
+        verifyLispEidTableMapDumpCalled(L3);
+
+        final VrfSubtable subtable = builder.build();
+        assertNotNull(subtable);
+        assertNull(subtable.getTableId());
+    }
+
+    @Test
+    public void testReadCurrentFailed() {
+        doThrowOnDump();
+        VrfSubtableBuilder builder = new VrfSubtableBuilder();
+        try {
+            customizer.readCurrentAttributes(validId, builder, ctx);
+        } catch (ReadFailedException e) {
+            assertTrue(e.getCause() instanceof DumpCallFailedException);
+            assertTrue(e.getCause().getCause() instanceof VppCallbackException);
+            assertTrue(builder.getTableId() == null);
+            verifyLispEidTableMapDumpNotCalled();
+
+            return;
+        }
+
+        fail("Test should throw ReadFailedException");
+    }
+
+    @Override
+    protected ReaderCustomizer<VrfSubtable, VrfSubtableBuilder> initCustomizer() {
+        return new VrfSubtableCustomizer(api);
+    }
+
+    @Test
+    public void testGetBuilder() {
+        final VrfSubtableBuilder builder = customizer.getBuilder(validId);
+
+        assertNotNull(builder);
+        assertNull(builder.getLocalMappings());
+        assertNull(builder.getRemoteMappings());
+        assertNull(builder.getTableId());
+    }
+
+    @Test
+    public void testMerge() {
+        VniTableBuilder parentBuilder = new VniTableBuilder();
+        VrfSubtable subtable = new VrfSubtableBuilder().build();
+
+        customizer.merge(parentBuilder, subtable);
+        assertEquals(subtable, parentBuilder.getVrfSubtable());
+    }
+}