package org.filestore.ejb.file;

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.persistence.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");
        }
    }

}