Exemplo n.º 1
0
  @Test
  public void testBatchWithoutProject() throws Exception {
    Event.EventContext api = Event.EventContext.apiKey(apiKeys.writeKey());
    ImmutableMap<String, Object> props =
        ImmutableMap.of("test0", "test", "test1", ImmutableList.of("test"), "test2", false);
    byte[] bytes =
        mapper.writeValueAsBytes(
            ImmutableMap.of(
                "api",
                api,
                "events",
                ImmutableList.of(
                    ImmutableMap.of("collection", "test", "properties", props),
                    ImmutableMap.of("collection", "test", "properties", props))));

    EventList events = mapper.readValue(bytes, EventList.class);

    assertEquals("test", events.project);
    assertEquals(api, events.api);

    for (Event event : events.events) {
      assertEquals("test", event.collection());

      assertEquals(eventBuilder.createEvent("test", props).properties(), event.properties());
    }
  }
Exemplo n.º 2
0
  @Test
  public void testEmptyMap() throws Exception {
    Event.EventContext api = Event.EventContext.apiKey(apiKeys.writeKey());
    byte[] bytes =
        mapper.writeValueAsBytes(
            ImmutableMap.of(
                "collection",
                "test",
                "api",
                api,
                "properties",
                ImmutableMap.of(
                    "test",
                    1,
                    "test2",
                    new HashMap<String, String>() {
                      {
                        put("a", null);
                      }
                    }),
                "test20",
                ImmutableMap.of(),
                "test3",
                true));

    Event event = mapper.readValue(bytes, Event.class);

    assertEquals("test", event.project());
    assertEquals("test", event.collection());
    assertEquals(api, event.api());
    assertEquals(
        eventBuilder.createEvent("test", ImmutableMap.of("test", 1.0, "test3", true)).properties(),
        event.properties());
  }
Exemplo n.º 3
0
  @Test
  public void testInvalidMap() throws Exception {
    Event.EventContext api = Event.EventContext.apiKey(apiKeys.writeKey());
    byte[] bytes =
        mapper.writeValueAsBytes(
            ImmutableMap.of(
                "collection",
                "test",
                "api",
                api,
                "properties",
                ImmutableMap.of("test1", ImmutableMap.of("test", 1, "test2", "test"))));

    Event event = mapper.readValue(bytes, Event.class);

    assertEquals("test", event.project());
    assertEquals("test", event.collection());
    assertEquals(api, event.api());
    assertEquals(
        eventBuilder
            .createEvent(
                "test", ImmutableMap.of("test1", ImmutableMap.of("test", 1.0, "test2", 0.0)))
            .properties(),
        event.properties());
  }
Exemplo n.º 4
0
  @Test(expectedExceptions = RakamException.class)
  public void testInvalidField() throws Exception {
    Event.EventContext api = Event.EventContext.apiKey(apiKeys.writeKey());
    byte[] bytes =
        mapper.writeValueAsBytes(
            ImmutableMap.of(
                "collection",
                "test",
                "api",
                api,
                "properties",
                ImmutableMap.of(
                    "test0", "test", "test1", ImmutableList.of("test", "test"), "test2", false),
                "test",
                "test"));

    Event event = mapper.readValue(bytes, Event.class);
    ;

    assertEquals("test", event.project());
    assertEquals("test", event.collection());
    assertEquals(api, event.api());
    assertEquals(
        eventBuilder.createEvent("test", ImmutableMap.of()).properties(), event.properties());
  }
Exemplo n.º 5
0
  @Test
  public void testMapType() throws Exception {
    Event.EventContext api = Event.EventContext.apiKey(apiKeys.writeKey());
    ImmutableMap<String, Object> properties =
        ImmutableMap.of(
            "test0",
            "test",
            "test1",
            ImmutableMap.of("a", 4.0, "b", 5.0, "c", 6.0, "d", 7.0),
            "test2",
            false);
    byte[] bytes =
        mapper.writeValueAsBytes(
            ImmutableMap.of(
                "collection", "test",
                "api", api,
                "properties", properties));

    Event event = mapper.readValue(bytes, Event.class);

    assertEquals("test", event.project());
    assertEquals("test", event.collection());
    assertEquals(api, event.api());
    assertEquals(eventBuilder.createEvent("test", properties).properties(), event.properties());
  }
Exemplo n.º 6
0
  @Test
  public void testPrimitiveTypes() throws Exception {
    Event.EventContext api = Event.EventContext.apiKey(apiKeys.writeKey());
    ImmutableMap<String, Object> properties =
        ImmutableMap.of(
            "test",
            1L,
            "test1",
            false,
            "test2",
            Instant.now(),
            "test3",
            "test",
            "test4",
            LocalDate.now());

    byte[] bytes =
        mapper.writeValueAsBytes(
            ImmutableMap.of(
                "collection", "test",
                "api", api,
                "properties", properties));

    Event event = mapper.readValue(bytes, Event.class);

    assertEquals("test", event.project());
    assertEquals("test", event.collection());
    assertEquals(api, event.api());
    assertEquals(eventBuilder.createEvent("test", properties).properties(), event.properties());

    assertEquals(
        ImmutableSet.copyOf(metastore.getCollection("test", "test")),
        ImmutableSet.of(
            new SchemaField("test", FieldType.DOUBLE),
            new SchemaField("_user", FieldType.STRING),
            new SchemaField("test1", FieldType.BOOLEAN),
            new SchemaField("test2", FieldType.TIMESTAMP),
            new SchemaField("test3", FieldType.STRING),
            new SchemaField("test4", FieldType.DATE)));
  }
Exemplo n.º 7
0
 @AfterMethod
 public void tearDownMethod() throws Exception {
   metastore.deleteProject("test");
   eventDeserializer.cleanCache();
   eventBuilder.cleanCache();
 }