--- /dev/null
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * Copyright © 2017-2018 Amdocs
+ * ================================================================================
+ * 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.crud.service;
+
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import javax.ws.rs.core.MediaType;
+import javax.ws.rs.core.MultivaluedHashMap;
+import javax.ws.rs.core.MultivaluedMap;
+import org.onap.aai.logging.LoggingContext;
+import org.onap.aai.restclient.client.OperationResult;
+import org.onap.aai.restclient.client.RestClient;
+import org.onap.crud.dao.champ.ChampDao;
+import org.slf4j.MDC;
+
+public class ChampDaoMockTest {
+ // @formatter:off
+ private final String champVertex = "{" +
+ "\"key\": \"test-uuid\"," +
+ "\"type\": \"vertexType\"," +
+ "\"properties\": {" +
+ "\"fqdn\": \"myhost.onap.com\"," +
+ "\"hostname\": \"myhost\" } }";
+
+ private final String champEdge = "{" +
+ "\"key\": \"test-uuid\"," +
+ "\"type\": \"edgeType\"," +
+ "\"properties\": {" +
+ "\"prevent-delete\": \"NONE\" }," +
+ "\"source\": {" +
+ "\"key\": \"50bdab41-ad1c-4d00-952c-a0aa5d827811\", \"type\": \"vserver\"}," +
+ "\"target\": {" +
+ "\"key\": \"1d326bc7-b985-492b-9604-0d5d1f06f908\", \"type\": \"pserver\"}" +
+ " }";
+
+ private final String edgePayload = "{" +
+ "\"type\":\"tosca.relationships.HostedOn\"," +
+ "\"properties\":{" +
+ "\"prevent-delete\":\"NONE\"}," +
+ "\"source\":{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"vserver\"," +
+ "\"properties\":{}}," +
+ "\"target\":{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "\"hostname\":\"myhost\"," +
+ "\"fqdn\":\"myhost.onap.com\"}}}";
+
+ private final String edgePayloadForPut = "{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"tosca.relationships.HostedOn\"," +
+ "\"properties\":{" +
+ "\"prevent-delete\":\"NONE\"}," +
+ "\"source\":{" +
+ "\"key\":\"50bdab41-ad1c-4d00-952c-a0aa5d827811\"," +
+ "\"type\":\"vserver\"," +
+ "\"properties\":{}}," +
+ "\"target\":{" +
+ "\"key\":\"1d326bc7-b985-492b-9604-0d5d1f06f908\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{}}}";
+
+ private final String edgePayloadForPatch = "{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"tosca.relationships.HostedOn\"," +
+ "\"properties\":{" +
+ "\"prevent-delete\":\"NONE\"}," +
+ "\"source\":{" +
+ "\"key\":\"50bdab41-ad1c-4d00-952c-a0aa5d827811\"," +
+ "\"type\":\"vserver\"}," +
+ "\"target\":{" +
+ "\"key\":\"1d326bc7-b985-492b-9604-0d5d1f06f908\"," +
+ "\"type\":\"pserver\"}}";
+
+ private final String edgePayloadForPost = "{" +
+ "\"type\":\"tosca.relationships.HostedOn\"," +
+ "\"properties\":{" +
+ "\"SVC-INFRA\":\"OUT\"," +
+ "\"prevent-delete\":\"IN\"," +
+ "\"delete-other-v\":\"NONE\"," +
+ "\"contains-other-v\":\"NONE\"" +
+ "}," +
+ "\"source\":{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"vserver\"," +
+ "\"properties\":{" +
+ "" +
+ "}" +
+ "}," +
+ "\"target\":{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "\"hostname\":\"myhost\"," +
+ "\"fqdn\":\"myhost.onap.com\"" +
+ "}" +
+ "}" +
+ "}";
+
+ private final String edgePayloadForPutNoProperties = "{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"tosca.relationships.HostedOn\"," +
+ "\"properties\":{" +
+ "\"SVC-INFRA\":\"OUT\"," +
+ "\"prevent-delete\":\"IN\"," +
+ "\"delete-other-v\":\"NONE\"," +
+ "\"contains-other-v\":\"NONE\"" +
+ "}," +
+ "\"source\":{" +
+ "\"key\":\"50bdab41-ad1c-4d00-952c-a0aa5d827811\"," +
+ "\"type\":\"vserver\"," +
+ "\"properties\":{" +
+ "" +
+ "}" +
+ "}," +
+ "\"target\":{" +
+ "\"key\":\"1d326bc7-b985-492b-9604-0d5d1f06f908\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "" +
+ "}" +
+ "}" +
+ "}";
+
+ private final String vertexPayload = "{" +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "\"hostname\":\"myhost\"," +
+ "\"in-maint\":false," +
+ "\"fqdn\":\"myhost.onap.com\"," +
+ "\"last-mod-source-of-truth\":\"source-of-truth\"," +
+ "\"source-of-truth\":\"source-of-truth\"," +
+ "\"aai-node-type\":\"pserver\"}}";
+
+ private final String vertexPayloadForVserver = "{" +
+ "\"type\":\"vserver\"," +
+ "\"properties\":{" +
+ "\"in-maint\":false," +
+ "\"vserver-name\":\"test-vserver\"," +
+ "\"vserver-id\":\"VSER1\"," +
+ "\"last-mod-source-of-truth\":\"source-of-truth\"," +
+ "\"vserver-name2\":\"alt-test-vserver\"," +
+ "\"source-of-truth\":\"source-of-truth\"," +
+ "\"vserver-selflink\":\"http://1.2.3.4/moreInfo\"," +
+ "\"is-closed-loop-disabled\":false," +
+ "\"aai-node-type\":\"vserver\"}}";
+
+ private final String vertexPayloadForPserver = "{" +
+ "\"key\":\"50bdab41-ad1c-4d00-952c-a0aa5d827811\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "\"ptnii-equip-name\":\"e-name\"," +
+ "\"hostname\":\"steve-host2\"," +
+ "\"equip-type\":\"server\"," +
+ "\"equip-vendor\":\"HP\"," +
+ "\"equip-model\":\"DL380p-nd\"," +
+ "\"in-maint\":false," +
+ "\"fqdn\":\"myhost.onap.net\"," +
+ "\"purpose\":\"my-purpose\"," +
+ "\"ipv4-oam-address\":\"1.2.3.4\"," +
+ "\"last-mod-source-of-truth\":\"source-of-truth\"," +
+ "\"aai-node-type\":\"pserver\"}}";
+
+ private final String vertexPayloadForPut = "{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "\"hostname\":\"myhost\"," +
+ "\"in-maint\":false," +
+ "\"fqdn\":\"myhost.onap.com\"," +
+ "\"last-mod-source-of-truth\":\"source-of-truth\"," +
+ "\"aai-node-type\":\"pserver\"}}";
+
+ private final String vertexPayloadForPatch = "{" +
+ "\"key\":\"test-uuid\"," +
+ "\"type\":\"pserver\"," +
+ "\"properties\":{" +
+ "\"hostname\":\"myhost\"," +
+ "\"fqdn\":\"myhost.onap.com\"," +
+ "\"last-mod-source-of-truth\":\"source-of-truth\"," +
+ "\"aai-node-type\":\"pserver\"}}";
+ // @formatter:on
+
+ private RestClient restClientMock;
+ private ChampDao champDao;
+
+ static final String CHAMP_URL = "https://host:9522/services/champ-service/v1/";
+ static final String OBJECT_SUB_URL = "objects";
+ static final String RELATIONSHIP_SUB_URL = "relationships";
+ static final String TRANSACTION_SUB_URL = "transaction";
+ static final String BASE_OBJECT_URL = CHAMP_URL + OBJECT_SUB_URL;
+ static final String HEADER_FROM_APP = "X-FromAppId";
+ static final String HEADER_TRANS_ID = "X-TransactionId";
+ static final String FROM_APP_NAME = "Gizmo";
+ static final String HEADER_TRANS_ID_VALUE = "1234567890";
+
+ ChampDaoMockTest() {
+ restClientMock = mock(RestClient.class);
+ init();
+ buildChampDao();
+ }
+
+ public void init() {
+
+ Map<String, String> queryParamsVertex = new HashMap<>();
+ queryParamsVertex.put("_reserved_version", "v11");
+ queryParamsVertex.put("hostname", "myhost");
+ queryParamsVertex.put("_reserved_aai-type", "pserver");
+
+ Map<String, String> queryParamsVertexV13 = new HashMap<>();
+ queryParamsVertexV13.put("_reserved_version", "v13");
+ queryParamsVertexV13.put("hostname", "myhost");
+ queryParamsVertexV13.put("_reserved_aai-type", "pserver");
+
+ Map<String, String> queryParamsVertices = new HashMap<>();
+ queryParamsVertices.put("_reserved_version", "v11");
+ queryParamsVertices.put("hostname", "myhost");
+ queryParamsVertices.put("_reserved_aai-type", "pserver");
+ queryParamsVertices.put("aai-node-type", "pserver");
+
+ Map<String, String> queryParamsVerticesV13 = new HashMap<>();
+ queryParamsVerticesV13.put("_reserved_version", "v13");
+ queryParamsVerticesV13.put("hostname", "myhost");
+ queryParamsVerticesV13.put("_reserved_aai-type", "pserver");
+ queryParamsVerticesV13.put("aai-node-type", "pserver");
+
+ Map<String, String> queryParamsEdge = new HashMap<>();
+ queryParamsEdge.put("_reserved_version", "v11");
+ queryParamsEdge.put("hostname", "myhost");
+ queryParamsEdge.put("_reserved_aai-type", "tosca.relationships.HostedOn");
+
+ Map<String, String> emptyQueryParams = null;
+
+ mockOpenTransaction();
+ mockRollbackTransaction("");
+ mockTransactionExists("");
+ mockCommitTransaction("");
+ mockGetVertex("872dd5df-0be9-4167-95e9-2cf4b21165ed", queryParamsVertex, "pserver");
+ mockGetVertex("872dd5df-0be9-4167-95e9-2cf4b21165ed", queryParamsVertexV13, "pserver");
+ mockGetVertex("50bdab41-ad1c-4d00-952c-a0aa5d827811", "", "vserver");
+ mockGetVertex("1d326bc7-b985-492b-9604-0d5d1f06f908", "", "pserver");
+ mockGetVertex("1d326bc7-b985-492b-9604-0d5d1f06f908", "?transactionId=", "pserver");
+ mockGetVertex("test-uuid", "", "pserver");
+ mockGetVertex("50bdab41-ad1c-4d00-952c-a0aa5d827811", "?transactionId=", "vserver");
+ mockGetVertices(queryParamsVertices, "pserver");
+ mockGetVertices(queryParamsVerticesV13, "pserver");
+ mockGetVertexEdges("872dd5df-0be9-4167-95e9-2cf4b21165ed", queryParamsVertex, "tosca.relationships.HostedOn");
+ mockGetVertexEdges("872dd5df-0be9-4167-95e9-2cf4b21165ed", queryParamsVertexV13,
+ "tosca.relationships.HostedOn");
+ mockGetVertexEdges("50bdab41-ad1c-4d00-952c-a0aa5d827811", emptyQueryParams, "tosca.relationships.HostedOn");
+ mockGetVertexEdges("1d326bc7-b985-492b-9604-0d5d1f06f908", emptyQueryParams, "tosca.relationships.HostedOn");
+ mockGetEdges("?", "tosca.relationships.HostedOn");
+ mockGetEdge("50bdab41-ad1c-4d00-952c-a0aa5d827811", "?transactionId=", "tosca.relationships.HostedOn");
+ mockGetEdge("872dd5df-0be9-4167-95e9-2cf4b21165ed", emptyQueryParams, "tosca.relationships.HostedOn");
+ mockGetEdge("872dd5df-0be9-4167-95e9-2cf4b21165ed", queryParamsEdge, "tosca.relationships.HostedOn");
+ mockGetEdge("my-uuid", emptyQueryParams, "tosca.relationships.HostedOn");
+ mockGetEdge("50bdab41-ad1c-4d00-952c-a0aa5d827811", queryParamsEdge, "tosca.relationships.HostedOn");
+ mockPostEdge("tosca.relationships.HostedOn", "", edgePayload);
+ mockPostEdge("tosca.relationships.HostedOn", "?transactionId=", edgePayloadForPost);
+ mockPostVertex("pserver", vertexPayload, "");
+ mockPostVertex("vserver", vertexPayloadForVserver, "?transactionId=");
+ mockPutVertex("test-uuid", "pserver", vertexPayloadForPut, "");
+ mockPutVertex("test-uuid", "pserver", vertexPayloadForPatch, "");
+ mockPutVertex("50bdab41-ad1c-4d00-952c-a0aa5d827811", "pserver", vertexPayloadForPserver, "?transactionId=");
+ mockPutEdge("test-uuid", "tosca.relationships.HostedOn", "", edgePayloadForPut);
+ mockPutEdge("test-uuid", "tosca.relationships.HostedOn", "", edgePayloadForPatch);
+ mockPutEdge("test-uuid", "tosca.relationships.HostedOn", "?transactionId=", edgePayloadForPutNoProperties);
+ mockDeleteVertex("872dd5df-0be9-4167-95e9-2cf4b21165ed", "pserver", "");
+ mockDeleteVertex("50bdab41-ad1c-4d00-952c-a0aa5d827811", "pserver", "?transactionId=");
+ mockDeleteEdge("872dd5df-0be9-4167-95e9-2cf4b21165ed", "tosca.relationships.HostedOn", "");
+ mockDeleteEdge("50bdab41-ad1c-4d00-952c-a0aa5d827811", "tosca.relationships.HostedOn", "?transactionId=");
+ }
+
+ public void buildChampDao() {
+ String baseRelationshipUrl = CHAMP_URL + RELATIONSHIP_SUB_URL;
+ String baseTransactionUrl = CHAMP_URL + TRANSACTION_SUB_URL;
+ champDao = new ChampDao(restClientMock, BASE_OBJECT_URL, baseRelationshipUrl, baseTransactionUrl);
+ }
+
+ public void mockOpenTransaction() {
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult("");
+ operationResult.setResultCode(200);
+ String url = CHAMP_URL + "transaction";
+
+
+ when(restClientMock.post(url, "", createHeaders(), MediaType.TEXT_PLAIN_TYPE, MediaType.TEXT_PLAIN_TYPE))
+ .thenReturn(operationResult);
+ }
+
+ public void mockRollbackTransaction(String id) {
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult("");
+ operationResult.setResultCode(200);
+ String url = CHAMP_URL + TRANSACTION_SUB_URL + "/" + id;
+
+
+ when(restClientMock.put(url, "{\"method\": \"rollback\"}", createHeaders(), MediaType.APPLICATION_JSON_TYPE,
+ MediaType.TEXT_PLAIN_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockCommitTransaction(String id) {
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult("");
+ operationResult.setResultCode(200);
+ String url = CHAMP_URL + TRANSACTION_SUB_URL + "/" + id;
+
+
+ when(restClientMock.put(url, "{\"method\": \"commit\"}", createHeaders(), MediaType.APPLICATION_JSON_TYPE,
+ MediaType.TEXT_PLAIN_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockTransactionExists(String id) {
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult("");
+ operationResult.setResultCode(200);
+ String url = CHAMP_URL + TRANSACTION_SUB_URL + "/" + id;
+
+ Map<String, List<String>> headers = new HashMap<>();
+ headers.put(HEADER_FROM_APP, Arrays.asList(FROM_APP_NAME));
+ headers.put(HEADER_TRANS_ID, Arrays.asList(MDC.get(LoggingContext.LoggingField.REQUEST_ID.toString())));
+
+ when(restClientMock.get(url, headers, MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockGetVertex(String id, String txId, String type) {
+ String vertexResponse = champVertex.replace("vertexType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(vertexResponse);
+ operationResult.setResultCode(200);
+
+ String url = BASE_OBJECT_URL + "/" + id + txId;
+
+
+ when(restClientMock.get(url, createHeaders(), MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockGetVertex(String id, Map<String, String> queryParams, String type) {
+ String vertexResponse = champVertex.replace("vertexType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(vertexResponse);
+ operationResult.setResultCode(200);
+
+ StringBuilder url = appendQueryParams(BASE_OBJECT_URL + "/" + id, queryParams);
+
+ when(restClientMock.get(url.toString(), createHeaders(), MediaType.APPLICATION_JSON_TYPE))
+ .thenReturn(operationResult);
+ }
+
+ public void mockGetVertexEdges(String id, Map<String, String> queryParams, String type) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ List<String> edgeResponselist = new ArrayList<>();
+ edgeResponselist.add(edgeResponse);
+ operationResult.setResult(edgeResponselist.toString());
+ operationResult.setResultCode(200);
+
+ StringBuilder url = appendQueryParams(BASE_OBJECT_URL + "/" + RELATIONSHIP_SUB_URL + "/" + id, queryParams);
+
+ when(restClientMock.get(url.toString(), createHeaders(), MediaType.APPLICATION_JSON_TYPE))
+ .thenReturn(operationResult);
+ }
+
+ public void mockGetVertices(Map<String, String> queryParams, String type) {
+ String vertexResponse = champVertex.replace("vertexType", type);
+ OperationResult operationResult = new OperationResult();
+ List<String> vertexResponselist = new ArrayList<>();
+ vertexResponselist.add(vertexResponse);
+ operationResult.setResult(vertexResponselist.toString());
+ operationResult.setResultCode(200);
+
+ StringBuilder url = appendQueryParams(BASE_OBJECT_URL + "/" + "filter", queryParams);
+
+ when(restClientMock.get(url.toString(), createHeaders(), MediaType.APPLICATION_JSON_TYPE))
+ .thenReturn(operationResult);
+ }
+
+ public void mockGetEdges(String queryParams, String type) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ List<String> edgeResponselist = new ArrayList<>();
+ edgeResponselist.add(edgeResponse);
+ operationResult.setResult(edgeResponselist.toString());
+ operationResult.setResultCode(200);
+
+ String url = CHAMP_URL + RELATIONSHIP_SUB_URL + "/" + "filter" + queryParams;
+
+
+ when(restClientMock.get(url, createHeaders(), MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockGetEdge(String id, String txId, String type) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(edgeResponse);
+ operationResult.setResultCode(200);
+
+ String url = CHAMP_URL + RELATIONSHIP_SUB_URL + "/" + id + txId;
+
+ when(restClientMock.get(url, createHeaders(), MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockGetEdge(String id, Map<String, String> queryParams, String type) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(edgeResponse);
+ operationResult.setResultCode(200);
+
+ StringBuilder url = appendQueryParams(CHAMP_URL + RELATIONSHIP_SUB_URL + "/" + id, queryParams);
+
+ when(restClientMock.get(url.toString(), createHeaders(), MediaType.APPLICATION_JSON_TYPE))
+ .thenReturn(operationResult);
+ }
+
+ public void mockPostEdge(String type, String txId, String payload) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(edgeResponse);
+ operationResult.setResultCode(201);
+ MultivaluedMap<String, String> headers = new MultivaluedHashMap<String, String>();
+ headers.add("etag", "test123");
+ operationResult.setHeaders(headers);
+
+ String baseRelationshipUrl = CHAMP_URL + RELATIONSHIP_SUB_URL + txId;
+ String url = baseRelationshipUrl;
+
+
+ when(restClientMock.post(url, payload, createHeaders(), MediaType.APPLICATION_JSON_TYPE,
+ MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockPostVertex(String type, String payload, String txId) {
+ String vertexResponse = champVertex.replace("vertexType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(vertexResponse);
+ operationResult.setResultCode(201);
+ MultivaluedMap<String, String> headers = new MultivaluedHashMap<String, String>();
+ headers.add("etag", "test123");
+ operationResult.setHeaders(headers);
+
+ String url = BASE_OBJECT_URL + txId;
+
+
+ when(restClientMock.post(url, payload, createHeaders(), MediaType.APPLICATION_JSON_TYPE,
+ MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockPutVertex(String id, String type, String payload, String txId) {
+ String vertexResponse = champVertex.replace("vertexType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(vertexResponse);
+ operationResult.setResultCode(200);
+ MultivaluedMap<String, String> headers = new MultivaluedHashMap<String, String>();
+ headers.add("etag", "test123");
+ operationResult.setHeaders(headers);
+
+ String url = BASE_OBJECT_URL + "/" + id + txId;
+
+
+ when(restClientMock.put(url, payload, createHeaders(), MediaType.APPLICATION_JSON_TYPE,
+ MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockPutEdge(String id, String type, String txId, String payload) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(edgeResponse);
+ operationResult.setResultCode(200);
+ MultivaluedMap<String, String> headers = new MultivaluedHashMap<String, String>();
+ headers.add("etag", "test123");
+ operationResult.setHeaders(headers);
+
+ String url = CHAMP_URL + RELATIONSHIP_SUB_URL + "/" + id + txId;
+
+
+ when(restClientMock.put(url, payload, createHeaders(), MediaType.APPLICATION_JSON_TYPE,
+ MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockDeleteVertex(String id, String type, String txId) {
+ String vertexResponse = champVertex.replace("vertexType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(vertexResponse);
+ operationResult.setResultCode(200);
+
+ String url = BASE_OBJECT_URL + "/" + id + txId;
+
+
+ when(restClientMock.delete(url, createHeaders(), MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ public void mockDeleteEdge(String id, String type, String txId) {
+ String edgeResponse = champEdge.replace("edgeType", type);
+ OperationResult operationResult = new OperationResult();
+ operationResult.setResult(edgeResponse);
+ operationResult.setResultCode(200);
+
+ String url = CHAMP_URL + RELATIONSHIP_SUB_URL + "/" + id + txId;
+
+
+ when(restClientMock.delete(url, createHeaders(), MediaType.APPLICATION_JSON_TYPE)).thenReturn(operationResult);
+ }
+
+ private Map<String, List<String>> createHeaders() {
+ Map<String, List<String>> headers = new HashMap<>();
+ List<String> listFromApp = new ArrayList<>();
+ List<String> listTransId = new ArrayList<>();
+ listFromApp.add(FROM_APP_NAME);
+ listTransId.add(HEADER_TRANS_ID_VALUE);
+ headers.put(HEADER_FROM_APP, listFromApp);
+ headers.put(HEADER_TRANS_ID, listTransId);
+
+ return headers;
+ }
+
+ private StringBuilder appendQueryParams(String url, Map<String, String> queryParams) {
+ StringBuilder strBuilder = new StringBuilder(url);
+
+ if (queryParams != null) {
+ String prefix = "?";
+ for (Map.Entry<String, String> entry : queryParams.entrySet()) {
+ strBuilder.append(prefix);
+ prefix = "&";
+ strBuilder.append(entry.getKey() + "=" + entry.getValue());
+ }
+ }
+ return strBuilder;
+ }
+
+ public ChampDao getChampDao() {
+ return champDao;
+ }
+
+ public void setChampDao(ChampDao champDao) {
+ this.champDao = champDao;
+ }
+}
Code Review / aai / gizmo.git / blobdiff