mod2/helm-generator/helmchartgenerator-core/src/test/java/org/onap/dcaegen2/platform/helmchartgenerator/KeyValueMergerTest.java

   1 /*
   2  * # ============LICENSE_START=======================================================
   3  * # Copyright (c) 2021 AT&T Intellectual Property. 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 org.onap.dcaegen2.platform.helmchartgenerator;
  20
  21 import org.apache.commons.io.FileUtils;
  22 import org.junit.jupiter.api.AfterEach;
  23 import org.junit.jupiter.api.BeforeEach;
  24 import org.junit.jupiter.api.Test;
  25 import org.junit.jupiter.api.extension.ExtendWith;
  26 import org.mockito.Mock;
  27 import org.mockito.Mockito;
  28 import org.mockito.junit.jupiter.MockitoExtension;
  29 import org.onap.dcaegen2.platform.helmchartgenerator.chartbuilder.KeyValueMerger;
  30 import org.onap.dcaegen2.platform.helmchartgenerator.models.chartinfo.ChartInfo;
  31 import org.onap.dcaegen2.platform.helmchartgenerator.models.chartinfo.Metadata;
  32 import org.yaml.snakeyaml.Yaml;
  33
  34 import java.io.File;
  35 import java.io.IOException;
  36 import java.io.InputStream;
  37 import java.io.PrintWriter;
  38 import java.nio.file.Files;
  39 import java.nio.file.Path;
  40 import java.util.HashMap;
  41 import java.util.Map;
  42
  43 import static org.mockito.ArgumentMatchers.any;
  44
  45 @ExtendWith(MockitoExtension.class)
  46 class KeyValueMergerTest {
  47
  48     private KeyValueMerger merger;
  49
  50     @Mock
  51     private Yaml yamlHelper;
  52
  53     private File chartDir;
  54
  55     @BeforeEach
  56     void setUp() {
  57         merger = new KeyValueMerger(yamlHelper);
  58     }
  59
  60     @Test
  61     void mergeValuesToChart() throws IOException {
  62         ChartInfo chartInfo = prepareChartInfo();
  63         chartDir = prepareChartDir();
  64
  65         Mockito.when(yamlHelper.load(any(InputStream.class))).thenReturn(new HashMap<String, Object>());
  66
  67         merger.mergeValuesToChart(chartInfo, chartDir);
  68         Mockito.verify(yamlHelper, Mockito.times(2)).dump(any(HashMap.class), any(PrintWriter.class));
  69     }
  70
  71     @AfterEach
  72     void tearDown(){
  73         FileUtils.deleteQuietly(chartDir);
  74     }
  75
  76     private File prepareChartDir() throws IOException {
  77         final Path chartDir = Files.createTempDirectory("chartDir");
  78         Files.createFile(chartDir.resolve("Chart.yaml"));
  79         Files.createFile(chartDir.resolve("values.yaml"));
  80         return chartDir.toFile();
  81     }
  82
  83     private ChartInfo prepareChartInfo() {
  84         ChartInfo chartInfo = new ChartInfo();
  85
  86         Metadata metadata = new Metadata();
  87         metadata.setName("someComponent");
  88         metadata.setVersion("someVersion");
  89         metadata.setDescription("someDescription");
  90
  91         Map<String, Object> values = new HashMap<>();
  92         values.put("someKey", "someValue");
  93
  94         chartInfo.setMetadata(metadata);
  95         chartInfo.setValues(values);
  96
  97         return chartInfo;
  98     }
  99 }
 100