@Test
public void testUpdateInvalidLoadBalancer() throws Exception {
EntityNotFoundException entityNotFoundException = new EntityNotFoundException();
when(objectMessage.getObject()).thenReturn(lb);
when(loadBalancerService.getWithUserPages(LOAD_BALANCER_ID, ACCOUNT_ID))
.thenThrow(entityNotFoundException);
deleteConnectionThrottleListener.doOnMessage(objectMessage);
verify(notificationService)
.saveAlert(
eq(ACCOUNT_ID),
eq(LOAD_BALANCER_ID),
eq(entityNotFoundException),
eq(AlertType.DATABASE_FAILURE.name()),
anyString());
verify(notificationService)
.saveConnectionLimitEvent(
eq(USERNAME),
eq(ACCOUNT_ID),
eq(LOAD_BALANCER_ID),
anyInt(),
anyString(),
anyString(),
eq(EventType.DELETE_CONNECTION_THROTTLE),
eq(CategoryType.DELETE),
eq(EventSeverity.CRITICAL));
}
@Test
public void testDeleteInvalidThrottle() throws Exception {
Exception exception = new Exception();
when(objectMessage.getObject()).thenReturn(lb);
when(loadBalancerService.getWithUserPages(LOAD_BALANCER_ID, ACCOUNT_ID)).thenReturn(lb);
doThrow(exception).when(reverseProxyLoadBalancerStmService).deleteConnectionThrottle(lb);
when(config.getString(Matchers.<ConfigurationKey>any())).thenReturn("REST");
deleteConnectionThrottleListener.doOnMessage(objectMessage);
verify(reverseProxyLoadBalancerStmService).deleteConnectionThrottle(lb);
verify(loadBalancerService).setStatus(lb, LoadBalancerStatus.ERROR);
verify(notificationService)
.saveAlert(
eq(ACCOUNT_ID),
eq(LOAD_BALANCER_ID),
eq(exception),
eq(AlertType.ZEUS_FAILURE.name()),
anyString());
verify(notificationService)
.saveConnectionLimitEvent(
eq(USERNAME),
eq(ACCOUNT_ID),
eq(LOAD_BALANCER_ID),
eq(CONNECTION_LIMIT_ID),
anyString(),
anyString(),
eq(EventType.DELETE_CONNECTION_THROTTLE),
eq(CategoryType.DELETE),
eq(EventSeverity.CRITICAL));
}
@Test
public void testDeleteThrottle() throws Exception {
when(objectMessage.getObject()).thenReturn(lb);
when(loadBalancerService.getWithUserPages(LOAD_BALANCER_ID, ACCOUNT_ID)).thenReturn(lb);
when(config.getString(Matchers.<ConfigurationKey>any())).thenReturn("REST");
deleteConnectionThrottleListener.doOnMessage(objectMessage);
verify(reverseProxyLoadBalancerStmService).deleteConnectionThrottle(lb);
verify(connectionThrottleService).delete(lb);
verify(loadBalancerService).setStatus(lb, LoadBalancerStatus.ACTIVE);
verify(notificationService)
.saveConnectionLimitEvent(
eq(USERNAME),
eq(ACCOUNT_ID),
eq(LOAD_BALANCER_ID),
eq(CONNECTION_LIMIT_ID),
anyString(),
anyString(),
eq(EventType.DELETE_CONNECTION_THROTTLE),
eq(CategoryType.DELETE),
eq(EventSeverity.INFO));
}
