完善测试。

abel533 · abel533 · commit 4c939d13816b · 2015-06-04T00:56:47.000+08:00
diff --git a/src/test/java/com/github/abel533/test/mysql/TestMysql.java b/src/test/java/com/github/abel533/test/mysql/TestMysql.java
@@ -5,7 +5,6 @@
 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;
@@ -15,14 +14,14 @@
  *
  * @author liuzh
  */
-public class TestInsertList {
+public class TestMysql {
 
     /**
      * 插入完整数据
      */
     //该方法测试需要mysql或者h2数据库，所以这里注释掉
-    @Test
-    public void testInsert() {
+    //@Test
+    public void testInsertList() {
         SqlSession sqlSession = MybatisHelper.getSqlSession();
         try {
             CountryMapper mapper = sqlSession.getMapper(CountryMapper.class);
@@ -43,4 +42,25 @@ public void testInsert() {
             sqlSession.close();
         }
     }
+
+    /**
+     * 插入完整数据
+     */
+    //该方法测试需要mysql或者h2数据库，所以这里注释掉
+    //@Test
+    public void testInsert() {
+        SqlSession sqlSession = MybatisHelper.getSqlSession();
+        try {
+            CountryMapper mapper = sqlSession.getMapper(CountryMapper.class);
+            Country country = new Country();
+            country.setCountrycode("CN");
+            country.setCountryname("天朝");
+            int count = mapper.InsertUseGeneratedKeysMapper(country);
+            Assert.assertEquals(1, count);
+            Assert.assertNotNull(country.getId());
+        } finally {
+            sqlSession.rollback();
+            sqlSession.close();
+        }
+    }
 }
