appc-oam/appc-oam-bundle/src/test/java/org/onap/appc/oam/AppcOamTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Copyright (C) 2017 Amdocs
   8  * ================================================================================
   9  * Modifications (C) 2018 Ericsson
  10  * =============================================================================
  11  * Licensed under the Apache License, Version 2.0 (the "License");
  12  * you may not use this file except in compliance with the License.
  13  * You may obtain a copy of the License at
  14  *
  15  *      http://www.apache.org/licenses/LICENSE-2.0
  16  *
  17  * Unless required by applicable law or agreed to in writing, software
  18  * distributed under the License is distributed on an "AS IS" BASIS,
  19  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  20  * See the License for the specific language governing permissions and
  21  * limitations under the License.
  22  *
  23  * ============LICENSE_END=========================================================
  24  */
  25
  26 package org.onap.appc.oam;
  27
  28 import static org.junit.Assert.assertEquals;
  29 import static org.mockito.Mockito.mock;
  30 import static org.powermock.api.mockito.PowerMockito.spy;
  31 import java.util.Collection;
  32 import java.util.concurrent.ExecutionException;
  33 import java.util.concurrent.Future;
  34 import org.junit.Assert;
  35 import org.junit.Before;
  36 import org.junit.Test;
  37 import org.mockito.Mockito;
  38 import org.onap.appc.exceptions.APPCException;
  39 import org.onap.appc.i18n.Msg;
  40 import org.onap.appc.metricservice.MetricRegistry;
  41 import org.onap.appc.metricservice.MetricService;
  42 import org.onap.appc.metricservice.metric.Metric;
  43 import org.onap.appc.metricservice.metric.MetricType;
  44 import org.onap.appc.metricservice.metric.impl.DefaultPrimitiveCounter;
  45 import org.onap.appc.metricservice.metric.impl.DispatchingFuntionMetricImpl;
  46 import org.onap.appc.oam.processor.OamMmodeProcessor;
  47 import org.onap.appc.oam.processor.OamRestartProcessor;
  48 import org.onap.appc.oam.processor.OamStartProcessor;
  49 import org.onap.appc.oam.processor.OamStopProcessor;
  50 import org.onap.appc.oam.util.AsyncTaskHelper;
  51 import org.onap.appc.oam.util.ConfigurationHelper;
  52 import org.onap.appc.oam.util.OperationHelper;
  53 import org.onap.appc.oam.util.StateHelper;
  54 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.AppcState;
  55 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.GetAppcStateOutput;
  56 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.GetMetricsOutput;
  57 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.MaintenanceModeInput;
  58 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.MaintenanceModeOutput;
  59 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.RestartInput;
  60 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.RestartOutput;
  61 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.StartInput;
  62 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.StartOutput;
  63 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.StopInput;
  64 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.StopOutput;
  65 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.common.header.CommonHeader;
  66 import org.opendaylight.yang.gen.v1.org.onap.appc.oam.rev170303.status.Status;
  67 import org.opendaylight.yangtools.yang.common.RpcError;
  68 import org.opendaylight.yangtools.yang.common.RpcResult;
  69 import org.powermock.reflect.Whitebox;
  70 import com.att.aft.dme2.internal.google.common.collect.Iterables;
  71 import com.att.eelf.configuration.EELFLogger;
  72 import com.google.common.collect.ImmutableMap;
  73
  74
  75 public class AppcOamTest {
  76
  77     private AppcOam appcOam;
  78     private CommonHeader mockCommonHeader = mock(CommonHeader.class);
  79     private Status mockStatus = mock(Status.class);
  80     private OperationHelper mockOperationHelper = mock(OperationHelper.class);
  81     private StateHelper mockStateHelper = mock(StateHelper.class);
  82
  83     @Before
  84     public void setUp() throws Exception {
  85         appcOam = spy(new AppcOam(null, null, null));
  86
  87         Whitebox.setInternalState(appcOam, "stateHelper", mockStateHelper);
  88         Whitebox.setInternalState(appcOam, "operationHelper", mockOperationHelper);
  89     }
  90
  91     @Test
  92     public void testMaintenanceMode() throws Exception {
  93         // mock processor creation
  94         OamMmodeProcessor mockProcessor = mock(OamMmodeProcessor.class);
  95         Mockito.doReturn(mockProcessor).when(appcOam).getOamMmodeProcessor(Mockito.any(EELFLogger.class),
  96                 Mockito.any(ConfigurationHelper.class), Mockito.any(StateHelper.class),
  97                 Mockito.any(AsyncTaskHelper.class), Mockito.any(OperationHelper.class));
  98         // mock input
  99         MaintenanceModeInput mockInput = mock(MaintenanceModeInput.class);
 100         Mockito.doReturn(mockCommonHeader).when(mockInput).getCommonHeader();
 101         // mock processor result
 102         Mockito.doReturn(mockStatus).when(mockProcessor).processRequest(mockInput);
 103
 104         Future<RpcResult<MaintenanceModeOutput>> response = appcOam.maintenanceMode(mockInput);
 105
 106         Assert.assertEquals("Should have common header", mockCommonHeader,
 107                 response.get().getResult().getCommonHeader());
 108         Assert.assertEquals("Should have status", mockStatus, response.get().getResult().getStatus());
 109     }
 110
 111     @Test
 112     public void testStart()  throws Exception {
 113         // mock processor creation
 114         OamStartProcessor mockProcessor = mock(OamStartProcessor.class);
 115         Mockito.doReturn(mockProcessor).when(appcOam).getOamStartProcessor(Mockito.any(EELFLogger.class),
 116                 Mockito.any(ConfigurationHelper.class), Mockito.any(StateHelper.class),
 117                 Mockito.any(AsyncTaskHelper.class), Mockito.any(OperationHelper.class));
 118         // mock input
 119         StartInput mockInput = mock(StartInput.class);
 120         Mockito.doReturn(mockCommonHeader).when(mockInput).getCommonHeader();
 121         // mock processor result
 122         Mockito.doReturn(mockStatus).when(mockProcessor).processRequest(mockInput);
 123
 124         Future<RpcResult<StartOutput>> response = appcOam.start(mockInput);
 125
 126         Assert.assertEquals("Should have common header", mockCommonHeader,
 127                 response.get().getResult().getCommonHeader());
 128         Assert.assertEquals("Should have status", mockStatus, response.get().getResult().getStatus());
 129     }
 130
 131     @Test
 132     public void testStop()  throws Exception {
 133         // mock processor creation
 134         OamStopProcessor mockProcessor = mock(OamStopProcessor.class);
 135         //Mockito.doNothing().when(mockProcessor).setInitialLogProperties();
 136         Mockito.doReturn(mockProcessor).when(appcOam).getOamStopProcessor(Mockito.any(EELFLogger.class),
 137                 Mockito.any(ConfigurationHelper.class), Mockito.any(StateHelper.class),
 138                 Mockito.any(AsyncTaskHelper.class), Mockito.any(OperationHelper.class));
 139         // mock input
 140         StopInput mockInput = mock(StopInput.class);
 141         Mockito.doReturn(mockCommonHeader).when(mockInput).getCommonHeader();
 142         // mock processor result
 143         Mockito.doReturn(mockStatus).when(mockProcessor).processRequest(mockInput);
 144
 145         Future<RpcResult<StopOutput>> response = appcOam.stop(mockInput);
 146
 147         Assert.assertEquals("Should have common header", mockCommonHeader,
 148                 response.get().getResult().getCommonHeader());
 149         Assert.assertEquals("Should have status", mockStatus, response.get().getResult().getStatus());
 150     }
 151
 152     @Test
 153     public void testRestart()  throws Exception {
 154         // mock processor creation
 155         OamRestartProcessor mockProcessor = mock(OamRestartProcessor.class);
 156         Mockito.doReturn(mockProcessor).when(appcOam).getOamRestartProcessor(Mockito.any(EELFLogger.class),
 157                 Mockito.any(ConfigurationHelper.class), Mockito.any(StateHelper.class),
 158                 Mockito.any(AsyncTaskHelper.class), Mockito.any(OperationHelper.class));
 159         // mock input
 160         RestartInput mockInput = mock(RestartInput.class);
 161         Mockito.doReturn(mockCommonHeader).when(mockInput).getCommonHeader();
 162         // mock processor result
 163         Mockito.doReturn(mockStatus).when(mockProcessor).processRequest(mockInput);
 164
 165         Future<RpcResult<RestartOutput>> response = appcOam.restart(mockInput);
 166
 167         Assert.assertEquals("Should have common header", mockCommonHeader,
 168                 response.get().getResult().getCommonHeader());
 169         Assert.assertEquals("Should have status", mockStatus, response.get().getResult().getStatus());
 170     }
 171
 172     @Test
 173     public void testGetAppcState() throws Exception {
 174         AppcState appcState = AppcState.Started;
 175         Mockito.doReturn(appcState).when(mockStateHelper).getCurrentOamYangState();
 176
 177         Future<RpcResult<GetAppcStateOutput>> state = appcOam.getAppcState();
 178         Assert.assertEquals("Should return the same state",
 179                 appcState, state.get().getResult().getState());
 180     }
 181
 182     @Test
 183     public void testGetMetricsMetricDisabled() throws InterruptedException, ExecutionException {
 184         Whitebox.setInternalState(appcOam, "isMetricEnabled", false);
 185         Future<RpcResult<GetMetricsOutput>> result = appcOam.getMetrics();
 186         assertEquals("Metric Service not enabled", Iterables.get(result.get().getErrors(), 0).getMessage());
 187     }
 188
 189     @Test
 190     public void testGetMetricsNoMetricsService() throws InterruptedException, ExecutionException, APPCException {
 191         Whitebox.setInternalState(appcOam, "isMetricEnabled", true);
 192         Mockito.doThrow(new APPCException()).when(mockOperationHelper).getService(MetricService.class);
 193         Future<RpcResult<GetMetricsOutput>> result = appcOam.getMetrics();
 194         assertEquals("Metric Service not found", Iterables.get(result.get().getErrors(), 0).getMessage());
 195     }
 196
 197     @Test
 198     public void testGetMetricsNoMetrics() throws InterruptedException, ExecutionException, APPCException {
 199         Whitebox.setInternalState(appcOam, "isMetricEnabled", true);
 200         MetricService mockMetricService = mock(MetricService.class);
 201         Mockito.doReturn(mockMetricService).when(mockOperationHelper).getService(MetricService.class);
 202         Future<RpcResult<GetMetricsOutput>> result = appcOam.getMetrics();
 203         assertEquals("No metrics Registered", Iterables.get(result.get().getErrors(), 0).getMessage());
 204     }
 205
 206     @Test
 207     public void testGetMetricsWithMetricRegistry() throws InterruptedException, ExecutionException, APPCException {
 208         Whitebox.setInternalState(appcOam, "isMetricEnabled", true);
 209         MetricService mockMetricService = mock(MetricService.class);
 210         MetricRegistry mockMetricRegistry = mock(MetricRegistry.class);
 211         Mockito.doReturn(mockMetricService).when(mockOperationHelper).getService(MetricService.class);
 212         Mockito.doReturn(ImmutableMap.of("TEST REGISTRY NAME", mockMetricRegistry)).when(mockMetricService).getAllRegistry();
 213         Metric metric = new DispatchingFuntionMetricImpl("TEST METRIC NAME", MetricType.COUNTER, 0, 0);
 214         Mockito.doReturn(new Metric[] {metric}).when(mockMetricRegistry).metrics();
 215         Future<RpcResult<GetMetricsOutput>> result = appcOam.getMetrics();
 216         assertEquals(1, result.get().getResult().getMetrics().size());
 217     }
 218
 219     @Test
 220     public void testClose() throws Exception {
 221         ConfigurationHelper mockConfigurationHelper = mock(ConfigurationHelper.class);
 222         Mockito.doReturn("TEST APP NAME").when(mockConfigurationHelper).getAppcName();
 223         Whitebox.setInternalState(appcOam, "configurationHelper", mockConfigurationHelper);
 224         EELFLogger mockLogger = Mockito.mock(EELFLogger.class);
 225         Whitebox.setInternalState(appcOam, "logger", mockLogger);
 226         appcOam.close();
 227         Mockito.verify(mockLogger).info(Msg.COMPONENT_TERMINATED, "TEST APP NAME", "oam");
 228     }
 229 }