cps-service/src/test/java/org/onap/cps/TestUtils.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2020 Nordix Foundation
   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  *  Unless required by applicable law or agreed to in writing, software
  11  *  distributed under the License is distributed on an "AS IS" BASIS,
  12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *  See the License for the specific language governing permissions and
  14  *  limitations under the License.
  15  *
  16  *  SPDX-License-Identifier: Apache-2.0
  17  *  ============LICENSE_END=========================================================
  18  */
  19
  20 package org.onap.cps;
  21
  22 import com.google.common.collect.ImmutableMap;
  23 import java.io.File;
  24 import java.io.IOException;
  25 import java.nio.file.Files;
  26 import java.util.Map;
  27
  28 /**
  29  * Common convenience methods for testing.
  30  */
  31 public class TestUtils {
  32
  33     /**
  34      * Convert a file in the test resource folder to file.
  35      *
  36      * @param filename to name of the file in test/resources
  37      * @return the file
  38      * @throws IOException when there is an IO issue
  39      */
  40     public static File readFile(final String filename) {
  41         return new File(ClassLoader.getSystemClassLoader().getResource(filename).getFile());
  42     }
  43
  44     /**
  45      * Convert a file in the test resource folder to a string.
  46      *
  47      * @param filename to name of the file in test/resources
  48      * @return the content of the file as a String
  49      * @throws IOException when there is an IO issue
  50      */
  51     public static String getResourceFileContent(final String filename) throws IOException {
  52         final File file = readFile(filename);
  53         return new String(Files.readAllBytes(file.toPath()));
  54     }
  55
  56     /**
  57      * Reads yang resources into map.
  58      *
  59      * @param resources list of file paths
  60      * @return yang resource map where key is filename and value is file content
  61      * @throws IOException when there an I/O issue
  62      */
  63     public static Map<String, String> getYangResourcesAsMap(final String... resources) throws IOException {
  64         final ImmutableMap.Builder<String, String> yangResourceNameToContentBuilder = new ImmutableMap.Builder<>();
  65         for (final String resourcePath : resources) {
  66             final File file = readFile(resourcePath);
  67             final String content = new String(Files.readAllBytes(file.toPath()));
  68             yangResourceNameToContentBuilder.put(file.getName(), content);
  69         }
  70         return yangResourceNameToContentBuilder.build();
  71     }
  72 }