完善测试。

abel533 · abel533 · commit 7e7474cce84f · 2015-06-04T00:42:31.000+08:00
diff --git a/src/test/java/com/github/abel533/mapper/MybatisJavaHelper.java b/src/test/java/com/github/abel533/mapper/MybatisJavaHelper.java
@@ -52,6 +52,7 @@ public class MybatisJavaHelper {
                 // 注册通用Mapper接口 - 可以自动注册继承的接口
                 mapperHelper.registerMapper(Mapper.class);
                 mapperHelper.registerMapper(HsqldbMapper.class);
+                mapperHelper.registerMapper(MySqlMapper.class);
                 //配置完成后，执行下面的操作
                 mapperHelper.processConfiguration(session.getConfiguration());
                 //OK - mapperHelper的任务已经完成，可以不管了
diff --git a/src/test/java/com/github/abel533/test/jdbc/TestJDBC.java b/src/test/java/com/github/abel533/test/jdbc/TestJDBC.java
@@ -31,8 +31,6 @@
 import org.junit.Assert;
 import org.junit.Test;
 
-import java.util.List;
-
 /**
  * Created by liuzh on 2014/11/21.
  */
@@ -46,19 +44,11 @@ public void testJDBC() {
         try {
             CountryJDBCMapper mapper = sqlSession.getMapper(CountryJDBCMapper.class);
             CountryJDBC country = new CountryJDBC();
-            country.setId(10086);
             country.setCountrycode("CN");
             Assert.assertEquals(1, mapper.insert(country));
-
-            //查询CN结果
-            country = new CountryJDBC();
-            country.setCountrycode("CN");
-            List<CountryJDBC> list = mapper.select(country);
-
-            Assert.assertEquals(1, list.size());
-            //删除插入的数据,以免对其他测试产生影响
-            Assert.assertEquals(1, mapper.deleteByPrimaryKey(10086));
+            Assert.assertNotNull(country.getId());
         } finally {
+            sqlSession.rollback();
             sqlSession.close();
         }
     }
diff --git a/src/test/java/com/github/abel533/test/mysql/TestInsertList.java b/src/test/java/com/github/abel533/test/mysql/TestInsertList.java
@@ -1,11 +1,11 @@
 package com.github.abel533.test.mysql;
 
-import com.github.abel533.entity.Example;
 import com.github.abel533.mapper.CountryMapper;
 import com.github.abel533.mapper.MybatisHelper;
 import com.github.abel533.model.Country;
 import org.apache.ibatis.session.SqlSession;
 import org.junit.Assert;
+import org.junit.Test;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -21,28 +21,23 @@ public class TestInsertList {
      * 插入完整数据
      */
     //该方法测试需要mysql或者h2数据库，所以这里注释掉
-    //@Test
+    @Test
     public void testInsert() {
         SqlSession sqlSession = MybatisHelper.getSqlSession();
         try {
             CountryMapper mapper = sqlSession.getMapper(CountryMapper.class);
             List<Country> countryList = new ArrayList<Country>();
             for (int i = 0; i < 10; i++) {
                 Country country = new Country();
-                country.setId(10086 + i);
                 country.setCountrycode("CN" + i);
                 country.setCountryname("天朝" + i);
                 countryList.add(country);
             }
             int count = mapper.insertList(countryList);
             Assert.assertEquals(10, count);
-
-
-            Example example = new Example(Country.class);
-            example.createCriteria().andGreaterThanOrEqualTo("id",10086);
-            List<Country> list = mapper.selectByExample(example);
-
-            Assert.assertEquals("CN2",list.get(2).getCountrycode());
+            for (Country country : countryList) {
+                Assert.assertNotNull(country.getId());
+            }
         } finally {
             sqlSession.rollback();
             sqlSession.close();
