package org.openecomp.sdc.be.components.distribution.engine;
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
-import static org.mockito.Mockito.when;
-
-import java.io.File;
-import java.util.ArrayList;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
-import java.util.concurrent.atomic.AtomicBoolean;
-
-import org.junit.BeforeClass;
+import fj.data.Either;
+import org.apache.commons.collections.CollectionUtils;
+import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.openecomp.sdc.be.config.DistributionEngineConfiguration.CreateTopicConfig;
import org.openecomp.sdc.be.distribution.api.client.CambriaOperationStatus;
import org.openecomp.sdc.be.impl.ComponentsUtils;
+import org.openecomp.sdc.be.resources.data.OperationalEnvironmentEntry;
import org.openecomp.sdc.common.api.ConfigurationSource;
import org.openecomp.sdc.common.impl.ExternalConfiguration;
import org.openecomp.sdc.common.impl.FSConfigurationSource;
-import fj.data.Either;
-
-public class DistributionEngineInitTaskTest {
-
- @Mock
- private ComponentsUtils componentsUtils = Mockito.mock(ComponentsUtils.class);
-
- @Mock
- private CambriaHandler cambriaHandler = Mockito.mock(CambriaHandler.class);
-
- // public static final IAuditingDao iAuditingDao =
- // Mockito.mock(AuditingDao.class);
-
- @BeforeClass
- public static void setup() {
- // ExternalConfiguration.setAppName("distribEngine1");
- ExternalConfiguration.setAppName("catalog-be");
- ExternalConfiguration.setConfigDir("src/test/resources/config");
- ExternalConfiguration.listenForChanges();
-
- ConfigurationSource configurationSource = new FSConfigurationSource(ExternalConfiguration.getChangeListener(), ExternalConfiguration.getConfigDir() + File.separator + ExternalConfiguration.getAppName());
-
- ConfigurationManager configurationManager = new ConfigurationManager(configurationSource);
- }
-
- @Test
- public void checkIncrement() {
-
- String envName = "PrOD";
-
- DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
- int retry = 2;
- int maxRetry = 40;
- deConfiguration.setInitRetryIntervalSec(retry);
- deConfiguration.setInitMaxIntervalSec(maxRetry);
- DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null);
-
- for (int i = 1; i < 5; i++) {
- initTask.incrementRetryInterval();
- assertEquals("check next retry interval", initTask.getCurrentRetryInterval(), retry * (long) Math.pow(2, i));
- }
-
- initTask.incrementRetryInterval();
- assertEquals("check next retry interval reach max retry interval", initTask.getCurrentRetryInterval(), maxRetry);
-
- }
-
- @Test
- public void testInitFlowScenarioSuccess() {
-
- String notifTopic = "notif";
- String statusTopic = "status";
-
- List<String> uebServers = new ArrayList<>();
- uebServers.add("server1");
- CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.NOT_FOUND);
- Either<Set<String>, CambriaErrorResponse> right = Either.right(cambriaErrorResponse);
- when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(right);
-
- String envName = "PrOD";
-
- DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
- deConfiguration.setUebServers(uebServers);
- int retry = 2;
- int maxRetry = 40;
- deConfiguration.setInitRetryIntervalSec(retry);
- deConfiguration.setInitMaxIntervalSec(maxRetry);
- deConfiguration.setDistributionNotifTopicName(notifTopic);
- deConfiguration.setDistributionStatusTopicName(statusTopic);
- CreateTopicConfig createTopic = new CreateTopicConfig();
- createTopic.setPartitionCount(1);
- createTopic.setReplicationCount(1);
- deConfiguration.setCreateTopic(createTopic);
-
- cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
-
- String realNotifTopic = notifTopic + "-" + envName.toUpperCase();
- String realStatusTopic = statusTopic + "-" + envName.toUpperCase();
- when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realNotifTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
- when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realStatusTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
-
- cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
- when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any()))
- .thenReturn(cambriaErrorResponse);
-
- DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null);
- initTask.setCambriaHandler(cambriaHandler);
-
- boolean initFlow = initTask.initFlow();
- assertTrue("check init flow succeed", initFlow);
-
- }
-
- @Test
- public void testInitFlowScenarioSuccessTopicsAlreadyExists() {
-
- String envName = "PrOD";
- String notifTopic = "notif";
- String statusTopic = "status";
-
- String realNotifTopic = notifTopic + "-" + envName.toUpperCase();
- String realStatusTopic = statusTopic + "-" + envName.toUpperCase();
+import java.io.File;
+import java.util.ArrayList;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.concurrent.atomic.AtomicBoolean;
- Set<String> topics = new HashSet<String>();
- topics.add(realNotifTopic);
- topics.add(realStatusTopic);
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.when;
- List<String> uebServers = new ArrayList<>();
- uebServers.add("server1");
- Either<Set<String>, CambriaErrorResponse> left = Either.left(topics);
-
- when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(left);
-
- DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
- deConfiguration.setUebServers(uebServers);
- int retry = 2;
- int maxRetry = 40;
- deConfiguration.setInitRetryIntervalSec(retry);
- deConfiguration.setInitMaxIntervalSec(maxRetry);
- deConfiguration.setDistributionNotifTopicName(notifTopic);
- deConfiguration.setDistributionStatusTopicName(statusTopic);
- CreateTopicConfig createTopic = new CreateTopicConfig();
- createTopic.setPartitionCount(1);
- createTopic.setReplicationCount(1);
- deConfiguration.setCreateTopic(createTopic);
-
- CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
- when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any()))
- .thenReturn(cambriaErrorResponse);
-
- DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null);
- initTask.setCambriaHandler(cambriaHandler);
-
- try {
- boolean initFlow = initTask.initFlow();
- assertTrue("check init flow succeed", initFlow);
- } catch (Exception e) {
- assertTrue("Should not throw exception", false);
- }
-
- }
+public class DistributionEngineInitTaskTest {
- @Test
- public void testInitFlowScenarioFailToRegister() {
+ @Mock
+ private ComponentsUtils componentsUtils;
- String notifTopic = "notif";
- String statusTopic = "status";
-
- List<String> uebServers = new ArrayList<>();
- uebServers.add("server1");
- CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.NOT_FOUND);
- Either<Set<String>, CambriaErrorResponse> right = Either.right(cambriaErrorResponse);
- when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(right);
-
- String envName = "PrOD";
-
- DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
- deConfiguration.setUebServers(uebServers);
- int retry = 2;
- int maxRetry = 40;
- deConfiguration.setInitRetryIntervalSec(retry);
- deConfiguration.setInitMaxIntervalSec(maxRetry);
- deConfiguration.setDistributionNotifTopicName(notifTopic);
- deConfiguration.setDistributionStatusTopicName(statusTopic);
- CreateTopicConfig createTopic = new CreateTopicConfig();
- createTopic.setPartitionCount(1);
- createTopic.setReplicationCount(1);
- deConfiguration.setCreateTopic(createTopic);
+ @Mock
+ private CambriaHandler cambriaHandler;
- cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
+ @Before
+ public void setup() {
+ ExternalConfiguration.setAppName("catalog-be");
+ ExternalConfiguration.setConfigDir("src/test/resources/config");
+ ExternalConfiguration.listenForChanges();
- String realNotifTopic = notifTopic + "-" + envName.toUpperCase();
- String realStatusTopic = statusTopic + "-" + envName.toUpperCase();
- when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realNotifTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
- when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realStatusTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
+ ConfigurationSource configurationSource = new FSConfigurationSource(ExternalConfiguration.getChangeListener(), ExternalConfiguration.getConfigDir() + File.separator + ExternalConfiguration.getAppName());
+ new ConfigurationManager(configurationSource);
- when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(SubscriberTypeEnum.class)))
- .thenReturn(new CambriaErrorResponse(CambriaOperationStatus.OK));
+ componentsUtils = Mockito.mock(ComponentsUtils.class);
+ cambriaHandler = Mockito.mock(CambriaHandler.class);
+ }
+
+ @Test
+ public void checkIncrement() {
- when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(SubscriberTypeEnum.class)))
- .thenReturn(new CambriaErrorResponse(CambriaOperationStatus.CONNNECTION_ERROR));
+ String envName = "PrOD";
+
+ DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
+ int retry = 2;
+ int maxRetry = 40;
+ deConfiguration.setInitRetryIntervalSec(retry);
+ deConfiguration.setInitMaxIntervalSec(maxRetry);
+ DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null, readEnvFromConfig(deConfiguration));
+
+ for (int i = 1; i < 5; i++) {
+ initTask.incrementRetryInterval();
+ assertEquals("check next retry interval", initTask.getCurrentRetryInterval(), retry * (long) Math.pow(2, i));
+ }
+
+ initTask.incrementRetryInterval();
+ assertEquals("check next retry interval reach max retry interval", initTask.getCurrentRetryInterval(), maxRetry);
+
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testInitFlowScenarioSuccess() {
+ String notifTopic = "notif";
+ String statusTopic = "status";
+
+ List<String> uebServers = new ArrayList<>();
+ uebServers.add("server1");
+ CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.NOT_FOUND);
+ Either<Set<String>, CambriaErrorResponse> right = Either.right(cambriaErrorResponse);
+ when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(right);
- DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null);
- initTask.setCambriaHandler(cambriaHandler);
+ String envName = "PrOD";
+
+ DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
+ deConfiguration.setUebServers(uebServers);
+ int retry = 2;
+ int maxRetry = 40;
+ deConfiguration.setInitRetryIntervalSec(retry);
+ deConfiguration.setInitMaxIntervalSec(maxRetry);
+ deConfiguration.setDistributionNotifTopicName(notifTopic);
+ deConfiguration.setDistributionStatusTopicName(statusTopic);
+ CreateTopicConfig createTopic = new CreateTopicConfig();
+ createTopic.setPartitionCount(1);
+ createTopic.setReplicationCount(1);
+ deConfiguration.setCreateTopic(createTopic);
+
+ cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
+
+ String realNotifTopic = notifTopic + "-" + envName.toUpperCase();
+ String realStatusTopic = statusTopic + "-" + envName.toUpperCase();
+ when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realNotifTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
+ when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realStatusTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
+
+ cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
+ when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any()))
+ .thenReturn(cambriaErrorResponse);
+
+ DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null, readEnvFromConfig(deConfiguration));
+ initTask.setCambriaHandler(cambriaHandler);
+
+ boolean initFlow = initTask.initFlow();
+ assertTrue("check init flow succeed", initFlow);
- boolean initFlow = initTask.initFlow();
- assertFalse("check init flow failed", initFlow);
+ }
- }
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testInitFlowScenarioSuccessTopicsAlreadyExists() {
- @Test
- public void testInitFlowScenario1GetTopicsFailed() {
-
- List<String> uebServers = new ArrayList<>();
- uebServers.add("server1");
- CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.CONNNECTION_ERROR);
- Either<Set<String>, CambriaErrorResponse> right = Either.right(cambriaErrorResponse);
- when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(right);
-
- String envName = "PrOD";
-
- DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
- deConfiguration.setUebServers(uebServers);
- int retry = 2;
- int maxRetry = 40;
- deConfiguration.setInitRetryIntervalSec(retry);
- deConfiguration.setInitMaxIntervalSec(maxRetry);
- DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null);
- initTask.setCambriaHandler(cambriaHandler);
-
- boolean initFlow = initTask.initFlow();
- assertFalse("check init flow failed", initFlow);
-
- }
+ String envName = "PrOD";
+ String notifTopic = "notif";
+ String statusTopic = "status";
+
+ String realNotifTopic = notifTopic + "-" + envName.toUpperCase();
+ String realStatusTopic = statusTopic + "-" + envName.toUpperCase();
+
+ Set<String> topics = new HashSet<String>();
+ topics.add(realNotifTopic);
+ topics.add(realStatusTopic);
+
+ List<String> uebServers = new ArrayList<>();
+ uebServers.add("server1");
+ Either<Set<String>, CambriaErrorResponse> left = Either.left(topics);
+
+ when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(left);
+
+ DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
+ deConfiguration.setUebServers(uebServers);
+ int retry = 2;
+ int maxRetry = 40;
+ deConfiguration.setInitRetryIntervalSec(retry);
+ deConfiguration.setInitMaxIntervalSec(maxRetry);
+ deConfiguration.setDistributionNotifTopicName(notifTopic);
+ deConfiguration.setDistributionStatusTopicName(statusTopic);
+ CreateTopicConfig createTopic = new CreateTopicConfig();
+ createTopic.setPartitionCount(1);
+ createTopic.setReplicationCount(1);
+ deConfiguration.setCreateTopic(createTopic);
+
+ CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
+ when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any()))
+ .thenReturn(cambriaErrorResponse);
+
+ DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null, readEnvFromConfig(deConfiguration));
+ initTask.setCambriaHandler(cambriaHandler);
+
+ try {
+ boolean initFlow = initTask.initFlow();
+ assertTrue("check init flow succeed", initFlow);
+ } catch (Exception e) {
+ assertTrue("Should not throw exception", false);
+ }
+
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testInitFlowScenarioFailToRegister() {
+
+ String notifTopic = "notif";
+ String statusTopic = "status";
+
+ List<String> uebServers = new ArrayList<>();
+ uebServers.add("server1");
+ CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.NOT_FOUND);
+ Either<Set<String>, CambriaErrorResponse> right = Either.right(cambriaErrorResponse);
+ when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(right);
+
+ String envName = "PrOD";
+
+ DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
+ deConfiguration.setUebServers(uebServers);
+ int retry = 2;
+ int maxRetry = 40;
+ deConfiguration.setInitRetryIntervalSec(retry);
+ deConfiguration.setInitMaxIntervalSec(maxRetry);
+ deConfiguration.setDistributionNotifTopicName(notifTopic);
+ deConfiguration.setDistributionStatusTopicName(statusTopic);
+ CreateTopicConfig createTopic = new CreateTopicConfig();
+ createTopic.setPartitionCount(1);
+ createTopic.setReplicationCount(1);
+ deConfiguration.setCreateTopic(createTopic);
+
+ cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.OK);
+
+ String realNotifTopic = notifTopic + "-" + envName.toUpperCase();
+ String realStatusTopic = statusTopic + "-" + envName.toUpperCase();
+ when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realNotifTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
+ when(cambriaHandler.createTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realStatusTopic), Mockito.eq(1), Mockito.eq(1))).thenReturn(cambriaErrorResponse);
+
+ when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realNotifTopic)))
+ .thenReturn(new CambriaErrorResponse(CambriaOperationStatus.OK));
+
+ when(cambriaHandler.registerToTopic(Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.eq(realStatusTopic)))
+ .thenReturn(new CambriaErrorResponse(CambriaOperationStatus.CONNNECTION_ERROR));
+
+
+ DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null, readEnvFromConfig(deConfiguration));
+ initTask.setCambriaHandler(cambriaHandler);
+
+ boolean initFlow = initTask.initFlow();
+ assertFalse("check init flow failed", initFlow);
+
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testInitFlowScenario1GetTopicsFailed() {
+
+ List<String> uebServers = new ArrayList<>();
+ uebServers.add("server1");
+ CambriaErrorResponse cambriaErrorResponse = new CambriaErrorResponse(CambriaOperationStatus.CONNNECTION_ERROR);
+ Either<Set<String>, CambriaErrorResponse> right = Either.right(cambriaErrorResponse);
+ when(cambriaHandler.getTopics(Mockito.any(List.class))).thenReturn(right);
+
+ String envName = "PrOD";
+
+ DistributionEngineConfiguration deConfiguration = new DistributionEngineConfiguration();
+ deConfiguration.setUebServers(uebServers);
+ int retry = 2;
+ int maxRetry = 40;
+ deConfiguration.setInitRetryIntervalSec(retry);
+ deConfiguration.setInitMaxIntervalSec(maxRetry);
+ DistributionEngineInitTask initTask = new DistributionEngineInitTask(0l, deConfiguration, envName, new AtomicBoolean(false), componentsUtils, null, readEnvFromConfig(deConfiguration));
+ initTask.setCambriaHandler(cambriaHandler);
+
+ boolean initFlow = initTask.initFlow();
+ assertFalse("check init flow failed", initFlow);
+
+ }
+
+ private OperationalEnvironmentEntry readEnvFromConfig(DistributionEngineConfiguration distributionEngineConfiguration) {
+ OperationalEnvironmentEntry entry = new OperationalEnvironmentEntry();
+ entry.setUebApikey(distributionEngineConfiguration.getUebPublicKey());
+ entry.setUebSecretKey(distributionEngineConfiguration.getUebSecretKey());
+ Set<String> puebEndpoints = new HashSet<>();
+ if(distributionEngineConfiguration.getUebServers() != null)
+ puebEndpoints.addAll(distributionEngineConfiguration.getUebServers());
+ entry.setDmaapUebAddress(puebEndpoints);
+ String envName = "UNKNOWN";
+ if(CollectionUtils.isNotEmpty(distributionEngineConfiguration.getEnvironments()))
+ envName = distributionEngineConfiguration.getEnvironments().get(0);
+ entry.setEnvironmentId(envName);
+ return entry;
+ }
}
Code Review / sdc.git / blobdiff