Fix infinite loop on cancel transaction

[mdsal.git] / dom / mdsal-dom-broker / src / test / java / org / opendaylight / mdsal / dom / broker / DOMForwardedWriteTransactionTest.java

diff --git a/dom/mdsal-dom-broker/src/test/java/org/opendaylight/mdsal/dom/broker/DOMForwardedWriteTransactionTest.java b/dom/mdsal-dom-broker/src/test/java/org/opendaylight/mdsal/dom/broker/DOMForwardedWriteTransactionTest.java
new file mode 100644 (file)
index 0000000..dc37003
--- /dev/null
+++ b/dom/mdsal-dom-broker/src/test/java/org/opendaylight/mdsal/dom/broker/DOMForwardedWriteTransactionTest.java
@@ -0,0 +1,76 @@
+/*
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.mdsal.dom.broker;
+
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.doThrow;
+
+import com.google.common.util.concurrent.CheckedFuture;
+import java.util.Collections;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
+import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
+import org.opendaylight.mdsal.dom.spi.store.DOMStoreWriteTransaction;
+
+public class DOMForwardedWriteTransactionTest {
+
+    @Mock
+    private AbstractDOMForwardedTransactionFactory abstractDOMForwardedTransactionFactory;
+
+    @Mock
+    private DOMStoreWriteTransaction domStoreWriteTransaction;
+
+    @Before
+    public void setup() {
+        MockitoAnnotations.initMocks(this);
+    }
+
+    @Test
+    public void readyRuntimeExceptionAndCancel() {
+        RuntimeException thrown = new RuntimeException();
+        doThrow(thrown).when(domStoreWriteTransaction).ready();
+        DOMForwardedWriteTransaction<DOMStoreWriteTransaction> domForwardedWriteTransaction =
+                new DOMForwardedWriteTransaction<>(
+                        new Object(),
+                        Collections.singletonMap(LogicalDatastoreType.OPERATIONAL, domStoreWriteTransaction),
+                        abstractDOMForwardedTransactionFactory);
+        CheckedFuture<Void, TransactionCommitFailedException> submitFuture = domForwardedWriteTransaction.submit();
+        try {
+            submitFuture.checkedGet();
+            Assert.fail("TransactionCommitFailedException expected");
+        } catch (TransactionCommitFailedException e) {
+            assertTrue(e.getCause() == thrown);
+            domForwardedWriteTransaction.cancel();
+        }
+    }
+
+    @Test
+    public void submitRuntimeExceptionAndCancel() {
+        RuntimeException thrown = new RuntimeException();
+        doReturn(null).when(domStoreWriteTransaction).ready();
+        doThrow(thrown).when(abstractDOMForwardedTransactionFactory).submit(any(), any());
+        DOMForwardedWriteTransaction<DOMStoreWriteTransaction> domForwardedWriteTransaction =
+                new DOMForwardedWriteTransaction<>(
+                        new Object(),
+                        Collections.singletonMap(LogicalDatastoreType.OPERATIONAL, domStoreWriteTransaction),
+                        abstractDOMForwardedTransactionFactory);
+        CheckedFuture<Void, TransactionCommitFailedException> submitFuture = domForwardedWriteTransaction.submit();
+        try {
+            submitFuture.checkedGet();
+            Assert.fail("TransactionCommitFailedException expected");
+        } catch (TransactionCommitFailedException e) {
+            assertTrue(e.getCause() == thrown);
+            domForwardedWriteTransaction.cancel();
+        }
+    }
+}
\ No newline at end of file