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%2FShardCommitCoordinationTaskTest.java;h=13f76b425ede750abf544f168f7f65f7b030fb5a;hb=7d8e59521efbfadd55de34f645db8a8692de6564;hp=24a2047b53eb25dd8071f5691a9c11233d2c96c8;hpb=e662b9c1d106916c3f5f537f4230738d69a6fcb4;p=mdsal.git

diff --git a/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/ShardCommitCoordinationTaskTest.java b/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/ShardCommitCoordinationTaskTest.java
index 24a2047b53..13f76b425e 100644
--- a/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/ShardCommitCoordinationTaskTest.java
+++ b/dom/mdsal-dom-inmemory-datastore/src/test/java/org/opendaylight/mdsal/dom/store/inmemory/ShardCommitCoordinationTaskTest.java
@@ -7,8 +7,11 @@
  */
 package org.opendaylight.mdsal.dom.store.inmemory;
 
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doNothing;
 import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.COHORTS;
 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.DOM_DATA_TREE_IDENTIFIER;
@@ -17,11 +20,20 @@ import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.LISTENABLE_FUT
 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.resetMocks;
 
 import org.junit.After;
+import org.junit.Before;
 import org.junit.Test;
 import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
 
 public class ShardCommitCoordinationTaskTest {
 
+    final InmemoryDOMDataTreeShardWriteTransaction mockTx = mock(InmemoryDOMDataTreeShardWriteTransaction.class);
+
+    @Before
+    public void setUp() throws Exception {
+        doReturn("MockedTx").when(mockTx).toString();
+        doNothing().when(mockTx).transactionCommited(any());
+    }
+
     @Test
     public void basicTest() throws Exception {
         doReturn(Void.TYPE).when(LISTENABLE_FUTURE).get();
@@ -30,7 +42,7 @@ public class ShardCommitCoordinationTaskTest {
         COHORTS.add(DOM_STORE_THREE_PHASE_COMMIT_COHORT);
 
         ShardCommitCoordinationTask shardCommitCoordinationTask =
-                new ShardCommitCoordinationTask(DOM_DATA_TREE_IDENTIFIER, COHORTS);
+                new ShardCommitCoordinationTask(DOM_DATA_TREE_IDENTIFIER, COHORTS, mockTx);
 
         shardCommitCoordinationTask.call();
         verify(DOM_STORE_THREE_PHASE_COMMIT_COHORT).commit();
@@ -43,7 +55,7 @@ public class ShardCommitCoordinationTaskTest {
 
         COHORTS.add(DOM_STORE_THREE_PHASE_COMMIT_COHORT);
         ShardCommitCoordinationTask shardCommitCoordinationTask =
-                new ShardCommitCoordinationTask(DOM_DATA_TREE_IDENTIFIER, COHORTS);
+                new ShardCommitCoordinationTask(DOM_DATA_TREE_IDENTIFIER, COHORTS, mockTx);
         shardCommitCoordinationTask.call();
     }