Add support for ABATED alerts within CDAP TCA

[dcaegen2/analytics/tca.git] / dcae-analytics-cdap-tca / src / test / java / org / openecomp / dcae / apod / analytics / cdap / tca / worker / TCADMaaPPublisherWorkerTest.java

diff --git a/dcae-analytics-cdap-tca/src/test/java/org/openecomp/dcae/apod/analytics/cdap/tca/worker/TCADMaaPPublisherWorkerTest.java b/dcae-analytics-cdap-tca/src/test/java/org/openecomp/dcae/apod/analytics/cdap/tca/worker/TCADMaaPPublisherWorkerTest.java
index f2eeef8..2324b38 100644 (file)
--- a/dcae-analytics-cdap-tca/src/test/java/org/openecomp/dcae/apod/analytics/cdap/tca/worker/TCADMaaPPublisherWorkerTest.java
+++ b/dcae-analytics-cdap-tca/src/test/java/org/openecomp/dcae/apod/analytics/cdap/tca/worker/TCADMaaPPublisherWorkerTest.java
@@ -20,10 +20,12 @@
 
 package org.openecomp.dcae.apod.analytics.cdap.tca.worker;
 
+import co.cask.cdap.api.app.ApplicationSpecification;
 import co.cask.cdap.api.worker.WorkerConfigurer;
 import co.cask.cdap.api.worker.WorkerContext;
 import org.junit.Before;
 import org.junit.Test;
+import org.mockito.Mockito;
 import org.openecomp.dcae.apod.analytics.cdap.common.CDAPComponentsConstants;
 import org.openecomp.dcae.apod.analytics.cdap.common.exception.CDAPSettingsException;
 import org.openecomp.dcae.apod.analytics.cdap.tca.BaseAnalyticsCDAPTCAUnitTest;
@@ -46,6 +48,7 @@ public class TCADMaaPPublisherWorkerTest extends BaseAnalyticsCDAPTCAUnitTest {
     private WorkerConfigurer workerConfigurer;
     private WorkerContext workerContext;
     private TCADMaaPPublisherWorker publisherWorker;
+    private ApplicationSpecification mockApplicationSpecification;
 
     @Before
     public void before() throws Exception {
@@ -53,6 +56,8 @@ public class TCADMaaPPublisherWorkerTest extends BaseAnalyticsCDAPTCAUnitTest {
         workerContext = mock(WorkerContext.class);
         doNothing().when(workerConfigurer).setName(anyString());
         doNothing().when(workerConfigurer).setDescription(anyString());
+        mockApplicationSpecification = Mockito.mock(ApplicationSpecification.class);
+        when(workerContext.getApplicationSpecification()).thenReturn(mockApplicationSpecification);
         publisherWorker = new TCADMaaPPublisherWorker(VES_ALERTS_TABLE_NAME);
 
     }
@@ -68,12 +73,14 @@ public class TCADMaaPPublisherWorkerTest extends BaseAnalyticsCDAPTCAUnitTest {
 
     @Test(expected = CDAPSettingsException.class)
     public void testInitializeWhenSettingsHaveErrors() throws Exception {
+        when(mockApplicationSpecification.getConfiguration()).thenReturn("{}");
         publisherWorker.initialize(workerContext);
     }
 
     @Test
     public void testInitializeWhenSettingsAreValid() throws Exception {
         when(workerContext.getRuntimeArguments()).thenReturn(getPreferenceMap());
+        when(mockApplicationSpecification.getConfiguration()).thenReturn(fromStream(TCA_APP_CONFIG_FILE_LOCATION));
         publisherWorker.initialize(workerContext);
     }