File Service Test

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.fail;

import java.sql.DriverManager;
import java.sql.SQLNonTransientConnectionException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;
import javax.transaction.RollbackException;

import org.filestore.ejb.file.entity.FileItem;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class FileServiceTest {

	private static Logger LOGGER = Logger.getLogger(FileServiceTest.class
			.getName());

	private static EntityManagerFactory factory;
	private static EntityManager em;
	private static FileService service;

	@BeforeClass
	public static void setUp() throws Exception {
		try {
			LOGGER.log(Level.INFO, "Starting memory database for unit tests");
			Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
			DriverManager.getConnection(
					"jdbc:derby:memory:unit-testing-jpa;create=true").close();
		} catch (Exception e) {
			LOGGER.log(Level.SEVERE, "unable to start database", e);
			fail("Exception during database startup.");
		}
		try {
			LOGGER.log(Level.INFO,
					"Building Hibernate EntityManager for unit tests");
			factory = Persistence.createEntityManagerFactory("testPU");
			em = factory.createEntityManager();
		} catch (Exception ex) {
			ex.printStackTrace();
			fail("Exception during JPA EntityManager instanciation.");
		}
		LOGGER.log(Level.INFO, "Building FileService");
		service = new FileServiceBean();
		((FileServiceBean) service).em = em;
	}

	@AfterClass
	public static void tearDown() throws Exception {
		LOGGER.log(Level.INFO, "Shuting Hibernate JPA layer.");
		if (em != null) {
			em.close();
		}
		if (factory != null) {
			factory.close();
		}
		LOGGER.log(Level.INFO, "Stopping memory database.");
		try {
			DriverManager.getConnection(
					"jdbc:derby:memory:unit-testing-jpa;shutdown=true").close();
		} catch (SQLNonTransientConnectionException ex) {
			if (ex.getErrorCode() != 45000) {
				throw ex;
			}
		}
	}

	@Test
    public void testPostAndDeleteFile() throws FileServiceException {
    	try {
	    	em.getTransaction().begin();

	    	List<String> receivers = new ArrayList<String> ();
	    	receivers.add("sheldon@test.com");
	    	receivers.add("rajesh@test.com");
	    	receivers.add("penny@test.com");
	    	String key = service.postFile("jayblanc@gmail.com", receivers, "Bazinga", "The.Big.Bang.Theory.S06E01.mkv", "this should be a uuid");
	    	assertNotNull(key);

	    	FileItem item = service.getFile(key);
	    	assertEquals("jayblanc@gmail.com", item.getOwner());
	    	assertEquals("Bazinga", item.getMessage());
	    	assertEquals("The.Big.Bang.Theory.S06E01.mkv", item.getName());

	    	service.deleteFile(key);
	    	try {
	    		item = service.getFile(key);
	    		fail("The file should not exists anymore !!");
	    	} catch ( FileServiceException e ) {
	    		//
	    	}

	    	em.getTransaction().commit();
    	}  catch (IllegalStateException | RollbackException e) {
            em.getTransaction().rollback();
            LOGGER.log(Level.SEVERE, "error during testing file service", e);
            fail("Exception during testing file service");
        }
    }