DruidDataSourceTest.java

package com.alibaba.druid.bvt.pool;

import java.sql.Connection;
import java.sql.SQLException;

import junit.framework.TestCase;

import org.junit.Assert;

import com.alibaba.druid.pool.DruidDataSource;

public class DruidDataSourceTest extends TestCase {

    private DruidDataSource dataSource;

    protected void setUp() throws Exception {
        dataSource = new DruidDataSource();
        dataSource.setUrl("jdbc:mock:xxx");
        dataSource.setTestOnBorrow(false);

        dataSource.init();
    }

    protected void tearDown() throws Exception {
        dataSource.close();
    }

    public void test_getInitStackTrace() {
        String stackTrace = dataSource.getInitStackTrace();
        Assert.assertTrue(stackTrace.indexOf("com.alibaba.druid.bvt.pool.DruidDataSourceTest.setUp") != -1);
    }

    public void test_restart() throws Exception {
        Assert.assertEquals(true, dataSource.isInited());
        {
            Connection conn = dataSource.getConnection();
            Assert.assertEquals(1, dataSource.getActiveCount());

            Exception error = null;
            try {
                dataSource.restart();
            } catch (SQLException ex) {
                error = ex;
            }
            Assert.assertNotNull(error);
            Assert.assertEquals(true, dataSource.isInited());

            conn.close();
            dataSource.restart();
        }

        Assert.assertEquals(0, dataSource.getActiveCount());
        Assert.assertEquals(false, dataSource.isInited());

        Connection conn = dataSource.getConnection();
        conn.close();
    }
}