private static String toJson(DataResponse<?> builder) throws IOException {

    ByteArrayOutputStream out = new ByteArrayOutputStream();

    try (JsonGenerator generator =
        new JacksonService().getJsonFactory().createJsonGenerator(out, JsonEncoding.UTF8)) {
      builder.getEncoder().encode(null, builder.getObjects(), generator);
    }

    return new String(out.toByteArray(), "UTF-8");
  }
  @Test
  public void testEncoder_FilterNoMatch() throws IOException {

    filters.add(
        new EncoderFilter() {

          @Override
          public boolean matches(ClientEntity<?> clientEntity) {
            return false;
          }

          @Override
          public boolean encode(
              String propertyName, Object object, JsonGenerator out, Encoder delegate)
              throws IOException {

            fail("Non matching filter was not supposed to be invoked");
            return false;
          }

          @Override
          public boolean willEncode(String propertyName, Object object, Encoder delegate) {
            fail("Non matching filter was not supposed to be invoked");
            return false;
          }
        });

    ClientEntity<E2> descriptor = getClientEntity(E2.class);
    descriptor.setIdIncluded(true);

    DataResponse<E2> builder = DataResponse.forType(E2.class).withClientEntity(descriptor);
    encoderService.makeEncoder(builder);

    ObjectContext context = cayenneService.newContext();
    E2 e21 = new E2();
    e21.setObjectId(new ObjectId("E2", E2.ID_PK_COLUMN, 7));
    e21.setName("XYZ");
    e21.setAddress("bla bla street");
    context.registerNewObject(e21);

    builder.getEncoder().encode(null, Collections.singletonList(e21), mock(JsonGenerator.class));
  }
  @Test
  public void testGetRootEncoder_ExcludedRelationshipAttributes() throws IOException {
    ClientEntity<E3> e3Descriptor = getClientEntity(E3.class);
    e3Descriptor.setIdIncluded(true);
    e3Descriptor.getAttributes().add(E3.NAME.getName());

    ClientEntity<E2> descriptor = getClientEntity(E2.class);
    descriptor.setIdIncluded(true);
    descriptor.getRelationships().put(E2.E3S.getName(), e3Descriptor);

    ObjectContext context = cayenneService.newContext();
    E2 e2 = new E2();
    e2.setObjectId(new ObjectId("E2", E2.ID_PK_COLUMN, 7));
    e2.setName("XYZ");
    e2.setAddress("bla bla street");
    context.registerNewObject(e2);

    E3 e31 = new E3();
    e31.setObjectId(new ObjectId("E3", E3.ID_PK_COLUMN, 5));
    e31.setName("31");
    e31.setPhoneNumber("+12345678");
    context.registerNewObject(e31);
    e2.addToE3s(e31);

    E3 e32 = new E3();
    e32.setObjectId(new ObjectId("E3", E3.ID_PK_COLUMN, 6));
    e32.setName("32");
    e31.setPhoneNumber("+87654321");
    context.registerNewObject(e32);
    e2.addToE3s(e32);

    DataResponse<E2> builder =
        DataResponse.forType(E2.class)
            .withClientEntity(descriptor)
            .withObjects(Collections.singletonList(e2));
    encoderService.makeEncoder(builder);

    assertEquals(
        "[{\"id\":7,\"e3s\":[{\"id\":5,\"name\":\"31\"},{\"id\":6,\"name\":\"32\"}]}]",
        toJson(builder));
  }
  @Test
  public void testGetRootEncoder_ExcludedAttributes() throws IOException {
    // empty filter - must only include id
    ClientEntity<E1> descriptor = getClientEntity(E1.class);
    descriptor.setIdIncluded(true);

    E1 e1 = new E1();
    e1.setObjectId(new ObjectId("E1", E1.ID_PK_COLUMN, 777));
    e1.setName("XYZ");
    e1.setAge(30);
    e1.setDescription("test");

    DataResponse<E1> builder =
        DataResponse.forType(E1.class)
            .withClientEntity(descriptor)
            .withObjects(Collections.singletonList(e1));

    encoderService.makeEncoder(builder);

    assertEquals("[{\"id\":777}]", toJson(builder));
  }
  @Test
  public void testEncoder_FilteredToOne() throws IOException {

    filters.add(
        new EncoderFilter() {

          @Override
          public boolean matches(ClientEntity<?> clientEntity) {
            return true;
          }

          @Override
          public boolean encode(
              String propertyName, Object object, JsonGenerator out, Encoder delegate)
              throws IOException {

            if (object instanceof E2) {
              E2 e2 = (E2) object;
              if (Cayenne.intPKForObject(e2) == 7) {
                return delegate.encode(propertyName, object, out);
              } else {
                return false;
              }
            } else {
              delegate.encode(propertyName, object, out);
              return true;
            }
          }

          @Override
          public boolean willEncode(String propertyName, Object object, Encoder delegate) {
            if (object instanceof E2) {
              E2 e2 = (E2) object;
              if (Cayenne.intPKForObject(e2) == 7) {
                return delegate.willEncode(propertyName, object);
              } else {
                return false;
              }
            } else {
              return delegate.willEncode(propertyName, object);
            }
          }
        });

    ClientEntity<E2> e2Descriptor = getClientEntity(E2.class);
    e2Descriptor.setIdIncluded(true);

    ClientEntity<E3> e3Descriptor = getClientEntity(E3.class);
    e3Descriptor.setIdIncluded(true);
    e3Descriptor.getRelationships().put(E3.E2.getName(), e2Descriptor);

    ObjectContext context = cayenneService.newContext();

    E2 e21 = new E2();
    e21.setObjectId(new ObjectId("E2", E2.ID_PK_COLUMN, 7));
    context.registerNewObject(e21);

    E3 e31 = new E3();
    e31.setObjectId(new ObjectId("E3", E3.ID_PK_COLUMN, 5));
    context.registerNewObject(e31);
    e31.setE2(e21);

    DataResponse<E3> builder =
        DataResponse.forType(E3.class).withClientEntity(e3Descriptor).withObject(e31);

    encoderService.makeEncoder(builder);

    assertEquals("[{\"id\":5,\"e2\":{\"id\":7},\"e2_id\":7}]", toJson(builder));

    E2 e22 = new E2();
    e22.setObjectId(new ObjectId("E2", E2.ID_PK_COLUMN, 8));
    context.registerNewObject(e22);

    E3 e32 = new E3();
    e32.setObjectId(new ObjectId("E3", E3.ID_PK_COLUMN, 6));
    context.registerNewObject(e32);
    e32.setE2(e22);

    builder.withObject(e32);

    assertEquals("[{\"id\":6}]", toJson(builder));
  }
  @Test
  public void testEncoder_FilteredRoots() throws IOException {

    filters.add(
        new EncoderFilter() {

          @Override
          public boolean matches(ClientEntity<?> clientEntity) {
            return true;
          }

          @Override
          public boolean encode(
              String propertyName, Object object, JsonGenerator out, Encoder delegate)
              throws IOException {

            E2 e2 = (E2) object;
            if (Cayenne.intPKForObject(e2) == 7) {
              delegate.encode(propertyName, object, out);
              return true;
            }

            return false;
          }

          @Override
          public boolean willEncode(String propertyName, Object object, Encoder delegate) {
            E2 e2 = (E2) object;
            if (Cayenne.intPKForObject(e2) == 7) {
              delegate.willEncode(propertyName, object);
              return true;
            }

            return false;
          }
        });

    ClientEntity<E2> descriptor = getClientEntity(E2.class);
    descriptor.setIdIncluded(true);

    ObjectContext context = cayenneService.newContext();
    E2 e21 = new E2();
    e21.setObjectId(new ObjectId("E2", E2.ID_PK_COLUMN, 7));
    e21.setName("XYZ");
    e21.setAddress("bla bla street");
    context.registerNewObject(e21);

    DataResponse<E2> builder =
        DataResponse.forType(E2.class).withClientEntity(descriptor).withObject(e21);

    encoderService.makeEncoder(builder);
    assertEquals("[{\"id\":7}]", toJson(builder));

    E2 e22 = new E2();
    e22.setObjectId(new ObjectId("E2", E2.ID_PK_COLUMN, 8));
    e22.setName("XYZ");
    e22.setAddress("bla bla street");
    context.registerNewObject(e22);

    builder.withObject(e22);
    encoderService.makeEncoder(builder);

    assertEquals("[]", toJson(builder));
  }
Esempio n. 7
0
 private String responseContents(DataResponse<?> response, PushPopVisitor visitor) {
   response.getEncoder().visitEntities(response.getObjects(), visitor);
   return visitor.ids.stream().collect(joining(";"));
 }