Search in sources :

Example 11 with UpdateAlertTriggerEntity

use of io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity in project gravitee-management-rest-api by gravitee-io.

the class ApplicationAlertServiceTest method shouldHandleNotificationTemplateUpdatedEventStatusAlert.

@Test
public void shouldHandleNotificationTemplateUpdatedEventStatusAlert() throws Exception {
    NotificationTemplateEntity notificationTemplate = new NotificationTemplateEntity();
    notificationTemplate.setHook(AlertHook.CONSUMER_HTTP_STATUS.name());
    notificationTemplate.setTitle("notification-template-title");
    notificationTemplate.setContent("notification-template-content");
    NotificationTemplateEvent notificationTemplateEvent = new NotificationTemplateEvent("org-id", notificationTemplate);
    final SimpleEvent<ApplicationAlertEventType, Object> event = new SimpleEvent<>(ApplicationAlertEventType.NOTIFICATION_TEMPLATE_UPDATE, notificationTemplateEvent);
    ApplicationListItem app1 = new ApplicationListItem();
    app1.setId("app1");
    ApplicationListItem app2 = new ApplicationListItem();
    app2.setId("app2");
    when(applicationService.findByOrganization("org-id")).thenReturn(new HashSet<>(Arrays.asList(app1, app2)));
    final AlertTriggerEntity alertTrigger = mock(AlertTriggerEntity.class);
    when(alertTrigger.getType()).thenReturn("METRICS_RATE");
    when(alertService.findByReferenceAndReferenceIds(AlertReferenceType.APPLICATION, Arrays.asList("app1", "app2"))).thenReturn(Collections.singletonList(alertTrigger));
    Notification notification = new Notification();
    notification.setType("default-email");
    notification.setConfiguration("");
    when(alertTrigger.getNotifications()).thenReturn(Collections.singletonList(notification));
    JsonNode emailNode = JsonNodeFactory.instance.objectNode().put("to", "to").put("from", "from").put("subject", "subject").put("body", "body");
    when(mapper.readTree(notification.getConfiguration())).thenReturn(emailNode);
    cut.handleEvent(event);
    ArgumentCaptor<UpdateAlertTriggerEntity> updatingCaptor = ArgumentCaptor.forClass(UpdateAlertTriggerEntity.class);
    verify(alertService, times(1)).update(updatingCaptor.capture());
    final UpdateAlertTriggerEntity updating = updatingCaptor.getValue();
    assertThat(updating.getNotifications().get(0).getConfiguration()).contains("notification-template-content");
    assertThat(updating.getNotifications().get(0).getConfiguration()).contains("notification-template-title");
}
Also used : UpdateAlertTriggerEntity(io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity) SimpleEvent(io.gravitee.common.event.impl.SimpleEvent) NotificationTemplateEntity(io.gravitee.rest.api.model.notification.NotificationTemplateEntity) JsonNode(com.fasterxml.jackson.databind.JsonNode) UpdateAlertTriggerEntity(io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity) NewAlertTriggerEntity(io.gravitee.rest.api.model.alert.NewAlertTriggerEntity) AlertTriggerEntity(io.gravitee.rest.api.model.alert.AlertTriggerEntity) Notification(io.gravitee.notifier.api.Notification) ApplicationListItem(io.gravitee.rest.api.model.application.ApplicationListItem) NotificationTemplateEvent(io.gravitee.rest.api.model.notification.NotificationTemplateEvent) ApplicationAlertEventType(io.gravitee.rest.api.model.alert.ApplicationAlertEventType) Test(org.junit.Test)

Example 12 with UpdateAlertTriggerEntity

use of io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity in project gravitee-management-rest-api by gravitee-io.

the class ApplicationAlertServiceTest method shouldDeleteMember.

@Test
public void shouldDeleteMember() throws Exception {
    AlertTriggerEntity trigger1 = mock(AlertTriggerEntity.class);
    Notification notification = new Notification();
    notification.setType("default-email");
    notification.setConfiguration("");
    Notification notification2 = new Notification();
    notification2.setType("default-email");
    notification2.setConfiguration("");
    when(trigger1.getNotifications()).thenReturn(Collections.singletonList(notification));
    AlertTriggerEntity trigger2 = mock(AlertTriggerEntity.class);
    when(trigger2.getNotifications()).thenReturn(Collections.singletonList(notification2));
    List<AlertTriggerEntity> triggers = new ArrayList<>();
    triggers.add(trigger1);
    triggers.add(trigger2);
    when(alertService.findByReference(AlertReferenceType.APPLICATION, APPLICATION_ID)).thenReturn(triggers);
    JsonNode emailNode = JsonNodeFactory.instance.objectNode().put("to", "to,delete@mail.gio").put("from", "from").put("subject", "subject").put("body", "body");
    when(mapper.readTree(notification.getConfiguration())).thenReturn(emailNode);
    when(mapper.readTree(notification2.getConfiguration())).thenReturn(emailNode);
    cut.deleteMemberFromApplication(APPLICATION_ID, "delete@mail.gio");
    ArgumentCaptor<UpdateAlertTriggerEntity> updatingCaptor = ArgumentCaptor.forClass(UpdateAlertTriggerEntity.class);
    verify(alertService, times(2)).update(updatingCaptor.capture());
    final UpdateAlertTriggerEntity updating = updatingCaptor.getValue();
    assertThat(updating.getNotifications().get(0).getConfiguration()).contains("to");
    assertThat(updating.getNotifications().get(0).getConfiguration()).doesNotContain("delete@mail.gio");
}
Also used : UpdateAlertTriggerEntity(io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity) ArrayList(java.util.ArrayList) JsonNode(com.fasterxml.jackson.databind.JsonNode) UpdateAlertTriggerEntity(io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity) NewAlertTriggerEntity(io.gravitee.rest.api.model.alert.NewAlertTriggerEntity) AlertTriggerEntity(io.gravitee.rest.api.model.alert.AlertTriggerEntity) Notification(io.gravitee.notifier.api.Notification) Test(org.junit.Test)

Aggregations

UpdateAlertTriggerEntity (io.gravitee.rest.api.model.alert.UpdateAlertTriggerEntity)12 Test (org.junit.Test)8 AlertTriggerEntity (io.gravitee.rest.api.model.alert.AlertTriggerEntity)7 NewAlertTriggerEntity (io.gravitee.rest.api.model.alert.NewAlertTriggerEntity)6 JsonNode (com.fasterxml.jackson.databind.JsonNode)4 Notification (io.gravitee.notifier.api.Notification)4 AlertInput (io.gravitee.rest.api.portal.rest.model.AlertInput)3 RateCondition (io.gravitee.alert.api.condition.RateCondition)2 ThresholdRangeCondition (io.gravitee.alert.api.condition.ThresholdRangeCondition)2 SimpleEvent (io.gravitee.common.event.impl.SimpleEvent)2 ApplicationAlertEventType (io.gravitee.rest.api.model.alert.ApplicationAlertEventType)2 ApplicationListItem (io.gravitee.rest.api.model.application.ApplicationListItem)2 NotificationTemplateEntity (io.gravitee.rest.api.model.notification.NotificationTemplateEntity)2 NotificationTemplateEvent (io.gravitee.rest.api.model.notification.NotificationTemplateEvent)2 ArrayList (java.util.ArrayList)2 AggregationCondition (io.gravitee.alert.api.condition.AggregationCondition)1 Alert (io.gravitee.rest.api.portal.rest.model.Alert)1 Permissions (io.gravitee.rest.api.portal.rest.security.Permissions)1 Consumes (javax.ws.rs.Consumes)1 PUT (javax.ws.rs.PUT)1