Examples with HookNotification org.apache.atlas.model.notification.HookNotification used on opensource projects

Example 6 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class KafkaConsumerTest method testReceive.

@Test
public void testReceive() throws Exception {
    Referenceable entity = getEntity(TRAIT_NAME);
    EntityUpdateRequest message = new EntityUpdateRequest("user1", entity);
    String json = AtlasType.toV1Json(new AtlasNotificationMessage<>(new MessageVersion("1.0.0"), message));
    TopicPartition tp = new TopicPartition("ATLAS_HOOK", 0);
    List<ConsumerRecord<String, String>> klist = Collections.singletonList(new ConsumerRecord<>("ATLAS_HOOK", 0, 0L, "mykey", json));
    Map mp = Collections.singletonMap(tp, klist);
    ConsumerRecords records = new ConsumerRecords(mp);
    when(kafkaConsumer.poll(100)).thenReturn(records);
    kafkaConsumer.assign(Collections.singletonList(tp));
    AtlasKafkaConsumer consumer = new AtlasKafkaConsumer(NotificationType.HOOK, kafkaConsumer, false, 100L);
    List<AtlasKafkaMessage<HookNotification>> messageList = consumer.receive();
    assertTrue(messageList.size() > 0);
    HookNotification consumedMessage = messageList.get(0).getMessage();
    assertMessagesEqual(message, consumedMessage, entity);
}

Example 7 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class KafkaNotificationTest method testReceiveKafkaMessages.

@Test
public void testReceiveKafkaMessages() throws Exception {
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u1", new Referenceable("type")));
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u2", new Referenceable("type")));
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u3", new Referenceable("type")));
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u4", new Referenceable("type")));
    NotificationConsumer<Object> consumer = kafkaNotification.createConsumers(NotificationInterface.NotificationType.HOOK, 1).get(0);
    List<AtlasKafkaMessage<Object>> messages = null;
    // fetch starting time
    long startTime = System.currentTimeMillis();
    while ((System.currentTimeMillis() - startTime) < 10000) {
        messages = consumer.receive();
        if (messages.size() > 0) {
            break;
        }
    }
    int i = 1;
    for (AtlasKafkaMessage<Object> msg : messages) {
        HookNotification message = (HookNotification) msg.getMessage();
        assertEquals(message.getUser(), "u" + i++);
    }
    consumer.close();
}

Example 8 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class HookNotificationDeserializerTest method testDeserializeSplitMessage.

@Test
public void testDeserializeSplitMessage() throws Exception {
    Referenceable entity = generateVeryLargeEntityWithTrait();
    EntityUpdateRequest message = new EntityUpdateRequest("user1", entity);
    List<String> jsonMsgList = new ArrayList<>();
    AbstractNotification.createNotificationMessages(message, jsonMsgList);
    assertTrue(jsonMsgList.size() > 1);
    HookNotification deserializedMessage = deserialize(jsonMsgList);
    assertEqualMessage(deserializedMessage, message);
}

Example 9 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class HookNotificationDeserializerTest method testDeserializeLegacyMessage.

// validate deserialization of legacy message, which doesn't use MessageVersion
@Test
public void testDeserializeLegacyMessage() throws Exception {
    Referenceable entity = generateEntityWithTrait();
    EntityUpdateRequest message = new EntityUpdateRequest("user1", entity);
    String jsonMsg = AtlasType.toV1Json(message);
    HookNotification deserializedMessage = deserialize(Collections.singletonList(jsonMsg));
    assertEqualMessage(deserializedMessage, message);
}

Example 10 with HookNotification

use of org.apache.atlas.model.notification.HookNotification in project atlas by apache.

the class HookNotificationDeserializerTest method testDeserializeCompressedMessage.

@Test
public void testDeserializeCompressedMessage() throws Exception {
    Referenceable entity = generateLargeEntityWithTrait();
    EntityUpdateRequest message = new EntityUpdateRequest("user1", entity);
    List<String> jsonMsgList = new ArrayList<>();
    AbstractNotification.createNotificationMessages(message, jsonMsgList);
    assertTrue(jsonMsgList.size() == 1);
    String compressedMsg = jsonMsgList.get(0);
    String uncompressedMsg = AtlasType.toV1Json(message);
    assertTrue(compressedMsg.length() < uncompressedMsg.length(), "Compressed message (" + compressedMsg.length() + ") should be shorter than uncompressed message (" + uncompressedMsg.length() + ")");
    HookNotification deserializedMessage = deserialize(jsonMsgList);
    assertEqualMessage(deserializedMessage, message);
}