Test same stateless bean on every ctx lookup

kubamarchwicki · kubamarchwicki · commit 2d74f53716b1 · 2013-12-08T19:11:26.000+01:00
diff --git a/ejb/stateless/src/main/java/org/javaee7/ejb/stateless/AccountSessionBean.java b/ejb/stateless/src/main/java/org/javaee7/ejb/stateless/AccountSessionBean.java
@@ -47,11 +47,19 @@
 @Stateless
 public class AccountSessionBean {
 
+    private float amount = 0;
+
     public String withdraw(float amount) {
+        this.amount -= amount;
         return "Withdrawn: " + amount;
     }
 
-    public String deposit(float amount) { 
+    public String deposit(float amount) {
+        this.amount += amount;
         return "Deposited: " + amount;
     }
+
+    public float getAmount() {
+        return this.amount;
+    }
 }
diff --git a/ejb/stateless/src/test/java/org/javaee7/ejb/stateless/AccountSessionStatelessnessTest.java b/ejb/stateless/src/test/java/org/javaee7/ejb/stateless/AccountSessionStatelessnessTest.java
@@ -0,0 +1,69 @@
+package org.javaee7.ejb.stateless;
+
+import org.jboss.arquillian.container.test.api.Deployment;
+import org.jboss.arquillian.junit.Arquillian;
+import org.jboss.arquillian.junit.InSequence;
+import org.jboss.shrinkwrap.api.Archive;
+import org.jboss.shrinkwrap.api.ShrinkWrap;
+import org.jboss.shrinkwrap.api.asset.EmptyAsset;
+import org.jboss.shrinkwrap.api.spec.JavaArchive;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import javax.naming.InitialContext;
+import javax.naming.NamingException;
+
+import static org.hamcrest.MatcherAssert.*;
+import static org.hamcrest.Matchers.*;
+
+/**
+ * @author Jakub Marchwicki
+ */
+@RunWith(Arquillian.class)
+public class AccountSessionStatelessnessTest {
+
+    final private float deposit_amount = 10f;
+	private AccountSessionBean account;
+
+	@Deployment
+	public static Archive<?> deployment() {
+		return ShrinkWrap.create(JavaArchive.class, "test.jar")
+				.addClass(AccountSessionBean.class)
+				.addAsManifestResource(EmptyAsset.INSTANCE, "beans.xml");
+	}
+
+    @Before
+    public void setup() throws NamingException {
+        InitialContext ctx = new InitialContext();
+        Object object = ctx.lookup("java:global/test/AccountSessionBean");
+
+        assertThat(object, instanceOf(AccountSessionBean.class));
+
+        AccountSessionBean account = (AccountSessionBean) object;
+        if (this.account != null) {
+            assertThat("Expect same instance",
+                    account.hashCode(),
+                    is(equalTo(this.account.hashCode())));
+        }
+
+        this.account = account;
+    }
+
+	@Test
+    @InSequence(1)
+	public void should_deposit_amount() {
+        assertThat(account.getAmount(), is(equalTo(0f)));
+
+        String actual = account.deposit(deposit_amount);
+
+        assertThat(actual, is(equalTo("Deposited: " + deposit_amount)));
+        assertThat(account.getAmount(), is(equalTo(deposit_amount)));
+	}
+
+	@Test
+    @InSequence(2)
+	public void should_contain_already_deposited_amount() {
+        assertThat(account.getAmount(), is(equalTo(deposit_amount)));
+	}
+}

Original file line number	Diff line number	Diff line change
`@@ -47,11 +47,19 @@`
`47`	`47`	`@Stateless`
`48`	`48`	`public class AccountSessionBean {`
`49`	`49`
	`50`	`+ private float amount = 0;`
	`51`	`+`
`50`	`52`	`public String withdraw(float amount) {`
	`53`	`+ this.amount -= amount;`
`51`	`54`	`return "Withdrawn: " + amount;`
`52`	`55`	`}`
`53`	`56`
`54`		`- public String deposit(float amount) {`
	`57`	`+ public String deposit(float amount) {`
	`58`	`+ this.amount += amount;`
`55`	`59`	`return "Deposited: " + amount;`
`56`	`60`	`}`
	`61`	`+`
	`62`	`+ public float getAmount() {`
	`63`	`+ return this.amount;`
	`64`	`+ }`
`57`	`65`	`}`