wamblee.org Git - utils/blob - system/general/src/test/java/org/wamblee/system/components/DatabaseComponent.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.system.components;
  17
  18 import java.util.Properties;
  19
  20 import org.wamblee.system.components.ORMappingConfig.DatabaseType;
  21 import org.wamblee.system.core.AbstractComponent;
  22 import org.wamblee.system.core.DefaultProvidedInterface;
  23 import org.wamblee.system.core.ProvidedInterface;
  24 import org.wamblee.system.core.Scope;
  25 import org.wamblee.test.persistence.Database;
  26 import org.wamblee.test.persistence.DerbyDatabase;
  27
  28 /**
  29  *
  30  * @author $author$
  31  * @version $Revision$
  32  */
  33 public class DatabaseComponent extends AbstractComponent<Database> {
  34     private static ProvidedInterface DB_PROPS = new DefaultProvidedInterface(
  35         "dbProps", Properties.class);
  36
  37     private Database database;
  38
  39     /**
  40      * Creates a new DatabaseComponent object.
  41      *
  42      */
  43     public DatabaseComponent(String aName, Database aDatabase) {
  44         super(aName);
  45         database = aDatabase;
  46         addProvidedInterface(DB_PROPS);
  47     }
  48
  49     @Override
  50     protected Database doStart(Scope aScope) {
  51         database.start();
  52
  53         Properties props = new Properties();
  54
  55         if (database instanceof DerbyDatabase) {
  56             props.put("database.type", DatabaseType.DERBY.toString());
  57         } else {
  58             throw new IllegalArgumentException("Unknown database type " +
  59                 database);
  60         }
  61
  62         // props.put("database.driver", database.getDriverClassName());
  63         props.put("database.url", database.getJdbcUrl());
  64         props.put("database.username", database.getUsername());
  65         props.put("database.password", database.getPassword());
  66
  67         addInterface(DB_PROPS, props, aScope);
  68
  69         return database;
  70     }
  71
  72     @Override
  73     protected void doStop(Database aRuntime) {
  74         database.stop();
  75     }
  76 }