[jboss-cvs] JBossAS SVN: r60063 - trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession.

[jboss-cvs-commits at lists.jboss.org]

Author: bstansberry at jboss.com
Date: 2007-01-27 18:02:02 -0500 (Sat, 27 Jan 2007)
New Revision: 60063

Modified:
   trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCart.java
   trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCartBean.java
Log:
Expose a VMID so client can ensure SFSB and SLSB are in same VM

Modified: trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCart.java
===================================================================
--- trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCart.java	2007-01-27 23:01:01 UTC (rev 60062)
+++ trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCart.java	2007-01-27 23:02:02 UTC (rev 60063)
@@ -21,6 +21,8 @@
   */
 package org.jboss.ejb3.test.clusteredsession;
 
+import java.rmi.dgc.VMID;
+
 import javax.ejb.Remove;
 
 /**
@@ -30,6 +32,8 @@
  */
 public interface ShoppingCart
 {
+   VMID getVMID();
+   
    long createCustomer();
 
    void update();

Modified: trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCartBean.java
===================================================================
--- trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCartBean.java	2007-01-27 23:01:01 UTC (rev 60062)
+++ trunk/ejb3/src/test/org/jboss/ejb3/test/clusteredsession/ShoppingCartBean.java	2007-01-27 23:02:02 UTC (rev 60063)
@@ -22,6 +22,8 @@
 package org.jboss.ejb3.test.clusteredsession;
 
 import java.io.Serializable;
+import java.rmi.dgc.VMID;
+
 import javax.ejb.EJB;
 import javax.ejb.Remote;
 import javax.ejb.Remove;
@@ -49,11 +51,16 @@
    @PersistenceContext(type=PersistenceContextType.EXTENDED) EntityManager em;
 
    @EJB StatelessLocal stateless;
-
+   
    private Customer customer;
 
    @EJB private Contained contained;
 
+   public VMID getVMID()
+   {
+      return stateless.getVMID();
+   }
+   
    public long createCustomer()
    {
       customer = new Customer();
@@ -123,5 +130,9 @@
 
 
    @Remove
-   public void checkout() { em.remove(customer);}
+   public void checkout() 
+   {
+      if (customer != null)
+         em.remove(customer);
+   }
 }