Search in sources :

Example 1 with ArraySplitOperation

use of com.nextdoor.bender.operation.json.array.ArraySplitOperation in project bender by Nextdoor.

the class ArraySplitOperationTest method testEventString.

@Test
public void testEventString() throws JsonSyntaxException, UnsupportedEncodingException, IOException {
    JsonParser parser = new JsonParser();
    JsonElement input = parser.parse(getResourceString("array_input.json"));
    GenericJsonEvent devent = new GenericJsonEvent(input.getAsJsonObject());
    ArraySplitOperation operation = new ArraySplitOperation("$.arr");
    InternalEvent ievent = new InternalEvent("", null, 0);
    ievent.setEventObj(devent);
    List<String> actual = operation.perform(ievent).stream().map(InternalEvent::getEventString).collect(Collectors.toList());
    List<String> expected = Arrays.asList("{\"foo\":1}", "{\"foo\":2}", "{\"foo\":3}");
    assertEquals(expected, actual);
}
Also used : ArraySplitOperation(com.nextdoor.bender.operation.json.array.ArraySplitOperation) GenericJsonEvent(com.nextdoor.bender.deserializer.json.GenericJsonEvent) JsonElement(com.google.gson.JsonElement) JsonParser(com.google.gson.JsonParser) InternalEvent(com.nextdoor.bender.InternalEvent) Test(org.junit.Test) OperationTest(com.nextdoor.bender.operations.json.OperationTest)

Example 2 with ArraySplitOperation

use of com.nextdoor.bender.operation.json.array.ArraySplitOperation in project bender by Nextdoor.

the class ArraySplitOperationTest method testTimeMatch.

@Test
public void testTimeMatch() throws JsonSyntaxException, UnsupportedEncodingException, IOException {
    JsonParser parser = new JsonParser();
    TestContext t = new TestContext();
    t.setFunctionName("foo");
    LambdaContext lctx = new LambdaContext(t);
    JsonElement input = parser.parse(getResourceString("array_input.json"));
    GenericJsonEvent devent = new GenericJsonEvent(input.getAsJsonObject());
    ArraySplitOperation operation = new ArraySplitOperation("$.arr");
    InternalEvent ievent = new InternalEvent("", lctx, 123);
    ievent.setEventObj(devent);
    ievent.setEventTime(124);
    List<InternalEvent> events = operation.perform(ievent);
    assertEquals(ievent.getArrivalTime(), events.get(0).getArrivalTime());
    assertEquals(ievent.getEventTime(), events.get(0).getEventTime());
    assertEquals(ievent.getArrivalTime(), events.get(1).getArrivalTime());
    assertEquals(ievent.getEventTime(), events.get(1).getEventTime());
    assertEquals(ievent.getArrivalTime(), events.get(2).getArrivalTime());
    assertEquals(ievent.getEventTime(), events.get(2).getEventTime());
}
Also used : ArraySplitOperation(com.nextdoor.bender.operation.json.array.ArraySplitOperation) GenericJsonEvent(com.nextdoor.bender.deserializer.json.GenericJsonEvent) JsonElement(com.google.gson.JsonElement) TestContext(com.nextdoor.bender.aws.TestContext) LambdaContext(com.nextdoor.bender.LambdaContext) JsonParser(com.google.gson.JsonParser) InternalEvent(com.nextdoor.bender.InternalEvent) Test(org.junit.Test) OperationTest(com.nextdoor.bender.operations.json.OperationTest)

Example 3 with ArraySplitOperation

use of com.nextdoor.bender.operation.json.array.ArraySplitOperation in project bender by Nextdoor.

the class ArraySplitOperationTest method testMetadataMatch.

@Test
public void testMetadataMatch() throws JsonSyntaxException, UnsupportedEncodingException, IOException {
    JsonParser parser = new JsonParser();
    TestContext t = new TestContext();
    t.setFunctionName("foo");
    LambdaContext lctx = new LambdaContext(t);
    JsonElement input = parser.parse(getResourceString("array_input.json"));
    GenericJsonEvent devent = new GenericJsonEvent(input.getAsJsonObject());
    ArraySplitOperation operation = new ArraySplitOperation("$.arr");
    InternalEvent ievent = new InternalEvent("", lctx, 1);
    ievent.setEventObj(devent);
    List<InternalEvent> events = operation.perform(ievent);
    assertEquals(new Long(1), events.get(0).getEventMetadata().get("arrivalEpochMs"));
    assertEquals("6a00541bfa24fd59884de557be71f3a5a1344613", events.get(0).getEventMetadata().get("eventSha1Hash"));
    assertEquals(new Long(1), events.get(1).getEventMetadata().get("arrivalEpochMs"));
    assertEquals("5cbc013a144777d9ce51385ac24a95ce05f77075", events.get(1).getEventMetadata().get("eventSha1Hash"));
    assertEquals(new Long(1), events.get(2).getEventMetadata().get("arrivalEpochMs"));
    assertEquals("6c3023db7ef03721c27df9f33b26e0d2cbc84d6b", events.get(2).getEventMetadata().get("eventSha1Hash"));
}
Also used : ArraySplitOperation(com.nextdoor.bender.operation.json.array.ArraySplitOperation) GenericJsonEvent(com.nextdoor.bender.deserializer.json.GenericJsonEvent) JsonElement(com.google.gson.JsonElement) TestContext(com.nextdoor.bender.aws.TestContext) LambdaContext(com.nextdoor.bender.LambdaContext) JsonParser(com.google.gson.JsonParser) InternalEvent(com.nextdoor.bender.InternalEvent) Test(org.junit.Test) OperationTest(com.nextdoor.bender.operations.json.OperationTest)

