Python + Selenium + Unittest自动化测试初学

一、封装公共代码：

1、封装浏览器路径：

from selenium import webdriver
import time
class  Common(object):
    #初始化
    def __init__(self):
        #获取浏览器路径
        executable_path = r"G:\BaiduNetdiskDownload\chromedriver.exe"
        #创建浏览器
        self.driver = webdriver.Chrome(executable_path)
        #浏览器最大化
        self.driver.maximize_window()

2、访问URL、关闭浏览器：

    #访问指定URL
    def open_url(self,url):
        self.driver.get(url)
        self.driver.implicitly_wait(2)
    #关闭浏览器
    def close_driver(self):
        self.driver.quit()

3、封装元素定位：

    #定位元素方式
    def LocateElement(self,locate_type,value):
        El = None
        if locate_type == "id":
            El = self.driver.find_element_by_id(value)
        elif locate_type == "name":
            El = self.driver.find_element_by_name(value)
        elif locate_type == "class":
            El = self.driver.find_element_by_class_name(value)
        elif locate_type == "tag":
            El = self.driver.find_element_by_tag_name(value)
        elif locate_type == "text":
            El = self.driver.find_element_by_link_text(value)
        elif locate_type == "partial":
            El = self.driver.find_element_by_partial_link_text(value)
        elif locate_type == "xpath":
            El = self.driver.find_element_by_xpath(value)
        elif locate_type == "css":
            El = self.driver.find_element_by_css_selector(value)
        # 返回定位元素
        if El is not None:
            return El

4、封装元素的基本操作：

#对点击元素的封装
    def click(self,locate_type,value):
        #调用LocateElement()
        el = self.LocateElement(locate_type,value)
        #执行点击操作
        el.click()

    #对定位到的元素进行文本输入
    def input_data(self,locate_type,value,data):
        el = self.LocateElement(locate_type, value)
        # 执行输入操作
        el.send_keys(data)

    #获取定位到的元素中的文本内容<a>xxx</a>
    def get_text(self,locate_type, value):
        el = self.LocateElement(locate_type, value)
        #返回值
        return el.text

    #获取定位到的元素中的标签属性
    def get_attr(self,locate_type, value,data):
        el = self.LocateElement(locate_type, value)
        #返回值
        return el.get_attribute(data)

5、结束清理：

    #结束的时候清理
    def __del__(self):
        time.sleep(3)
        self.driver.close()

6、举例百度：

if __name__ == '__main__':
    com = Common()
    com.open_url('https://www.baidu.com/')
    com.input_data('id','kw','Selenium')
    com.click('id','su')

二、引用公共代码

1、创建登录文件：

from Common.Common import Common
class Login(Common):
    def login(self,user_name,pwd):
        #以CSDN网址为例
        self.open_url("https://www.csdn.net/")
        #定位登录元素
        self.click('xpath', "//*[@id='csdn-toolbar']/div/div/div[3]/div/div[1]/a")
        #定位账号密码登录元素
        self.click('xpath', "//*[@id='app']/div/div/div[1]/div[2]/div[5]/ul/li[2]/a")
        #定位用户名
        self.input_data('id', "all", user_name)
        #定位密码
        self.input_data('name', "pwd", pwd)
        #定位登录按钮
        self.click('xpath', "//*[@id='app']/div/div/div[1]/div[2]/div[5]/div/div[6]/div/button")

if __name__ == '__main__':
    log = Login()
    #输入有效的账号名和密码实现登录
    log.login('18066xxxxxx','xxxxxxxxxxx')  

2、创建测试用例

#导入单元测试框架
import unittest
#从Business包中的Login文件导入Login类
from Business.Login import Login
import time

测试用例代码示例：（CSDN网址）

#继承TestCase类
class TestCases(unittest.TestCase):
	#setUp()用于测试用例执行之前
    def setUp(self):
        print("Test start")
	#tearDown()用于测试用例执行之后
    def tearDown(self):
        print("Test end")
        
    #登录成功的情况：
    def test_001(self):
        log = Login()
        #正确账号用户密码登录
        log.login('18066xxxxxx','xxxxxxxx')
        #点击登录成功后的头像进入新的界面
        log.click('xpath', "//*[@id='csdn-toolbar']/div/div/div[3]/div/div[1]/a/img")
        time.sleep(1)
        #获取用于断言判断登录后的情况
        data = log.get_text('xpath', "/html/body/header/div/div/div[1]/h1/a")
        #进行断言判断
        self.assertEqual("xxxxxx的博客",data)
        
    #只输入账号不输入密码时，登录按钮为灰色显示；
    def test_002(self):
        log = Login()
        #仅输入账号不输入密码
        log.login('18066xxxxxx','')
        data = log.get_attr('xpath', "//*[@id='app']/div/div/div[1]/div[2]/div[5]/div/div[6]/div/button",'disabled')
        #进行断言判断
        self.assertTrue(data,msg = "登录按钮灰色显示")
        
    #账号正确、密码不正确时登录情况；
    def test_003(self):
        log = Login()
        #仅输入账号不输入密码
        log.login('18066xxxxxx','AAAAAAA')
        data = log.get_text('xpath', "//*[@id='js_err_dom']")
        #进行断言判断
        self.assertIsNotNone(data,msg="提示用户名或密码错误/输入次数太多，请稍后再试")
        
if __name__ == '__main__':
    unittest.main()