1 package org.openecomp.sdc.be.components.distribution.engine;
4 import org.mockito.Mock;
5 import org.mockito.Mockito;
6 import org.openecomp.sdc.be.components.BeConfDependentTest;
7 import org.openecomp.sdc.be.components.distribution.engine.report.DistributionCompleteReporter;
8 import org.openecomp.sdc.be.config.DistributionEngineConfiguration;
9 import org.openecomp.sdc.be.impl.ComponentsUtils;
10 import org.openecomp.sdc.be.resources.data.OperationalEnvironmentEntry;
11 import org.openecomp.sdc.be.resources.data.auditing.AuditingActionEnum;
13 import mockit.Deencapsulation;
15 public class DistributionEnginePollingTaskTest extends BeConfDependentTest {
18 private ComponentsUtils componentsUtils;
20 private DistributionEnginePollingTask createTestSubject() {
21 componentsUtils = Mockito.mock(ComponentsUtils.class);
22 DistributionEngineConfiguration distributionEngineConfiguration = configurationManager
23 .getDistributionEngineConfiguration();
25 return new DistributionEnginePollingTask(distributionEngineConfiguration,
26 new DistributionCompleteReporterMock(), componentsUtils, new DistributionEngineClusterHealth(),
27 new OperationalEnvironmentEntry());
31 public void testStartTask() throws Exception {
32 DistributionEnginePollingTask testSubject;
33 String topicName = "";
36 testSubject = createTestSubject();
37 testSubject.startTask(topicName);
41 public void testStopTask() throws Exception {
42 DistributionEnginePollingTask testSubject;
45 testSubject = createTestSubject();
46 testSubject.stopTask();
50 public void testDestroy() throws Exception {
51 DistributionEnginePollingTask testSubject;
54 testSubject = createTestSubject();
55 testSubject.destroy();
59 public void testRun() throws Exception {
60 DistributionEnginePollingTask testSubject;
63 testSubject = createTestSubject();
68 public void testHandleDistributionNotificationMsg() throws Exception {
69 DistributionEnginePollingTask testSubject;
70 DistributionStatusNotification notification = new DistributionStatusNotification();
71 notification.setDistributionID("mock");
72 notification.setConsumerID("mock");
73 notification.setArtifactURL("mock");
74 notification.setTimestamp(435435);
75 notification.setStatus(DistributionStatusNotificationEnum.ALREADY_DEPLOYED);
76 notification.setErrorReason("mock");
81 testSubject = createTestSubject();
82 Mockito.doNothing().when(componentsUtils).auditDistributionStatusNotification(Mockito.any(AuditingActionEnum.class),
83 Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyString(),
84 Mockito.anyString(), Mockito.anyString(), Mockito.anyString());
85 Deencapsulation.invoke(testSubject, "handleDistributionNotificationMsg",
90 public void testShutdownExecutor() throws Exception {
91 DistributionEnginePollingTask testSubject;
94 testSubject = createTestSubject();
95 Deencapsulation.invoke(testSubject, "shutdownExecutor");
98 private class DistributionCompleteReporterMock implements DistributionCompleteReporter {
101 public void reportDistributionComplete(DistributionStatusNotification distributionStatusNotification) {
102 // TODO Auto-generated method stub