use of org.infinispan.protostream.annotations.impl.testdomain.TestClass3 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);
}
Aggregations