svn commit: r1144118 - /openejb/trunk/openejb3/container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java

[dblevins]

Author: dblevins
Date: Fri Jul  8 01:18:24 2011
New Revision: 1144118

URL: http://svn.apache.org/viewvc?rev=1144118&view=rev
Log:
OPENEJB-1594: Regression related to NewBean support

Modified:
    
openejb/trunk/openejb3/container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java

Modified: 
openejb/trunk/openejb3/container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java
URL: 
http://svn.apache.org/viewvc/openejb/trunk/openejb3/container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java?rev=1144118&r1=1144117&r2=1144118&view=diff
==============================================================================
--- 
openejb/trunk/openejb3/container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java
 (original)
+++ 
openejb/trunk/openejb3/container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java
 Fri Jul  8 01:18:24 2011
@@ -56,6 +56,7 @@ import org.apache.openejb.OpenEJBExcepti
 import org.apache.openejb.ProxyInfo;
 import org.apache.openejb.RpcContainer;
 import org.apache.openejb.SystemException;
+import org.apache.openejb.cdi.CdiEjbBean;
 import org.apache.openejb.config.rules.CheckDependsOn;
 import org.apache.openejb.monitoring.StatsInterceptor;
 import org.apache.openejb.monitoring.ObjectNameBuilder;
@@ -442,7 +443,7 @@ public class StatefulContainer implement
     protected Object removeEJBObject(BeanContext beanContext, Object primKey, 
Class callInterface, Method callMethod, Object[] args, InterfaceType 
interfaceType) throws OpenEJBException {
         if (primKey == null) throw new NullPointerException("primKey is null");
 
-        final Class scope = beanContext.get(Bean.class).getScope();
+        final Class scope = beanContext.get(CdiEjbBean.class).getScope();
         if(scope != Dependent.class) {
             throw new UnsupportedOperationException("Can not call EJB 
Statefull Bean Remove Method without scoped @Dependent.  Found scope: 
@"+scope.getSimpleName());
         }