Examples with ViewedVideo org.datanucleus.tests.embedded.ViewedVideo used on opensource projects

Example 1 with ViewedVideo

use of org.datanucleus.tests.embedded.ViewedVideo in project tests by datanucleus.

the class EmbeddedTest method testEmbeddedCollectionWithReferenceToUnembedded.

public void testEmbeddedCollectionWithReferenceToUnembedded() {
    try {
        PersistenceManager pm = pmf.getPersistenceManager();
        Transaction tx = pm.currentTransaction();
        try {
            tx.begin();
            Video vid1 = new Video();
            vid1.setIdVideo(456L);
            vid1.setVideoTitle("Inception");
            Video vid2 = new Video();
            vid2.setIdVideo(789L);
            vid2.setVideoTitle("The Matrix");
            Video vid3 = new Video();
            vid3.setIdVideo(123L);
            vid3.setVideoTitle("Saving Private Ryan");
            LibraryUser user1 = new LibraryUser();
            user1.setName("Andy");
            user1.getViewedVideos().add(new ViewedVideo(vid1));
            user1.getViewedVideos().add(new ViewedVideo(vid2));
            LibraryUser user2 = new LibraryUser();
            user2.setName("Chris");
            user2.getViewedVideos().add(new ViewedVideo(vid1));
            user2.getViewedVideos().add(new ViewedVideo(vid2));
            user2.getViewedVideos().add(new ViewedVideo(vid3));
            // Persist all elements :
            pm.makePersistentAll(user1, user2);
            tx.commit();
        } catch (Throwable thr) {
            LOG.error(">> Exception in persist", thr);
            fail("Exception in persist of data : " + thr.getMessage());
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
        // Query the data
        pm = pmf.getPersistenceManager();
        tx = pm.currentTransaction();
        try {
            tx.begin();
            Query q = pm.newQuery("SELECT FROM " + LibraryUser.class.getName() + " ORDER BY name");
            List<LibraryUser> users = (List<LibraryUser>) q.execute();
            assertEquals("Number of users is wrong", 2, users.size());
            LibraryUser user1 = users.get(0);
            assertNotNull(user1);
            assertEquals("Andy", user1.getName());
            Set<ViewedVideo> vids1 = user1.getViewedVideos();
            assertNotNull(vids1);
            assertEquals(2, vids1.size());
            boolean vid1Present = false;
            boolean vid2Present = false;
            boolean vid3Present = false;
            for (ViewedVideo vid : vids1) {
                if (vid.getVideo().getIdVideo() == 456) {
                    vid1Present = true;
                } else if (vid.getVideo().getIdVideo() == 789) {
                    vid2Present = true;
                } else if (vid.getVideo().getIdVideo() == 123) {
                    vid3Present = true;
                }
            }
            assertTrue(vid1Present);
            assertTrue(vid2Present);
            assertFalse(vid3Present);
            LibraryUser user2 = users.get(1);
            assertNotNull(user2);
            assertEquals("Chris", user2.getName());
            Set<ViewedVideo> vids2 = user2.getViewedVideos();
            assertNotNull(vids2);
            assertEquals(3, vids2.size());
            vid1Present = false;
            vid2Present = false;
            vid3Present = false;
            for (ViewedVideo vid : vids2) {
                if (vid.getVideo().getIdVideo() == 456) {
                    vid1Present = true;
                } else if (vid.getVideo().getIdVideo() == 789) {
                    vid2Present = true;
                } else if (vid.getVideo().getIdVideo() == 123) {
                    vid3Present = true;
                }
            }
            assertTrue(vid1Present);
            assertTrue(vid2Present);
            assertTrue(vid3Present);
            tx.commit();
        } catch (Throwable thr) {
            LOG.error(">> Exception in query", thr);
            fail("Exception in query of data : " + thr.getMessage());
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
    } finally {
        PersistenceManager pm = pmf.getPersistenceManager();
        Collection<Object> toDelete;
        Query qVideo = pm.newQuery(Video.class);
        Query qUser = pm.newQuery(LibraryUser.class);
        toDelete = new ArrayList<Object>((Collection<Object>) qVideo.execute());
        toDelete.addAll(new ArrayList<Object>((Collection<Object>) qUser.execute()));
        pm.deletePersistentAll(toDelete);
        pm.close();
    }
}