001 /** 002 * Copyright 2005-2011 The Kuali Foundation 003 * 004 * Licensed under the Educational Community License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.opensource.org/licenses/ecl2.php 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 package org.kuali.rice.kim.test; 017 018 import org.kuali.rice.core.api.lifecycle.BaseLifecycle; 019 import org.kuali.rice.core.api.lifecycle.Lifecycle; 020 import org.kuali.rice.core.framework.resourceloader.SpringResourceLoader; 021 import org.kuali.rice.kim.api.services.KimApiServiceLocator; 022 import org.kuali.rice.kim.api.type.KimType; 023 import org.kuali.rice.kim.impl.permission.PermissionTemplateBo; 024 import org.kuali.rice.krad.service.KRADServiceLocator; 025 import org.kuali.rice.test.BaselineTestCase; 026 import org.kuali.rice.test.BaselineTestCase.BaselineMode; 027 import org.kuali.rice.test.BaselineTestCase.Mode; 028 import org.kuali.rice.test.SQLDataLoader; 029 import org.kuali.rice.test.lifecycles.KEWXmlDataLoaderLifecycle; 030 031 import javax.xml.namespace.QName; 032 import java.util.ArrayList; 033 import java.util.HashMap; 034 import java.util.List; 035 import java.util.Map; 036 037 import static org.junit.Assert.fail; 038 039 /** 040 * This is test base that should be used for all KIM unit tests. All non-web unit tests for KIM should extend this base 041 * class. 042 * 043 * @author Kuali Rice Team (rice.collab@kuali.org) 044 */ 045 @BaselineMode(Mode.ROLLBACK_CLEAR_DB) 046 public abstract class KIMTestCase extends BaselineTestCase { 047 048 private static final String KIM_MODULE_NAME = "kim"; 049 050 public KIMTestCase() { 051 super(KIM_MODULE_NAME); 052 } 053 054 @Override 055 protected List<Lifecycle> getSuiteLifecycles() { 056 List<Lifecycle> suiteLifecycles = super.getSuiteLifecycles(); 057 suiteLifecycles.add(new KEWXmlDataLoaderLifecycle("classpath:org/kuali/rice/kim/test/DefaultSuiteTestData.xml")); 058 return suiteLifecycles; 059 } 060 061 @Override 062 protected void loadSuiteTestData() throws Exception { 063 super.loadSuiteTestData(); 064 new SQLDataLoader("classpath:org/kuali/rice/kim/test/DefaultSuiteTestData.sql", "/").runSql(); 065 new SQLDataLoader("classpath:org/kuali/rice/kim/test/CircularRolesTestData.sql", "/").runSql(); 066 new SQLDataLoader("classpath:org/kuali/rice/kim/test/CircularGroupsTestData.sql", "/").runSql(); 067 } 068 069 @Override 070 protected Lifecycle getLoadApplicationLifecycle() { 071 SpringResourceLoader springResourceLoader = new SpringResourceLoader(new QName("KIMTestHarnessApplicationResourceLoader"), "classpath:KIMTestHarnessSpringBeans.xml", null); 072 springResourceLoader.setParentSpringResourceLoader(getTestHarnessSpringResourceLoader()); 073 return springResourceLoader; 074 } 075 076 /** 077 * Override the standard per-test lifecycles to prepend ClearDatabaseLifecycle and ClearCacheLifecycle 078 * @see org.kuali.rice.test.RiceTestCase#getPerTestLifecycles() 079 */ 080 @Override 081 protected List<Lifecycle> getPerTestLifecycles() { 082 List<Lifecycle> lifecycles = super.getPerTestLifecycles(); 083 lifecycles.add(new ClearCacheLifecycle()); 084 return lifecycles; 085 } 086 087 public class ClearCacheLifecycle extends BaseLifecycle { 088 @Override 089 public void stop() throws Exception { 090 //KimApiServiceLocator.getIdentityManagementService().flushAllCaches(); 091 //KimApiServiceLocator.getRoleService().flushRoleCaches(); 092 super.stop(); 093 } 094 095 } 096 097 protected List<String> getPerTestTablesNotToClear() { 098 List<String> tablesNotToClear = new ArrayList<String>(); 099 tablesNotToClear.add("KRIM_.*"); 100 tablesNotToClear.add("KRNS_.*"); 101 tablesNotToClear.add("KRCR_.*"); 102 tablesNotToClear.add("KREW_.*"); 103 return tablesNotToClear; 104 } 105 106 107 /** 108 * @see org.kuali.rice.test.RiceTestCase#getModuleName() 109 */ 110 @Override 111 protected String getModuleName() { 112 return KIM_MODULE_NAME; 113 } 114 115 protected KimType getDefaultKimType() { 116 KimType type = KimApiServiceLocator.getKimTypeInfoService().getKimType("1"); 117 if (type == null) { 118 fail("Failed to locate the default Kim Type."); 119 } 120 return type; 121 } 122 123 protected PermissionTemplateBo getDefaultPermissionTemplate() { 124 Map<String, Object> fieldValues = new HashMap<String, Object>(); 125 fieldValues.put("namespaceCode", "KUALI"); 126 fieldValues.put("name", "Default"); 127 PermissionTemplateBo template = KRADServiceLocator.getBusinessObjectService().findByPrimaryKey(PermissionTemplateBo.class, fieldValues); 128 if (template == null) { 129 fail("Failed to locate the default Permission Template."); 130 } 131 return template; 132 } 133 134 protected String getNewRoleId() { 135 return getIdFromSequence("KRIM_ROLE_ID_S"); 136 } 137 138 protected String getNewRoleMemberId() { 139 return getIdFromSequence("KRIM_ROLE_MBR_ID_S"); 140 } 141 142 protected String getNewRolePermissionId() { 143 return getIdFromSequence("KRIM_ROLE_ID_S"); 144 } 145 146 protected String getIdFromSequence(String sequenceName) { 147 Long sequenceId = KRADServiceLocator.getSequenceAccessorService().getNextAvailableSequenceNumber(sequenceName); 148 return "" + sequenceId; 149 } 150 }