src/test/java/org/onap/clamp/loop/LoopRepositoriesItCase.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP CLAMP
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. All rights
   6  *                             reserved.
   7  * ================================================================================
   8  * Modifications Copyright (c) 2019 Samsung
   9  * ================================================================================
  10  *  Licensed under the Apache License, Version 2.0 (the "License");
  11  * you may not use this file except in compliance with the License.
  12  * You may obtain a copy of the License at
  13  *
  14  * http://www.apache.org/licenses/LICENSE-2.0
  15  *
  16  * Unless required by applicable law or agreed to in writing, software
  17  * distributed under the License is distributed on an "AS IS" BASIS,
  18  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  19  * See the License for the specific language governing permissions and
  20  * limitations under the License.
  21  * ============LICENSE_END============================================
  22  * ===================================================================
  23  *
  24  */
  25
  26 package org.onap.clamp.loop;
  27
  28 import static org.assertj.core.api.Assertions.assertThat;
  29
  30 import com.google.gson.Gson;
  31 import com.google.gson.GsonBuilder;
  32 import com.google.gson.JsonObject;
  33
  34 import java.time.Instant;
  35 import java.util.HashSet;
  36
  37 import org.junit.Test;
  38 import org.junit.runner.RunWith;
  39 import org.onap.clamp.clds.Application;
  40 import org.onap.clamp.loop.log.LogType;
  41 import org.onap.clamp.loop.log.LoopLog;
  42 import org.onap.clamp.loop.log.LoopLogRepository;
  43 import org.onap.clamp.loop.service.Service;
  44 import org.onap.clamp.loop.service.ServicesRepository;
  45 import org.onap.clamp.loop.template.LoopTemplate;
  46 import org.onap.clamp.loop.template.LoopTemplatesRepository;
  47 import org.onap.clamp.loop.template.MicroServiceModel;
  48 import org.onap.clamp.loop.template.MicroServiceModelsRepository;
  49 import org.onap.clamp.loop.template.PolicyModel;
  50 import org.onap.clamp.loop.template.PolicyModelId;
  51 import org.onap.clamp.loop.template.PolicyModelsRepository;
  52 import org.onap.clamp.policy.microservice.MicroServicePolicy;
  53 import org.onap.clamp.policy.microservice.MicroServicePolicyService;
  54 import org.onap.clamp.policy.operational.OperationalPolicy;
  55 import org.onap.clamp.policy.operational.OperationalPolicyService;
  56 import org.springframework.beans.factory.annotation.Autowired;
  57 import org.springframework.boot.test.context.SpringBootTest;
  58 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  59 import org.springframework.transaction.annotation.Transactional;
  60
  61 @RunWith(SpringJUnit4ClassRunner.class)
  62 @SpringBootTest(classes = Application.class)
  63 public class LoopRepositoriesItCase {
  64
  65     private Gson gson = new GsonBuilder().setPrettyPrinting().excludeFieldsWithoutExposeAnnotation().create();
  66
  67     @Autowired
  68     private LoopsRepository loopRepository;
  69
  70     @Autowired
  71     private MicroServicePolicyService microServicePolicyService;
  72
  73     @Autowired
  74     private OperationalPolicyService operationalPolicyService;
  75
  76     @Autowired
  77     private LoopLogRepository loopLogRepository;
  78
  79     @Autowired
  80     private LoopTemplatesRepository loopTemplateRepository;
  81
  82     @Autowired
  83     private MicroServiceModelsRepository microServiceModelsRepository;
  84
  85     @Autowired
  86     private PolicyModelsRepository policyModelsRepository;
  87
  88     @Autowired
  89     private ServicesRepository servicesRepository;
  90
  91     private Service getService(String serviceDetails, String resourceDetails) {
  92         return new Service(serviceDetails, resourceDetails);
  93     }
  94
  95     private OperationalPolicy getOperationalPolicy(String configJson, String name) {
  96         return new OperationalPolicy(name, null, new Gson().fromJson(configJson, JsonObject.class));
  97     }
  98
  99     private MicroServiceModel getMicroServiceModel(String yaml, String name, String policyType, String createdBy,
 100             PolicyModel policyModel) {
 101         MicroServiceModel model = new MicroServiceModel(name, policyType, yaml, policyModel);
 102         return model;
 103     }
 104
 105     private PolicyModel getPolicyModel(String policyType, String policyModelTosca, String version, String policyAcronym,
 106             String policyVariant, String createdBy) {
 107         return new PolicyModel(policyType, policyModelTosca, version, policyAcronym, policyVariant);
 108     }
 109
 110     private LoopTemplate getLoopTemplate(String name, String blueprint, String svgRepresentation, String createdBy,
 111             Integer maxInstancesAllowed) {
 112         LoopTemplate template = new LoopTemplate(name, blueprint, svgRepresentation, maxInstancesAllowed, null);
 113         template.addMicroServiceModel(getMicroServiceModel("yaml", "microService1", "org.onap.policy.drools", createdBy,
 114                 getPolicyModel("org.onap.policy.drools", "yaml", "1.0.0", "Drools", "type1", createdBy)));
 115         return template;
 116     }
 117
 118     private Loop getLoop(String name, String svgRepresentation, String blueprint, String globalPropertiesJson,
 119             String dcaeId, String dcaeUrl, String dcaeBlueprintId) {
 120         Loop loop = new Loop();
 121         loop.setName(name);
 122         loop.setSvgRepresentation(svgRepresentation);
 123         loop.setBlueprint(blueprint);
 124         loop.setGlobalPropertiesJson(new Gson().fromJson(globalPropertiesJson, JsonObject.class));
 125         loop.setLastComputedState(LoopState.DESIGN);
 126         loop.setDcaeDeploymentId(dcaeId);
 127         loop.setDcaeDeploymentStatusUrl(dcaeUrl);
 128         loop.setDcaeBlueprintId(dcaeBlueprintId);
 129         loop.setLoopTemplate(getLoopTemplate("templateName", "yaml", "svg", "toto", 1));
 130         return loop;
 131     }
 132
 133     private MicroServicePolicy getMicroServicePolicy(String name, String modelType, String jsonRepresentation,
 134             String policyTosca, String jsonProperties, boolean shared) {
 135         MicroServicePolicy microService = new MicroServicePolicy(name, modelType, policyTosca, shared,
 136                 gson.fromJson(jsonRepresentation, JsonObject.class), new HashSet<>());
 137         microService.setProperties(new Gson().fromJson(jsonProperties, JsonObject.class));
 138         return microService;
 139     }
 140
 141     private LoopLog getLoopLog(LogType type, String message, Loop loop) {
 142         return new LoopLog(message, type, "CLAMP", loop);
 143     }
 144
 145     @Test
 146     @Transactional
 147     public void crudTest() {
 148         // Setup
 149         Loop loopTest = getLoop("ControlLoopTest", "<xml></xml>", "yamlcontent", "{\"testname\":\"testvalue\"}",
 150                 "123456789", "https://dcaetest.org", "UUID-blueprint");
 151         OperationalPolicy opPolicy = this.getOperationalPolicy("{\"type\":\"GUARD\"}", "GuardOpPolicyTest");
 152         loopTest.addOperationalPolicy(opPolicy);
 153         MicroServicePolicy microServicePolicy = getMicroServicePolicy("configPolicyTest", "",
 154                 "{\"configtype\":\"json\"}", "tosca_definitions_version: tosca_simple_yaml_1_0_0",
 155                 "{\"param1\":\"value1\"}", true);
 156         loopTest.addMicroServicePolicy(microServicePolicy);
 157         LoopLog loopLog = getLoopLog(LogType.INFO, "test message", loopTest);
 158         loopTest.addLog(loopLog);
 159         Service service = getService(
 160                 "{\"name\": \"vLoadBalancerMS\", \"UUID\": \"63cac700-ab9a-4115-a74f-7eac85e3fce0\"}", "{\"CP\": {}}");
 161         loopTest.setModelService(service);
 162
 163         // Attempt to save into the database the entire loop
 164         Loop loopInDb = loopRepository.save(loopTest);
 165         assertThat(loopInDb).isNotNull();
 166         assertThat(loopRepository.findById(loopInDb.getName()).get()).isNotNull();
 167         assertThat(loopInDb.getCreatedDate()).isNotNull();
 168         assertThat(loopInDb.getUpdatedDate()).isNotNull();
 169         assertThat(loopInDb.getUpdatedDate()).isEqualTo(loopInDb.getCreatedDate());
 170         assertThat(loopInDb.getName()).isEqualTo("ControlLoopTest");
 171         // Autogen id so now set the ID in the previous model so that we can compare the
 172         // objects
 173         loopLog.setId(((LoopLog) loopInDb.getLoopLogs().toArray()[0]).getId());
 174
 175         assertThat(loopInDb).isEqualToIgnoringGivenFields(loopTest, "components", "createdDate", "updatedDate",
 176                 "createdBy", "updatedBy");
 177         assertThat(loopRepository.existsById(loopTest.getName())).isEqualTo(true);
 178         assertThat(operationalPolicyService.isExisting(opPolicy.getName())).isEqualTo(true);
 179         assertThat(microServicePolicyService.isExisting(microServicePolicy.getName())).isEqualTo(true);
 180         assertThat(loopLogRepository.existsById(loopLog.getId())).isEqualTo(true);
 181         assertThat(loopTemplateRepository.existsById(loopInDb.getLoopTemplate().getName())).isEqualTo(true);
 182         assertThat(loopTemplateRepository.existsById(loopInDb.getLoopTemplate().getName())).isEqualTo(true);
 183         assertThat(servicesRepository.existsById(loopInDb.getModelService().getServiceUuid())).isEqualTo(true);
 184         assertThat(microServiceModelsRepository.existsById(
 185                 loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getName()))
 186                         .isEqualTo(true);
 187         assertThat(policyModelsRepository.existsById(new PolicyModelId(
 188                 loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getPolicyModel()
 189                         .getPolicyModelType(),
 190                 loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getPolicyModel()
 191                         .getVersion()))).isEqualTo(true);
 192
 193         // Now attempt to read from database
 194         Loop loopInDbRetrieved = loopRepository.findById(loopTest.getName()).get();
 195         assertThat(loopInDbRetrieved).isEqualToIgnoringGivenFields(loopTest, "components", "createdDate", "updatedDate",
 196                 "createdBy", "updatedBy");
 197         assertThat(loopInDbRetrieved).isEqualToComparingOnlyGivenFields(loopInDb, "createdDate", "updatedDate",
 198                 "createdBy", "updatedBy");
 199         assertThat((LoopLog) loopInDbRetrieved.getLoopLogs().toArray()[0]).isEqualToComparingFieldByField(loopLog);
 200         assertThat((OperationalPolicy) loopInDbRetrieved.getOperationalPolicies().toArray()[0])
 201                 .isEqualToComparingFieldByField(opPolicy);
 202         assertThat((MicroServicePolicy) loopInDbRetrieved.getMicroServicePolicies().toArray()[0])
 203                 .isEqualToIgnoringGivenFields(microServicePolicy, "createdDate", "updatedDate", "createdBy",
 204                         "updatedBy");
 205
 206         // Attempt an update
 207         ((LoopLog) loopInDbRetrieved.getLoopLogs().toArray()[0]).setLogInstant(Instant.now());
 208         loopInDbRetrieved.setBlueprint("yaml2");
 209         Loop loopInDbRetrievedUpdated = loopRepository.saveAndFlush(loopInDbRetrieved);
 210         // Loop loopInDbRetrievedUpdated =
 211         // loopRepository.findById(loopTest.getName()).get();
 212         assertThat(loopInDbRetrievedUpdated.getBlueprint()).isEqualTo("yaml2");
 213         assertThat((LoopLog) loopInDbRetrievedUpdated.getLoopLogs().toArray()[0])
 214                 .isEqualToComparingFieldByField(loopInDbRetrieved.getLoopLogs().toArray()[0]);
 215         // UpdatedDate should have been changed
 216         assertThat(loopInDbRetrievedUpdated.getUpdatedDate()).isNotEqualTo(loopInDbRetrievedUpdated.getCreatedDate());
 217         // createdDate should have NOT been changed
 218         assertThat(loopInDbRetrievedUpdated.getCreatedDate()).isEqualTo(loopInDb.getCreatedDate());
 219         // other audit are the same
 220         assertThat(loopInDbRetrievedUpdated.getCreatedBy()).isEqualTo("");
 221         assertThat(loopInDbRetrievedUpdated.getUpdatedBy()).isEqualTo("");
 222
 223         // Attempt to delete the object and check it has well been cascaded
 224
 225         loopRepository.delete(loopInDbRetrieved);
 226         assertThat(loopRepository.existsById(loopTest.getName())).isEqualTo(false);
 227         assertThat(operationalPolicyService.isExisting(opPolicy.getName())).isEqualTo(false);
 228         assertThat(microServicePolicyService.isExisting(microServicePolicy.getName())).isEqualTo(true);
 229         assertThat(loopLogRepository.existsById(loopLog.getId())).isEqualTo(false);
 230         assertThat(loopTemplateRepository.existsById(loopInDb.getLoopTemplate().getName())).isEqualTo(true);
 231         assertThat(servicesRepository.existsById(loopInDb.getModelService().getServiceUuid())).isEqualTo(true);
 232         assertThat(microServiceModelsRepository.existsById(
 233                 loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getName()))
 234                         .isEqualTo(true);
 235
 236         assertThat(policyModelsRepository.existsById(new PolicyModelId(
 237                 loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getPolicyModel()
 238                         .getPolicyModelType(),
 239                 loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getPolicyModel()
 240                         .getVersion()))).isEqualTo(true);
 241
 242         // Cleanup
 243         // microServiceModelsRepository
 244         // .delete(loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel());
 245         //
 246         // policyModelsRepository.delete(
 247         // loopInDb.getLoopTemplate().getMicroServiceModelUsed().first().getMicroServiceModel().getPolicyModel());
 248         // loopTemplateRepository.delete(loopInDb.getLoopTemplate());
 249         // servicesRepository.delete(service);
 250
 251     }
 252 }