@Test
public void shouldRunHibernateLazyInitialization() throws Exception {
LazyInitializer initializer = mock(LazyInitializer.class);
SomeProxy proxy = new SomeProxy(initializer);
proxy.name = "my name";
proxy.aField = "abc";
when(initializer.getPersistentClass()).thenReturn(Client.class);
when(proxy.getHibernateLazyInitializer().getImplementation()).thenReturn(proxy);
serialization.from(proxy).serialize();
assertThat(result(), is("{\"client\":{\"aField\":\"abc\",\"name\":\"my name\"}}"));
}
protected Class<?> getClass(ExtendedObjectOutput out, Object v) {
Class<?> cls = v.getClass();
if (v instanceof HibernateProxy) {
LazyInitializer initializer = ((HibernateProxy) v).getHibernateLazyInitializer();
String className =
(initializer.isUninitialized()
? initializer.getEntityName()
: initializer.getImplementation().getClass().getName());
if (className != null && className.length() > 0) {
try {
cls = out.getReflection().loadClass(className);
} catch (ClassNotFoundException e) {
cls = initializer.getPersistentClass();
}
}
}
return cls;
}