git.aero2k.de Git - tmp/jakarta-migration.git/blob

   1 /**
   2  *  This document is a part of the source code and related artifacts
   3  *  for CollectionSpace, an open source collections management system
   4  *  for museums and related institutions:
   5
   6  *  http://www.collectionspace.org
   7  *  http://wiki.collectionspace.org
   8
   9  *  Copyright 2009 University of California at Berkeley
  10
  11  *  Licensed under the Educational Community License (ECL), Version 2.0.
  12  *  You may not use this file except in compliance with this License.
  13
  14  *  You may obtain a copy of the ECL 2.0 License at
  15
  16  *  https://source.collectionspace.org/collection-space/LICENSE.txt
  17
  18  *  Unless required by applicable law or agreed to in writing, software
  19  *  distributed under the License is distributed on an "AS IS" BASIS,
  20  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  21  *  See the License for the specific language governing permissions and
  22  *  limitations under the License.
  23  *//**
  24  *  This document is a part of the source code and related artifacts
  25  *  for CollectionSpace, an open source collections management system
  26  *  for museums and related institutions:
  27
  28  *  http://www.collectionspace.org
  29  *  http://wiki.collectionspace.org
  30
  31  *  Copyright 2009 University of California at Berkeley
  32
  33  *  Licensed under the Educational Community License (ECL), Version 2.0.
  34  *  You may not use this file except in compliance with this License.
  35
  36  *  You may obtain a copy of the ECL 2.0 License at
  37
  38  *  https://source.collectionspace.org/collection-space/LICENSE.txt
  39
  40  *  Unless required by applicable law or agreed to in writing, software
  41  *  distributed under the License is distributed on an "AS IS" BASIS,
  42  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  43  *  See the License for the specific language governing permissions and
  44  *  limitations under the License.
  45  */
  46 /*
  47  * To change this template, choose Tools | Templates
  48  * and open the template in the editor.
  49  */
  50 package org.collectionspace.services.authorization.test;
  51
  52 import java.io.File;
  53 import java.io.FileInputStream;
  54 import java.io.InputStream;
  55 import org.slf4j.Logger;
  56 import org.slf4j.LoggerFactory;
  57 import java.lang.reflect.Method;
  58 import java.util.HashSet;
  59 import javax.xml.bind.JAXBContext;
  60 import javax.xml.bind.Marshaller;
  61 import javax.xml.bind.Unmarshaller;
  62 import org.collectionspace.services.authorization.AuthZ;
  63 import org.springframework.context.support.ClassPathXmlApplicationContext;
  64 import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
  65 import org.springframework.security.core.Authentication;
  66 import org.springframework.security.core.GrantedAuthority;
  67 import org.springframework.security.core.authority.GrantedAuthorityImpl;
  68 import org.springframework.security.core.context.SecurityContextHolder;
  69 import org.springframework.transaction.TransactionDefinition;
  70 import org.springframework.transaction.TransactionStatus;
  71 import org.springframework.transaction.support.DefaultTransactionDefinition;
  72 import org.testng.annotations.DataProvider;
  73 import org.testng.annotations.Test;
  74
  75 /**
  76  *
  77  * @author
  78  */
  79 public abstract class AbstractAuthorizationTestImpl {
  80
  81     final Logger logger = LoggerFactory.getLogger(AbstractAuthorizationTestImpl.class);
  82     private org.springframework.jdbc.datasource.DataSourceTransactionManager txManager;
  83     final static String testDataDir = "src/test/resources/test-data/";
  84
  85     /**
  86      * Returns the name of the currently running test.
  87      *
  88      * Note: although the return type is listed as Object[][],
  89      * this method instead returns a String.
  90      *
  91      * @param   m  The currently running test method.
  92      *
  93      * @return  The name of the currently running test method.
  94      */
  95     @DataProvider(name = "testName")
  96     protected static Object[][] testName(Method m) {
  97         return new Object[][]{
  98                     new Object[]{m.getName()}
  99                 };
 100     }
 101
 102     protected void setup() {
 103         ClassPathXmlApplicationContext appContext = new ClassPathXmlApplicationContext(
 104                 new String[]{"applicationContext-authorization-test.xml"});
 105         login();
 106         AuthZ authZ = AuthZ.get();
 107         txManager = (org.springframework.jdbc.datasource.DataSourceTransactionManager) appContext.getBean("transactionManager");
 108     }
 109
 110     protected void login() {
 111         GrantedAuthority gauth = new GrantedAuthorityImpl("ROLE_ADMINISTRATOR");
 112         HashSet<GrantedAuthority> gauths = new HashSet<GrantedAuthority>();
 113         gauths.add(gauth);
 114         Authentication authRequest = new UsernamePasswordAuthenticationToken("test", "test", gauths);
 115         SecurityContextHolder.getContext().setAuthentication(authRequest);
 116     }
 117
 118     protected void logout() {
 119         SecurityContextHolder.getContext().setAuthentication(null);
 120     }
 121
 122     protected TransactionStatus beginTransaction(String name) {
 123         DefaultTransactionDefinition def = new DefaultTransactionDefinition();
 124         // explicitly setting the transaction name is something that can only be done programmatically
 125         def.setName(name);
 126         def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED);
 127         return txManager.getTransaction(def);
 128     }
 129
 130     protected void rollbackTransaction(TransactionStatus status) {
 131         txManager.rollback(status);
 132     }
 133
 134     protected void commitTransaction(TransactionStatus status) {
 135         txManager.commit(status);
 136     }
 137
 138     static void toFile(Object o, Class jaxbClass, String fileName) {
 139         File f = new File(fileName);
 140         try {
 141             JAXBContext jc = JAXBContext.newInstance(jaxbClass);
 142             Marshaller m = jc.createMarshaller();
 143             m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT,
 144                     Boolean.TRUE);
 145             m.marshal(o, f);
 146         } catch (Exception e) {
 147             e.printStackTrace();
 148         }
 149     }
 150
 151     static Object fromFile(Class jaxbClass, String fileName) throws Exception {
 152         InputStream is = new FileInputStream(fileName);
 153         try {
 154             JAXBContext context = JAXBContext.newInstance(jaxbClass);
 155             Unmarshaller unmarshaller = context.createUnmarshaller();
 156             //note: setting schema to null will turn validator off
 157             unmarshaller.setSchema(null);
 158             return jaxbClass.cast(unmarshaller.unmarshal(is));
 159         } finally {
 160             if (is != null) {
 161                 try {
 162                     is.close();
 163                 } catch (Exception e) {
 164                 }
 165             }
 166         }
 167     }
 168
 169     @Test(dataProvider = "testName", dataProviderClass = AbstractAuthorizationTestImpl.class)
 170     public void test(String testName) {
 171         if (logger.isDebugEnabled()) {
 172             logger.debug(testName);
 173         }
 174     }
 175 }