Search in sources :

Example 6 with Event

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

the class ParseTreeEvaluatorListenerTest method testSimpleEqualityOperatorExpressionWithLiteralType.

@Test
void testSimpleEqualityOperatorExpressionWithLiteralType() {
    final String equalStatement = "\"a\" == \"a\"";
    final String notEqualStatement = "\"a\" != \"b\"";
    final Event testEvent = createTestEvent(new HashMap<>());
    assertThat(evaluateStatementOnEvent(equalStatement, testEvent), is(true));
    assertThat(evaluateStatementOnEvent(notEqualStatement, testEvent), is(true));
}
Also used : Event(com.amazon.dataprepper.model.event.Event) JacksonEvent(com.amazon.dataprepper.model.event.JacksonEvent) Test(org.junit.jupiter.api.Test)

Example 7 with Event

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

the class ParseTreeEvaluatorListenerTest method testSimpleEqualityOperatorExpressionWithJsonPointerType.

@Test
void testSimpleEqualityOperatorExpressionWithJsonPointerType() {
    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));
}
Also used : Event(com.amazon.dataprepper.model.event.Event) JacksonEvent(com.amazon.dataprepper.model.event.JacksonEvent) Test(org.junit.jupiter.api.Test)

Example 8 with Event

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

the class ParseTreeEvaluatorListenerTest method testSimpleNotOperatorExpressionWithJsonPointerTypeInValidValue.

@Test
void testSimpleNotOperatorExpressionWithJsonPointerTypeInValidValue() {
    final String testKey = "testKey";
    final int testValue = random.nextInt(1000);
    final Map<String, Integer> data = Map.of(testKey, testValue);
    final String notStatement = String.format("not /%s", testKey);
    final Event testEvent = createTestEvent(data);
    assertThrows(ExpressionEvaluationException.class, () -> evaluateStatementOnEvent(notStatement, testEvent));
}
Also used : Event(com.amazon.dataprepper.model.event.Event) JacksonEvent(com.amazon.dataprepper.model.event.JacksonEvent) Test(org.junit.jupiter.api.Test)

Example 9 with Event

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

the class ParseTreeCoercionServiceTest method testCoerceTerminalNodeStringType.

@Test
void testCoerceTerminalNodeStringType() {
    when(token.getType()).thenReturn(DataPrepperExpressionParser.String);
    final String testString = "test string";
    final String testNodeStringValue = String.format("\"%s\"", testString);
    when(terminalNode.getSymbol()).thenReturn(token);
    when(terminalNode.getText()).thenReturn(testNodeStringValue);
    final Event testEvent = createTestEvent(new HashMap<>());
    final Object result = objectUnderTest.coercePrimaryTerminalNode(terminalNode, testEvent);
    assertThat(result, instanceOf(String.class));
    assertThat(result, equalTo(testString));
}
Also used : Event(com.amazon.dataprepper.model.event.Event) TestObject(org.opensearch.dataprepper.expression.util.TestObject) ArgumentMatchers.anyString(org.mockito.ArgumentMatchers.anyString) Test(org.junit.jupiter.api.Test) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest)

Example 10 with Event

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

the class ParseTreeCoercionServiceTest method testCoerceTerminalNodeFloatType.

@Test
void testCoerceTerminalNodeFloatType() {
    when(token.getType()).thenReturn(DataPrepperExpressionParser.Float);
    final Float testFloat = new Random().nextFloat();
    when(terminalNode.getSymbol()).thenReturn(token);
    when(terminalNode.getText()).thenReturn(String.valueOf(testFloat));
    final Event testEvent = createTestEvent(new HashMap<>());
    final Object result = objectUnderTest.coercePrimaryTerminalNode(terminalNode, testEvent);
    assertThat(result, instanceOf(Float.class));
    assertThat(result, equalTo(testFloat));
}
Also used : Random(java.util.Random) Event(com.amazon.dataprepper.model.event.Event) TestObject(org.opensearch.dataprepper.expression.util.TestObject) Test(org.junit.jupiter.api.Test) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest)

Aggregations

Event (com.amazon.dataprepper.model.event.Event)81 Test (org.junit.jupiter.api.Test)52 JacksonEvent (com.amazon.dataprepper.model.event.JacksonEvent)46 Record (com.amazon.dataprepper.model.record.Record)23 ParameterizedTest (org.junit.jupiter.params.ParameterizedTest)22 List (java.util.List)12 TestObject (org.opensearch.dataprepper.expression.util.TestObject)11 ArgumentMatchers.anyString (org.mockito.ArgumentMatchers.anyString)10 ArrayList (java.util.ArrayList)7 HashMap (java.util.HashMap)7 MethodSource (org.junit.jupiter.params.provider.MethodSource)6 ZonedDateTime (java.time.ZonedDateTime)5 Map (java.util.Map)5 ParseTree (org.antlr.v4.runtime.tree.ParseTree)4 AggregateActionInput (com.amazon.dataprepper.plugins.processor.aggregate.AggregateActionInput)3 InputStream (java.io.InputStream)3 LinkedList (java.util.LinkedList)3 Random (java.util.Random)3 Test (org.junit.Test)3 ValueSource (org.junit.jupiter.params.provider.ValueSource)3