1 files changed, 107 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/DistributionEnginePollingTaskTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/DistributionEnginePollingTaskTest.java
new file mode 100644
index 0000000000..ecbc1f90f6
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/DistributionEnginePollingTaskTest.java
@@ -0,0 +1,107 @@
+package org.openecomp.sdc.be.components.distribution.engine;
+
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.openecomp.sdc.be.components.BeConfDependentTest;
+import org.openecomp.sdc.be.components.distribution.engine.report.DistributionCompleteReporter;
+import org.openecomp.sdc.be.config.DistributionEngineConfiguration;
+import org.openecomp.sdc.be.impl.ComponentsUtils;
+import org.openecomp.sdc.be.resources.data.OperationalEnvironmentEntry;
+import org.openecomp.sdc.be.resources.data.auditing.AuditingActionEnum;
+
+import mockit.Deencapsulation;
+
+public class DistributionEnginePollingTaskTest extends BeConfDependentTest {
+
+	@Mock
+	private ComponentsUtils componentsUtils;
+
+	private DistributionEnginePollingTask createTestSubject() {
+		componentsUtils = Mockito.mock(ComponentsUtils.class);
+		DistributionEngineConfiguration distributionEngineConfiguration = configurationManager
+				.getDistributionEngineConfiguration();
+
+		return new DistributionEnginePollingTask(distributionEngineConfiguration,
+				new DistributionCompleteReporterMock(), componentsUtils, new DistributionEngineClusterHealth(),
+				new OperationalEnvironmentEntry());
+	}
+
+	@Test
+	public void testStartTask() throws Exception {
+		DistributionEnginePollingTask testSubject;
+		String topicName = "";
+
+		// default test
+		testSubject = createTestSubject();
+		testSubject.startTask(topicName);
+	}
+
+	@Test
+	public void testStopTask() throws Exception {
+		DistributionEnginePollingTask testSubject;
+
+		// default test
+		testSubject = createTestSubject();
+		testSubject.stopTask();
+	}
+
+	@Test
+	public void testDestroy() throws Exception {
+		DistributionEnginePollingTask testSubject;
+
+		// default test
+		testSubject = createTestSubject();
+		testSubject.destroy();
+	}
+
+	@Test
+	public void testRun() throws Exception {
+		DistributionEnginePollingTask testSubject;
+
+		// default test
+		testSubject = createTestSubject();
+		testSubject.run();
+	}
+	
+	@Test
+	public void testHandleDistributionNotificationMsg() throws Exception {
+		DistributionEnginePollingTask testSubject;
+		DistributionStatusNotification notification = new DistributionStatusNotification();
+		notification.setDistributionID("mock");
+		notification.setConsumerID("mock");
+		notification.setArtifactURL("mock");
+		notification.setTimestamp(435435);
+		notification.setStatus(DistributionStatusNotificationEnum.ALREADY_DEPLOYED);
+		notification.setErrorReason("mock");
+		
+		
+		
+		// default test
+		testSubject = createTestSubject();
+		Mockito.doNothing().when(componentsUtils).auditDistributionStatusNotification(Mockito.any(AuditingActionEnum.class), 
+				Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), 
+				Mockito.anyString(), Mockito.anyString(), Mockito.anyString());
+		Deencapsulation.invoke(testSubject, "handleDistributionNotificationMsg",
+				notification);
+	}
+
+	@Test
+	public void testShutdownExecutor() throws Exception {
+		DistributionEnginePollingTask testSubject;
+
+		// default test
+		testSubject = createTestSubject();
+		Deencapsulation.invoke(testSubject, "shutdownExecutor");
+	}
+
+	private class DistributionCompleteReporterMock implements DistributionCompleteReporter {
+
+		@Override
+		public void reportDistributionComplete(DistributionStatusNotification distributionStatusNotification) {
+			// TODO Auto-generated method stub
+
+		}
+
+	}
+}
+\ No newline at end of file