Example 4 with ArraySplitOperation

use of com.nextdoor.bender.operation.json.array.ArraySplitOperation in project bender by Nextdoor.

the class ArraySplitOperationTest method testMissingField.

@Test(expected = OperationException.class)
public void testMissingField() throws JsonSyntaxException, UnsupportedEncodingException, IOException {
    JsonParser parser = new JsonParser();
    TestContext t = new TestContext();
    t.setFunctionName("foo");
    LambdaContext lctx = new LambdaContext(t);
    JsonElement input = parser.parse(getResourceString("array_input.json"));
    GenericJsonEvent devent = new GenericJsonEvent(input.getAsJsonObject());
    ArraySplitOperation operation = new ArraySplitOperation("$.bar]");
    InternalEvent ievent = new InternalEvent("", lctx, 123);
    ievent.setEventObj(devent);
    ievent.setEventTime(124);
    List<InternalEvent> events = operation.perform(ievent);
}
Also used : ArraySplitOperation(com.nextdoor.bender.operation.json.array.ArraySplitOperation) GenericJsonEvent(com.nextdoor.bender.deserializer.json.GenericJsonEvent) JsonElement(com.google.gson.JsonElement) TestContext(com.nextdoor.bender.aws.TestContext) LambdaContext(com.nextdoor.bender.LambdaContext) JsonParser(com.google.gson.JsonParser) InternalEvent(com.nextdoor.bender.InternalEvent) Test(org.junit.Test) OperationTest(com.nextdoor.bender.operations.json.OperationTest)

Example 5 with ArraySplitOperation

use of com.nextdoor.bender.operation.json.array.ArraySplitOperation in project bender by Nextdoor.

the class ArraySplitOperationTest method testNonArray.

@Test(expected = OperationException.class)
public void testNonArray() throws JsonSyntaxException, UnsupportedEncodingException, IOException {
    JsonParser parser = new JsonParser();
    TestContext t = new TestContext();
    t.setFunctionName("foo");
    LambdaContext lctx = new LambdaContext(t);
    JsonElement input = parser.parse(getResourceString("array_input.json"));
    GenericJsonEvent devent = new GenericJsonEvent(input.getAsJsonObject());
    ArraySplitOperation operation = new ArraySplitOperation("$.arr[0]");
    InternalEvent ievent = new InternalEvent("", lctx, 123);
    ievent.setEventObj(devent);
    ievent.setEventTime(124);
    List<InternalEvent> events = operation.perform(ievent);
}
Also used : ArraySplitOperation(com.nextdoor.bender.operation.json.array.ArraySplitOperation) GenericJsonEvent(com.nextdoor.bender.deserializer.json.GenericJsonEvent) JsonElement(com.google.gson.JsonElement) TestContext(com.nextdoor.bender.aws.TestContext) LambdaContext(com.nextdoor.bender.LambdaContext) JsonParser(com.google.gson.JsonParser) InternalEvent(com.nextdoor.bender.InternalEvent) Test(org.junit.Test) OperationTest(com.nextdoor.bender.operations.json.OperationTest)

Aggregations

JsonElement (com.google.gson.JsonElement)8 InternalEvent (com.nextdoor.bender.InternalEvent)8 GenericJsonEvent (com.nextdoor.bender.deserializer.json.GenericJsonEvent)8 ArraySplitOperation (com.nextdoor.bender.operation.json.array.ArraySplitOperation)8 OperationTest (com.nextdoor.bender.operations.json.OperationTest)8 Test (org.junit.Test)8 JsonParser (com.google.gson.JsonParser)7 LambdaContext (com.nextdoor.bender.LambdaContext)7 TestContext (com.nextdoor.bender.aws.TestContext)7 JsonSyntaxException (com.google.gson.JsonSyntaxException)1 OperationException (com.nextdoor.bender.operation.OperationException)1 IOException (java.io.IOException)1 UnsupportedEncodingException (java.io.UnsupportedEncodingException)1 Arrays (java.util.Arrays)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Map (java.util.Map)1 Collectors (java.util.stream.Collectors)1 Assert.assertEquals (org.junit.Assert.assertEquals)1 Assert.assertTrue (org.junit.Assert.assertTrue)1