Simplification. No need for @SesssionScope bean. Inject @stateful to arq test

Author: kubamarchwicki

jpa/extended-pc/src/main/java/org/javaee7/jpa/extended/pc/BigBangTheoryService.java

@@ -1,16 +1,14 @@
 package org.javaee7.jpa.extended.pc
-
 import org.jboss.arquillian.container.test.api.Deployment
 import org.jboss.arquillian.spock.ArquillianSputnik
 import org.jboss.shrinkwrap.api.ShrinkWrap
 import org.jboss.shrinkwrap.api.spec.WebArchive
 import org.junit.runner.RunWith
 import spock.lang.Specification
 
-import javax.inject.Inject
+import javax.ejb.EJB
 import javax.persistence.EntityManager
 import javax.persistence.PersistenceContext
-
 /**
  * @author Kuba Marchwicki
  */
@@ -20,8 +18,8 @@ class ExtendedPersistenceContextSpecification extends Specification {
     @PersistenceContext
     EntityManager em;
 
-    @Inject
-    BigBangTheoryService service;
+    @EJB
+    CharactersBean bean;
 
     @Deployment
     def static WebArchive deploy() {
@@ -30,12 +28,19 @@ class ExtendedPersistenceContextSpecification extends Specification {
                 .addAsResource("META-INF/persistence.xml")
                 .addAsResource("META-INF/create.sql")
                 .addAsResource("META-INF/drop.sql")
-                .addAsResource("META-INF/load.sql");
+                .addAsResource("META-INF/load.sql")
     }
 
     def setup() {
-        service.addWilWheaton();
-        service.updateRaj();
+        Character wil = new Character(8, "Wil Wheaton")
+        bean.save(wil)
+
+        for (Character c : bean.get()) {
+            if ("Raj".equals(c.getName())) {
+                c.setName("Rajesh Ramayan")
+                bean.save(c)
+            }
+        }
     }
 
     def "should not persist changes without transaction flush"() {
@@ -46,7 +51,7 @@ class ExtendedPersistenceContextSpecification extends Specification {
 
     def "should update characters after transaction flush"() {
         when:
-        service.proceed();
+        bean.commitChanges()
 
         then:
         8 == em.createNamedQuery(Character.FIND_ALL, Character.class).getResultList().size();