EntityRelationshipsTest.java

/*
 ###############################################################################
 #                                                                             #
 #    Copyright (C) 2011-2016 OpenMEAP, Inc.                                   #
 #    Credits to Jonathan Schang & Rob Thacher                                 #
 #                                                                             #
 #    Released under the LGPLv3                                                #
 #                                                                             #
 #    OpenMEAP is free software: you can redistribute it and/or modify         #
 #    it under the terms of the GNU Lesser General Public License as published #
 #    by the Free Software Foundation, either version 3 of the License, or     #
 #    (at your option) any later version.                                      #
 #                                                                             #
 #    OpenMEAP is distributed in the hope that it will be useful,              #
 #    but WITHOUT ANY WARRANTY; without even the implied warranty of           #
 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            #
 #    GNU Lesser General Public License for more details.                      #
 #                                                                             #
 #    You should have received a copy of the GNU Lesser General Public License #
 #    along with OpenMEAP.  If not, see <http://www.gnu.org/licenses/>.        #
 #                                                                             #
 ###############################################################################
 */

package com.openmeap.model;

import org.apache.log4j.*;

import java.util.*;

import org.junit.*;
import javax.persistence.*;
import org.hibernate.ejb.*;
import org.junit.Assert;

import com.openmeap.model.dto.Application;
import com.openmeap.model.dto.ApplicationArchive;
import com.openmeap.model.dto.ApplicationInstallation;
import com.openmeap.model.dto.ApplicationVersion;

/**
 * This body of tests really just validates that the model relationships function the way i expect.
 * 
 * NOTE: in windows, close the test.db file before running
 * 
 * @author schang
 */
public class EntityRelationshipsTest {
	
	private static EntityManager em = null;
	
	@BeforeClass static public void beforeClass() {
		BasicConfigurator.configure();
		Logger.getRootLogger().setLevel(Level.TRACE);
		
		ModelTestUtils.resetTestDb();
		ModelTestUtils.createModel(null);
		em = ModelTestUtils.createEntityManager();
	}
	
	@AfterClass static public void afterClass() {
		ModelTestUtils.resetTestDb();
	}
	
	@Test public void testModel() {
		
		assertInserts(em);
		
		makeModifications(em);
		
		ApplicationArchive aa = null;
		ApplicationVersion av = null;
		Application app = null;
		
		// Verify that deleting an application version 
		// deletes only the version and archive
		av = em.find(ApplicationVersion.class, 1L);
		app = em.find(Application.class, 1L);
		
		em.getTransaction().begin();
		app.removeVersion(av);
		em.remove(av);
		em.getTransaction().commit();
		
		app = em.find(Application.class, 1L);
		aa = em.find(ApplicationArchive.class, 1L);
		Assert.assertTrue(app!=null);
		Assert.assertTrue(aa!=null);		
		
		// Verify that deleting an application deletes all the crap associated to it
		app = em.find(Application.class, 1L);
		Assert.assertTrue(app!=null);
		
		em.getTransaction().begin();
		em.remove(app);
		em.getTransaction().commit();
		
		av = em.find(ApplicationVersion.class, 2L);
		Assert.assertTrue(av!=null);
		
		em.clear();
	}
	
	public void makeModifications(EntityManager em) {
		Application app = em.find(Application.class,1L);
		app.setDescription("BUNKER");
		em.persist(app);
	}

	public void assertInserts(EntityManager em) {
		// TODO: flush this whole test out much more
		
		Query q = em.createQuery("select distinct a from Application a");
		
		@SuppressWarnings("unchecked")
		List<Application> result = (List<Application>)q.getResultList();
		
		Assert.assertTrue(result.size()==1);
		Application a = result.get(0);
		
		// Application simple types
		//Assert.assertTrue(a.getCurrentVersion()!=null);
		Assert.assertTrue(a.getName().compareTo("Application.name")==0);
		Assert.assertTrue(a.getDescription().compareTo("Application.description")==0);
		
		// Versions and CurrentVersion
		Assert.assertTrue(a.getVersions()!=null && a.getVersions().entrySet().size()==2);
		Assert.assertTrue(a.getDeployments().size()==3);
	}
}