gerrit.o-ran-sc Code Review - smo/teiv.git/blob

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2024 Ericsson
   4  *  Modifications Copyright (C) 2024 OpenInfra Foundation Europe
   5  *  ================================================================================
   6  *  Licensed under the Apache License, Version 2.0 (the "License");
   7  *  you may not use this file except in compliance with the License.
   8  *  You may obtain a copy of the License at
   9  *
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *
  18  *  SPDX-License-Identifier: Apache-2.0
  19  *  ============LICENSE_END=========================================================
  20  */
  21 package org.oran.smo.yangtools.parser.input.test;
  22
  23 import static org.junit.Assert.assertTrue;
  24
  25 import java.util.ArrayList;
  26 import java.util.List;
  27
  28 import org.junit.Test;
  29
  30 import org.oran.smo.yangtools.parser.ParserExecutionContext;
  31 import org.oran.smo.yangtools.parser.YangDeviceModel;
  32 import org.oran.smo.yangtools.parser.findings.FindingsManager;
  33 import org.oran.smo.yangtools.parser.findings.ModifyableFindingSeverityCalculator;
  34 import org.oran.smo.yangtools.parser.input.ByteArrayYangInput;
  35 import org.oran.smo.yangtools.parser.model.ConformanceType;
  36 import org.oran.smo.yangtools.parser.model.YangModel;
  37 import org.oran.smo.yangtools.parser.model.statements.yang.YContainer;
  38 import org.oran.smo.yangtools.parser.model.statements.yang.YLeaf;
  39 import org.oran.smo.yangtools.parser.model.statements.yang.YModule;
  40
  41 public class ByteArrayYangInputTest {
  42
  43     private YangDeviceModel yangDeviceModel;
  44     private ModifyableFindingSeverityCalculator severityCalculator;
  45     private FindingsManager findingsManager;
  46     private ParserExecutionContext context;
  47
  48     @Test
  49     public void test_byte_array_input() {
  50
  51         final String moduleContent = getModuleContent();
  52         final byte[] bytes = moduleContent.getBytes();
  53         final ByteArrayYangInput byteArrayYangInput = new ByteArrayYangInput(bytes, "test");
  54
  55         // First run
  56
  57         yangDeviceModel = new YangDeviceModel("Yang Parser JAR Test Device Model");
  58         severityCalculator = new ModifyableFindingSeverityCalculator();
  59         findingsManager = new FindingsManager(severityCalculator);
  60         context = new ParserExecutionContext(findingsManager);
  61
  62         List<YangModel> yangModelInputs = new ArrayList<>();
  63         yangModelInputs.add(new YangModel(byteArrayYangInput, ConformanceType.IMPLEMENT));
  64         yangDeviceModel.parseIntoYangModels(context, yangModelInputs);
  65
  66         YModule module = yangModelInputs.get(0).getYangModelRoot().getModule();
  67         assertTrue(module.getModuleName().equals("byte-array-test-module"));
  68
  69         YContainer cont1 = module.getContainers().get(0);
  70         assertTrue(cont1.getContainerName().equals("cont1"));
  71
  72         YLeaf leaf1 = cont1.getLeafs().get(0);
  73         assertTrue(leaf1.getLeafName().equals("leaf1"));
  74
  75         // Set up the whole lot again for second run, except for the ByteArrayYangInput, to test repeated parsing.
  76
  77         yangDeviceModel = new YangDeviceModel("Yang Parser JAR Test Device Model");
  78         severityCalculator = new ModifyableFindingSeverityCalculator();
  79         findingsManager = new FindingsManager(severityCalculator);
  80         context = new ParserExecutionContext(findingsManager);
  81
  82         yangModelInputs = new ArrayList<>();
  83         yangModelInputs.add(new YangModel(byteArrayYangInput, ConformanceType.IMPLEMENT));
  84         yangDeviceModel.parseIntoYangModels(context, yangModelInputs);
  85
  86         module = yangModelInputs.get(0).getYangModelRoot().getModule();
  87         assertTrue(module.getModuleName().equals("byte-array-test-module"));
  88
  89         cont1 = module.getContainers().get(0);
  90         assertTrue(cont1.getContainerName().equals("cont1"));
  91
  92         cont1.getLeafs().get(0);
  93         assertTrue(leaf1.getLeafName().equals("leaf1"));
  94     }
  95
  96     @SuppressWarnings("unlikely-arg-type")
  97     @Test
  98     public void test_byte_array_input_equals() {
  99
 100         final byte[] bytes1 = new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
 101         final byte[] bytes2 = new byte[] { 9, 8, 7, 6 };
 102
 103         final ByteArrayYangInput input1 = new ByteArrayYangInput(bytes1, "test");
 104         final ByteArrayYangInput input2 = new ByteArrayYangInput(bytes1, "test");
 105         final ByteArrayYangInput input3 = new ByteArrayYangInput(bytes1, "other");
 106         final ByteArrayYangInput input4 = new ByteArrayYangInput(bytes2, "test");
 107
 108         assertTrue(input1.equals(input2) == true);
 109         assertTrue(input1.equals(null) == false);
 110         assertTrue(input1.equals("") == false);
 111         assertTrue(input1.equals(input3) == false);
 112         assertTrue(input1.equals(input4) == false);
 113
 114         assertTrue(input1.getName().equals("test"));
 115         assertTrue(input1.getFile() == null);
 116     }
 117
 118     private static String getModuleContent() {
 119
 120         final StringBuilder sb = new StringBuilder(1000);
 121
 122         sb.append("module byte-array-test-module {\n");
 123         sb.append("\n");
 124         sb.append("    namespace \"test:byte-array-test-module\";\n");
 125         sb.append("    prefix \"this\";\n");
 126         sb.append("\n");
 127         sb.append("    revision \"2021-04-06\" {\n");
 128         sb.append("        description \"initial revision\";\n");
 129         sb.append("    }\n");
 130         sb.append("\n");
 131         sb.append("container cont1 {\n");
 132         sb.append("\n");
 133         sb.append("             leaf leaf1 {\n");
 134         sb.append("                     type int8;\n");
 135         sb.append("             }\n");
 136         sb.append("     }n");
 137         sb.append("\n");
 138         sb.append("}\n");
 139
 140         return sb.toString();
 141     }
 142 }