1 files changed, 128 insertions, 0 deletions

diff --git a/infra/data-impl/src/test/java/io/fd/honeycomb/data/impl/WriteTransactionTest.java b/infra/data-impl/src/test/java/io/fd/honeycomb/data/impl/WriteTransactionTest.java
new file mode 100644
index 000000000..1dab5524d
--- /dev/null
+++ b/infra/data-impl/src/test/java/io/fd/honeycomb/data/impl/WriteTransactionTest.java
@@ -0,0 +1,128 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.honeycomb.data.impl;
+
+import static junit.framework.TestCase.assertTrue;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.fail;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.MockitoAnnotations.initMocks;
+
+import com.google.common.util.concurrent.CheckedFuture;
+import io.fd.honeycomb.data.DataModification;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataValidationFailedException;
+
+public class WriteTransactionTest {
+
+    @Mock
+    private DataModification configSnapshot;
+    @Mock
+    private YangInstanceIdentifier path;
+    @Mock
+    private NormalizedNode<?,?> data;
+
+    private WriteTransaction writeTx;
+
+    @Before
+    public void setUp() {
+        initMocks(this);
+        writeTx = WriteTransaction.createConfigOnly(configSnapshot);
+    }
+
+    @Test
+    public void testPut() {
+        writeTx.put(LogicalDatastoreType.CONFIGURATION, path, data);
+        verify(configSnapshot).write(path, data);
+    }
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testPutOperational() {
+        writeTx.put(LogicalDatastoreType.OPERATIONAL, path, data);
+        verify(configSnapshot).write(path, data);
+    }
+
+    @Test(expected = IllegalStateException.class)
+    public void testOnFinishedTx() {
+        writeTx.submit();
+        writeTx.put(LogicalDatastoreType.CONFIGURATION, path, data);
+        verify(configSnapshot).write(path, data);
+    }
+
+    @Test
+    public void testMerge() {
+        writeTx.merge(LogicalDatastoreType.CONFIGURATION, path, data);
+        verify(configSnapshot).merge(path, data);
+    }
+
+    @Test
+    public void testCancel() {
+        assertTrue(writeTx.cancel());
+    }
+
+    @Test
+    public void testCancelFinished() {
+        writeTx.submit();
+        assertFalse(writeTx.cancel());
+    }
+
+    @Test
+    public void testDelete() {
+        writeTx.delete(LogicalDatastoreType.CONFIGURATION, path);
+        verify(configSnapshot).delete(path);
+    }
+
+    @Test
+    public void testSubmit() throws Exception {
+        writeTx.submit();
+        verify(configSnapshot).validate();
+        verify(configSnapshot).commit();
+    }
+
+    @Test
+    public void testSubmitFailed() throws Exception {
+        doThrow(mock(DataValidationFailedException.class)).when(configSnapshot).commit();
+        final CheckedFuture<Void, TransactionCommitFailedException> future = writeTx.submit();
+        try {
+            future.get();
+        } catch (Exception e) {
+            assertTrue(e.getCause() instanceof TransactionCommitFailedException);
+            return;
+        }
+        fail("Expected exception to be thrown");
+
+    }
+
+    @Test(expected = UnsupportedOperationException.class)
+    public void testCommit() {
+        writeTx.commit();
+    }
+
+    @Test
+    public void testGetIdentifier() {
+        assertNotNull(writeTx.getIdentifier());
+    }
+}
\ No newline at end of file