wamblee.org Git - utils/blob - test/hibernate/src/main/java/org/wamblee/test/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.test.persistence.hibernate;
  17
  18 import java.util.Map;
  19
  20 import org.dbunit.dataset.filter.ITableFilterSimple;
  21 import org.wamblee.test.persistence.JpaCustomizer;
  22 import org.wamblee.test.persistence.PersistenceUnitDescription;
  23
  24 /**
  25  *
  26  * @author $author$
  27  * @version $Revision$
  28  */
  29 public class HibernateJpaCustomizer implements JpaCustomizer {
  30     /**
  31      * Creates a new HibernateJpaCustomizer object.
  32      */
  33     public HibernateJpaCustomizer() {
  34         // Empty
  35     }
  36
  37     @Override
  38     public void customize(PersistenceUnitDescription aPersistenceUnit,
  39         Map<String, String> aJpaProperties) {
  40
  41         System.setProperty("hibernate.temp.use_jdbc_metadata_defaults", "true");
  42
  43         // Set non-JPA connection properties for older versions of hibernate
  44         System.getProperties().remove("hibernate.connection.datasource");
  45         if (System.getProperty("hibernate.connection.datasource") != null ) {
  46             throw new RuntimeException("ERROR");
  47         }
  48         System.setProperty("hibernate.connection.url", aJpaProperties.get("javax.persistence.jdbc.url"));
  49         System.setProperty("hibernate.connection.username", aJpaProperties.get("javax.persistence.jdbc.user"));
  50         System.setProperty("hibernate.connection.password", aJpaProperties.get("javax.persistence.jdbc.password"));
  51
  52         // Hibernate schema generation
  53         aJpaProperties.put("hibernate.hbm2ddl.auto", "create-drop");
  54     }
  55
  56     @Override
  57     public ITableFilterSimple getJpaTables() {
  58         return new HibernateTables();
  59     }
  60 }