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

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):
        # 清空cookies
        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()