使用Parameters+DataProvider完成数据驱动接口测试

向右看齐 2022-05-18 08:10 107阅读 0赞

之前思考过几种不同的接口自动化框架的搭建结构，这是其中一种，实现的效果是：通过xml里存储的sql语句，去遍历mysql获得相应的测试数据，然后完成测试。  
好处是这样做，完全是靠数据驱动测试，在有新增接口时，无须改动代码，只需要在mysql内添加测试数据，然后新的测试场景只需要修改xml就可以了。  
下面讲一下简单的实现思路：

1.配置DataProvider\_ForMysql类  
用于链接数据库获取测试数据

import java.sql.*;
    import java.util.*;
    
    /**
     *数据库链接配置
     */
    public class DataProvider_ForMysql implements Iterator<Object[]> {
    
        ResultSet result;  //结果集
        List<Map<String, String>> dataList = new ArrayList<Map<String, String>>();  //生成存放结果集的list
        int rowNum=0;     //总行数
        int curRowNo=0;   //当前行数
    
        public DataProvider_ForMysql(String ip, String port, String baseName,
                                     String userName, String password, String sql) throws ClassNotFoundException, SQLException {
    
            Class.forName("com.mysql.cj.jdbc.Driver");
            String url = String.format("jdbc:mysql://%s:%s/%s?serverTimezone=UTC", ip, port, baseName);
            //获取连接
            Connection conn = DriverManager.getConnection(url, userName, password);
            //获取创建语句对象
            Statement stmt = conn.createStatement();
            //执行sql语句，获取查询结果集
            result = stmt.executeQuery(sql);
            //获取当前行数据
            ResultSetMetaData rd = result.getMetaData();
    
            //循环每行
            while (result.next()) {
                Map<String, String> map = new HashMap<String, String>();
                //循环每列，如果不要id，则讲i设为2
                for (int i = 1; i <= rd.getColumnCount(); i++) {
                    String key = result.getMetaData().getColumnName(i);
                    String value = result.getString(i);
                    map.put(key,value);
                }
                dataList.add(map);
            }
    
            this.rowNum = dataList.size();
    
            conn.close();
            stmt.close();
        }
    
    
        @Override
        public boolean hasNext() {
            if(rowNum==0||curRowNo>=rowNum){
                return false;
            }else{
                return true;
            }
        }
    
        @Override
        public Object[] next() {
            Map<String,String> s=dataList.get(curRowNo);
            Object[] d=new Object[1];
            d[0]=s;
            this.curRowNo++;
            return d;
        }
    
        @Override
        public void remove() {
            throw new UnsupportedOperationException("remove unsupported");
        }
    }

2.配置xml文件

<?xml version="1.0" encoding="UTF-8" ?>
    <suite name="suite">
        <test name="test">
            <parameter  name="valueName" value="select * from new_model"/>
            <classes>
                <class name="com.course.TestCase"/>
            </classes>
        </test>
        <listeners>
            <listener class-name="com.course.config.ExtentTestNGIReporterListener"></listener>
        </listeners>
    </suite>

3.测试执行

public class TestCase {
    
        String sql;   //xml中读出的sql语句
    
    
    	/**
    	* 通过name来取xml文件中对应的sql语句
    	*/
        @Parameters({"valueName"})
        @BeforeClass()
        public void beforeClass(String sql) {
            this.sql = sql;
        }
    
       
        /**
         * XML中的SQL决定了执行什么用例, 执行多少条用例, SQL的搜索结果为需要测试的测试用例
         */
        @DataProvider(name = "testData")
        private Iterator<Object[]> getData(){
            try {
                return new DataProvider_ForMysql(TestConfig.DB_IP, TestConfig.DB_PORT,
                    TestConfig.DB_BASE_NAME,TestConfig.DB_USERNAME, TestConfig.DB_PASSWORD, sql);
            } catch (ClassNotFoundException e || SQLException e) {
                e.printStackTrace();
            } 
            return null;
        }
    
    
    	/**
    	*完整的测试数据，会以map的形式传递到测试方法里
    	*/
        @Test(dataProvider = "testData")
        public void test(Map<String, String> dataMap) {
            System.out.println(dataMap.toString());
        }
    }

无论mysql中读出多少数据，都会一条条的去执行，所以在测试场景中，需要考虑的是如何通过sql语句筛选测试数据，以及用例与用例间的数据依赖。在之后的文章中，也会有这方面内容的分享。  
有疑问的小伙伴可以在下方留言，我会针对问题不断优化文章内容，希望能有效的帮到更多人，谢谢~