@Test
public void getAggregatorsTest() throws Exception {
List<Aggregator> aggregators = new LinkedList<>();
Aggregator aggregator = new Aggregator();
aggregator.setAggregatorId("*****@*****.**");
aggregator.setAggregatorName("aggregatorName");
aggregators.add(aggregator);
when(userManager.isAdmin()).thenReturn(true);
when(aggregatorManager.getAPIAggregators()).thenReturn(aggregators);
Response response = toTest.getAggregators();
Assert.assertEquals(200, response.getStatus());
Assert.assertEquals(aggregators, response.getEntity());
}
@Test
public void getAggregatorsNotAdminTest() throws Exception {
Aggregator aggregator = new Aggregator();
aggregator.setAggregatorId("*****@*****.**");
aggregator.setAggregatorName("aggregatorName");
RSUser user = new RSUser();
user.setEmail("*****@*****.**");
when(userManager.isAdmin()).thenReturn(false);
when(userManager.getCurrentUser()).thenReturn(user);
when(aggregatorManager.getAggregator("*****@*****.**")).thenReturn(aggregator);
Response response = toTest.getAggregators();
Assert.assertEquals(200, response.getStatus());
List listResponse = (List) response.getEntity();
Assert.assertEquals(aggregator, listResponse.get(0));
}
