use of com.blazebit.persistence.view.testsuite.basic.model.FetchesDocumentView2 in project blaze-persistence by Blazebit.
the class EntityFetchesTest method entityViewFetchesInitializeEntities.
@Test
// NOTE: Datanucleus doesn't seem to support fetching non-root relations
@Category({ NoEclipselink.class, NoDatanucleus.class })
public void entityViewFetchesInitializeEntities() {
CriteriaBuilder<Person> criteria = cbf.create(em, Person.class).where("id").eq(pers1.getId());
EntityViewSetting<FetchesPersonView2, CriteriaBuilder<FetchesPersonView2>> setting = EntityViewSetting.create(FetchesPersonView2.class);
CriteriaBuilder<FetchesPersonView2> cb = evm.applySetting(setting, criteria);
List<FetchesPersonView2> results = cb.getResultList();
// Close the em to make sure this was fetched properly
em.getTransaction().rollback();
em.close();
// Only a single entity view is produced
assertEquals(1, results.size());
FetchesPersonView2 view = results.get(0);
// Friends match
assertEquals(friend.getId(), view.getFriend().getId());
// Friend is properly initialized
assertEquals(2, view.getFriend().getOwnedDocuments().size());
assertEquals(friend.getOwnedDocuments(), view.getFriend().getOwnedDocuments());
// Owned documents have the correct size
assertEquals(2, view.getOwnedDocuments().size());
// Extract ownedDocuments from view
Iterator<Document> documentIterator = view.getOwnedDocuments().iterator();
Document ownedDoc1 = documentIterator.next();
Document ownedDoc2 = documentIterator.next();
// Extract ownedDocuments from pers1
documentIterator = pers1.getOwnedDocuments().iterator();
Document expectedDoc1 = documentIterator.next();
Document expectedDoc2;
if (!expectedDoc1.getId().equals(ownedDoc1.getId())) {
expectedDoc2 = expectedDoc1;
expectedDoc1 = documentIterator.next();
} else {
expectedDoc2 = documentIterator.next();
}
// Check contacts have been initialized of owned documents
assertEquals(2, ownedDoc1.getContacts().size());
assertEquals(expectedDoc1.getContacts(), ownedDoc1.getContacts());
assertEquals(2, ownedDoc2.getContacts().size());
assertEquals(expectedDoc2.getContacts(), ownedDoc2.getContacts());
// Extract partner documents
Document partnerDocument = pers1.getPartnerDocument();
FetchesDocumentView2 partnerDocumentView = view.getPartnerDocument();
// Check contacts and personList are properly initialized
assertEquals(2, partnerDocumentView.getContacts().size());
assertEquals(partnerDocument.getContacts(), partnerDocumentView.getContacts());
assertEquals("p6", partnerDocumentView.getContacts().get(0).getName());
assertEquals(0, partnerDocumentView.getContacts().get(0).getOwnedDocuments().size());
assertEquals("p7", partnerDocumentView.getContacts().get(1).getName());
assertEquals(0, partnerDocumentView.getContacts().get(1).getOwnedDocuments().size());
assertEquals(2, partnerDocumentView.getPeople().size());
assertEquals(partnerDocument.getPeople(), partnerDocumentView.getPeople());
assertEquals(pers1, partnerDocumentView.getPeople().get(0));
assertEquals(2, partnerDocumentView.getPeople().get(0).getOwnedDocuments().size());
assertEquals(pers1.getOwnedDocuments(), partnerDocumentView.getPeople().get(0).getOwnedDocuments());
assertEquals(friend, partnerDocumentView.getPeople().get(1));
assertEquals(2, partnerDocumentView.getPeople().get(1).getOwnedDocuments().size());
assertEquals(friend.getOwnedDocuments(), partnerDocumentView.getPeople().get(1).getOwnedDocuments());
}
Aggregations