Add ClientTransactionCursorTest

Change-Id: I8ea618446f8785f89221ee5c02f2f7e1f67bf40e
Signed-off-by: Andrej Mak <andrej.mak@pantheon.tech>

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransactionCursorTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransactionCursorTest.java
new file mode 100644 (file)
index 0000000..14bc0de
--- /dev/null
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransactionCursorTest.java
@@ -0,0 +1,130 @@
+/*
+ * Copyright (c) 2017 Pantheon Technologies s.r.o. and others.  All rights reserved.
+ *
+ * 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.controller.cluster.databroker.actors.dds;
+
+import static org.mockito.Mockito.verify;
+
+import java.util.Arrays;
+import java.util.stream.Collectors;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
+import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
+
+public class ClientTransactionCursorTest {
+
+    private static final QName NODE_1 = QName.create("ns-1", "node-1");
+    private static final QName NODE_2 = QName.create(NODE_1, "node-2");
+    private static final QName NODE_3 = QName.create(NODE_1, "node-3");
+
+    @Mock
+    private ClientTransaction transaction;
+    private ClientTransactionCursor cursor;
+
+    @Before
+    public void setUp() throws Exception {
+        MockitoAnnotations.initMocks(this);
+        cursor = new ClientTransactionCursor(transaction);
+    }
+
+    @Test
+    public void testEnterOneNode() throws Exception {
+        cursor.enter(YangInstanceIdentifier.NodeIdentifier.create(NODE_1));
+        cursor.delete(YangInstanceIdentifier.NodeIdentifier.create(NODE_2));
+        final YangInstanceIdentifier expected = createId(NODE_1, NODE_2);
+        verify(transaction).delete(expected);
+    }
+
+    @Test
+    public void testEnterNodeIterables() throws Exception {
+        final Iterable<YangInstanceIdentifier.PathArgument> collect = toPathArg(NODE_1, NODE_2);
+        cursor.enter(collect);
+        cursor.delete(YangInstanceIdentifier.NodeIdentifier.create(NODE_3));
+        final YangInstanceIdentifier expected = createId(NODE_1, NODE_2, NODE_3);
+        verify(transaction).delete(expected);
+    }
+
+    @Test
+    public void testEnterNodeVarargs() throws Exception {
+        cursor.enter(YangInstanceIdentifier.NodeIdentifier.create(NODE_1),
+                YangInstanceIdentifier.NodeIdentifier.create(NODE_2));
+        cursor.delete(YangInstanceIdentifier.NodeIdentifier.create(NODE_3));
+        final YangInstanceIdentifier expected = createId(NODE_1, NODE_2, NODE_3);
+        verify(transaction).delete(expected);
+    }
+
+    @Test
+    public void testExitOneLevel() throws Exception {
+        cursor.enter(toPathArg(NODE_1, NODE_2));
+        cursor.exit();
+        cursor.delete(YangInstanceIdentifier.NodeIdentifier.create(NODE_2));
+        final YangInstanceIdentifier expected = createId(NODE_1, NODE_2);
+        verify(transaction).delete(expected);
+    }
+
+    @Test
+    public void testExitTwoLevels() throws Exception {
+        cursor.enter(toPathArg(NODE_1, NODE_2, NODE_3));
+        cursor.exit(2);
+        cursor.delete(YangInstanceIdentifier.NodeIdentifier.create(NODE_2));
+        final YangInstanceIdentifier expected = createId(NODE_1, NODE_2);
+        verify(transaction).delete(expected);
+    }
+
+    @Test
+    public void testClose() throws Exception {
+        cursor.close();
+        verify(transaction).closeCursor(cursor);
+    }
+
+    @Test
+    public void testDelete() throws Exception {
+        cursor.delete(YangInstanceIdentifier.NodeIdentifier.create(NODE_1));
+        final YangInstanceIdentifier expected = createId(NODE_1);
+        verify(transaction).delete(expected);
+    }
+
+    @Test
+    public void testMerge() throws Exception {
+        final YangInstanceIdentifier.NodeIdentifier path = YangInstanceIdentifier.NodeIdentifier.create(NODE_1);
+        final ContainerNode data = createData(path.getNodeType());
+        cursor.merge(path, data);
+        final YangInstanceIdentifier expected = createId(NODE_1);
+        verify(transaction).merge(expected, data);
+    }
+
+    @Test
+    public void testWrite() throws Exception {
+        final YangInstanceIdentifier.NodeIdentifier path = YangInstanceIdentifier.NodeIdentifier.create(NODE_1);
+        final ContainerNode data = createData(path.getNodeType());
+        cursor.write(path, data);
+        final YangInstanceIdentifier expected = createId(NODE_1);
+        verify(transaction).write(expected, data);
+    }
+
+    private static Iterable<YangInstanceIdentifier.PathArgument> toPathArg(final QName... pathArguments) {
+        return Arrays.stream(pathArguments)
+                .map(YangInstanceIdentifier.NodeIdentifier::create)
+                .collect(Collectors.toList());
+    }
+
+    private static YangInstanceIdentifier createId(final QName... pathArguments) {
+        return YangInstanceIdentifier.create(toPathArg(pathArguments));
+    }
+
+    private static ContainerNode createData(final QName id) {
+        return Builders.containerBuilder()
+                .withNodeIdentifier(YangInstanceIdentifier.NodeIdentifier.create(id))
+                .build();
+    }
+
+}
\ No newline at end of file