构建测试体系

pom.xml

@@ -57,7 +57,25 @@
         <dependency>
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>
-            <version>3.8.1</version>
+            <version>4.12</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.mybatis</groupId>
+            <artifactId>mybatis</artifactId>
+            <version>3.4.0</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+            <version>1.1.7</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>mysql</groupId>
+            <artifactId>mysql-connector-java</artifactId>
+            <version>5.1.40</version>
             <scope>test</scope>
         </dependency>
     </dependencies>

src/test/java/com/itfsw/mybatis/generator/plugins/DBHelperTest.java

+/*
+ * Copyright (c) 2017.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.itfsw.mybatis.generator.plugins;
+
+import com.itfsw.mybatis.generator.plugins.tools.DBHelper;
+import org.apache.ibatis.session.SqlSession;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+
+/**
+ * ---------------------------------------------------------------------------
+ *
+ * ---------------------------------------------------------------------------
+ * @author: hewei
+ * @time:2017/6/26 17:19
+ * ---------------------------------------------------------------------------
+ */
+public class DBHelperTest {
+    /**
+     * 测试 getSqlSession
+     *
+     * @throws IOException
+     * @throws SQLException
+     */
+    @Test
+    public void testGetSqlSession() throws IOException, SQLException {
+        DBHelper helper = DBHelper.getHelper("scripts/test_init.sql");
+        SqlSession sqlSession = helper.getSqlSession();
+        Connection connection = sqlSession.getConnection();
+        Statement statement = connection.createStatement();
+
+        // 执行查询
+        statement.execute("SELECT COUNT(*) as total FROM tb");
+        ResultSet resultSet = statement.getResultSet();
+
+        resultSet.first();
+        Assert.assertEquals(resultSet.getInt("total"), 4);
+
+        statement.close();
+        connection.close();
+        sqlSession.close();
+    }
+}

src/test/java/com/itfsw/mybatis/generator/plugins/tools/DBHelper.java

+/*
+ * Copyright (c) 2017.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.itfsw.mybatis.generator.plugins.tools;
+
+import org.apache.ibatis.io.Resources;
+import org.apache.ibatis.session.SqlSession;
+import org.apache.ibatis.session.SqlSessionFactory;
+import org.apache.ibatis.session.SqlSessionFactoryBuilder;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.sql.Statement;
+
+/**
+ * ---------------------------------------------------------------------------
+ *
+ * ---------------------------------------------------------------------------
+ * @author: hewei
+ * @time:2017/6/26 16:52
+ * ---------------------------------------------------------------------------
+ */
+public class DBHelper {
+    private static final String MYBATIS_CONFIG = "mybatis-config.xml";    // 配置文件
+    private static DBHelper helper; // helper
+
+    /**
+     * 构造函数
+     */
+    private DBHelper() {
+    }
+
+    /**
+     * 获取数据库操作工具
+     *
+     * @param initSql
+     * @return
+     */
+    public static DBHelper getHelper(String initSql) throws IOException, SQLException {
+        if (helper == null){
+            helper = new DBHelper();
+            helper.initDB(initSql);
+        }
+        return helper;
+    }
+
+    /**
+     * 获取SqlSession
+     *
+     * @return
+     * @throws IOException
+     */
+    public SqlSession getSqlSession() throws IOException {
+        InputStream inputStream = Resources.getResourceAsStream(MYBATIS_CONFIG);
+        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
+        inputStream.close();
+        return sqlSessionFactory.openSession(true);
+    }
+
+    /**
+     * 初始化数据库
+     *
+     * @param initSql
+     * @throws IOException
+     * @throws SQLException
+     */
+    private void initDB(String initSql) throws IOException, SQLException {
+        SqlSession sqlSession = this.getSqlSession();
+        Connection connection = sqlSession.getConnection();
+        Statement statement = connection.createStatement();
+        // 获取建表和初始化sql
+        InputStream inputStream = Resources.getResourceAsStream(initSql);
+        InputStreamReader inputStreamReader = new InputStreamReader(inputStream, "UTF-8");
+        BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
+        // 读取sql语句执行
+        StringBuffer sb = new StringBuffer();
+        String line;
+        while ((line = bufferedReader.readLine()) != null){
+            sb.append(line).append("\n");
+            if (line.matches(".*;$")){
+                System.out.println(sb.toString());
+                statement.execute(sb.toString());
+                sb.setLength(0);
+            }
+        }
+        bufferedReader.close();
+        inputStreamReader.close();
+        inputStream.close();
+        statement.close();
+        connection.close();
+        sqlSession.close();
+    }
+}

src/test/resources/mybatis-config.xml

+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+  ~ Copyright (c) 2017.
+  ~
+  ~ Licensed under the Apache License, Version 2.0 (the "License");
+  ~ you may not use this file except in compliance with the License.
+  ~ You may obtain a copy of the License at
+  ~
+  ~     http://www.apache.org/licenses/LICENSE-2.0
+  ~
+  ~ Unless required by applicable law or agreed to in writing, software
+  ~ distributed under the License is distributed on an "AS IS" BASIS,
+  ~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  ~ See the License for the specific language governing permissions and
+  ~ limitations under the License.
+  -->
+
+<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTDSQL Map Config 3.0//EN"  "http://mybatis.org/dtd/mybatis-3-config.dtd">
+<configuration>
+    <!-- 配置环境，制定数据库连接信息 -->
+    <environments default="test">
+        <environment id="test">
+            <transactionManager type="JDBC" />
+            <dataSource type="POOLED">
+                <property name="driver" value="com.mysql.jdbc.Driver"/>
+                <property name="url" value="jdbc:mysql://localhost:3306/mybatis-generator-plugin?characterEncoding=UTF-8&amp;allowMultiQueries=true"/>
+                <property name="username" value="root"/>
+                <property name="password" value="root"/>
+            </dataSource>
+        </environment>
+    </environments>
+
+    <mappers>
+        <!-- Mapper扫描包，必须同目录同名称下-->
+        <package name="com.itfsw.mybatis.generator.plugins.dao"/>
+    </mappers>
+</configuration>
\ No newline at end of file

src/test/resources/scripts/test_init.sql

+/*
+Navicat MySQL Data Transfer
+
+Source Server         : localhost
+Source Server Version : 50617
+Source Host           : localhost:3306
+Source Database       : mybatis-generator-plugin
+
+Target Server Type    : MYSQL
+Target Server Version : 50617
+File Encoding         : 65001
+
+Date: 2017-06-26 17:30:13
+*/
+
+SET FOREIGN_KEY_CHECKS=0;
+
+-- ----------------------------
+-- Table structure for tb
+-- ----------------------------
+DROP TABLE IF EXISTS `tb`;
+CREATE TABLE `tb` (
+  `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '注释1',
+  `field1` varchar(255) DEFAULT NULL COMMENT '注释2',
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=latin1;
+
+-- ----------------------------
+-- Records of tb
+-- ----------------------------
+INSERT INTO `tb` VALUES ('1', 'f1');
+INSERT INTO `tb` VALUES ('2', 'f2');
+INSERT INTO `tb` VALUES ('3', 'f3');
+INSERT INTO `tb` VALUES ('4', 'f4');