selenium+python自动化测试--数据驱动

之前selenium+python自动化测试--登录  中，几个账号和密码登录就要写几个用例，感觉很麻烦~，所以下面介绍一下用数据驱动实现不同用例的登录。

话不多说，直接贴代码~~

文件名称：test_ddtlogin.py

import ddt import unittest from selenium import webdriver from common.base import Base testdata = [ {'user':'','psw':'','exp':'请输入手机号码'}, {'user':'admin','psw':'','exp':'请输入密码'}, {'user':'','psw':'666666','exp':'请输入手机号码'}, {'user':'admin','psw':'666666','exp':'账号不存在'} ] @ddt.ddt class Testddt(unittest.TestCase): # 定位手机号
    username = ("name", "loginName") # 定位密码
    psw = ("name", "loginPassWord") # 定位登录按钮
    loginbutton = ("id", "loginBtn") # 定位提示信息
    message = ("className", "toast-message") @classmethod def setUpClass(cls): cls.driver = webdriver.Firefox() cls.baseddt = Base(cls.driver) cls.driver.get("不愿透露姓名的url") def setUp(self): self.baseddt.clear(self.username) self.baseddt.clear(self.psw) @classmethod def tearDownClass(cls): cls.driver.quit() # 加载测试数据
    @ddt.data(*testdata) def test_01(self,data): username = data['user'] password = data['psw'] exp = data['exp'] #输入手机号
 self.baseddt.send(self.username,username) #输入密码
 self.baseddt.send(self.psw,password) #点击登录按钮
 self.baseddt.click(self.loginbutton) #获取结果
        result = self.baseddt.find(self.message).text print("我是结果：", result) self.assertEqual(result, exp) if __name__ == '__main__': unittest.main()

 上面代码运行多次后，发现最后一个用例执行失败了，觉得可能是缓存问题，所以还是改成下面的代码：

 

import ddt import unittest from selenium import webdriver from common.base import Base testdata = [ {'user':'','psw':'','exp':'请输入手机号码'}, {'user':'admin','psw':'','exp':'请输入密码'}, {'user':'','psw':'666666','exp':'请输入手机号码'}, {'user':'admin','psw':'666666','exp':'账号不存在'} ] @ddt.ddt class Testddt(unittest.TestCase): # 定位手机号
    username = ("name", "loginName") # 定位密码
    psw = ("name", "loginPassWord") # 定位登录按钮
    loginbutton = ("id", "loginBtn") # 定位提示信息
    message = ("className", "toast-message") @classmethod def setUpClass(cls): cls.driver = webdriver.Firefox() cls.baseddt = Base(cls.driver) def setUp(self): self.driver.get("不愿透露姓名的url") def tearDown(self): self.driver.delete_all_cookies() @classmethod def tearDownClass(cls): cls.driver.quit() # 加载测试数据
    @ddt.data(*testdata) def test_01(self,data): username = data['user'] password = data['psw'] exp = data['exp'] #输入手机号
 self.baseddt.send(self.username,username) #输入密码
 self.baseddt.send(self.psw,password) #点击登录按钮
 self.baseddt.click(self.loginbutton) #获取结果
        result = self.baseddt.find(self.message).text print("我是结果：", result) self.assertEqual(result, exp) if __name__ == '__main__': unittest.main()

 

运行结果：

 

纯属个人思路，如有错误请在评论区指出，一定立即纠正。非常感谢~ 

读取文件操作请看下回分解😜