use of com.hortonworks.streamline.streams.notification.Notifier in project streamline by hortonworks.
the class NotificationBoltTest method testAck.
@Test
public void testAck() throws Exception {
Map<String, Object> fieldsAndValues = new HashMap<>();
fieldsAndValues.put("temperature", "100");
final StreamlineEvent event = StreamlineEventImpl.builder().fieldsAndValues(fieldsAndValues).dataSourceId("srcid").build();
final Notification notification = new StreamlineEventAdapter(event);
new MockUp<NotificationQueueHandler>() {
@Mock
public void enqueue(Notifier notifier, Notification notification1) {
notifier.notify(notification);
}
};
new Expectations() {
{
mockProxyUtil.loadClassFromJar(anyString, "TestClass");
result = notifier;
tuple.getValueByField(anyString);
result = event;
}
};
Map<String, String> stormConf = new HashMap<>();
stormConf.put("catalog.root.url", "http://localhost:8080/api/v1/catalog");
stormConf.put("local.notifier.jar.path", "/tmp");
bolt.prepare(stormConf, null, collector);
bolt.execute(tuple);
new Verifications() {
{
hBaseNotificationStore.store(notification);
times = 1;
hBaseNotificationStore.updateNotificationStatus(notification.getId(), Notification.Status.DELIVERED);
times = 1;
collector.ack(tuple);
times = 1;
collector.fail(tuple);
times = 0;
}
};
}
use of com.hortonworks.streamline.streams.notification.Notifier in project streamline by hortonworks.
the class NotificationBoltTest method setUp.
@Before
public void setUp() throws Exception {
bolt = new NotificationBolt(new NotificationSink() {
@Override
public String getNotifierName() {
return NOTIFIER_NAME;
}
@Override
public String getNotifierJarFileName() {
return NOTIFIER_NAME + ".jar";
}
@Override
public String getNotifierClassName() {
return "TestClass";
}
@Override
public Map<String, Object> getNotifierProperties() {
return new HashMap<>();
}
@Override
public Map<String, Object> getNotifierFieldValues() {
return new HashMap<>();
}
});
bolt.withNotificationStoreClass("com.hortonworks.streamline.streams.notification.store.hbase.HBaseNotificationStore");
notifier = new Notifier() {
private NotificationContext myCtx;
@Override
public void open(NotificationContext ctx) {
System.out.println("Notifier open called with ctx " + ctx);
Assert.assertEquals(ctx.getConfig().getProperties(), NOTIFIER_PROPS);
Assert.assertEquals(ctx.getConfig().getDefaultFieldValues(), NOTIFIER_KV);
myCtx = ctx;
}
@Override
public void notify(Notification notification) {
System.out.println("Notifier notify called with notification " + notification);
String temp = (String) notification.getFieldsAndValues().get("temperature");
if (temp == null) {
myCtx.fail(notification.getId());
} else {
myCtx.ack(notification.getId());
}
}
@Override
public void close() {
System.out.println("Notifier Close called");
}
@Override
public boolean isPull() {
return false;
}
@Override
public List<String> getFields() {
return new ArrayList<>();
}
@Override
public NotificationContext getContext() {
return myCtx;
}
};
}
use of com.hortonworks.streamline.streams.notification.Notifier in project streamline by hortonworks.
the class NotificationServiceImplTest method testDuplicateRegister.
@Test
public void testDuplicateRegister() throws Exception {
new Expectations() {
{
mockCtx.getConfig();
times = 3;
result = mockNotifierConfig;
mockNotifierConfig.getClassName();
times = 1;
result = "Test";
mockNotifierConfig.getJarPath();
times = 1;
result = "/tmp/test.jar";
mockProxyUtil.loadClassFromJar("/tmp/test.jar", "Test");
result = mockNotifier;
}
};
notificationService.register("test_notifier", mockCtx);
new Expectations() {
{
mockCtx.getConfig();
times = 2;
result = mockNotifierConfig;
mockNotifierConfig.getClassName();
times = 1;
result = "Test";
mockNotifierConfig.getJarPath();
times = 1;
result = "/tmp/test.jar";
mockProxyUtil.loadClassFromJar("/tmp/test.jar", "Test");
result = mockNotifier;
}
};
Notifier result = notificationService.register("test_notifier", mockCtx);
assertEquals(mockNotifier, result);
new Verifications() {
{
mockNotifier.open(mockCtx);
times = 0;
}
};
}
use of com.hortonworks.streamline.streams.notification.Notifier in project streamline by hortonworks.
the class NotificationServiceImplTest method testRemove.
@Test
public void testRemove() throws Exception {
new Expectations() {
{
mockCtx.getConfig();
times = 3;
result = mockNotifierConfig;
mockNotifierConfig.getClassName();
times = 1;
result = "Test";
mockNotifierConfig.getJarPath();
times = 1;
result = "/tmp/test.jar";
mockProxyUtil.loadClassFromJar("/tmp/test.jar", "Test");
result = mockNotifier;
}
};
notificationService.register("test_notifier", mockCtx);
Notifier result = notificationService.remove("test_notifier");
assertEquals(mockNotifier, result);
new Verifications() {
{
mockNotifier.close();
times = 1;
}
};
// removing again should return null
result = notificationService.remove("test_notifier");
assertEquals(null, result);
}
use of com.hortonworks.streamline.streams.notification.Notifier in project streamline by hortonworks.
the class NotificationServiceImpl method notify.
@Override
public Future<?> notify(String notifierName, Notification notification) {
LOG.debug("Notify notifierName {}, notification {}", notifierName, notification);
notificationStore.ifPresent(s -> s.store(notification));
Notifier notifier = notifiers.get(notifierName);
if (notifier == null) {
throw new NoSuchNotifierException("Notifier not found for id " + notification.getNotifierName());
}
return queueHandler.enqueue(notifier, notification);
}
Aggregations