Search in sources :

Example 1 with TopologySummary

use of org.apache.metron.rest.model.TopologySummary in project metron by apache.

the class StormStatusServiceImplTest method activateTopologyShouldReturnActiveTopologyResponse.

@Test
public void activateTopologyShouldReturnActiveTopologyResponse() {
    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.Test)

Example 2 with TopologySummary

use of org.apache.metron.rest.model.TopologySummary in project metron by apache.

the class StormStatusServiceImplTest method activateTopologyShouldReturnErrorTopologyResponse.

@Test
public void activateTopologyShouldReturnErrorTopologyResponse() {
    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", "error message");
        }
    });
    TopologyResponse expected = new TopologyResponse();
    expected.setErrorMessage("error message");
    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.Test)

Example 3 with TopologySummary

use of org.apache.metron.rest.model.TopologySummary in project metron by apache.

the class StormStatusServiceImplTest method deactivateTopologyShouldReturnActiveTopologyResponse.

@Test
public void deactivateTopologyShouldReturnActiveTopologyResponse() {
    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/deactivate", null, Map.class)).thenReturn(new HashMap() {

        {
            put("status", "success");
        }
    });
    TopologyResponse expected = new TopologyResponse();
    expected.setSuccessMessage(TopologyStatusCode.INACTIVE.toString());
    assertEquals(expected, stormStatusService.deactivateTopology("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.Test)

Example 4 with TopologySummary

use of org.apache.metron.rest.model.TopologySummary in project metron by apache.

the class StormStatusServiceImplTest method deactivateTopologyShouldReturnTopologyNotFoundTopologyResponse.

@Test
public void deactivateTopologyShouldReturnTopologyNotFoundTopologyResponse() {
    when(environment.getProperty(STORM_UI_SPRING_PROPERTY)).thenReturn(HTTP_STORM_UI);
    when(restTemplate.getForObject(HTTP_STORM_UI + TOPOLOGY_SUMMARY_URL, TopologySummary.class)).thenReturn(new TopologySummary());
    TopologyResponse expected = new TopologyResponse();
    expected.setErrorMessage(TopologyStatusCode.TOPOLOGY_NOT_FOUND.toString());
    assertEquals(expected, stormStatusService.deactivateTopology("bro"));
}
Also used : TopologyResponse(org.apache.metron.rest.model.TopologyResponse) TopologySummary(org.apache.metron.rest.model.TopologySummary) Test(org.junit.Test)

Example 5 with TopologySummary

use of org.apache.metron.rest.model.TopologySummary in project metron by apache.

the class StormStatusServiceImplTest method deactivateTopologyShouldReturnErrorTopologyResponse.

@Test
public void deactivateTopologyShouldReturnErrorTopologyResponse() {
    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/deactivate", null, Map.class)).thenReturn(new HashMap() {

        {
            put("status", "error message");
        }
    });
    TopologyResponse expected = new TopologyResponse();
    expected.setErrorMessage("error message");
    assertEquals(expected, stormStatusService.deactivateTopology("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.Test)

Aggregations

TopologySummary (org.apache.metron.rest.model.TopologySummary)9 Test (org.junit.Test)9 TopologyStatus (org.apache.metron.rest.model.TopologyStatus)7 TopologyResponse (org.apache.metron.rest.model.TopologyResponse)6 HashMap (java.util.HashMap)4 ArrayList (java.util.ArrayList)1