package producer;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import javax.persistence.EntityManager;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.jboss.weld.context.http.HttpRequestContext;
import org.junit.Test;
import org.junit.runner.RunWith;
import test.Tests;
import br.gov.frameworkdemoiselle.internal.proxy.EntityManagerProxy;
import br.gov.frameworkdemoiselle.util.Beans;
import br.gov.frameworkdemoiselle.util.NameQualifier;
@RunWith(Arquillian.class)
public class RequestScopedProducerTest {
private static final String PATH = "src/test/resources/producer";
@Deployment
public static WebArchive createDeployment() {
WebArchive deployment = Tests.createDeployment(RequestScopedProducerTest.class);
deployment.addAsResource(Tests.createFileAsset(PATH + "/persistence.xml"), "META-INF/persistence.xml");
deployment.addAsResource(Tests.createFileAsset(PATH + "/demoiselle.properties"), "demoiselle.properties");
return deployment;
}
@Test
public void produceEntityManager() {
EntityManager manager = Beans.getReference(EntityManager.class);
assertNotNull(manager);
assertEquals(EntityManagerProxy.class, manager.getClass());
}
@Test
public void produceMultipleEntityManagers() {
EntityManager m1 = Beans.getReference(EntityManager.class, new NameQualifier("pu"));
assertNotNull(m1);
assertEquals(EntityManagerProxy.class, m1.getClass());
EntityManager m2 = Beans.getReference(EntityManager.class, new NameQualifier("pu2"));
assertNotNull(m2);
assertEquals(EntityManagerProxy.class, m2.getClass());
}
@Test
public void produceOneEntityManagerPerRequest() {
EntityManager m1 = Beans.getReference(EntityManager.class, new NameQualifier("pu"));
assertNotNull(m1);
assertEquals(EntityManagerProxy.class, m1.getClass());
EntityManager m2 = Beans.getReference(EntityManager.class, new NameQualifier("pu"));
assertNotNull(m2);
assertEquals(EntityManagerProxy.class, m2.getClass());
MyEntity entity = new MyEntity();
entity.setId(createId("testID"));
m1.persist(entity);
assertTrue(m2.contains(entity));
}
@Test
public void produceDifferentEntityManagerPerRequest() {
HttpRequestContext weldContext = Beans.getReference(HttpRequestContext.class);
boolean wasNotActive = false;
if (!weldContext.isActive()){
wasNotActive = true;
weldContext.activate();
}
EntityManager m1 = Beans.getReference(EntityManager.class, new NameQualifier("pu"));
assertNotNull(m1);
assertEquals(EntityManagerProxy.class, m1.getClass());
MyEntity entity = new MyEntity();
entity.setId(createId("testID"));
m1.persist(entity);
assertTrue(m1.contains(entity));
weldContext.invalidate();
weldContext.deactivate();
if (!weldContext.isActive()){
weldContext.activate();
}
EntityManager m2 = Beans.getReference(EntityManager.class, new NameQualifier("pu"));
assertTrue( m2.isOpen() );
assertTrue( !m2.contains(entity));
if (wasNotActive && weldContext.isActive()){
weldContext.invalidate();
weldContext.deactivate();
}
}
private String createId(String id) {
return this.getClass().getName() + "_" + id;
}
}