2 * ===============================LICENSE_START======================================
4 * ================================================================================
5 * Copyright © 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.openecomp.dcae.apod.analytics.cdap.tca.worker;
23 import co.cask.cdap.api.app.ApplicationSpecification;
24 import co.cask.cdap.api.worker.WorkerConfigurer;
25 import co.cask.cdap.api.worker.WorkerContext;
26 import org.junit.Before;
27 import org.junit.Test;
28 import org.mockito.Mockito;
29 import org.openecomp.dcae.apod.analytics.cdap.common.CDAPComponentsConstants;
30 import org.openecomp.dcae.apod.analytics.cdap.common.exception.CDAPSettingsException;
31 import org.openecomp.dcae.apod.analytics.cdap.tca.BaseAnalyticsCDAPTCAUnitTest;
33 import static org.mockito.ArgumentMatchers.anyString;
34 import static org.mockito.ArgumentMatchers.eq;
35 import static org.mockito.Mockito.doNothing;
36 import static org.mockito.Mockito.mock;
37 import static org.mockito.Mockito.times;
38 import static org.mockito.Mockito.verify;
39 import static org.mockito.Mockito.when;
42 * @author Rajiv Singla . Creation Date: 12/20/2016.
44 public class TCADMaaPPublisherWorkerTest extends BaseAnalyticsCDAPTCAUnitTest {
46 private static final String VES_ALERTS_TABLE_NAME = "vesAlertsTable";
48 private WorkerConfigurer workerConfigurer;
49 private WorkerContext workerContext;
50 private TCADMaaPPublisherWorker publisherWorker;
51 private ApplicationSpecification mockApplicationSpecification;
54 public void before() throws Exception {
55 workerConfigurer = mock(WorkerConfigurer.class);
56 workerContext = mock(WorkerContext.class);
57 doNothing().when(workerConfigurer).setName(anyString());
58 doNothing().when(workerConfigurer).setDescription(anyString());
59 mockApplicationSpecification = Mockito.mock(ApplicationSpecification.class);
60 when(workerContext.getApplicationSpecification()).thenReturn(mockApplicationSpecification);
61 publisherWorker = new TCADMaaPPublisherWorker(VES_ALERTS_TABLE_NAME);
66 public void testConfigure() throws Exception {
67 publisherWorker.configure(workerConfigurer);
68 verify(workerConfigurer, times(1))
69 .setName(eq(CDAPComponentsConstants.TCA_FIXED_DMAAP_PUBLISHER_WORKER));
70 verify(workerConfigurer, times(1))
71 .setDescription(eq(CDAPComponentsConstants.TCA_FIXED_DMAAP_PUBLISHER_DESCRIPTION_WORKER));
74 @Test(expected = CDAPSettingsException.class)
75 public void testInitializeWhenSettingsHaveErrors() throws Exception {
76 when(mockApplicationSpecification.getConfiguration()).thenReturn("{}");
77 publisherWorker.initialize(workerContext);
81 public void testInitializeWhenSettingsAreValid() throws Exception {
82 when(workerContext.getRuntimeArguments()).thenReturn(getPreferenceMap());
83 when(mockApplicationSpecification.getConfiguration()).thenReturn(fromStream(TCA_APP_CONFIG_FILE_LOCATION));
84 publisherWorker.initialize(workerContext);