Bug 4149: Implement per-shard DatastoreContext settings

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / DatastoreContextConfigAdminOverlayTest.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DatastoreContextConfigAdminOverlayTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DatastoreContextConfigAdminOverlayTest.java
index ecfcdefcb28d2eeecc24e934180372178ba7678f..700b96a87e35040c9821996d595f7a64b4c23623 100644 (file)
--- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DatastoreContextConfigAdminOverlayTest.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DatastoreContextConfigAdminOverlayTest.java
@@ -96,6 +96,8 @@ public class DatastoreContextConfigAdminOverlayTest {
 
         DatastoreContext context = DatastoreContext.newBuilder().build();
         doReturn(context).when(mockIntrospector).getContext();
+        DatastoreContextFactory contextFactory = new DatastoreContextFactory(mockIntrospector);
+        doReturn(contextFactory).when(mockIntrospector).newContextFactory();
 
         DatastoreContextConfigAdminOverlay.Listener mockListener =
                 mock(DatastoreContextConfigAdminOverlay.Listener.class);
@@ -122,7 +124,7 @@ public class DatastoreContextConfigAdminOverlayTest {
 
         verify(mockIntrospector).update(properties);
 
-        verify(mockListener).onDatastoreContextUpdated(context);
+        verify(mockListener).onDatastoreContextUpdated(contextFactory);
 
         verify(mockBundleContext, times(2)).ungetService(mockConfigAdminServiceRef);