gerrit.onap Code Review - policy/distribution.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2019 Intel Corp. All rights reserved.
   4  *  Modifications Copyright (C) 2019-2020 AT&T Intellectual Property.
   5  *  Modifications Copyright (C) 2019 Nordix Foundation.
   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
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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  *
  19  * SPDX-License-Identifier: Apache-2.0
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.policy.distribution.forwarding.file;
  24
  25 import static org.junit.Assert.assertTrue;
  26 import static org.junit.Assert.fail;
  27
  28 import java.io.IOException;
  29 import java.nio.file.Files;
  30 import java.nio.file.Path;
  31 import java.nio.file.Paths;
  32 import java.util.ArrayList;
  33 import java.util.Collection;
  34
  35 import org.junit.AfterClass;
  36 import org.junit.BeforeClass;
  37 import org.junit.ClassRule;
  38 import org.junit.Test;
  39 import org.junit.rules.TemporaryFolder;
  40 import org.junit.runner.RunWith;
  41 import org.mockito.Mockito;
  42 import org.mockito.junit.MockitoJUnitRunner;
  43 import org.onap.policy.common.parameters.ParameterService;
  44 import org.onap.policy.models.tosca.authorative.concepts.ToscaEntity;
  45 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
  46
  47
  48 /**
  49  * Class to perform unit test of {@link FilePolicyForwarder}.
  50  *
  51  */
  52 @RunWith(MockitoJUnitRunner.class)
  53 public class FilePolicyForwarderTest {
  54     @ClassRule
  55     public static TemporaryFolder tempFolder = new TemporaryFolder();
  56
  57     private static final boolean VERBOSE = true;
  58     private static final String GROUP_NAME = "fileConfiguration";
  59
  60     /**
  61      * Set up.
  62      */
  63     @BeforeClass
  64     public static void setUp() {
  65         final FilePolicyForwarderParameterGroup configurationParameters = new FilePolicyForwarderParameterGroup();
  66         configurationParameters.setPath(tempFolder.getRoot().getAbsolutePath().toString());
  67         configurationParameters.setVerbose(VERBOSE);
  68         configurationParameters.setName(GROUP_NAME);
  69         ParameterService.register(configurationParameters);
  70     }
  71
  72     /**
  73      * Tear down.
  74      */
  75     @AfterClass
  76     public static void tearDown() {
  77         ParameterService.deregister(GROUP_NAME);
  78     }
  79
  80     @Test
  81     public void testForwardPolicy() {
  82         final Collection<ToscaEntity> policies = new ArrayList<>();
  83         final ToscaPolicy policy = createPolicy(policies, "test", "test");
  84
  85         final FilePolicyForwarder forwarder = new FilePolicyForwarder();
  86         forwarder.configure(GROUP_NAME);
  87
  88         try {
  89             forwarder.forward(policies);
  90             final Path path = Paths.get(tempFolder.getRoot().getAbsolutePath().toString(), policy.getName());
  91             assertTrue(Files.exists(path));
  92         } catch (final Exception exp) {
  93             fail("Test must not throw an exception");
  94         }
  95     }
  96
  97     @Test
  98     public void testForwardPolicyError() {
  99         final Collection<ToscaEntity> policies = new ArrayList<>();
 100         final ToscaPolicy policy = createPolicy(policies, "test", "test");
 101
 102         final ToscaPolicy spy = Mockito.spy(policy);
 103         Mockito.when(spy.toString()).thenAnswer(invocation -> {
 104             throw new IOException();
 105         });
 106         policies.add(spy);
 107
 108         final FilePolicyForwarder forwarder = new FilePolicyForwarder();
 109         forwarder.configure(GROUP_NAME);
 110
 111         try {
 112             forwarder.forward(policies);
 113             fail("Test must throw an exception");
 114         } catch (final Exception exp) {
 115             assertTrue(exp.getMessage().contains("Error sending policy"));
 116         }
 117     }
 118
 119     @Test
 120     public void testForwardUnsupportedPolicy() {
 121         final Collection<ToscaEntity> policies = new ArrayList<>();
 122         final FilePolicyForwarder forwarder = new FilePolicyForwarder();
 123         forwarder.configure(GROUP_NAME);
 124
 125         final ToscaEntity policy = new UnsupportedPolicy();
 126         policies.add(policy);
 127
 128         try {
 129             forwarder.forward(policies);
 130             fail("Test must throw an exception");
 131         } catch (final Exception exp) {
 132             assertTrue(exp.getMessage().contains("Cannot forward policy"));
 133         }
 134     }
 135
 136     class UnsupportedPolicy extends ToscaEntity {
 137
 138         @Override
 139         public String getName() {
 140             return "unsupported";
 141         }
 142     }
 143
 144     private ToscaPolicy createPolicy(final Collection<ToscaEntity> policies, final String name,
 145             final String description) {
 146         final ToscaPolicy policy = new ToscaPolicy();
 147         policy.setName("test");
 148         policy.setDescription("test");
 149         policies.add(policy);
 150         return policy;
 151     }
 152 }