Fix InMemory shard transaction chaining.

[mdsal.git] / dom / mdsal-dom-inmemory-datastore / src / test / java / org / opendaylight / mdsal / dom / store / inmemory / InmemoryDOMDataTreeShardWriteTransactionTest.java

diff --git a/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InmemoryDOMDataTreeShardWriteTransactionTest.java b/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InmemoryDOMDataTreeShardWriteTransactionTest.java
index 7d657e62aa31863bc0cd82255867112c7eacd76e..08eb6adfb2125f73edfb1aa50b1a2b972d003990 100644 (file)
--- a/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InmemoryDOMDataTreeShardWriteTransactionTest.java
+++ b/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/InmemoryDOMDataTreeShardWriteTransactionTest.java
@@ -58,6 +58,7 @@ public class InmemoryDOMDataTreeShardWriteTransactionTest {
             new ChildShardContext(DOM_DATA_TREE_IDENTIFIER, READABLE_WRITEABLE_DOM_DATA_TREE_SHARD);
     private static final Map<DOMDataTreeIdentifier, ChildShardContext> CHILD_SHARDS =
             ImmutableMap.of(DOM_DATA_TREE_IDENTIFIER, CHILD_SHARD_CONTEXT);
+    private InMemoryDOMDataTreeShardProducer mockProducer;
 
     @Before
     public void setUp() throws Exception {
@@ -81,9 +82,13 @@ public class InmemoryDOMDataTreeShardWriteTransactionTest {
         final InMemoryDOMDataTreeShardChangePublisher inMemoryDOMDataTreeShardChangePublisher =
                 new InMemoryDOMDataTreeShardChangePublisher(MoreExecutors.newDirectExecutorService(), 1, DATA_TREE,
                         YANG_INSTANCE_IDENTIFIER, CHILD_SHARDS);
+        mockProducer = mock(InMemoryDOMDataTreeShardProducer.class);
+        doNothing().when(mockProducer).transactionReady(any(), any());
+        doNothing().when(mockProducer).onTransactionCommited(any());
+        doNothing().when(mockProducer).transactionAborted(any());
 
         inmemoryDOMDataTreeShardWriteTransaction =
-                new InmemoryDOMDataTreeShardWriteTransaction(shardDataModification, DATA_TREE,
+                new InmemoryDOMDataTreeShardWriteTransaction(mockProducer, shardDataModification, DATA_TREE,
                         inMemoryDOMDataTreeShardChangePublisher,
                         MoreExecutors.listeningDecorator(Executors.newSingleThreadExecutor()));
     }