X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=dom%2Fmdsal-dom-inmemory-datastore%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fmdsal%2Fdom%2Fstore%2Finmemory%2FInMemoryDOMDataTreeShardProducerTest.java;h=2286082a9fa34d07e30fdc1f7e9701f038c5b5a6;hb=f17c5ab39d43de758c3f3c7bd642d7fa436a6983;hp=0115866eb4f422033d5115ac5550c39ad9695dbd;hpb=7723a349513ae47974fa014586e887cc731f69ce;p=mdsal.git

diff --git a/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InMemoryDOMDataTreeShardProducerTest.java b/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InMemoryDOMDataTreeShardProducerTest.java
index 0115866eb4..2286082a9f 100644
--- a/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InMemoryDOMDataTreeShardProducerTest.java
+++ b/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InMemoryDOMDataTreeShardProducerTest.java
@@ -9,16 +9,15 @@ package org.opendaylight.mdsal.dom.store.inmemory;
 
 import static org.junit.Assert.assertNotNull;
 import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.anyCollectionOf;
 import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.DOM_DATA_TREE_IDENTIFIER;
 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.resetMocks;
 
+import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
 import org.junit.Test;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
 import org.opendaylight.yangtools.yang.data.api.schema.tree.CursorAwareDataTreeSnapshot;
 
 public class InMemoryDOMDataTreeShardProducerTest {
@@ -33,18 +32,19 @@ public class InMemoryDOMDataTreeShardProducerTest {
 
         doReturn(inmemoryDOMDataTreeShardWriteTransaction).when(inMemoryDOMDataTreeShard)
                 .createTransaction(any(String.class), any(InMemoryDOMDataTreeShardProducer.class),
-                        anyCollectionOf((DOMDataTreeIdentifier.class)), any(CursorAwareDataTreeSnapshot.class));
+                        any(CursorAwareDataTreeSnapshot.class));
 
         final InMemoryDOMDataTreeShardProducer inMemoryDOMDataTreeShardProducer =
                 new InMemoryDOMDataTreeShardProducer(inMemoryDOMDataTreeShard,
-                        ImmutableSet.of(DOM_DATA_TREE_IDENTIFIER));
+                        ImmutableSet.of(DOM_DATA_TREE_IDENTIFIER),
+                        new ShardDataModificationFactory(DOM_DATA_TREE_IDENTIFIER, ImmutableMap.of(),
+                            ImmutableMap.of()));
 
         assertNotNull(inMemoryDOMDataTreeShardProducer.createTransaction());
         verify(inMemoryDOMDataTreeShard).createTransaction(
                 any(String.class),
                 any(InMemoryDOMDataTreeShardProducer.class),
-                anyCollectionOf(DOMDataTreeIdentifier.class),
                 any(CursorAwareDataTreeSnapshot.class));
         resetMocks();
     }
-}
\ No newline at end of file
+}