Improve the performance of resoures microservice

[aai/aai-common.git] / aai-core / src / test / java / org / onap / aai / rest / EntitlementTest.java

diff --git a/aai-core/src/test/java/org/onap/aai/rest/EntitlementTest.java b/aai-core/src/test/java/org/onap/aai/rest/EntitlementTest.java
new file mode 100644 (file)
index 0000000..3559b9d
--- /dev/null
+++ b/aai-core/src/test/java/org/onap/aai/rest/EntitlementTest.java
@@ -0,0 +1,96 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * 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.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.aai.rest;
+
+import com.jayway.jsonpath.JsonPath;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+import org.onap.aai.AAISetup;
+import org.onap.aai.HttpTestUtil;
+import org.onap.aai.PayloadUtil;
+import org.onap.aai.exceptions.AAIException;
+import org.onap.aai.serialization.engines.QueryStyle;
+import org.skyscreamer.jsonassert.JSONAssert;
+import org.springframework.test.annotation.DirtiesContext;
+
+import javax.ws.rs.core.Response;
+import java.io.IOException;
+import java.io.UnsupportedEncodingException;
+import java.util.Arrays;
+import java.util.Collection;
+
+import static org.junit.Assert.assertEquals;
+
+@RunWith(value = Parameterized.class)
+@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)
+public class EntitlementTest extends AAISetup {
+
+    private HttpTestUtil httpTestUtil;
+
+    @Parameterized.Parameter(value = 0)
+    public QueryStyle queryStyle;
+
+    private String vnfPayload;
+
+    private String vnfUri;
+
+    @Parameterized.Parameters(name = "QueryStyle.{0}")
+    public static Collection<Object[]> data() {
+        return Arrays.asList(new Object[][]{
+            {QueryStyle.TRAVERSAL_URI}
+        });
+    }
+
+    @Before
+    public void setUp() throws IOException {
+        httpTestUtil = new HttpTestUtil(queryStyle);
+        vnfPayload = PayloadUtil.getResourcePayload("vnf.json");
+        vnfUri = "/aai/v14/network/generic-vnfs/generic-vnf/vnf1";
+    }
+
+    @Test
+    public void testPutGenericVnfAndThenInsertEntitlement() throws IOException, AAIException {
+        String entitlementPayload = PayloadUtil.getResourcePayload("entitlement.json");
+        String entitlementUri = "/aai/v14/network/generic-vnfs/generic-vnf/vnf1/entitlements/entitlement/g1/r1";
+        Response response     = httpTestUtil.doPut(vnfUri, vnfPayload);
+        assertEquals("Expected the Generic Vnf to be created", 201, response.getStatus());
+
+        response = httpTestUtil.doGet(vnfUri);
+        assertEquals("Expected the Generic Vnf to be found", 200, response.getStatus());
+        String jsonResponse = response.getEntity().toString();
+        JSONAssert.assertEquals(vnfPayload, jsonResponse, false);
+
+        response     = httpTestUtil.doPut(entitlementUri, entitlementPayload);
+        assertEquals("Expected the Entitlement to be created", 201, response.getStatus());
+    }
+
+    @After
+    public void tearDown() throws UnsupportedEncodingException, AAIException {
+        Response response = httpTestUtil.doGet(vnfUri);
+        assertEquals("Expected the Generic Vnf to be found", 200, response.getStatus());
+        String jsonResponse = response.getEntity().toString();
+        String resourceVersion = JsonPath.read(jsonResponse, "$.resource-version");
+        response = httpTestUtil.doDelete(vnfUri, resourceVersion);
+        assertEquals("Expected the cloud region to be deleted", 204, response.getStatus());
+    }
+}