Search in sources :

Example 1 with TopologyResponse

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"));
}
Also used : TopologyResponse(org.apache.metron.rest.model.TopologyResponse) Test(org.junit.jupiter.api.Test)

Example 2 with TopologyResponse

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());
}
Also used : TopologyResponse(org.apache.metron.rest.model.TopologyResponse) SensorParserGroup(org.apache.metron.common.configuration.SensorParserGroup) SensorParserConfig(org.apache.metron.common.configuration.SensorParserConfig) Test(org.junit.jupiter.api.Test)

Example 3 with TopologyResponse

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));
}
Also used : TopologyResponse(org.apache.metron.rest.model.TopologyResponse) Test(org.junit.jupiter.api.Test)

Example 4 with TopologyResponse

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"));
}
Also used : TopologyResponse(org.apache.metron.rest.model.TopologyResponse) Test(org.junit.jupiter.api.Test)

Example 5 with TopologyResponse

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"));
}
Also used : HashMap(java.util.HashMap) TopologyResponse(org.apache.metron.rest.model.TopologyResponse) TopologySummary(org.apache.metron.rest.model.TopologySummary) TopologyStatus(org.apache.metron.rest.model.TopologyStatus) Test(org.junit.jupiter.api.Test)

Aggregations

TopologyResponse (org.apache.metron.rest.model.TopologyResponse)18 Test (org.junit.jupiter.api.Test)15 TopologySummary (org.apache.metron.rest.model.TopologySummary)6 TopologyStatus (org.apache.metron.rest.model.TopologyStatus)5 HashMap (java.util.HashMap)4 Map (java.util.Map)2 SensorParserConfig (org.apache.metron.common.configuration.SensorParserConfig)2 SensorParserGroup (org.apache.metron.common.configuration.SensorParserGroup)2