Java接口自动化测试之集成MyBatis和MySQL (五)

pom.xml新增dependency

 <dependency>
             <groupId>org.mybatis</groupId>
             <artifactId>mybatis</artifactId>
             <version>3.4.4</version>
         </dependency>
         <dependency>
             <groupId>mysql</groupId>
             <artifactId>mysql-connector-java</artifactId>
             <version>5.1.6</version>
 </dependency>

mybatis.xml

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <!-- 注册对象的空间命名 -->
    <environments default="development">
        <environment id="development">
            <transactionManager type="JDBC"/>
            <dataSource type="POOLED">
                <property name="driver" value="com.mysql.jdbc.Driver"/>
                <property name="url" value="jdbc:mysql://localhost:3306/test"/>
                <property name="username" value="root"/>
                <property name="password" value="root"/>
            </dataSource>
        </environment>
    </environments>
    <!-- 注册映射文件：java对象与数据库之间的xml文件路径！ -->
    <mappers>
        <mapper resource="mapper/SQLMapper.xml"/>
    </mappers>
</configuration>

mapper文件SQLMapper.xml

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<!-- 命名空间mapper,如果有多个mapper文件，这个必须唯一 -->
<mapper namespace="com.testng.model">

    <select id="doGet" parameterType="Integer" resultType="com.testng.model.DoGetCase">
        select * from dogetcase where id = #{id};
    </select>

    <select id="doPost" parameterType="Integer" resultType="com.testng.model.DoPostCase">
        select * from dopostcase where id = #{id};
    </select>

</mapper>

DoGetCase.java / DoPostCase.java 等JavaBean, 根据被测接口参数设计字段, 在MySQL中创建同样的表, 表字段也是根据被测接口的参数设计

 package com.testng.model;

 import lombok.Data;

 @Data
 public class DoGetCase {
     private int id;
     private String name;
     private String password;
     private String expected;
 }

 package com.testng.model;

 import lombok.Data;

 @Data
 public class DoPostCase {
     private int id;
     private String name;
     private String password;
     private String expected;
 }

公共类,连接数据用DatabaseUtil.java

 package com.testng.utils;

 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.IOException;
 import java.io.Reader;

 public class DatabaseUtil {
     public static SqlSession getSqlSession() throws IOException {
         // 获取配置的资源文件
         Reader reader = Resources.getResourceAsReader("mybatis.xml");
         SqlSessionFactory factory = new SqlSessionFactoryBuilder().build(reader);
         // sqlSession能够执行配置文件中的SQL语句
         SqlSession sqlSession = factory.openSession();
         return sqlSession;
     }
 }

修改优化TestNG测试用例, 对于每个被测接口可以单独提取出来一个方法, 唯一的变量就是数据库用例的id编号, 这里就不演示了

 package com.testng.cases;

 import com.testng.model.DoGetCase;
 import com.testng.utils.DatabaseUtil;
 import com.testng.utils.HttpUtils;
 import com.testng.utils.ReadConfig;
 import org.apache.ibatis.session.SqlSession;
 import org.testng.Assert;
 import org.testng.Reporter;
 import org.testng.annotations.Test;

 import java.io.IOException;

 public class DoGetTest {

     @Test(description = "成功的案例")
     public void getTest() throws IOException {
         SqlSession session = DatabaseUtil.getSqlSession();
         DoGetCase doGetCase = session.selectOne("doGet", 1);
         String uri = ReadConfig.URI + ReadConfig.GET_PATH1;
         String param = "name=" + doGetCase.getName() + "&password=" + doGetCase.getPassword();
         String url = uri + param;
         Reporter.log("请求地址" + url);
         String result = HttpUtils.doGet(url);
         Reporter.log("请求结果" + result);
         Assert.assertEquals(result, doGetCase.getExpected());
     }

     @Test(description = "失败的案例")
     public void getTest2() throws IOException {
         SqlSession session = DatabaseUtil.getSqlSession();
         DoGetCase doGetCase = session.selectOne("doGet", 2);
         String uri = ReadConfig.URI + ReadConfig.GET_PATH2;
         String param = "name=" + doGetCase.getName() + "&password=" + doGetCase.getPassword();
         String url = uri + param;
         Reporter.log("请求地址" + url);
         String result = HttpUtils.doGet(url);
         Reporter.log("请求结果" + result);
         Assert.assertEquals(result, doGetCase.getExpected());
     }
 }

 package com.testng.cases;

 import com.alibaba.fastjson.JSONObject;
 import com.testng.model.DoPostCase;
 import com.testng.utils.DatabaseUtil;
 import com.testng.utils.HttpUtils;
 import com.testng.utils.ReadConfig;

 import org.apache.ibatis.session.SqlSession;
 import org.testng.Assert;
 import org.testng.Reporter;
 import org.testng.annotations.Test;

 import java.io.IOException;

 public class DoPostTest {

     @Test
     public void postTest() throws IOException {
         SqlSession session = DatabaseUtil.getSqlSession();
         DoPostCase doPostCase = session.selectOne("doPost", 1);
         String url = ReadConfig.URI + ReadConfig.POST_PATH1;
         Reporter.log("请求地址" + url);
         JSONObject param = new JSONObject();
         param.put("name", doPostCase.getName());
         param.put("password", doPostCase.getPassword());
         Reporter.log("请求体" + param.toString());
         String result = HttpUtils.doPost(url, param.toString());
         Reporter.log("请求结果" + result);
         Assert.assertEquals(result, doPostCase.getExpected());
     }

     @Test
     public void postTest2() throws IOException {
         SqlSession session = DatabaseUtil.getSqlSession();
         DoPostCase doPostCase = session.selectOne("doPost", 2);
         String url = ReadConfig.URI + ReadConfig.POST_PATH2;
         Reporter.log("请求地址" + url);
         JSONObject param = new JSONObject();
         param.put("name", doPostCase.getName());
         param.put("password", doPostCase.getPassword());
         Reporter.log("请求体" + param.toString());
         String result = HttpUtils.doPost(url, param.toString());
         Reporter.log("请求结果" + result);
         Assert.assertEquals(result, doPostCase.getExpected());
     }

 }

再次执行testng.xml, 得到的结果跟之前的效果一样. 这样设计的好处是测试用例可以用MySQL数据库管理起来