1 files changed, 3 insertions, 13 deletions
diff --git a/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/initializers/LispInitializerTest.java b/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/initializers/LispInitializerTest.java
index 087f2cdee..4da6593e0 100644
--- a/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/initializers/LispInitializerTest.java
+++ b/lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/initializers/LispInitializerTest.java
@@ -9,10 +9,7 @@ import org.opendaylight.controller.md.sal.binding.api.DataBroker;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.Lisp;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.LispState;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.LispStateBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.eid.table.grouping.EidTableBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.locator.sets.grouping.LocatorSetsBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.map.resolvers.grouping.MapResolversBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.pitr.cfg.grouping.PitrCfgBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev160520.lisp.feature.data.grouping.LispFeatureDataBuilder;
 
 
 public class LispInitializerTest {
@@ -22,21 +19,14 @@ public class LispInitializerTest {
         final LispInitializer initializer = new LispInitializer(mock(DataBroker.class));
         final LispState state = new LispStateBuilder()
                 .setEnable(true)
-                .setEidTable(new EidTableBuilder().build())
-                .setLocatorSets(new LocatorSetsBuilder().build())
-                .setMapResolvers(new MapResolversBuilder().build())
-                .setPitrCfg(new PitrCfgBuilder().build())
+                .setLispFeatureData(new LispFeatureDataBuilder().build())
                 .build();
 
         final Lisp operational = initializer.convert(state);
 
         assertNotNull(operational);
-        assertEquals(operational.getPitrCfg(), state.getPitrCfg());
         assertEquals(operational.isEnable(), state.isEnable());
-        assertEquals(operational.getEidTable(), state.getEidTable());
-        assertEquals(operational.getLocatorSets(), state.getLocatorSets());
-        assertEquals(operational.getMapResolvers(), state.getMapResolvers());
-
+        assertEquals(operational.getLispFeatureData(), state.getLispFeatureData());
     }
 
 }
 \ No newline at end of file