Bug 5679 - ietf-yang-library module implemetation

[netconf.git] / restconf / sal-rest-connector / src / test / java / org / opendaylight / restconf / rest / services / impl / RestconfOperationsServiceTest.java

diff --git a/restconf/sal-rest-connector/src/test/java/org/opendaylight/restconf/rest/services/impl/RestconfOperationsServiceTest.java b/restconf/sal-rest-connector/src/test/java/org/opendaylight/restconf/rest/services/impl/RestconfOperationsServiceTest.java
index eea5db5d55283953ae9ebc8bcacf926bc2a08e07..7138f1dc0b5008fe2fd888382c8b373a4ebf6df2 100644 (file)
--- a/restconf/sal-rest-connector/src/test/java/org/opendaylight/restconf/rest/services/impl/RestconfOperationsServiceTest.java
+++ b/restconf/sal-rest-connector/src/test/java/org/opendaylight/restconf/rest/services/impl/RestconfOperationsServiceTest.java
@@ -7,19 +7,30 @@
  */
 package org.opendaylight.restconf.rest.services.impl;
 
-import java.util.ArrayList;
-import java.util.List;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+import com.google.common.collect.ImmutableSet;
+import com.google.common.util.concurrent.CheckedFuture;
+import java.net.URI;
+import java.util.Set;
 import javax.ws.rs.core.UriInfo;
-import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.mockito.Mock;
+import org.mockito.Mockito;
 import org.mockito.MockitoAnnotations;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
+import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
 import org.opendaylight.controller.md.sal.rest.common.TestRestconfUtils;
 import org.opendaylight.netconf.sal.restconf.impl.NormalizedNodeContext;
+import org.opendaylight.restconf.base.services.impl.RestconfOperationsServiceImpl;
 import org.opendaylight.restconf.handlers.DOMMountPointServiceHandler;
 import org.opendaylight.restconf.handlers.SchemaContextHandler;
+import org.opendaylight.restconf.handlers.TransactionChainHandler;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.common.QNameModule;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
@@ -37,35 +48,51 @@ public class RestconfOperationsServiceTest {
     private SchemaContextHandler schemaContextHandler;
     private DOMMountPointServiceHandler domMountPointServiceHandler;
 
-    private static final List<String> listOfRpcsNames = new ArrayList<>();
+    private Set<QName> listOfRpcsNames;
 
     @Before
     public void init() throws Exception {
         MockitoAnnotations.initMocks(this);
         this.schemaContext = TestRestconfUtils.loadSchemaContext("/modules");
-        this.schemaContextHandler = new SchemaContextHandler();
+
+        final TransactionChainHandler txHandler = Mockito.mock(TransactionChainHandler.class);
+        final DOMTransactionChain domTx = Mockito.mock(DOMTransactionChain.class);
+        Mockito.when(txHandler.get()).thenReturn(domTx);
+        final DOMDataWriteTransaction wTx = Mockito.mock(DOMDataWriteTransaction.class);
+        Mockito.when(domTx.newWriteOnlyTransaction()).thenReturn(wTx);
+        final CheckedFuture checked = Mockito.mock(CheckedFuture.class);
+        Mockito.when(wTx.submit()).thenReturn(checked);
+        final Object valueObj = null;
+        Mockito.when(checked.checkedGet()).thenReturn(valueObj);
+        this.schemaContextHandler = new SchemaContextHandler(txHandler);
         this.schemaContextHandler.onGlobalContextUpdated(this.schemaContext);
+
         this.domMountPointServiceHandler = new DOMMountPointServiceHandler(this.domMountPointService);
-        listOfRpcsNames.add("module2:dummy-rpc2-module2");
-        listOfRpcsNames.add("module2:dummy-rpc1-module2");
-        listOfRpcsNames.add("module1:dummy-rpc2-module1");
-        listOfRpcsNames.add("module1:dummy-rpc1-module1");
+
+        final QNameModule module1 = QNameModule.create(new URI("module:1"), null);
+        final QNameModule module2 = QNameModule.create(new URI("module:2"), null);
+
+        this.listOfRpcsNames = ImmutableSet.of(QName.create(module1, "dummy-rpc1-module1"),
+                QName.create(module1, "dummy-rpc2-module1"), QName.create(module2, "dummy-rpc1-module2"),
+                QName.create(module2, "dummy-rpc2-module2"));
     }
 
     @Test
     public void getOperationsTest() {
-        final RestconfOperationsServiceImpl oper = new RestconfOperationsServiceImpl(this.schemaContextHandler, this.domMountPointServiceHandler);
+        final RestconfOperationsServiceImpl oper =
+                new RestconfOperationsServiceImpl(this.schemaContextHandler, this.domMountPointServiceHandler);
         final NormalizedNodeContext operations = oper.getOperations(this.uriInfo);
         final ContainerNode data = (ContainerNode) operations.getData();
-        Assert.assertTrue(
-                data.getNodeType().getNamespace().toString().equals("urn:ietf:params:xml:ns:yang:ietf-restconf"));
-        Assert.assertTrue(data.getNodeType().getLocalName().equals("operations"));
-        for (final DataContainerChild<? extends PathArgument, ?> dataContainerChild : data.getValue()) {
-            Assert.assertTrue(dataContainerChild.getNodeType().getNamespace().toString()
-                    .equals("urn:ietf:params:xml:ns:yang:ietf-restconf"));
-            Assert.assertTrue(listOfRpcsNames.contains(dataContainerChild.getNodeType().getLocalName()));
-            Assert.assertTrue(dataContainerChild.getValue() == null);
-        }
+        assertEquals("urn:ietf:params:xml:ns:yang:ietf-restconf", data.getNodeType().getNamespace().toString());
+        assertEquals("operations", data.getNodeType().getLocalName());
+
+        assertEquals(4, data.getValue().size());
 
+        for (final DataContainerChild<? extends PathArgument, ?> child : data.getValue()) {
+            assertNull(child.getValue());
+
+            final QName qname = child.getNodeType().withoutRevision();
+            assertTrue(this.listOfRpcsNames.contains(qname));
+        }
     }
 }