Bump upstreams for Silicon
[netconf.git] / restconf / restconf-nb-bierman02 / src / test / java / org / opendaylight / controller / sal / restconf / impl / test / RestGetOperationTest.java
index e09f6c2bdbbc4b7365cc3366e5c8ebc09eab1119..2714f33425215d72c67f32765721afc0d2675a74 100644 (file)
@@ -26,6 +26,7 @@ import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.Set;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -42,6 +43,8 @@ import org.junit.Ignore;
 import org.junit.Test;
 import org.opendaylight.controller.md.sal.rest.common.TestRestconfUtils;
 import org.opendaylight.mdsal.dom.api.DOMMountPoint;
+import org.opendaylight.mdsal.dom.api.DOMSchemaService;
+import org.opendaylight.mdsal.dom.spi.FixedDOMSchemaService;
 import org.opendaylight.netconf.sal.rest.impl.JsonNormalizedNodeBodyReader;
 import org.opendaylight.netconf.sal.rest.impl.NormalizedNodeJsonBodyWriter;
 import org.opendaylight.netconf.sal.rest.impl.NormalizedNodeXmlBodyWriter;
@@ -170,7 +173,8 @@ public class RestGetOperationTest extends JerseyTest {
     public void getDataWithUrlMountPoint() throws Exception {
         when(brokerFacade.readConfigurationData(any(DOMMountPoint.class), any(YangInstanceIdentifier.class),
                 isNull())).thenReturn(prepareCnDataForMountPointTest(false));
-        when(mountInstance.getSchemaContext()).thenReturn(schemaContextTestModule);
+        when(mountInstance.getService(DOMSchemaService.class))
+                .thenReturn(Optional.of(FixedDOMSchemaService.of(schemaContextTestModule)));
 
         String uri = "/config/ietf-interfaces:interfaces/interface/0/yang-ext:mount/test-module:cont/cont1";
         assertEquals(200, get(uri, MediaType.APPLICATION_XML));
@@ -192,8 +196,8 @@ public class RestGetOperationTest extends JerseyTest {
         final YangInstanceIdentifier awaitedInstanceIdentifier = prepareInstanceIdentifierForList();
         when(brokerFacade.readConfigurationData(any(DOMMountPoint.class), eq(awaitedInstanceIdentifier),
                 isNull())).thenReturn(prepareCnDataForSlashesBehindMountPointTest());
-
-        when(mountInstance.getSchemaContext()).thenReturn(schemaContextTestModule);
+        when(mountInstance.getService(DOMSchemaService.class))
+                .thenReturn(Optional.of(FixedDOMSchemaService.of(schemaContextTestModule)));
 
         final String uri = "/config/ietf-interfaces:interfaces/interface/0/yang-ext:mount/"
                 + "test-module:cont/lst1/GigabitEthernet0%2F0%2F0%2F0";
@@ -221,8 +225,8 @@ public class RestGetOperationTest extends JerseyTest {
     public void getDataMountPointIntoHighestElement() throws Exception {
         when(brokerFacade.readConfigurationData(any(DOMMountPoint.class), any(YangInstanceIdentifier.class),
                 isNull())).thenReturn(prepareCnDataForMountPointTest(true));
-
-        when(mountInstance.getSchemaContext()).thenReturn(schemaContextTestModule);
+        when(mountInstance.getService(DOMSchemaService.class))
+                .thenReturn(Optional.of(FixedDOMSchemaService.of(schemaContextTestModule)));
 
         final String uri = "/config/ietf-interfaces:interfaces/interface/0/yang-ext:mount/test-module:cont";
         assertEquals(200, get(uri, MediaType.APPLICATION_XML));
@@ -369,7 +373,7 @@ public class RestGetOperationTest extends JerseyTest {
     public void getOperationsBehindMountPointTest() throws Exception {
         setControllerContext(schemaContextModules);
 
-        when(mountInstance.getSchemaContext()).thenReturn(schemaContextBehindMountPoint);
+        mockMountPoint();
 
         final String uri = "/operations/ietf-interfaces:interfaces/interface/0/yang-ext:mount/";
 
@@ -403,7 +407,7 @@ public class RestGetOperationTest extends JerseyTest {
     public void getModulesBehindMountPoint() throws Exception {
         setControllerContext(schemaContextModules);
 
-        when(mountInstance.getSchemaContext()).thenReturn(schemaContextBehindMountPoint);
+        mockMountPoint();
 
         final String uri = "/modules/ietf-interfaces:interfaces/interface/0/yang-ext:mount/";
 
@@ -431,7 +435,7 @@ public class RestGetOperationTest extends JerseyTest {
     public void getModuleBehindMountPoint() throws Exception {
         setControllerContext(schemaContextModules);
 
-        when(mountInstance.getSchemaContext()).thenReturn(schemaContextBehindMountPoint);
+        mockMountPoint();
 
         final String uri = "/modules/module/ietf-interfaces:interfaces/interface/0/yang-ext:mount/"
                 + "module1-behind-mount-point/2014-02-03";
@@ -456,8 +460,6 @@ public class RestGetOperationTest extends JerseyTest {
         assertEquals("module1-behind-mount-point", module.getLocalName());
         assertEquals("2014-02-03", module.getRevision().get().toString());
         assertEquals("module:1:behind:mount:point", module.getNamespace().toString());
-
-
     }
 
     private static void validateModulesResponseXml(final Response response, final SchemaContext schemaContext) {
@@ -744,4 +746,8 @@ public class RestGetOperationTest extends JerseyTest {
         }
     }
 
+    private void mockMountPoint() {
+        when(mountInstance.getService(DOMSchemaService.class))
+            .thenReturn(Optional.of(FixedDOMSchemaService.of(schemaContextBehindMountPoint)));
+    }
 }