Update module tests after introducing YangDAG

Change-Id: I3ce66e2e1fbed7b9a333382c664b5dd8ed501df0
Signed-off-by: Marek Gradzki <mgradzki@cisco.com>

2 files changed, 4 insertions, 2 deletions

diff --git a/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/VppManagementModuleTest.java b/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/VppManagementModuleTest.java
index eb2c3a933..8e06e0bc3 100644
--- a/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/VppManagementModuleTest.java
+++ b/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/VppManagementModuleTest.java
@@ -27,6 +27,7 @@ import com.google.inject.testing.fieldbinder.Bind;
 import com.google.inject.testing.fieldbinder.BoundFieldModule;
 import io.fd.honeycomb.translate.impl.read.registry.CompositeReaderRegistryBuilder;
 import io.fd.honeycomb.translate.read.ReaderFactory;
+import io.fd.honeycomb.translate.util.YangDAG;
 import io.fd.vpp.jvpp.core.future.FutureJVppCore;
 import java.util.HashSet;
 import java.util.Set;
@@ -57,7 +58,7 @@ public class VppManagementModuleTest {
         assertFalse(readerFactories.isEmpty());
 
         // Test registration process (all dependencies present, topological order of readers does exist, etc.)
-        final CompositeReaderRegistryBuilder registryBuilder = new CompositeReaderRegistryBuilder();
+        final CompositeReaderRegistryBuilder registryBuilder = new CompositeReaderRegistryBuilder(new YangDAG());
         readerFactories.stream().forEach(factory -> factory.init(registryBuilder));
         assertNotNull(registryBuilder.build());
     }
diff --git a/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/state/VppStateTest.java b/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/state/VppStateTest.java
index 81db88511..7e03d8c31 100644
--- a/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/state/VppStateTest.java
+++ b/vpp-management/impl/src/test/java/io/fd/hc2vpp/management/state/VppStateTest.java
@@ -32,6 +32,7 @@ import io.fd.honeycomb.translate.impl.read.GenericReader;
 import io.fd.honeycomb.translate.impl.read.registry.CompositeReaderRegistryBuilder;
 import io.fd.honeycomb.translate.read.ReadContext;
 import io.fd.honeycomb.translate.read.registry.ReaderRegistry;
+import io.fd.honeycomb.translate.util.YangDAG;
 import io.fd.vpp.jvpp.core.dto.ShowVersion;
 import io.fd.vpp.jvpp.core.dto.ShowVersionReply;
 import io.fd.vpp.jvpp.core.future.FutureJVppCore;
@@ -64,7 +65,7 @@ public class VppStateTest implements FutureProducer {
      * Create root VppState reader with all its children wired.
      */
     private static ReaderRegistry getVppStateReader(@Nonnull final FutureJVppCore jVpp) {
-        final CompositeReaderRegistryBuilder registry = new CompositeReaderRegistryBuilder();
+        final CompositeReaderRegistryBuilder registry = new CompositeReaderRegistryBuilder(new YangDAG());
 
         // VppState(Structural)
         final InstanceIdentifier<VppState> vppStateId = InstanceIdentifier.create(VppState.class);