Search in sources :

Example 1 with AlertResponder

use of com.thinkbiganalytics.alerts.api.AlertResponder in project kylo by Teradata.

the class AggregatingAlertProvider method accept.

/* (non-Javadoc)
     * @see reactor.fn.Consumer#accept(java.lang.Object)
     */
@Override
public void accept(Event<Alert> event) {
    final Alert alert = unwrapAlert(event.getData());
    final AlertManager mgr = (AlertManager) alert.getSource();
    final List<AlertResponder> responders = snapshotResponderts();
    responders.forEach(responder -> {
        AlertResponse resp = mgr.getResponse(alert);
        AlertResponseWrapper wrapper = new AlertResponseWrapper(resp);
        responder.alertChange(alert, wrapper);
    });
}
Also used : AlertManager(com.thinkbiganalytics.alerts.spi.AlertManager) AlertResponse(com.thinkbiganalytics.alerts.api.AlertResponse) Alert(com.thinkbiganalytics.alerts.api.Alert) SourceAlert(com.thinkbiganalytics.alerts.api.SourceAlert) AlertResponder(com.thinkbiganalytics.alerts.api.AlertResponder)

Example 2 with AlertResponder

use of com.thinkbiganalytics.alerts.api.AlertResponder in project kylo by Teradata.

the class AggregatingAlertProviderTest method testRespondToChange.

@Test
public void testRespondToChange() throws InterruptedException {
    TestAlert initMgrAlert = new TestAlert(this.manager, true);
    CountDownLatch latch = new CountDownLatch(1);
    this.provider.addAlertManager(this.manager);
    when(this.manager.getAlert(any(Alert.ID.class))).thenReturn(Optional.of(initMgrAlert));
    when(this.manager.getAlertAsServiceAccount(any(Alert.ID.class))).thenReturn(Optional.of(initMgrAlert));
    when(this.manager.getResponse(any(Alert.class))).thenReturn(this.response);
    when(this.response.handle(any(String.class), any(Serializable.class))).thenReturn(initMgrAlert);
    doAnswer(new Answer<Void>() {

        @Override
        public Void answer(InvocationOnMock invocation) throws Throwable {
            latch.countDown();
            return null;
        }
    }).when(this.listener).alertChange(any(Alert.class));
    this.provider.respondTo(new SourceAlertID(initMgrAlert.getId(), this.manager), new AlertResponder() {

        @Override
        public void alertChange(Alert alert, AlertResponse response) {
            response.handle(null, "handled");
        }
    });
    latch.await(10, TimeUnit.SECONDS);
    verify(this.response).handle(eq(null), eq("handled"));
    verify(this.listener, atLeastOnce()).alertChange(any(Alert.class));
}
Also used : Serializable(java.io.Serializable) AlertResponse(com.thinkbiganalytics.alerts.api.AlertResponse) CountDownLatch(java.util.concurrent.CountDownLatch) AlertResponder(com.thinkbiganalytics.alerts.api.AlertResponder) SourceAlertID(com.thinkbiganalytics.alerts.api.core.AggregatingAlertProvider.SourceAlertID) InvocationOnMock(org.mockito.invocation.InvocationOnMock) Alert(com.thinkbiganalytics.alerts.api.Alert) SourceAlertID(com.thinkbiganalytics.alerts.api.core.AggregatingAlertProvider.SourceAlertID) Test(org.junit.Test)

Aggregations

Alert (com.thinkbiganalytics.alerts.api.Alert)2 AlertResponder (com.thinkbiganalytics.alerts.api.AlertResponder)2 AlertResponse (com.thinkbiganalytics.alerts.api.AlertResponse)2 SourceAlert (com.thinkbiganalytics.alerts.api.SourceAlert)1 SourceAlertID (com.thinkbiganalytics.alerts.api.core.AggregatingAlertProvider.SourceAlertID)1 AlertManager (com.thinkbiganalytics.alerts.spi.AlertManager)1 Serializable (java.io.Serializable)1 CountDownLatch (java.util.concurrent.CountDownLatch)1 Test (org.junit.Test)1 InvocationOnMock (org.mockito.invocation.InvocationOnMock)1