gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.be.model.jsontitan.operations;
   2
   3 import static org.assertj.core.api.Assertions.assertThat;
   4 import static org.mockito.ArgumentMatchers.eq;
   5 import static org.mockito.Mockito.verifyZeroInteractions;
   6 import static org.mockito.Mockito.when;
   7
   8 import java.util.Arrays;
   9 import java.util.Collections;
  10
  11 import org.junit.Before;
  12 import org.junit.Test;
  13 import org.junit.runner.RunWith;
  14 import org.mockito.ArgumentCaptor;
  15 import org.mockito.Captor;
  16 import org.mockito.InjectMocks;
  17 import org.mockito.Mock;
  18 import org.mockito.junit.MockitoJUnitRunner;
  19 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  20 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
  21 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
  22 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
  23 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
  24 import org.openecomp.sdc.be.model.Component;
  25 import org.openecomp.sdc.be.model.PolicyDefinition;
  26 import org.openecomp.sdc.be.model.Resource;
  27 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  28
  29 import fj.data.Either;
  30
  31 @RunWith(MockitoJUnitRunner.class)
  32 public class PolicyOperationTest {
  33
  34     private static final String CONTAINER_ID = "containerId";
  35     private static final String POLICY_ID = "policyId";
  36     @InjectMocks
  37     private PolicyOperation testInstance;
  38     @Mock
  39     private TitanDao titanDao;
  40     @Mock
  41     private TopologyTemplateOperation topologyTemplateOperation;
  42     @Captor
  43     private ArgumentCaptor<PolicyDefinition> policyDefCaptor;
  44     private Component component;
  45     private PolicyDefinition policyDefinition;
  46     private PropertyDataDefinition prop1, prop2;
  47
  48     @Before
  49     public void setUp() throws Exception {
  50         component = new Resource();
  51         component.setUniqueId(CONTAINER_ID);
  52         policyDefinition = new PolicyDefinition();
  53         policyDefinition.setUniqueId(POLICY_ID);
  54         prop1 = new PropertyDataDefinition();
  55         prop1.setName("prop1");
  56         prop1.setValue("prop1");
  57
  58         prop2 = new PropertyDataDefinition();
  59         prop2.setName("prop2");
  60         prop2.setValue("prop2");
  61         policyDefinition.setProperties(Arrays.asList(prop1, prop2));
  62         component.setPolicies(Collections.singletonMap(POLICY_ID, policyDefinition));
  63     }
  64
  65     @Test
  66     public void updatePolicyProperties_failedToFetchContainer() {
  67         when(titanDao.getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse)).thenReturn(Either.right(TitanOperationStatus.NOT_FOUND));
  68         StorageOperationStatus storageOperationStatus = testInstance.updatePolicyProperties(component, POLICY_ID, Collections.emptyList());
  69         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.NOT_FOUND);
  70         verifyZeroInteractions(topologyTemplateOperation);
  71     }
  72
  73     @Test
  74     public void updatePolicyProperties_updateFailed() {
  75         GraphVertex cmptVertex = new GraphVertex();
  76         when(titanDao.getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse)).thenReturn(Either.left(cmptVertex));
  77         when(topologyTemplateOperation.updatePolicyOfToscaElement(cmptVertex, policyDefinition)).thenReturn(StorageOperationStatus.GENERAL_ERROR);
  78         StorageOperationStatus storageOperationStatus = testInstance.updatePolicyProperties(component, POLICY_ID, Collections.emptyList());
  79         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.GENERAL_ERROR);
  80     }
  81
  82     @Test
  83     public void updatePolicyProperties() {
  84         GraphVertex cmptVertex = new GraphVertex();
  85         when(titanDao.getVertexById(CONTAINER_ID, JsonParseFlagEnum.NoParse)).thenReturn(Either.left(cmptVertex));
  86         when(topologyTemplateOperation.updatePolicyOfToscaElement(eq(cmptVertex), policyDefCaptor.capture())).thenReturn(StorageOperationStatus.OK);
  87         PropertyDataDefinition prop1Copy = new PropertyDataDefinition(prop1);
  88         prop1Copy.setValue("prop1Copy");
  89         StorageOperationStatus storageOperationStatus = testInstance.updatePolicyProperties(component, POLICY_ID, Collections.singletonList(prop1Copy));
  90         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.OK);
  91         assertThat(policyDefCaptor.getValue().getProperties()).usingElementComparatorOnFields("value")
  92                 .containsExactlyInAnyOrder(prop1Copy, prop2);
  93     }
  94 }