catalog-be/src/test/java/org/openecomp/sdc/be/datamodel/UiComponentDataConverterTest.java

   1 package org.openecomp.sdc.be.datamodel;
   2
   3 import org.junit.Before;
   4 import org.junit.Test;
   5 import org.openecomp.sdc.be.components.utils.PolicyDefinitionBuilder;
   6 import org.openecomp.sdc.be.components.utils.ResourceBuilder;
   7 import org.openecomp.sdc.be.components.utils.ServiceBuilder;
   8 import org.openecomp.sdc.be.datamodel.utils.UiComponentDataConverter;
   9 import org.openecomp.sdc.be.model.PolicyDefinition;
  10 import org.openecomp.sdc.be.model.Resource;
  11 import org.openecomp.sdc.be.model.Service;
  12 import org.openecomp.sdc.be.ui.model.UiComponentDataTransfer;
  13
  14 import java.util.Collections;
  15
  16 import static org.assertj.core.api.Assertions.assertThat;
  17
  18 public class UiComponentDataConverterTest {
  19
  20     private PolicyDefinition policy1, policy2;
  21
  22     @Before
  23     public void setUp() throws Exception {
  24         policy1 = PolicyDefinitionBuilder.create()
  25                 .setName("policy1")
  26                 .setUniqueId("uid1")
  27                 .build();
  28
  29         policy2 = PolicyDefinitionBuilder.create()
  30                 .setName("policy2")
  31                 .setUniqueId("uid2")
  32                 .build();
  33     }
  34
  35     @Test
  36     public void getUiDataTransferFromResourceByParams_policies_noPoliciesForResource() {
  37         UiComponentDataTransfer componentDTO = UiComponentDataConverter.getUiDataTransferFromResourceByParams(new Resource(), Collections.singletonList("policies"));
  38         assertThat(componentDTO.getPolicies()).isEmpty();
  39     }
  40
  41     @Test
  42     public void getUiDataTransferFromServiceByParams_policies_noPoliciesForResource() {
  43         UiComponentDataTransfer componentDTO = UiComponentDataConverter.getUiDataTransferFromServiceByParams(new Service(), Collections.singletonList("policies"));
  44         assertThat(componentDTO.getPolicies()).isEmpty();
  45     }
  46
  47     @Test
  48     public void getUiDataTransferFromResourceByParams_policies() {
  49         Resource resourceWithPolicies = buildResourceWithPolicies();
  50         UiComponentDataTransfer componentDTO = UiComponentDataConverter.getUiDataTransferFromResourceByParams(resourceWithPolicies, Collections.singletonList("policies"));
  51         assertThat(componentDTO.getPolicies()).isEqualTo(resourceWithPolicies.resolvePoliciesList());
  52     }
  53
  54     @Test
  55     public void getUiDataTransferFromServiceByParams_policies() {
  56         Service resourceWithPolicies = buildServiceWithPolicies();
  57         UiComponentDataTransfer componentDTO = UiComponentDataConverter.getUiDataTransferFromServiceByParams(resourceWithPolicies, Collections.singletonList("policies"));
  58         assertThat(componentDTO.getPolicies()).isEqualTo(resourceWithPolicies.resolvePoliciesList());
  59     }
  60
  61     private Resource buildResourceWithPolicies() {
  62         return new ResourceBuilder()
  63                 .addPolicy(policy1)
  64                 .addPolicy(policy2)
  65                 .build();
  66     }
  67
  68     private Service buildServiceWithPolicies() {
  69         return new ServiceBuilder()
  70                 .addPolicy(policy1)
  71                 .addPolicy(policy2)
  72                 .build();
  73     }
  74 }