上一章我们讲了一些比较基础的应用,如何进行元素定位,如何进行时间等待,还有鼠标键盘操作等等,那么本章我们就将上一节所讲的全部用函数封装起来使用。
所谓的封装大体上来说就是将你想要的操作,写入到一个函数方法里面,或者类里面,想要使用的时候直接调用即可,如果想要更改参数也是比较方便的,举个例子:登录注册功能,涉及到了输入错误的账号密码,正确的账号密码以及非法字符,超出要求的长度或小于要求的长度等一系列的参数,写入到函数体里面,就可以很便捷的通过调用,将这些写在一块,也便于修改查询等。
一句话:简洁明了且实用
简单的封装
拿百度搜索的例子我们先来封装看看效果
from selenium import webdriver
class Baidu():
# 初始化工作
def __init__(self,url):
self.fox = webdriver.Firefox()
self.url= url
self.fox.get(url)
self.fox.implicitly_wait(5)
# 退出网址
def quit_url(self):
self.fox.quit()
# 定位搜索操作
def search(self,element_value,value):
self.fox.find_element_by_id(element_value).send_keys(value)
# 定位点击操作
def click_search(self,element_value):
self.fox.find_element_by_id(element_value).click()
# 传值
write = Baidu('https://baidu.com')
write.search('kw','python')
write.click_search('su')
write.quit_url()
看到这里是不是想说这不比昨天的多了那么多代码,好鸡肋啊。这里告诉你,这只是两个操作看起来的感觉,等写的代码多了,不封装写就会杂乱不堪,甚至难以维护。
写到这其实这算很基础的封装了,因为里面固定的方法太多了,随着代码的越来越多,后面的维护也是一种问题。如果我想通过输入元素让它自己去定位,又该怎么写,那我们就加上if--elif--else判断:
进阶你的封装
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
class Sele():
def __init__(self): # 初始化浏览器,并访问url
self.fox = webdriver.Firefox()
self.fox.implicitly_wait(10)
def open_url(self, url):
self.url = url
try:
self.fox.get(url)
except:
print('浏览器地址打开失败')
def quit_S(self): # 关闭浏览器
self.fox.quit()
def text_name(self, type, value):
if type == 'id':
loc = self.fox.find_element_by_id(value)
elif type == 'name':
loc = self.fox.find_element_by_name(value)
elif type == 'class_name':
loc = self.fox.find_element_by_class_name(value)
elif type == 'tag_name':
loc = self.fox.find_elements_by_tag_name(value)
elif type == 'xpath':
loc = self.fox.find_element_by_xpath(value)
elif type == 'css':
loc = self.fox.find_element_by_partial_link_text(value)
elif type == 'link_text':
loc = self.fox.find_element_by_link_text(value)
elif type == 'css_selector':
loc = self.fox.find_elements_by_css_selector(value)
else:
print('输入元素无效')
return loc
def txt(self, type, value, text): # 查找元素输入任意文本值
loc = self.text_name(type, value)
loc.send_keys(text)
def click(self, type, value): # 点击操作
loc = self.text_name(type, value)
loc.click()
def _title(self, title_text): # 判断标题是否正确
if self.fox.title == title_text:
print('标题正确')
else:
print('打开失败')
def url_(self, url): # 判断地址是否正确
if self.fox.current_url == url:
print('地址正确')
else:
print('地址错误')
def wait_time(self, type, value): # 显示等待
loc = WebDriverWait(self.fox, 10).until(EC.presence_of_element_located(type, value),message='元素未找到')
return loc
open = Sele()
open.open_url('https://www.baidu.com')
open.text_name('id','kw')
open.txt('id','kw','python')
open.click('id','su')
open.quit_S()
这样你是不是会觉得这样好麻烦啊,代码量好多。这样写还不如我自己之前的写法,不用类也不用函数方法。这样写之后你不用每次都把定位步骤来过一遍,并且这样的代码量还不算多。这只能说是入门阶段的一种阶段方法。
这里只试验了输入搜索功能,后面还有很多的模块,很多的功能,这样写一、是为了便于出错后查找出错的地方。二、便于自己浏览代码,后续有其他地方或项目需要用到了,可以很快的找到所需要修改的地方。三、重复利用性强。
unittest
unittest是Python单元测试框架,类似于JUnit框架
什么是单元测试?
单元测试是指,对软件中的最小可测试单元在与程序其他部分相隔离的情况下进行检查和验证 的工作,这里的最小可测试单元通常是指函数或者类,一般是开发来做的,按照测试阶段来分, 就是单元测试、集成测试、系统测试以及验收测试。
为什么要做单元测试?
1. 灵活的组织ui自动化/接口测试用例
2. 让用例高效的执行
3. 断言:方便验证测试用例的结果
4. 集成html形式测试报告
核心概念
最核心的四个概念:
1. test case:测试用例
2. test fixure:测试夹具,用于测试用例环境的搭建和销毁
3. test suite:测试套件,用来把需要一起执行的测试用例集中放到一块执行,相当于一个篮子。 我们可以使用TestLoader来加载测试用例到测试套件中。
4. test runner:就是用来执行测试用例的,并且可以生成相应的测试报告
概念只做了解,实践才是道理:
先创建一个py文件,将所需要封装的定位方法写在一起,测试部分写在一起,如果你还需要加日志等全部都分开来写,思路理顺,写起来也会相对而言顺畅一些。
class T_Test():
def __init__(self, fox):
self.fox = fox
def get_url(self, url):
self.url = url
self.fox.get(url)
return url
def click_my(self, name, value): # 元素点击
self.fox.find_element(name, value).click()
def send_key(self, name, value, txt): # 元素输入值
self.fox.find_element(name, value).send_keys(txt)
def get_iframe(self): # 切换窗口,定位框架
frame_0 = self.fox.find_elements_by_id('login_frame')[0]
self.fox.switch_to.frame(frame_0)
"""显示等待------ele为定位的元素"""
def wait_time(self, name, value):
WebDriverWait(self.fox, 10, 0.5).until(lambda x: x.find_element(name, value), message='元素为找到')
"""EC模块搭配显示等待"""
def EC_wait_time(self, name, value):
res = (name, value)
WebDriverWait(self.fox, 0.5).until(EC.presence_of_element_located(res), message='元素未找到')
"""判断标题是否与预期相等"""
def title_judge(self, title):
ec = EC.title_is(title)(self.fox)
print(ec)
"""eles定位"""
def clicks_my(self, name, value, num):
self.fox.find_elements(name, value)[num].click()
def sends_key(self, name, value, num, txt):
self.fox.find_elements(name, value)[num].send_keys(txt)
def name_1(self):
lg = Loggin()
self.res = lg.log_name('Log_file.log')
def log_url(self, url):
self.res.warning(f"打开网址:{self.get_url(url)}")
这里我把定位,切换窗口等一系列操作封装在了一起,写好注释也便于自己看,说不定后面自己写着写着就看迷糊了,有注释不迷糊。
我们来写测试部分,测试部分有一定的讲究。也就是你所要测试的部分你得test开头,类中也必须带有unittest。
from selenium import webdriver
import unittest
from selenium.webdriver.common.by import By
from Unittest_frame.QQ_home import T_Test
from ddt import ddt, data, unpack, file_data
# 数据驱动
@ddt
class LoginTest(unittest.TestCase, T_Test):
@classmethod
def setUpClass(cls) -> None: # 只打开一次浏览器
global fox
fox = webdriver.Firefox()
fox.implicitly_wait(5)
@classmethod
def tearDownClass(cls) -> None: # 只关闭一次浏览器
fox.quit()
# 读取文件
@file_data('./test.yaml')
def test01(self):
self.fox = fox
self.get_url('https://qzone.qq.com/')
self.get_iframe()
self.wait_time(By.ID, 'switcher_plogin')
self.EC_wait_time(By.ID, 'switcher_plogin')
self.click_my(By.ID, 'switcher_plogin')
self.send_key(By.ID, 'u', txt='1234568')
self.send_key(By.ID, 'p', txt='12345679')
self.click_my(By.ID, 'login_button')
self.title_judge('QQ空间-分享生活,留住感动')
self.name_1()
self.log_url(self.url)
# def test02(self):
# #测试eles定位
# self.fox.get('https://baidu.com')
# self.clicks_my(By.CLASS_NAME,'mnav',num=0)
if __name__ == '__main__':
unittest.main()
这里我用的是QQ空间来进行的书写,并且只写了一个测试用例,这样写其实还是比较不方便的,我们可以把定位方法再写到其他的.py文件中,测试部分只留测试数据即可。