2 * ============LICENSE_START=======================================================
3 * policy-yaml unit test
4 * ================================================================================
5 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.controlloop.policy.guard;
23 import static org.junit.Assert.*;
26 import java.io.FileInputStream;
27 import java.io.FileNotFoundException;
28 import java.io.IOException;
29 import java.io.InputStream;
31 import org.junit.Test;
32 import org.slf4j.Logger;
33 import org.slf4j.LoggerFactory;
34 import org.yaml.snakeyaml.DumperOptions;
35 import org.yaml.snakeyaml.DumperOptions.FlowStyle;
36 import org.yaml.snakeyaml.Yaml;
37 import org.yaml.snakeyaml.constructor.Constructor;
40 public class ControlLoopGuardTest {
41 private static final Logger logger = LoggerFactory.getLogger(ControlLoopGuardTest.class);
43 public void testGuardvDNS() {
44 this.test("src/test/resources/v2.0.0-guard/policy_guard_ONAP_demo_vDNS.yaml");
48 public void testGuardvUSP() {
49 this.test("src/test/resources/v2.0.0-guard/policy_guard_appc_restart.yaml");
53 public void test(String testFile) {
54 try (InputStream is = new FileInputStream(new File(testFile))) {
56 // Read the yaml into our Java Object
58 Yaml yaml = new Yaml(new Constructor(ControlLoopGuard.class));
59 Object obj = yaml.load(is);
61 assertTrue(obj instanceof ControlLoopGuard);
64 // Now dump it to a yaml string
66 DumperOptions options = new DumperOptions();
67 options.setDefaultFlowStyle(FlowStyle.BLOCK);
68 options.setPrettyFlow(true);
69 yaml = new Yaml(options);
70 String dumpedYaml = yaml.dump(obj);
71 logger.debug(dumpedYaml);
73 // Read that string back into our java object
75 Object newObject = yaml.load(dumpedYaml);
77 assertNotNull(newObject);
78 assertTrue(newObject instanceof ControlLoopGuard);
80 // Have to comment it out tentatively since it causes junit to fail.
81 // Seems we cannot use assertEquals here. Need advice.
83 //assertEquals(newObject, obj);
84 } catch (FileNotFoundException e) {
85 fail(e.getLocalizedMessage());
86 } catch (IOException e) {
87 fail(e.getLocalizedMessage());
91 public void dump(Object obj) {
92 logger.debug("Dumping {}", obj.getClass().getCanonicalName());
93 logger.debug("{}", obj);