Search in sources :

Example 1 with FetchesDocumentView2

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());
}
Also used : CriteriaBuilder(com.blazebit.persistence.CriteriaBuilder) FetchesPersonView2(com.blazebit.persistence.view.testsuite.basic.model.FetchesPersonView2) Document(com.blazebit.persistence.testsuite.entity.Document) FetchesDocumentView2(com.blazebit.persistence.view.testsuite.basic.model.FetchesDocumentView2) Person(com.blazebit.persistence.testsuite.entity.Person) Category(org.junit.experimental.categories.Category) Test(org.junit.Test) AbstractEntityViewTest(com.blazebit.persistence.view.testsuite.AbstractEntityViewTest)

Aggregations

CriteriaBuilder (com.blazebit.persistence.CriteriaBuilder)1 Document (com.blazebit.persistence.testsuite.entity.Document)1 Person (com.blazebit.persistence.testsuite.entity.Person)1 AbstractEntityViewTest (com.blazebit.persistence.view.testsuite.AbstractEntityViewTest)1 FetchesDocumentView2 (com.blazebit.persistence.view.testsuite.basic.model.FetchesDocumentView2)1 FetchesPersonView2 (com.blazebit.persistence.view.testsuite.basic.model.FetchesPersonView2)1 Test (org.junit.Test)1 Category (org.junit.experimental.categories.Category)1