com.taobao.tddl.jdbc.group.integration.SpringTest.java Source code

Introduction

Here is the source code for com.taobao.tddl.jdbc.group.integration.SpringTest.java

Source

/*(C) 2007-2012 Alibaba Group Holding Limited.   
 *This program is free software; you can redistribute it and/or modify   
*it under the terms of the GNU General Public License version 2 as   
* published by the Free Software Foundation.   
* Authors:   
*   junyu <junyu@taobao.com> , shenxun <shenxun@taobao.com>,   
*   linxuan <linxuan@taobao.com> ,qihao <qihao@taobao.com>    
*/
package com.taobao.tddl.jdbc.group.integration;

import static org.junit.Assert.*;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.Statement;

import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.springframework.context.support.ClassPathXmlApplicationContext;

import com.taobao.tddl.jdbc.group.SpringTGroupDataSource;
import com.taobao.tddl.jdbc.group.testutil.DBHelper;

/**
 * TAtomDataSourceorg.apache.commons.dbcp.BasicDataSourcecrud
 * 
 * @author yangzhu
 *
 */
public class SpringTest {
    private static ClassPathXmlApplicationContext ctx;
    private static SpringTGroupDataSource ds;

    @BeforeClass
    public static void setUpBeforeClass() throws Exception {
        ctx = new ClassPathXmlApplicationContext(new String[] { "conf/springTGroupDataSource.xml" });
        ds = (SpringTGroupDataSource) ctx.getBean("ds");
    }

    @AfterClass
    public static void tearDownAfterClass() {

    }

    @Before
    public void setUp() throws Exception {
        DBHelper.deleteAll();
    }

    //dbGroup: db1:r10w, db2:r20, db3:r30
    @Test
    public void springTest() throws Exception {

        Connection conn = ds.getConnection();

        //Statementcrud
        Statement stmt = conn.createStatement();
        assertEquals(stmt.executeUpdate("insert into crud(f1,f2) values(10,'str')"), 1);
        assertEquals(stmt.executeUpdate("update crud set f2='str2'"), 1);
        ResultSet rs = stmt.executeQuery("select f1,f2 from crud");
        rs.next();
        assertEquals(rs.getInt(1), 10);
        assertEquals(rs.getString(2), "str2");
        assertEquals(stmt.executeUpdate("delete from crud"), 1);
        rs.close();
        stmt.close();

        //PreparedStatementcrud
        String sql = "insert into crud(f1,f2) values(10,'str')";
        PreparedStatement ps = conn.prepareStatement(sql);
        assertEquals(ps.executeUpdate(), 1);
        ps.close();

        sql = "update crud set f2='str2'";
        ps = conn.prepareStatement(sql);
        assertEquals(ps.executeUpdate(), 1);
        ps.close();

        sql = "select f1,f2 from crud";
        ps = conn.prepareStatement(sql);
        rs = ps.executeQuery();
        rs.next();
        assertEquals(rs.getInt(1), 10);
        assertEquals(rs.getString(2), "str2");
        rs.close();
        ps.close();

        sql = "delete from crud";
        ps = conn.prepareStatement(sql);
        assertEquals(ps.executeUpdate(), 1);
        ps.close();

        conn.close();
    }
}