Update Query catalog db test

Issue-ID: SO-2758

Signed-off-by: zm330 <zhangminyj@chinamobile.com>
Change-Id: I0bace7c854a064d0c584341a5ffd844d5a22be55

2 files changed, 109 insertions, 0 deletions

diff --git a/adapters/mso-catalog-db-adapter/src/test/java/org/onap/so/adapters/catalogdb/catalogrest/QueryServiceArtifactTest.java b/adapters/mso-catalog-db-adapter/src/test/java/org/onap/so/adapters/catalogdb/catalogrest/QueryServiceArtifactTest.java
new file mode 100644
index 0000000000..2266eafa12
--- /dev/null
+++ b/adapters/mso-catalog-db-adapter/src/test/java/org/onap/so/adapters/catalogdb/catalogrest/QueryServiceArtifactTest.java
@@ -0,0 +1,55 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * Copyright (c) 2019, CMCC Technologies Co., Ltd.
+ * ================================================================================
+ * 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.so.adapters.catalogdb.catalogrest;
+
+import org.assertj.core.api.Assertions;
+import org.junit.Test;
+import org.onap.so.db.catalog.beans.Service;
+import org.onap.so.db.catalog.beans.ServiceArtifact;
+import org.onap.so.jsonpath.JsonPathUtil;
+import java.util.ArrayList;
+import java.util.List;
+import static org.mockito.Mockito.mock;
+
+public class QueryServiceArtifactTest {
+
+    @Test
+    public void ServiceArtifactTest() {
+        QueryServiceArtifact queryServiceArtifact = new QueryServiceArtifact(createList());
+        String jsonResult = queryServiceArtifact.JSON2(true, false);
+        Assertions.assertThat(JsonPathUtil.getInstance().locateResult(jsonResult, "$.serviceArtifact[0].name").get())
+                .isEqualTo("eMBB.zip");
+    }
+
+    private List<ServiceArtifact> createList() {
+        List<ServiceArtifact> artifacts = new ArrayList<>();
+        Service service = mock(Service.class);
+        ServiceArtifact artifact = new ServiceArtifact();
+        artifact.setService(service);
+        artifact.setArtifactUUID("b170dbeb-2954-4a4f-ad12-6bc84b3e089e");
+        artifact.setChecksum("ZWRkMGM3NzNjMmE3NzliYTFiZGNmZjVlMDE4OWEzMTA=");
+        artifact.setDescription("embbCn");
+        artifact.setType("OTHER");
+        artifact.setName("eMBB.zip");
+        artifact.setVersion("1");
+        artifacts.add(artifact);
+        return artifacts;
+    }
+}
diff --git a/adapters/mso-catalog-db-adapter/src/test/java/org/onap/so/adapters/catalogdb/catalogrest/QueryServiceInfoTest.java b/adapters/mso-catalog-db-adapter/src/test/java/org/onap/so/adapters/catalogdb/catalogrest/QueryServiceInfoTest.java
new file mode 100644
index 0000000000..e948141b89
--- /dev/null
+++ b/adapters/mso-catalog-db-adapter/src/test/java/org/onap/so/adapters/catalogdb/catalogrest/QueryServiceInfoTest.java
@@ -0,0 +1,54 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * Copyright (c) 2019, CMCC Technologies Co., Ltd.
+ * ================================================================================
+ * 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.so.adapters.catalogdb.catalogrest;
+
+import org.assertj.core.api.Assertions;
+import org.junit.Test;
+import org.onap.so.db.catalog.beans.ServiceInfo;
+import org.onap.so.db.catalog.beans.Service;
+import org.onap.so.jsonpath.JsonPathUtil;
+import java.util.ArrayList;
+import java.util.List;
+import static org.mockito.Mockito.mock;
+
+public class QueryServiceInfoTest {
+
+    @Test
+    public void serviceInfoTest() {
+        QueryServiceInfo queryServiceInfo = new QueryServiceInfo(createList());
+        String jsonResult = queryServiceInfo.JSON2(true, false);
+        String serviceInfo = jsonResult.substring(jsonResult.indexOf("{"), jsonResult.length());
+        Assertions.assertThat(JsonPathUtil.getInstance().locateResult(serviceInfo, "$.id").get()).isEqualTo("1");
+    }
+
+    private List<ServiceInfo> createList() {
+        Service service = mock(Service.class);
+        ServiceInfo serviceInfo = new ServiceInfo();
+        serviceInfo.setId(1);
+        serviceInfo.setService(service);
+        serviceInfo.setServiceInput(null);
+        serviceInfo.setServiceProperties(null);
+
+        List<ServiceInfo> serviceInfos = new ArrayList<>();
+        serviceInfos.add(serviceInfo);
+        return serviceInfos;
+    }
+
+}