org.mybatis.guice.sample.SampleTestBase.java Source code

Introduction

Here is the source code for org.mybatis.guice.sample.SampleTestBase.java

Source

/*
 *    Copyright 2010-2012 The MyBatis Team
 *
 *    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.mybatis.guice.sample;

import static org.apache.ibatis.io.Resources.getResourceAsReader;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.io.IOException;
import java.sql.SQLException;
import java.util.Properties;

import javax.sql.DataSource;

import org.apache.ibatis.jdbc.ScriptRunner;
import org.apache.ibatis.mapping.Environment;
import org.apache.ibatis.session.SqlSessionFactory;
import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;
import org.mybatis.guice.CustomException;
import org.mybatis.guice.sample.domain.User;
import org.mybatis.guice.sample.service.FooService;

import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Module;

/**
 * Example of MyBatis-Guice basic integration usage.
 *
 * This is the recommended scenario.
 *
 * @version $Id$
 */
@Ignore
public abstract class SampleTestBase {
    private FooService fooService;

    @Before
    public void setupMyBatisGuice() throws Exception {
        Injector injector = Guice.createInjector(createModules());

        createTestDb(injector);

        fooService = injector.getInstance(FooService.class);
    }

    protected abstract Module[] createModules();

    protected Properties createTestProperties() {
        Properties myBatisProperties = new Properties();
        myBatisProperties.setProperty("mybatis.environment.id", "test");
        myBatisProperties.setProperty("JDBC.username", "sa");
        myBatisProperties.setProperty("JDBC.password", "");
        myBatisProperties.setProperty("JDBC.autoCommit", "false");
        return myBatisProperties;
    }

    private void createTestDb(Injector injector) throws IOException, SQLException {
        // prepare the test db
        Environment environment = injector.getInstance(SqlSessionFactory.class).getConfiguration().getEnvironment();
        DataSource dataSource = environment.getDataSource();

        ScriptRunner runner = new ScriptRunner(dataSource.getConnection());
        runner.setAutoCommit(true);
        runner.setStopOnError(true);
        runner.runScript(getResourceAsReader("org/mybatis/guice/sample/db/database-schema.sql"));
        runner.runScript(getResourceAsReader("org/mybatis/guice/sample/db/database-test-data.sql"));
        runner.closeConnection();
    }

    @Test
    public void testFooService() {
        User user = fooService.doSomeBusinessStuff("u1");
        assertNotNull(user);
        assertEquals("Pocoyo", user.getName());
    }

    @Test(expected = IllegalArgumentException.class)
    public void testTransactionalOnClassAndMethod() {
        User user = new User();
        user.setName("Christian Poitras");
        fooService.brokenInsert(user);
    }

    @Test(expected = CustomException.class)
    public void testTransactionalOnClass() {
        User user = new User();
        user.setName("Christian Poitras");
        fooService.brokenInsert2(user);
    }
}