catalog-model/src/test/java/org/openecomp/sdc/be/model/jsontitan/operations/GroupsOperationTest.java

   1 package org.openecomp.sdc.be.model.jsontitan.operations;
   2
   3 import static java.util.Arrays.asList;
   4 import static org.assertj.core.api.Assertions.assertThat;
   5
   6 import fj.data.Either;
   7 import java.util.List;
   8 import org.junit.After;
   9 import org.junit.Before;
  10 import org.junit.BeforeClass;
  11 import org.junit.Test;
  12 import org.junit.runner.RunWith;
  13 import org.openecomp.sdc.be.dao.config.TitanSpringConfig;
  14 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  15 import org.openecomp.sdc.be.dao.jsongraph.HealingTitanDao;
  16 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
  17 import org.openecomp.sdc.be.model.Component;
  18 import org.openecomp.sdc.be.model.ComponentParametersView;
  19 import org.openecomp.sdc.be.model.GroupDefinition;
  20 import org.openecomp.sdc.be.model.ModelTestBase;
  21 import org.openecomp.sdc.be.model.Resource;
  22 import org.openecomp.sdc.be.model.config.ModelOperationsSpringConfig;
  23 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  24 import org.springframework.beans.factory.annotation.Autowired;
  25 import org.springframework.test.context.ContextConfiguration;
  26 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  27
  28
  29 @RunWith(SpringJUnit4ClassRunner.class)
  30 @ContextConfiguration(classes = {TitanSpringConfig.class, ModelOperationsSpringConfig.class})
  31 public class GroupsOperationTest extends ModelTestBase {
  32
  33     @Autowired
  34     private GroupsOperation groupsOperation;
  35
  36     @Autowired
  37     HealingTitanDao titanDao;
  38
  39     @Autowired
  40     private ToscaOperationFacade toscaOperationFacade;
  41     private Component container;
  42
  43     @BeforeClass
  44     public static void initClass() {
  45         ModelTestBase.init();
  46     }
  47
  48     @Before
  49     public void setUp() throws Exception {
  50         container = new Resource();
  51         container.setUniqueId(CONTAINER_ID);
  52         Either<GraphVertex, TitanOperationStatus> createdCmpt = titanDao.createVertex(createBasicContainerGraphVertex());
  53         assertThat(createdCmpt.isLeft()).isTrue();
  54
  55     }
  56
  57     @After
  58     public void tearDown() throws Exception {
  59         titanDao.rollback();
  60     }
  61
  62     @Test
  63     public void addGroups_whenContainerHasNoGroups_associateContainerWithGroup() {
  64         GroupDefinition g1 = createGroupDefinition("g1");
  65         GroupDefinition g2 = createGroupDefinition("g2");
  66         Either<List<GroupDefinition>, StorageOperationStatus> createGroups = groupsOperation.addGroups(container, asList(g1, g2));
  67         assertThat(createGroups.isLeft()).isTrue();
  68
  69         ComponentParametersView getGroupsFilter = new ComponentParametersView(true);
  70         getGroupsFilter.setIgnoreGroups(false);
  71         Component cmptWithGroups = toscaOperationFacade.getToscaElement(CONTAINER_ID, getGroupsFilter).left().value();
  72         assertThat(cmptWithGroups.getGroups())
  73                 .usingElementComparatorOnFields("name", "uniqueId")
  74                 .containsExactlyInAnyOrder(g1, g2);
  75     }
  76
  77     @Test
  78     public void addGroups_whenContainerHasGroups_addTheGivenGroupsToTheGroupsList() {
  79         GroupDefinition g1 = createGroupDefinition("g1");
  80         GroupDefinition g2 = createGroupDefinition("g2");
  81         groupsOperation.addGroups(container, asList(g1, g2)).left().value();
  82
  83         GroupDefinition g3 = createGroupDefinition("g3");
  84         GroupDefinition g4 = createGroupDefinition("g4");
  85
  86         groupsOperation.addGroups(container, asList(g3, g4)).left().value();
  87
  88         ComponentParametersView getGroupsFilter = new ComponentParametersView(true);
  89         getGroupsFilter.setIgnoreGroups(false);
  90         Component cmptWithGroups = toscaOperationFacade.getToscaElement(CONTAINER_ID, getGroupsFilter).left().value();
  91         assertThat(cmptWithGroups.getGroups())
  92                 .usingElementComparatorOnFields("name", "uniqueId")
  93                 .containsExactlyInAnyOrder(g1, g2, g3, g4);
  94
  95     }
  96
  97     private GroupDefinition createGroupDefinition(String id) {
  98         GroupDefinition groupDefinition = new GroupDefinition();
  99         groupDefinition.setUniqueId(id);
 100         groupDefinition.setName("name" + id);
 101         return groupDefinition;
 102     }
 103
 104
 105 }