wamblee.org Git - utils/blob - support/spring/src/test/java/org/wamblee/test/spring/HibernateUtils.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.spring;
  17
  18 import org.apache.oro.io.AwkFilenameFilter;
  19
  20 import org.hibernate.cfg.Configuration;
  21
  22 import org.wamblee.io.ClassPathResource;
  23 import org.wamblee.io.InputResource;
  24
  25 import java.io.File;
  26 import java.io.FileFilter;
  27 import java.io.IOException;
  28
  29 import java.util.Map;
  30 import java.util.Properties;
  31 import java.util.TreeMap;
  32
  33 /**
  34  * Hibernate utilities.
  35  *
  36  * @author Erik Brakkee
  37  */
  38 public final class HibernateUtils {
  39     private static final String DATABASE_PROPS = "test.database.properties";
  40
  41     /**
  42      * Disabled.
  43      *
  44      */
  45     private HibernateUtils() {
  46         // Empty
  47     }
  48
  49     /**
  50      *
  51      * @param aDir
  52      *
  53      * @return
  54      *
  55      */
  56     public static Configuration getConfiguration(File aDir) throws IOException {
  57         Configuration conf = new Configuration();
  58         File[] files = aDir.listFiles((FileFilter) (new AwkFilenameFilter(
  59             ".*\\.hbm\\.xml")));
  60
  61         for (File f : files) {
  62             System.out.println("Mapping file: " + f);
  63             conf.addFile(f);
  64         }
  65
  66         Map<String, String> dbProps = getHibernateProperties();
  67
  68         for (Map.Entry<String, String> entry : dbProps.entrySet()) {
  69             System.out.println("Property: " + entry.getKey() + "=" +
  70                 entry.getValue());
  71             conf.setProperty(entry.getKey(), entry.getValue());
  72         }
  73
  74         return conf;
  75     }
  76
  77     private static Map<String, String> getHibernateProperties()
  78         throws IOException {
  79         System.out.println("Reading properties file: " + DATABASE_PROPS);
  80
  81         InputResource lPropFile = new ClassPathResource(DATABASE_PROPS);
  82         Properties props = new Properties();
  83         props.load(lPropFile.getInputStream());
  84
  85         Map<String, String> result = new TreeMap<String, String>();
  86         result.put("hibernate.connection.driver_class", props
  87             .getProperty("database.driver"));
  88         result.put("hibernate.connection.url", props
  89             .getProperty("database.url"));
  90         result.put("hibernate.connection.username", props
  91             .getProperty("database.username"));
  92         result.put("hibernate.connection.password", props
  93             .getProperty("database.password"));
  94
  95         return result;
  96     }
  97 }