java单元测试(JUnit)(转)

类名：TestGetOneData.java

TestGetOneData.java代码如下：

package com.whh.test.test;

 

import static org.junit.Assert.assertEquals;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.Statement;

import org.junit.After;

import org.junit.Before;

import org.junit.Test;

 

/**

 * 测试从数据库中的某表中取出一条记录

 *

 * @author wanghonghai

 *

 */

public class TestGetOneData {

 

    private Connection conn = null;

    private String url = "jdbc:oracle:thin:@localhost:1521:whhdb";

    private String driverClassName = "oracle.jdbc.driver.OracleDriver";

    private String user = "whh";

    private String pass = "whh";

 

//setup()方法中放置标准资源初始化

    @Before

    public void setUp() throws Exception {

       Class.forName(driverClassName).newInstance();

       conn = DriverManager.getConnection(url, user, pass);

    }

 

//tearDown()方法中放置标准资源回收

 

    @After

    public void tearDown() throws Exception {

       conn = null;

    }

 

//testGetOneData()方法中放置的是需要测试代码或要测试的另一个类中的方法

 

    @Test

    public void testGetOneData() {

       try {

           Statement st = conn.createStatement();

           String sql = "select u_name from WHH_USER_INFO where U_ID = 1";

           ResultSet rs = st.executeQuery(sql);

           String str = null;

           while (rs.next()) {

              str = rs.getString("u_name");

           }

           assertEquals("管理员", str);

       } catch (Exception e) {

           e.printStackTrace();

       }

    }

}