Examples with TestClass2 org.infinispan.protostream.annotations.impl.testdomain.subpackage.TestClass2 used on opensource projects

Example 1 with TestClass2

use of org.infinispan.protostream.annotations.impl.testdomain.subpackage.TestClass2 in project protostream by infinispan.

the class ProtoSchemaBuilderTest method testGeneration.

@Test
public void testGeneration() throws Exception {
    SerializationContext ctx = createContext();
    ProtoSchemaBuilder protoSchemaBuilder = new ProtoSchemaBuilder();
    protoSchemaBuilder.fileName("test.proto").packageName("test_package").addClass(TestClass.class).addClass(TestClass3.class).addClass(Simple.class).build(ctx);
    assertTrue(ctx.canMarshall(TestEnum.class));
    assertTrue(ctx.canMarshall(Simple.class));
    assertTrue(ctx.canMarshall(TestClass.class));
    assertTrue(ctx.canMarshall(TestClass.InnerClass.class));
    assertTrue(ctx.canMarshall(TestClass.InnerClass2.class));
    assertTrue(ctx.canMarshall(TestClass2.class));
    assertTrue(ctx.canMarshall(TestClass3.class));
    assertTrue(ctx.canMarshall("test_package.TestEnumABC"));
    assertTrue(ctx.canMarshall("test_package.Simple"));
    assertTrue(ctx.canMarshall("test_package.TestClass2"));
    assertTrue(ctx.canMarshall("test_package.TestClass3"));
    assertTrue(ctx.canMarshall("test_package.TestClass"));
    assertTrue(ctx.canMarshall("test_package.TestClass.InnerClass"));
    assertTrue(ctx.canMarshall("test_package.TestClass.InnerClass2"));
    Simple simple = new Simple();
    simple.afloat = 3.14f;
    byte[] bytes = ProtobufUtil.toWrappedByteArray(ctx, simple);
    Object unmarshalled = ProtobufUtil.fromWrappedByteArray(ctx, bytes);
    assertTrue(unmarshalled instanceof Simple);
    Simple unwrapped = (Simple) unmarshalled;
    assertEquals(3.14f, unwrapped.afloat, 0.001);
    TestClass testClass = new TestClass();
    testClass.surname = "test";
    testClass.longField = 100L;
    testClass.testClass2 = new TestClass2();
    testClass.testClass2.address = "test address";
    bytes = ProtobufUtil.toWrappedByteArray(ctx, testClass);
    unmarshalled = ProtobufUtil.fromWrappedByteArray(ctx, bytes);
    assertTrue(unmarshalled instanceof TestClass);
    assertEquals("test", ((TestClass) unmarshalled).surname);
    assertEquals(100L, ((TestClass) unmarshalled).longField);
    assertEquals("test address", ((TestClass) unmarshalled).testClass2.address);
}

Example 2 with TestClass2

use of org.infinispan.protostream.annotations.impl.testdomain.subpackage.TestClass2 in project protostream by infinispan.

the class ProtoSchemaBuilderTest method testAutoImport.

@Test
public void testAutoImport() throws Exception {
    SerializationContext ctx = createContext();
    new ProtoSchemaBuilder().fileName("test1.proto").packageName("test_package1").addClass(TestEnum.class).build(ctx);
    assertTrue(ctx.canMarshall(TestEnum.class));
    assertTrue(ctx.canMarshall("test_package1.TestEnumABC"));
    ProtoSchemaBuilder protoSchemaBuilder2 = new ProtoSchemaBuilder();
    protoSchemaBuilder2.autoImportClasses(false).fileName("test2.proto").packageName("test_package2").addClass(TestClass.class);
    try {
        // TestClass2 was not added explicitly and is not auto-added because autoImportClasses == false so we expect an exception
        protoSchemaBuilder2.build(ctx);
        fail("ProtoSchemaBuilderException expected");
    } catch (ProtoSchemaBuilderException e) {
        assertTrue(e.getMessage().contains("Found a reference to class org.infinispan.protostream.annotations.impl.testdomain.subpackage.TestClass2 which was not added to the builder and 'autoImportClasses' is disabled."));
    }
    assertFalse(ctx.canMarshall(TestClass.class));
    assertFalse(ctx.canMarshall(TestClass.InnerClass.class));
    assertFalse(ctx.canMarshall(TestClass.InnerClass2.class));
    assertFalse(ctx.canMarshall(TestClass2.class));
    // it must work after explicitly adding TestClass2
    protoSchemaBuilder2.addClass(TestClass2.class).build(ctx);
    assertTrue(ctx.canMarshall(TestClass.class));
    assertTrue(ctx.canMarshall("test_package2.TestClass"));
    assertTrue(ctx.canMarshall(TestClass.InnerClass.class));
    assertTrue(ctx.canMarshall(TestClass.InnerClass2.class));
    assertTrue(ctx.canMarshall(TestClass2.class));
    assertTrue(ctx.canMarshall("test_package1.TestEnumABC"));
    assertFalse(ctx.canMarshall("test_package2.TestEnumABC"));
}