Examples with Event com.amazon.dataprepper.model.event.Event used on opensource projects

Example 76 with Event

use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.

the class RemoveDuplicatesAggregateActionTest method concludeGroup_with_empty_groupState_returns_empty_Optional.

@Test
void concludeGroup_with_empty_groupState_returns_empty_Optional() {
    removeDuplicatesAggregateAction = createObjectUnderTest();
    final AggregateActionInput aggregateActionInput = new AggregateActionTestUtils.TestAggregateActionInput();
    final Optional<Event> result = removeDuplicatesAggregateAction.concludeGroup(aggregateActionInput);
    assertThat(result.isPresent(), equalTo(false));
}

Example 77 with Event

use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.

the class MetricsPluginSumTest method test.

@Test
public void test() throws JsonProcessingException {
    final KeyValue childAttr1 = KeyValue.newBuilder().setKey("statement").setValue(AnyValue.newBuilder().setIntValue(1_000).build()).build();
    final KeyValue childAttr2 = KeyValue.newBuilder().setKey("statement.params").setValue(AnyValue.newBuilder().setStringValue("us-east-1").build()).build();
    final KeyValue attribute1 = KeyValue.newBuilder().setKey("db.details").setValue(AnyValue.newBuilder().setKvlistValue(KeyValueList.newBuilder().addAllValues(Arrays.asList(childAttr1, childAttr2)).build()).build()).build();
    final KeyValue attribute2 = KeyValue.newBuilder().setKey("http.status").setValue(AnyValue.newBuilder().setStringValue("4xx").build()).build();
    final AnyValue anyValue1 = AnyValue.newBuilder().setStringValue("asdf").build();
    final AnyValue anyValue2 = AnyValue.newBuilder().setDoubleValue(2000.123).build();
    final AnyValue anyValue3 = AnyValue.newBuilder().setKvlistValue(KeyValueList.newBuilder().addAllValues(Arrays.asList(childAttr1, childAttr2))).build();
    final ArrayValue arrayValue = ArrayValue.newBuilder().addAllValues(Arrays.asList(anyValue1, anyValue2, anyValue3)).build();
    final KeyValue attribute3 = KeyValue.newBuilder().setKey("aws.details").setValue(AnyValue.newBuilder().setArrayValue(arrayValue)).build();
    NumberDataPoint dataPoint = NumberDataPoint.newBuilder().setAsInt(3).addAllAttributes(Arrays.asList(attribute1, attribute2, attribute3)).build();
    Sum sum = Sum.newBuilder().addAllDataPoints(Collections.singletonList(dataPoint)).build();
    Metric metric = Metric.newBuilder().setSum(sum).setUnit("seconds").setName("name").setDescription("description").build();
    InstrumentationLibraryMetrics instLib = InstrumentationLibraryMetrics.newBuilder().setInstrumentationLibrary(InstrumentationLibrary.newBuilder().setVersion("v1").setName("name").build()).addMetrics(metric).build();
    Resource resource = Resource.newBuilder().addAttributes(KeyValue.newBuilder().setKey("service.name").setValue(AnyValue.newBuilder().setStringValue("service").build())).build();
    ResourceMetrics resourceMetrics = ResourceMetrics.newBuilder().setResource(resource).addInstrumentationLibraryMetrics(instLib).build();
    ExportMetricsServiceRequest exportMetricRequest = ExportMetricsServiceRequest.newBuilder().addResourceMetrics(resourceMetrics).build();
    Record record = new Record<>(exportMetricRequest);
    List<Record<Event>> rec = (List<Record<Event>>) rawProcessor.doExecute(Arrays.asList(record));
    Record<Event> firstRecord = rec.get(0);
    ObjectMapper objectMapper = new ObjectMapper();
    Map map = objectMapper.readValue(firstRecord.getData().toJsonString(), Map.class);
    assertSumProcessing(map);
}

Example 78 with Event

use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.

the class ParseTreeEvaluatorListenerTest method testSimpleEqualityOperatorExpressionWithJsonPointerTypeExistingKey.

@Test
void testSimpleEqualityOperatorExpressionWithJsonPointerTypeExistingKey() {
    final String testKey = "testKey";
    final Integer testValue = random.nextInt(1000);
    final Map<String, Integer> data = Map.of(testKey, testValue);
    final Event testEvent = createTestEvent(data);
    final String equalStatement = String.format("/%s == %d", testKey, testValue);
    final String notEqualStatement = String.format("/%s != %d", testKey, testValue + 1);
    assertThat(evaluateStatementOnEvent(equalStatement, testEvent), is(true));
    assertThat(evaluateStatementOnEvent(notEqualStatement, testEvent), is(true));
}

Example 79 with Event

use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.

the class ParseTreeCoercionServiceTest method testCoerceTerminalNodeJsonPointerTypeMissingKey.

@Test
void testCoerceTerminalNodeJsonPointerTypeMissingKey() {
    final String testMissingKey = "missingKey";
    final String testJsonPointerKey = "/" + testMissingKey;
    final Event testEvent = createTestEvent(new HashMap<>());
    when(token.getType()).thenReturn(DataPrepperExpressionParser.JsonPointer);
    when(terminalNode.getSymbol()).thenReturn(token);
    when(terminalNode.getText()).thenReturn(testJsonPointerKey);
    final Object result = objectUnderTest.coercePrimaryTerminalNode(terminalNode, testEvent);
    assertThat(result, nullValue());
}

Example 80 with Event

use of com.amazon.dataprepper.model.event.Event in project data-prepper by opensearch-project.

the class ParseTreeCoercionServiceTest method testCoerceTerminalNodeEscapeJsonPointerType.

@ParameterizedTest
@MethodSource("provideKeys")
void testCoerceTerminalNodeEscapeJsonPointerType(final String testKey, final String testEscapeJsonPointer) throws ExpressionCoercionException {
    final String testValue = "test value";
    final Event testEvent = createTestEvent(Map.of(testKey, testValue));
    when(token.getType()).thenReturn(DataPrepperExpressionParser.EscapedJsonPointer);
    when(terminalNode.getSymbol()).thenReturn(token);
    when(terminalNode.getText()).thenReturn(testEscapeJsonPointer);
    final Object result = objectUnderTest.coercePrimaryTerminalNode(terminalNode, testEvent);
    assertThat(result, instanceOf(String.class));
    assertThat(result, equalTo(testValue));
}