rapp-manager-dme/src/test/java/com/oransc/rappmanager/dme/service/DmeAcmInterceptorTest.java

   1 /*-
   2  * ============LICENSE_START======================================================================
   3  * Copyright (C) 2023-2024 OpenInfra Foundation Europe. All rights reserved.
   4  * ===============================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  * ============LICENSE_END========================================================================
  17  */
  18
  19 package com.oransc.rappmanager.dme.service;
  20
  21 import static org.junit.jupiter.api.Assertions.assertEquals;
  22
  23 import com.fasterxml.jackson.databind.ObjectMapper;
  24 import com.google.gson.JsonArray;
  25 import com.google.gson.JsonParser;
  26 import com.oransc.rappmanager.models.csar.RappCsarConfigurationHandler;
  27 import com.oransc.rappmanager.models.rapp.Rapp;
  28 import com.oransc.rappmanager.models.rapp.RappDmeResourceBuilder;
  29 import com.oransc.rappmanager.models.rapp.RappState;
  30 import com.oransc.rappmanager.models.rappinstance.RappInstance;
  31 import java.util.HashMap;
  32 import java.util.HashSet;
  33 import java.util.Map;
  34 import java.util.Set;
  35 import java.util.UUID;
  36 import java.util.stream.Stream;
  37 import org.junit.jupiter.api.Test;
  38 import org.junit.jupiter.params.ParameterizedTest;
  39 import org.junit.jupiter.params.provider.Arguments;
  40 import org.junit.jupiter.params.provider.MethodSource;
  41 import org.onap.policy.clamp.models.acm.concepts.AutomationComposition;
  42 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionElement;
  43 import org.onap.policy.models.tosca.authorative.concepts.ToscaDataType;
  44 import org.onap.policy.models.tosca.authorative.concepts.ToscaNodeTemplate;
  45 import org.onap.policy.models.tosca.authorative.concepts.ToscaNodeType;
  46 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
  47 import org.onap.policy.models.tosca.authorative.concepts.ToscaTopologyTemplate;
  48 import org.springframework.beans.factory.annotation.Autowired;
  49 import org.springframework.boot.test.context.SpringBootTest;
  50
  51 @SpringBootTest(classes = {ObjectMapper.class, RappCsarConfigurationHandler.class, DmeAcmInterceptor.class})
  52 class DmeAcmInterceptorTest {
  53
  54     @Autowired
  55     DmeAcmInterceptor dmeAcmInterceptor;
  56
  57     RappDmeResourceBuilder rappDmeResourceBuilder = new RappDmeResourceBuilder();
  58     private static final String validRappFile = "valid-rapp-package.csar";
  59     String validCsarFileLocation = "src/test/resources/";
  60
  61     @ParameterizedTest
  62     @MethodSource("getrAppInstances")
  63     void testInjectAutomationComposition(RappInstance rAppInstance) {
  64         AutomationComposition automationComposition = new AutomationComposition();
  65         Map<UUID, AutomationCompositionElement> elements =
  66                 new HashMap<>(Map.of(UUID.randomUUID(), new AutomationCompositionElement()));
  67         automationComposition.setElements(elements);
  68         assertEquals(1, automationComposition.getElements().size());
  69         Rapp rApp = Rapp.builder().rappId(UUID.randomUUID()).name("").packageName(validRappFile)
  70                             .packageLocation(validCsarFileLocation).state(RappState.COMMISSIONED)
  71                             .rappResources(rappDmeResourceBuilder.getResources()).build();
  72         dmeAcmInterceptor.injectAutomationComposition(automationComposition, rApp, rAppInstance);
  73         assertEquals(2, automationComposition.getElements().size());
  74     }
  75
  76     @Test
  77     void testInjectToscaServiceTemplate() {
  78         ToscaServiceTemplate toscaServiceTemplate = new ToscaServiceTemplate();
  79         toscaServiceTemplate.setDataTypes(new HashMap<>(Map.of("datatype1", new ToscaDataType())));
  80         toscaServiceTemplate.setNodeTypes(new HashMap<>(Map.of("nodetype1", new ToscaNodeType())));
  81         ToscaTopologyTemplate toscaTopologyTemplate = new ToscaTopologyTemplate();
  82         ToscaNodeTemplate toscaNodeTemplate = new ToscaNodeTemplate();
  83         String elements = "elements";
  84         toscaNodeTemplate.setProperties(new HashMap<>(Map.of(elements, "[{}]")));
  85         String nodeTemplateKey = "onap.policy.clamp.ac.element.AutomationCompositionDefinition";
  86
  87         toscaTopologyTemplate.setNodeTemplates(new HashMap<>(Map.of(nodeTemplateKey, toscaNodeTemplate)));
  88         toscaServiceTemplate.setToscaTopologyTemplate(toscaTopologyTemplate);
  89         Object o =
  90                 toscaServiceTemplate.getToscaTopologyTemplate().getNodeTemplates().get(nodeTemplateKey).getProperties()
  91                         .get(elements);
  92         JsonArray elementArray = JsonParser.parseString(o.toString()).getAsJsonArray();
  93         assertEquals(1, toscaServiceTemplate.getDataTypes().size());
  94         assertEquals(1, toscaServiceTemplate.getNodeTypes().size());
  95         assertEquals(1, elementArray.size());
  96         dmeAcmInterceptor.injectToscaServiceTemplate(toscaServiceTemplate);
  97         Object modObject =
  98                 toscaServiceTemplate.getToscaTopologyTemplate().getNodeTemplates().get(nodeTemplateKey).getProperties()
  99                         .get(elements);
 100         JsonArray newElementArray = JsonParser.parseString(modObject.toString()).getAsJsonArray();
 101         assertEquals(4, toscaServiceTemplate.getDataTypes().size());
 102         assertEquals(2, toscaServiceTemplate.getNodeTypes().size());
 103         assertEquals(2, newElementArray.size());
 104     }
 105
 106     private static Stream<Arguments> getrAppInstances() {
 107         RappDmeResourceBuilder rappDmeResourceBuilder = new RappDmeResourceBuilder();
 108         RappInstance rappInstanceProducerEmpty = rappDmeResourceBuilder.getRappInstance();
 109         RappInstance rappInstanceConsumerEmpty = rappDmeResourceBuilder.getRappInstance();
 110         RappInstance rappInstanceSameInfoType = rappDmeResourceBuilder.getRappInstance();
 111         rappInstanceProducerEmpty.getDme().setInfoTypesProducer(null);
 112         rappInstanceProducerEmpty.getDme().setInfoProducer(null);
 113         rappInstanceConsumerEmpty.getDme().setInfoTypeConsumer(null);
 114         rappInstanceConsumerEmpty.getDme().setInfoConsumer(null);
 115         Set<String> infoTypesProducer = new HashSet<>(rappInstanceSameInfoType.getDme().getInfoTypesProducer());
 116         infoTypesProducer.remove(rappInstanceSameInfoType.getDme().getInfoConsumer());
 117         rappInstanceSameInfoType.getDme().setInfoTypesProducer(infoTypesProducer);
 118         return Stream.of(Arguments.of(rappDmeResourceBuilder.getRappInstance()),
 119                 Arguments.of(rappInstanceProducerEmpty), Arguments.of(rappInstanceConsumerEmpty),
 120                 Arguments.of(rappInstanceSameInfoType));
 121     }
 122 }