1 files changed, 224 insertions, 0 deletions
diff --git a/sca-java-2.x/tags/2.0.1-RC1/modules/policy-security-jsr250/src/test/java/org/apache/tuscany/sca/policy/security/jsr250/PolicyProcessorTestCase.java b/sca-java-2.x/tags/2.0.1-RC1/modules/policy-security-jsr250/src/test/java/org/apache/tuscany/sca/policy/security/jsr250/PolicyProcessorTestCase.java
new file mode 100644
index 0000000000..de8e0ccacf
--- /dev/null
+++ b/sca-java-2.x/tags/2.0.1-RC1/modules/policy-security-jsr250/src/test/java/org/apache/tuscany/sca/policy/security/jsr250/PolicyProcessorTestCase.java
@@ -0,0 +1,224 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.tuscany.sca.policy.security.jsr250;
+
+import java.lang.reflect.Method;
+
+import javax.annotation.security.DenyAll;
+import javax.annotation.security.PermitAll;
+import javax.annotation.security.RolesAllowed;
+import javax.annotation.security.RunAs;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import org.apache.tuscany.sca.assembly.DefaultAssemblyFactory;
+import org.apache.tuscany.sca.core.DefaultExtensionPointRegistry;
+import org.apache.tuscany.sca.core.ExtensionPointRegistry;
+import org.apache.tuscany.sca.implementation.java.DefaultJavaImplementationFactory;
+import org.apache.tuscany.sca.implementation.java.JavaImplementation;
+import org.apache.tuscany.sca.implementation.java.JavaImplementationFactory;
+import org.apache.tuscany.sca.implementation.java.introspect.impl.PolicyProcessor;
+import org.apache.tuscany.sca.implementation.java.introspect.impl.ServiceProcessor;
+import org.apache.tuscany.sca.interfacedef.Operation;
+import org.apache.tuscany.sca.interfacedef.java.DefaultJavaInterfaceFactory;
+import org.apache.tuscany.sca.interfacedef.java.JavaInterface;
+import org.apache.tuscany.sca.interfacedef.java.JavaOperation;
+import org.apache.tuscany.sca.interfacedef.java.impl.PolicyJavaInterfaceVisitor;
+import org.apache.tuscany.sca.policy.DefaultPolicyFactory;
+import org.oasisopen.sca.annotation.Service;
+
+/**
+ * @version $Rev: 662474 $ $Date: 2008-06-02 17:18:28 +0100 (Mon, 02 Jun 2008) $
+ */
+public class PolicyProcessorTestCase extends TestCase {
+    private ExtensionPointRegistry registry;
+    private ServiceProcessor serviceProcessor;
+    private PolicyProcessor policyProcessor;
+    private JSR250PolicyProcessor jsr250Processor;
+    private PolicyJavaInterfaceVisitor visitor;
+    private JavaImplementation type;   
+    
+    private interface Interface1 {
+        int method1();
+
+        int method2();
+
+        int method3();
+
+        int method4();
+    }
+    
+    @RunAs("Role1")
+    @Service(Interface1.class)
+    private class Service1 implements Interface1 {
+        public int method1() {
+            return 0;
+        }
+
+        public int method2() {
+            return 0;
+        }
+
+        public int method3() {
+            return 0;
+        }
+
+        public int method4() {
+            return 0;
+        }
+    }   
+    
+    @RolesAllowed({"Role2", "Role3"})
+    @Service(Interface1.class)
+    private class Service2 implements Interface1 {
+        public int method1() {
+            return 0;
+        }
+
+        public int method2() {
+            return 0;
+        }
+
+        public int method3() {
+            return 0;
+        }
+
+        public int method4() {
+            return 0;
+        }
+    } 
+    
+    @PermitAll()
+    @Service(Interface1.class)
+    private class Service3 implements Interface1 {
+        public int method1() {
+            return 0;
+        }
+
+        public int method2() {
+            return 0;
+        }
+
+        public int method3() {
+            return 0;
+        }
+
+        public int method4() {
+            return 0;
+        }
+    }    
+    
+
+    @Service(Interface1.class)
+    private class Service4 implements Interface1 {
+        public int method1() {
+            return 0;
+        }
+
+        @RolesAllowed({"Role4", "Role5"})
+        public int method2() {
+            return 0;
+        }
+
+        @PermitAll
+        public int method3() {
+            return 0;
+        }
+
+        @DenyAll
+        public int method4() {
+            return 0;
+        }
+    }       
+    
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        registry = new DefaultExtensionPointRegistry();
+        registry.start();
+        serviceProcessor = new ServiceProcessor(new DefaultAssemblyFactory(), new DefaultJavaInterfaceFactory(registry));
+        policyProcessor = new PolicyProcessor(registry);
+        jsr250Processor = new JSR250PolicyProcessor(registry);
+        visitor = new PolicyJavaInterfaceVisitor(registry);
+        JavaImplementationFactory javaImplementationFactory = new DefaultJavaImplementationFactory();
+        type = javaImplementationFactory.createJavaImplementation();
+    }
+
+    public void testSingleInterfaceWithRunAsAtClassLevel() throws Exception {
+        runProcessors(Service1.class, null, type);
+        Assert.assertEquals(1, type.getPolicySets().size());
+    }  
+    
+    public void testSingleInterfaceWithRolesAllowedsAtClassLevel() throws Exception {
+        runProcessors(Service2.class, null, type);
+        Assert.assertEquals(1, type.getPolicySets().size());
+    }  
+    
+    public void testSingleInterfaceWithPermitAllAtClassLevel() throws Exception {
+        runProcessors(Service3.class, null, type);
+        Assert.assertEquals(1, type.getPolicySets().size());
+    }     
+    
+    public void testSingleInterfaceWithRolesAllowedAtMethodLevel() throws Exception {
+        Method aMethod = Service4.class.getDeclaredMethod("method2");
+        runProcessors(Service4.class, aMethod, type);
+        Operation op = getOperationModel(aMethod, type);
+        Assert.assertEquals(1, op.getPolicySets().size());
+    } 
+    
+    public void testSingleInterfaceWithPermitAllAtMethodLevel() throws Exception {
+        Method aMethod = Service4.class.getDeclaredMethod("method3");
+        runProcessors(Service4.class, aMethod, type);
+        Operation op = getOperationModel(aMethod, type);
+        Assert.assertEquals(1, op.getPolicySets().size());
+    }     
+
+    public void testSingleInterfaceWithDenyAllAtMethodLevel() throws Exception {
+        Method aMethod = Service4.class.getDeclaredMethod("method4");
+        runProcessors(Service4.class, aMethod, type);
+        Operation op = getOperationModel(aMethod, type);
+        Assert.assertEquals(1, op.getPolicySets().size());
+    }  
+    
+    public void testSingleInterfaceWithNothingAtMethodLevel() throws Exception {
+        Method aMethod = Service4.class.getDeclaredMethod("method1");
+        runProcessors(Service4.class, aMethod, type);
+        Operation op = getOperationModel(aMethod, type);
+        Assert.assertEquals(0, op.getPolicySets().size());
+    }      
+    
+    private void runProcessors(Class clazz, Method method, JavaImplementation type)throws Exception {
+        serviceProcessor.visitClass(clazz, type);
+        policyProcessor.visitClass(clazz, type);
+        jsr250Processor.visitClass(clazz, type);
+        if (method != null){
+            jsr250Processor.visitMethod(method, type);
+        }
+    }
+    
+    private Operation getOperationModel(Method method, JavaImplementation type){
+        for(Operation op : type.getOperations()){
+            if (((JavaOperation)op).getJavaMethod().equals(method)){
+                return op;
+            } 
+        }
+        return null;
+    }
+}