junit4测试数据库源码

package mysqldb; import static org.junit.Assert.*; import java.sql.Connection; import java.sql.PreparedStatement; import java.sql.ResultSet; import org.junit.AfterClass; import org.junit.Before; import org.junit.BeforeClass; import org.junit.Test; public class PersonDBTest { private static Connection con = null; private PersonDB personDB = null; @BeforeClass public static void setUpBeforeClass() throws Exception { con = ConDB.getConnection();//创建连接 } @AfterClass public static void tearDownAfterClass() throws Exception { con.close(); } @Before public void setUp() throws Exception { personDB = new PersonDB(); } @Test public void testInsert() { Person person = new Person();//创建Person对象,插入.. person.setId(2); person.setUsername("chensi"); person.setPassword("123456"); person.setAge(20); personDB.insert(person); Person person2 = new Person(); person2 = personDB.getById(2);//这里用到了获取当前最大ID的方法.这个方法我设定为private类型.是帮助方法..数据库的ID我设定为自增..每次加1.. Assert(person, person2); personDB.delete(getMaxId());//最后删除这条数据...测试不要给数据库增加垃圾数据. } @Test public void testGetById() { Person person = new Person(); person.setId(2); person.setUsername("chensi"); person.setPassword("123456"); person.setAge(20); personDB.insert(person); Person person2 = new Person(); person2 = personDB.getById(2); Assert(person, person2); personDB.delete(2); } @Test public void testDelete() { Person person = new Person(); person.setId(2); person.setUsername("han"); person.setPassword("111"); person.setAge(11); personDB.insert(person); Person person2 = new Person(); person2 = personDB.getById(2); Assert(person, person2); personDB.delete(2); Person person3 = new Person(); person3 = personDB.getById(person2.getId()); assertNull(person3.getUsername()); } @Test public void testUpdate() { Person person = new Person(); person.setId(2); person.setUsername("1111"); person.setPassword("pass"); person.setAge(111); personDB.insert(person); Person person2 = new Person(); person2 = personDB.getById(2); this.Assert(person, person2); Person person3 = new Person(); person3.setId(person2.getId()); person3.setUsername("2222"); person3.setPassword("word"); person3.setAge(0); personDB.update(person3); Person person4 = new Person(); person4 = personDB.getById(2); Assert(person3, person4); personDB.delete(person2.getId()); } private void Assert(Person person1, Person person2)//帮助代码.起到提取代码的作用. { assertEquals(person1.getUsername(), person2.getUsername()); assertEquals(person1.getPassword(), person2.getPassword()); assertEquals(person1.getAge(), person2.getAge()); } private int getMaxId()//帮助方法.所以设置为private { int id = 0; try { String sql = "select max(id) from person "; PreparedStatement st = con.prepareStatement(sql); ResultSet rs = st.executeQuery(); if (rs.next()) { id = rs.getInt(1); } } catch (Exception ex) { ex.printStackTrace(); } return id; } }