catalog-be/src/test/java/org/openecomp/sdc/be/servlets/ComponentServletTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.servlets;
  22
  23 import fj.data.Either;
  24 import org.glassfish.jersey.server.ResourceConfig;
  25 import org.json.JSONArray;
  26 import org.json.JSONObject;
  27 import org.junit.Before;
  28 import org.junit.Test;
  29 import org.openecomp.sdc.be.components.impl.ComponentBusinessLogicProvider;
  30 import org.openecomp.sdc.be.components.impl.ResourceBusinessLogic;
  31 import org.openecomp.sdc.be.components.utils.PolicyDefinitionBuilder;
  32 import org.openecomp.sdc.be.model.PolicyDefinition;
  33 import org.openecomp.sdc.be.model.User;
  34 import org.openecomp.sdc.be.ui.model.UiComponentDataTransfer;
  35 import org.openecomp.sdc.common.api.Constants;
  36
  37 import javax.ws.rs.client.Invocation;
  38 import javax.ws.rs.core.MediaType;
  39 import javax.ws.rs.core.Response;
  40 import java.util.Collections;
  41
  42 import static java.util.Arrays.asList;
  43 import static org.assertj.core.api.Assertions.assertThat;
  44 import static org.mockito.ArgumentMatchers.any;
  45 import static org.mockito.ArgumentMatchers.eq;
  46 import static org.mockito.Mockito.mock;
  47 import static org.mockito.Mockito.when;
  48
  49 public class ComponentServletTest extends JerseySpringBaseTest{
  50
  51     private static final String USER_ID = "userId";
  52     private static final String RESOURCE_ID = "resourceId";
  53     private ResourceBusinessLogic resourceBusinessLogic;
  54     private PolicyDefinition policy1, policy2;
  55
  56     @Override
  57     @Before
  58     public void setUp() throws Exception {
  59         super.setUp();
  60         policy1 = buildPolicy("p1");
  61         policy2 = buildPolicy("p2");
  62     }
  63
  64     @Override
  65     protected ResourceConfig configure() {
  66         resourceBusinessLogic = mock(ResourceBusinessLogic.class);
  67         ComponentServlet componentServlet = new ComponentServlet(new ComponentBusinessLogicProvider(resourceBusinessLogic, null, null));
  68         return super.configure().register(componentServlet);
  69     }
  70
  71     @Test
  72     public void filterDataByParam_getPolicies_returnOnlyNameTargetsAndIdFields() {
  73         UiComponentDataTransfer dataTransfer = buildDataTransferWithPolicies();
  74         when(resourceBusinessLogic.getComponentDataFilteredByParams(eq(RESOURCE_ID.toLowerCase()), any(User.class), eq(Collections.singletonList("policies")))).thenReturn(Either.left(dataTransfer));
  75         UiComponentDataTransfer uiComponentDataTransfer = buildGetPolicyTypesCall().get(UiComponentDataTransfer.class);
  76         assertThat(uiComponentDataTransfer.getPolicies())
  77                 .usingElementComparatorOnFields("name", "uniqueId", "targets")
  78                 .containsExactlyInAnyOrder(policy1, policy2)
  79                 .extracting("properties")//properties is not returned in the response
  80                 .containsExactly(null, null);
  81     }
  82
  83     @Test
  84     public void filterDataByParam_getPolicies_policyTypeNameFieldShouldReturnAsType() {
  85         UiComponentDataTransfer dataTransfer = buildDataTransferWithPolicies();
  86         when(resourceBusinessLogic.getComponentDataFilteredByParams(eq(RESOURCE_ID.toLowerCase()), any(User.class), eq(Collections.singletonList("policies")))).thenReturn(Either.left(dataTransfer));
  87         Response uiComponentDataTransfer = buildGetPolicyTypesCall().get();
  88         verifyPolicyTypeFieldUsingJsonResponse(uiComponentDataTransfer);
  89     }
  90
  91     private void verifyPolicyTypeFieldUsingJsonResponse(Response uiComponentDataTransfer) {
  92         JSONObject json = new JSONObject(uiComponentDataTransfer.readEntity(String.class));
  93         JSONArray policies = json.getJSONArray("policies");
  94         for (int i = 0; i < policies.length(); i++) {
  95             JSONObject policy = policies.getJSONObject(i);
  96             String policyId = policy.get("uniqueId").toString();
  97             assertThat(policy.get("type")).isEqualTo("type" + policyId);
  98         }
  99     }
 100
 101     private UiComponentDataTransfer buildDataTransferWithPolicies() {
 102         UiComponentDataTransfer res = new UiComponentDataTransfer();
 103         res.setPolicies(asList(policy1, policy2));
 104         return res;
 105     }
 106
 107     private PolicyDefinition buildPolicy(String id) {
 108         return PolicyDefinitionBuilder.create()
 109                 .setUniqueId(id)
 110                 .setName("name" + id)
 111                 .setType("type" + id)
 112                 .addGroupTarget("group1")
 113                 .addGroupTarget("group2")
 114                 .addComponentInstanceTarget("inst1")
 115                 .addComponentInstanceTarget("inst2")
 116                 .addProperty("prop1")
 117                 .build();
 118     }
 119
 120     private Invocation.Builder buildGetPolicyTypesCall() {
 121         return target("/v1/catalog/resources/{id}/filteredDataByParams")
 122                 .queryParam("include", "policies")
 123                 .resolveTemplate("id", RESOURCE_ID)
 124                 .request(MediaType.APPLICATION_JSON)
 125                 .header(Constants.USER_ID_HEADER, USER_ID);
 126     }
 127 }