| 1 |  |   | 
  | 2 |  |   | 
  | 3 |  |   | 
  | 4 |  |   | 
  | 5 |  |   | 
  | 6 |  |   | 
  | 7 |  |   | 
  | 8 |  |   | 
  | 9 |  |   | 
  | 10 |  |   | 
  | 11 |  |   | 
  | 12 |  |   | 
  | 13 |  |   | 
  | 14 |  |   | 
  | 15 |  |   | 
  | 16 |  |  package org.kuali.student.common.test.spring; | 
  | 17 |  |   | 
  | 18 |  |  import java.lang.reflect.AccessibleObject; | 
  | 19 |  |  import java.lang.reflect.Field; | 
  | 20 |  |   | 
  | 21 |  |  import org.springframework.beans.factory.config.BeanDefinition; | 
  | 22 |  |  import org.springframework.beans.factory.support.BeanDefinitionRegistry; | 
  | 23 |  |  import org.springframework.beans.factory.support.RootBeanDefinition; | 
  | 24 |  |  import org.springframework.context.ApplicationContext; | 
  | 25 |  |  import org.springframework.test.context.TestContext; | 
  | 26 |  |  import org.springframework.test.context.support.DependencyInjectionTestExecutionListener; | 
  | 27 |  |   | 
  | 28 | 1 |  public class DaoTestDependencyInjectorListener extends | 
  | 29 |  |                  DependencyInjectionTestExecutionListener { | 
  | 30 |  |   | 
  | 31 |  |          @Override | 
  | 32 |  |          protected void injectDependencies(TestContext testContext) throws Exception { | 
  | 33 |  |   | 
  | 34 | 2 |                  ApplicationContext context = testContext.getApplicationContext(); | 
  | 35 |  |   | 
  | 36 | 2 |                  BeanDefinitionRegistry registry = ((BeanDefinitionRegistry) context); | 
  | 37 |  |   | 
  | 38 | 2 |                  Field[] fields = testContext.getTestClass().getDeclaredFields(); | 
  | 39 | 6 |                  for (Field f : fields) { | 
  | 40 | 4 |                          Dao a = f.getAnnotation(Dao.class); | 
  | 41 | 4 |                          if (a != null) { | 
  | 42 | 2 |                                  Class<?> clazz = Class.forName(a.value()); | 
  | 43 | 2 |                                  BeanDefinition definition = new RootBeanDefinition(clazz); | 
  | 44 | 2 |                                  registry.registerBeanDefinition(f.getName(), definition); | 
  | 45 | 2 |                                  Field.setAccessible(new AccessibleObject[]{f}, true); | 
  | 46 | 2 |                                  f.set(testContext.getTestInstance(), testContext | 
  | 47 |  |                                                  .getApplicationContext().getBean(f.getName())); | 
  | 48 |  |                          } | 
  | 49 |  |                  } | 
  | 50 |  |   | 
  | 51 | 2 |                  super.injectDependencies(testContext); | 
  | 52 |  |   | 
  | 53 | 2 |          } | 
  | 54 |  |  } |