package com.alibaba.druid.bvt.console;


import java.sql.Connection;
import java.sql.Statement;
import com.alibaba.druid.support.console.Option;
import org.junit.runner.notification.Failure;
import org.junit.runner.JUnitCore;
import org.junit.runner.Result;
import com.alibaba.druid.support.console.DruidStat;
import com.alibaba.druid.util.JdbcUtils;
import com.alibaba.druid.pool.DruidDataSource;

import java.lang.management.ManagementFactory;
import junit.framework.TestCase;

public class DruidStatTest extends TestCase {

    private DruidDataSource dataSource;
    private DruidDataSource dataSource2;

    private static String getSelfPid() {
        String pid = ManagementFactory.getRuntimeMXBean().getName().split("@")[0];
        return pid;
    }

    protected void setUp() throws Exception {
        dataSource = new DruidDataSource();
        dataSource.setMinIdle(1);
        dataSource.setUrl("jdbc:h2:mem:test;");
        dataSource.setTestOnBorrow(false);
        dataSource.setFilters("stat");
        dataSource.setRemoveAbandoned(true);
        dataSource.setRemoveAbandonedTimeout(600);
        JdbcUtils.execute(dataSource, "CREATE TABLE user (id INT, name VARCHAR(40))");
        JdbcUtils.execute(dataSource, "insert into user values(20,'name1')");
        JdbcUtils.execute(dataSource, "insert into user values(30,'name2')");

        dataSource2 = new DruidDataSource();
        dataSource2.setMinIdle(1);
        dataSource2.setUrl("jdbc:h2:mem:test2;");
        dataSource2.setTestOnBorrow(false);
        dataSource2.setFilters("stat");
        dataSource2.setRemoveAbandoned(true);
        dataSource2.setRemoveAbandonedTimeout(600);

        JdbcUtils.execute(dataSource2, "CREATE TABLE user (id INT, name VARCHAR(40))");
        JdbcUtils.execute(dataSource2, "insert into user values(20,'name1')");

    }

    protected void tearDown() throws Exception {
        JdbcUtils.execute(dataSource, "DROP TABLE user");
        JdbcUtils.close(dataSource);

        JdbcUtils.execute(dataSource2, "DROP TABLE user");
        JdbcUtils.close(dataSource2);
    }

    public void test_printDruidStat() throws Exception {
        String pid = getSelfPid();
        String[] cmdArray = {"-sql", pid};
        Option opt = Option.parseOptions(cmdArray);
        DruidStat.printDruidStat(opt);
    }

    public void test_printDruidStat2() throws Exception {
        String pid = getSelfPid();
        String[] cmdArray = {"-ds","-s2", pid};
        Option opt = Option.parseOptions(cmdArray);
        DruidStat.printDruidStat(opt);
    }

    public void test_printDruidStat3() throws Exception {
        String pid = getSelfPid();
        String[] cmdArray = {"-act", pid};
        Option opt = Option.parseOptions(cmdArray);

        Connection conn = null;
        Statement stmt = null;
        try {
            conn = dataSource.getConnection();
            stmt = conn.createStatement();
            stmt.execute("insert into user values(30,'name2')");
            DruidStat.printDruidStat(opt);
        } finally {
            if (stmt != null ) try { stmt.close(); } catch (Exception e) {}
            if (conn != null ) try { conn.close(); } catch (Exception e) {}
        }
    }

    public static void main(String[] args) {
		Result result = JUnitCore.runClasses(DruidStatTest.class);
		for (Failure failure : result.getFailures()) {
			System.out.println(failure.toString());
		}
	}

}