/**

  * Copyright 2010 The Kuali Foundation Licensed under the

  * Educational Community 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.osedu.org/licenses/ECL-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.kuali.student.common.test.spring;

 import org.springframework.beans.BeansException;

 import org.springframework.beans.factory.config.BeanDefinition;

 import org.springframework.beans.factory.config.BeanFactoryPostProcessor;

 import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;

 import org.springframework.beans.factory.support.BeanDefinitionRegistry;

 import org.springframework.beans.factory.support.RootBeanDefinition;

 import org.springframework.orm.jpa.support.PersistenceAnnotationBeanPostProcessor;

 public class DaoPostProcessor implements BeanFactoryPostProcessor {

         private String daoImplClasses;

         /**

          * @return the daoImplClasses

          */

         public String getDaoImplClasses() {

                 return daoImplClasses;

         }

         /**

          * @param daoImplClasses

          *            the daoImplClasses to set

          */

         public void setDaoImplClasses(String daoImplClasses) {

                 this.daoImplClasses = daoImplClasses;

         }

         @Override

         public void postProcessBeanFactory(

                         ConfigurableListableBeanFactory beanFactory) throws BeansException {

                 String[] classes = daoImplClasses.split(",");

                 BeanDefinitionRegistry registry = ((BeanDefinitionRegistry) beanFactory);

                 PersistenceAnnotationBeanPostProcessor pabpp = new PersistenceAnnotationBeanPostProcessor();

                 if (classes != null && classes.length > 0) {

                         for (String line : classes) {

                                 if (!"".equals(line)) {

                                         try {

                                                 String[] split = line.split("\\|");

                                                 String className = split[0];

                                                 Class<?> clazz = Class.forName(className);

                                                 BeanDefinition definition = new RootBeanDefinition(

                                                                 clazz);

                                                 registry.registerBeanDefinition(clazz.getSimpleName(),

                                                                 definition);

                                                 pabpp.postProcessMergedBeanDefinition(

                                                                 (RootBeanDefinition) definition, clazz, clazz

                                                                                 .getSimpleName());

                                         } catch (Exception e) {

                                                 throw new RuntimeException(e);

                                         }

                                 }

                         }

                 }

         }

 }