use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.
the class PutAllAggregateActionTest method concludeGroup_should_return_groupState_As_An_Event_correctly.
@Test
void concludeGroup_should_return_groupState_As_An_Event_correctly() {
combineAggregateAction = createObjectUnderTest();
final AggregateActionInput aggregateActionInput = new AggregateActionTestUtils.TestAggregateActionInput();
final GroupState groupState = aggregateActionInput.getGroupState();
for (final Map<String, Object> eventMap : eventMaps) {
groupState.putAll(eventMap);
}
final Optional<Event> result = combineAggregateAction.concludeGroup(aggregateActionInput);
assertThat(result.isPresent(), equalTo(true));
assertThat(result.get().getMetadata().getEventType(), equalTo(PutAllAggregateAction.EVENT_TYPE));
assertThat(result.get().toMap(), equalTo(groupState));
}
use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.
the class RemoveDuplicatesAggregateActionTest method concludeGroup_with_non_empty_groupState_returns_empty_Optional.
@Test
void concludeGroup_with_non_empty_groupState_returns_empty_Optional() {
removeDuplicatesAggregateAction = createObjectUnderTest();
final AggregateActionInput aggregateActionInput = new AggregateActionTestUtils.TestAggregateActionInput();
final GroupState groupState = aggregateActionInput.getGroupState();
groupState.put(UUID.randomUUID().toString(), UUID.randomUUID().toString());
final Optional<Event> result = removeDuplicatesAggregateAction.concludeGroup(aggregateActionInput);
assertThat(result.isPresent(), equalTo(false));
}
use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.
the class StringPrepperTests method testStringPrepperDefault.
@Test
public void testStringPrepperDefault() {
final StringPrepper stringPrepper = createObjectUnderTest();
final List<Record<Event>> modifiedRecords = (List<Record<Event>>) stringPrepper.execute(TEST_RECORDS);
stringPrepper.shutdown();
final List<Event> modifiedRecordEvents = modifiedRecords.stream().map(Record::getData).collect(Collectors.toList());
assertThat(modifiedRecordEvents.size(), equalTo(2));
final Event firstEvent = modifiedRecordEvents.get(0);
final Event secondEvent = modifiedRecordEvents.get(1);
assertTrue(firstEvent.containsKey(TEST_KEY));
assertThat(firstEvent.getMetadata().getEventType(), equalTo(TEST_EVENT_TYPE));
assertThat(firstEvent.get(TEST_KEY, String.class), equalTo(UPPERCASE_TEST_STRING.toUpperCase()));
assertTrue(secondEvent.containsKey(TEST_KEY));
assertThat(secondEvent.getMetadata().getEventType(), equalTo(TEST_EVENT_TYPE));
assertThat(secondEvent.get(TEST_KEY, String.class), equalTo(LOWERCASE_TEST_STRING));
}
use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.
the class StringPrepperTests method testStringPrepperLowerCase.
@Test
public void testStringPrepperLowerCase() {
configuration.setUpperCase(false);
final StringPrepper stringPrepper = createObjectUnderTest();
final List<Record<Event>> modifiedRecords = (List<Record<Event>>) stringPrepper.execute(TEST_RECORDS);
stringPrepper.shutdown();
final List<Event> modifiedRecordEvents = modifiedRecords.stream().map(Record::getData).collect(Collectors.toList());
assertThat(modifiedRecordEvents.size(), equalTo(2));
final Event firstEvent = modifiedRecordEvents.get(0);
final Event secondEvent = modifiedRecordEvents.get(1);
assertTrue(firstEvent.containsKey(TEST_KEY));
assertThat(firstEvent.getMetadata().getEventType(), equalTo(TEST_EVENT_TYPE));
assertThat(firstEvent.get(TEST_KEY, String.class), equalTo(UPPERCASE_TEST_STRING));
assertTrue(secondEvent.containsKey(TEST_KEY));
assertThat(secondEvent.getMetadata().getEventType(), equalTo(TEST_EVENT_TYPE));
assertThat(secondEvent.get(TEST_KEY, String.class), equalTo(LOWERCASE_TEST_STRING.toLowerCase()));
}
use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.
the class AggregateActionResponseTest method AggregateActionResponse_fromEvent_returns_correct_AggregateActionResponse.
@Test
void AggregateActionResponse_fromEvent_returns_correct_AggregateActionResponse() {
final Event event = JacksonEvent.builder().withEventType("event").withData(Collections.singletonMap(UUID.randomUUID().toString(), UUID.randomUUID().toString())).build();
final AggregateActionResponse aggregateActionResponse = AggregateActionResponse.fromEvent(event);
assertThat(aggregateActionResponse.getEvent(), equalTo(event));
}
Aggregations