use of org.apache.metron.rest.model.TopologyResponse in project metron by apache.
the class StormAdminServiceImplTest method startParserTopologyShouldReturnGlobalConfigMissingError.
@Test
public void startParserTopologyShouldReturnGlobalConfigMissingError() throws Exception {
when(globalConfigService.get()).thenReturn(null);
TopologyResponse expected = new TopologyResponse();
expected.setErrorMessage(TopologyStatusCode.GLOBAL_CONFIG_MISSING.toString());
assertEquals(expected, stormAdminService.startParserTopology("bro"));
}
use of org.apache.metron.rest.model.TopologyResponse in project metron by apache.
the class StormAdminServiceImplTest method startParserTopologyByGroupShouldProperlyReturnSuccessTopologyResponse.
@Test
public void startParserTopologyByGroupShouldProperlyReturnSuccessTopologyResponse() throws Exception {
SensorParserGroup group = new SensorParserGroup();
group.setName("group");
group.setSensors(new HashSet<String>() {
{
add("bro");
add("snort");
}
});
when(sensorParserGroupService.findOne("group")).thenReturn(group);
when(stormCLIClientWrapper.startParserTopology("bro,snort")).thenReturn(0);
when(globalConfigService.get()).thenReturn(new HashMap<String, Object>());
when(sensorParserConfigService.findOne("bro")).thenReturn(new SensorParserConfig());
when(sensorParserConfigService.findOne("snort")).thenReturn(new SensorParserConfig());
TopologyResponse expected = new TopologyResponse();
expected.setSuccessMessage(TopologyStatusCode.STARTED.toString());
TopologyResponse actual = stormAdminService.startParserTopology("group");
assertEquals(expected, actual);
assertEquals(expected.hashCode(), actual.hashCode());
}
use of org.apache.metron.rest.model.TopologyResponse in project metron by apache.
the class StormAdminServiceImplTest method stopIndexingTopologyShouldProperlyReturnSuccessTopologyResponse.
@Test
public void stopIndexingTopologyShouldProperlyReturnSuccessTopologyResponse() throws Exception {
when(stormCLIClientWrapper.stopIndexingTopology("random_access_indexing", false)).thenReturn(0);
TopologyResponse expected = new TopologyResponse();
expected.setSuccessMessage(TopologyStatusCode.STOPPED.toString());
assertEquals(expected, stormAdminService.stopIndexingTopology("random_access_indexing", false));
}
use of org.apache.metron.rest.model.TopologyResponse in project metron by apache.
the class StormAdminServiceImplTest method startIndexingTopologyShouldProperlyReturnSuccessTopologyResponse.
@Test
public void startIndexingTopologyShouldProperlyReturnSuccessTopologyResponse() throws Exception {
when(stormCLIClientWrapper.startIndexingTopology("random_access_indexing_script_path")).thenReturn(0);
TopologyResponse expected = new TopologyResponse();
expected.setSuccessMessage(TopologyStatusCode.STARTED.toString());
assertEquals(expected, stormAdminService.startIndexingTopology("random_access_indexing_script_path"));
}
use of org.apache.metron.rest.model.TopologyResponse in project metron by apache.
the class StormStatusServiceImplTest method activateTopologyShouldReturnActiveTopologyResponse.
@Test
public void activateTopologyShouldReturnActiveTopologyResponse() throws Exception {
final TopologyStatus topologyStatus = new TopologyStatus();
topologyStatus.setName("bro");
topologyStatus.setId("bro_id");
final TopologySummary topologySummary = new TopologySummary();
topologySummary.setTopologies(new TopologyStatus[] { topologyStatus });
when(environment.getProperty(STORM_UI_SPRING_PROPERTY)).thenReturn(HTTP_STORM_UI);
when(restTemplate.getForObject(HTTP_STORM_UI + TOPOLOGY_SUMMARY_URL, TopologySummary.class)).thenReturn(topologySummary);
when(restTemplate.postForObject(HTTP_STORM_UI + TOPOLOGY_URL + "/bro_id/activate", null, Map.class)).thenReturn(new HashMap() {
{
put("status", "success");
}
});
TopologyResponse expected = new TopologyResponse();
expected.setSuccessMessage(TopologyStatusCode.ACTIVE.toString());
assertEquals(expected, stormStatusService.activateTopology("bro"));
}
Aggregations