/*
 * Copyright 2008,  Unitils.org
 *
 * Licensed under the Apache 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.apache.org/licenses/LICENSE-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.unitils.database;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertSame;
import java.util.Properties;

import javax.sql.DataSource;

import org.junit.Before;
import org.junit.Test;
import org.unitils.UnitilsJUnit4;
import org.unitils.core.ConfigurationLoader;
import org.unitils.core.dbsupport.SQLHandler;
import org.unitils.database.annotations.TestDataSource;

/**
 * Tests for the DatabaseModule
 * 
 * @author Filip Neven
 * @author Tim Ducheyne
 */
public class DatabaseModuleTest extends UnitilsJUnit4 {

	/* Tested object */
	private TestDatabaseModule databaseModule;


	/**
	 * Initializes the test fixture.
	 */
	@Before
	public void setUp() throws Exception {
		Properties configuration = new ConfigurationLoader().loadConfiguration();
		configuration.setProperty(DatabaseModule.PROPERTY_UPDATEDATABASESCHEMA_ENABLED, "true");

		databaseModule = new TestDatabaseModule();
		databaseModule.init(configuration);
		DataSourceWrapper wrapper = databaseModule.getWrapper("");
		databaseModule.setWrapper(wrapper);
		wrapper.setTransactionManager(databaseModule.getTransactionManager());
	}


	/**
	 * Test the injection of the dataSource into a test object. This should also have triggered the DbMaintainer.
	 */
	@Test
	public void testInjectDataSource() throws Exception {
		DbTest dbTest = new DbTest();
		databaseModule.injectDataSource(dbTest);

		assertNotNull(dbTest.dataSourceFromField);
		assertNotNull(dbTest.dataSourceFromMethod);
		assertSame(dbTest.dataSourceFromField, dbTest.dataSourceFromMethod);
		//assertTrue(databaseModule.updateDataSchemaCalled);
	}


	/**
	 * Object that plays the role of database test object in this class's tests.
	 */
	public static class DbTest {

		private DataSource dataSourceFromMethod;

		@TestDataSource
		private DataSource dataSourceFromField = null;

		@TestDataSource
		public void setDataSource(DataSource dataSource) {
			this.dataSourceFromMethod = dataSource;
		}
	}


	/**
	 * Database module that intercepts the updating of the database schema.
	 */
	public class TestDatabaseModule extends DatabaseModule {

		private boolean updateDataSchemaCalled = false;

		public void updateDatabase(SQLHandler sqlHandler) {
			updateDataSchemaCalled = true;
		}
	}

}