自动化测试数据驱动之csv文件读取

--------------------------------------------------------------------

用例1，用户名密码为空。

用例2，用户名为空。

用例3，密码为空。

用例4，用户名密码正确。

----------------------------------------------------------

 

这里以你们喜闻乐见的读取csv文件为例。

读取数据文件，并得到相应的数据，把这些数据用到具体的某个用例当中。

from selenium import webdriver
import unittest
import csv


# 读取本地 CSV 文件
data = csv.reader(open('login_test.csv', 'r'))

# 读取整个文件的数据放到users数组
users = []
for i in data:
    user = []
    for j in i:
        user.append(j)
    users.append(user)


class loginTest(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.get("http://xxx.login.page")

    # 封装用户登录
    def user_login(self, username, password):
        self.driver.find_element_by_id("idInput").send_keys(username)
        self.driver.find_element_by_id("pwdInput").send_keys(password)
        self.driver.find_element_by_id("loginBtn").click()

    def test_login1(self):
        '''用户名、密码为空登录'''
        i = 0
        for user in users:
            print(user[0])
            if user[0] == 'user_pawd_null':
                print(i)
                username = users[i][1]
                password = users[i][2]
                break;
            else:
                i +=1
        self.user_login(username, password)
    
    def test_login2(self):
        '''用户名正确,密码为空'''
        username = users[1][1]
        password = users[1][2]
        self.user_login(username, password)

    def test_login3(self):
        '''用户名为空,密码正确'''
        username = users[2][1]
        password = users[2][2]
        self.user_login(username, password)

    def test_login4(self):
        '''用户名密码正确 '''
        username = users[3][1]
        password = users[3][2]
        self.user_login(username, password)


if __name__ == '__main__':
    unittest.main()

1、创建了一CSV文件，然后把登录用的测试数据写到了文件了。--->创建了一个专门存放数据的文件，这多有逼格，自我感觉良好。

2、读取CSV文件，并且通过for循环，把所有数据组装成一个二维数组，并放users数组中。--->这没什么呀，只是多写了个for循环而已，继续自我感觉良好。

3、test_login1用例，为用户名密码都为空的用例，判断users数组中某一行的第一列是否为“user_pawd_null”,是的话，说明这一行就是我想要的。取这一行的第二、第三列的测试数据，进行登录测试。--->这个取数据的方式有点。。。有点麻烦！

4、没关系！没关系！麻烦的话，我们看test_login2 ，用users[1][1]和users[1][2]也能取到CSV表第二行的数据。  ---->这不就不那么麻烦了！嗯，是不那么麻烦了，不过，有点傻逼。你确定你清楚的知道users[1][1]和users[1][2] 代表的啥？别急！别急！我打开CSV文件看看第2行对应是什么数据。