wamblee.org Git - utils/blob - test/hibernate/src/main/java/org/wamblee/support/persistence/hibernate/HibernateJpaCustomizer.java

   1 /*
   2  * Copyright 2005-2010 the original author or authors.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.wamblee.support.persistence.hibernate;
  17
  18 import org.dbunit.dataset.filter.ITableFilterSimple;
  19
  20 import org.wamblee.support.persistence.JpaCustomizer;
  21 import org.wamblee.support.persistence.PersistenceUnitDescription;
  22
  23 import java.util.Map;
  24
  25 /**
  26  *
  27  * @author $author$
  28  * @version $Revision$
  29  */
  30 public class HibernateJpaCustomizer implements JpaCustomizer {
  31     /**
  32      * Creates a new HibernateJpaCustomizer object.
  33      */
  34     public HibernateJpaCustomizer() {
  35         // Empty
  36     }
  37
  38     @Override
  39     public void customize(PersistenceUnitDescription aPersistenceUnit,
  40         Map<String, String> aJpaProperties) {
  41
  42         System.setProperty("hibernate.temp.use_jdbc_metadata_defaults", "true");
  43
  44         // Set non-JPA connection properties for older versions of hibernate
  45         System.getProperties().remove("hibernate.connection.datasource");
  46         if (System.getProperty("hibernate.connection.datasource") != null ) {
  47             throw new RuntimeException("ERROR");
  48         }
  49         System.setProperty("hibernate.connection.url", aJpaProperties.get("javax.persistence.jdbc.url"));
  50         System.setProperty("hibernate.connection.username", aJpaProperties.get("javax.persistence.jdbc.user"));
  51         System.setProperty("hibernate.connection.password", aJpaProperties.get("javax.persistence.jdbc.password"));
  52
  53         // Hibernate schema generation
  54         aJpaProperties.put("hibernate.hbm2ddl.auto", "create-drop");
  55     }
  56
  57     @Override
  58     public ITableFilterSimple getJpaTables() {
  59         return new HibernateTables();
  60     }
  61 }