目标:封装一个基类和介绍如何实现POM。关于基类,是这样定义的:把一些常见的页面操作的selenium封装到base_page.py这个类文件,以后每个POM中的页面类,都继承这个基类,这样每个页面类都有基类的方法。
1.在实现封装基类里,我们实现了元素八大方式的定位和截图类封装。具体项目层级结构如下图。
2.基类base_page.py的具体实现代码,这里就封装了几个常用方法,其他方法,你自己去练习封装下。
import time from selenium.common.exceptions import NoSuchElementException import os.path from framework.logger import Logger logger = Logger(logger="BasePage").getlog() class BasePage(object): """ 定义一个页面基类,让所有页面都能继承这个类,封装一些常用的页面操作方法到这个类 """ def __init__(self,driver): self.driver = driver def quit_browser(self): """ 退出浏览器 :return: """ self.driver.quit() logger.info("退出浏览器。") def forward(self): """ 单击当前页上的【前进】按钮 :return: """ self.driver.forward() logger.info("单击当前页上的【前进】按钮。 ") def back(self): """ 单击当前页上的【后退】按钮。 :return: """ self.driver.back() logger.info("单击当前页上的【后退】按钮。") def wait(self,seconds): """ 隐式等待 :return: """ self.driver.implicitly_wait(seconds) logger.info("隐式等待 %d 秒。"%seconds) def close(self): """ 点击关闭当前窗口 :return: """ try: self.driver.close() logger.info("点击关闭当前窗口。") except NameError as e: logger.error("点击关闭当前窗口失败:%s"%e) def get_window_img(self): """ 在这里我们把file_path这个参数写死,直接保存到我们项目根目录的一个文件夹.\screenshots下 :return: """ file_path = os.path.dirname(os.path.abspath('.')) + '/screenshots/' rq = time.strftime('%Y%m%d%H%M',time.localtime(time.time())) screen_name = file_path + rq + '.png' try: self.driver.get_screenshot_as_file(screen_name) logger.info("成功截图并保存到这里:/screenshots") except NameError as e: logger.error("截图失败! %s"%e) self.get_window_img() def find_element(self,selector): """ 这个地方为什么是根据=>来切割字符串,请看页面里定位元素的方法 submit_btn = "id=>su" login_lnk = "xpath => //*[@id='u1']/a[7]" # 百度首页登录链接定位 如果采用等号,结果很多xpath表达式中包含一个=,这样会造成切割不准确,影响元素定位 :param selector: :return: """ element = '' if '=>' not in selector: return self.driver.find_element_by_id(selector) selector_by = selector.split('=>')[0] selector_value = selector.split('=>')[1] if selector_by == "i" or selector_by == 'id': try: element = self.driver.find_element_by_id(selector_value) logger.info("Had find the element \' %s \' successful" "by %s via value:%s " % (element.text,selector_by,selector_value)) except NoSuchElementException as e: logger.error("NoSuchElementException: %s" %e) self.get_window_img() elif selector_by == "n" or selector_by == 'name': element = self.driver.find_element_by_name(selector_value) elif selector_by == "c" or selector_by == 'class_name': element = self.driver.find_element_by_class_mane(selector_value) elif selector_by == "l" or selector_by == 'link_text': element = self.driver.find_element_by_link_text(selector_value) elif selector_by == "p" or selector_by == 'partial_link_text': element = self.driver.find_element_by_partial_link_text(selector_value) elif selector_by == "t" or selector_by == 'tag_name': element = self.driver.find_element_by_tag_name(selector_value) elif selector_by == "x" or selector_by == 'xpath': try: element = self.driver.find_element_by_xpath(selector_value) logger.info("Had find the element \' %s \' successful" "by %s via value: %s "% (element.text,selector_by,selector_value)) except NoSuchElementException as e: logger.error("NoSuchElementException: %s" %e) self.get_window_img() elif selector_by == "s" or selector_by == 'selector_selector': element = self.driver.find_element_by_css_selector(selector_value) else: raise NameError("Please enter a valid type of targeting elements.") return element def type(self,selector,text): """ 输入 :param selector: :param text: :return: """ el = self.find_element(selector) el.clear() try: el.send_keys(text) logger.info("Had type \' %s \' in inputBox" %text) except NameError as e: logger.error("Failed to type in input box with %s"%e) self.get_window_img() def clear(self,selector): """ 清除文本框 :param selector: :return: """ el = self.find_element(selector) try: el.clear() logger.info("Clear text in input box before typing.") except NameError as e: logger.error("Fail to clear in input box with %s"%e) self.get_window_img() def click(self,selector): """ 点击元素 :param selector: :return: """ el = self.find_element(selector) try: el.click() logger.info("The element \' %s \' was clicked." %el.text) except NameError as e: logger.error("Fail to click the element with %s" %e) def get_pape_title(self): """ 获取网页标题 :return: """ logger.info("Current page title is %s" %self.driver.title) return self.driver.title @staticmethod def sleep(seconds): time.sleep(seconds) logger.info("Sleep for %d seconds"%seconds)
3. 页面对象中,百度主页的元素定位和简单的操作函数,页面类主要是元素定位和页面操作写成函数,供测试类调用。
baidu_homepage.py
from framework.base_page import BasePage class HomePage(BasePage): input_box = "id=>kw" search_submit_btn = "xpath=>//*[@id='su']" def type_search(self,text): self.type(self.input_box,text) def send_submit_btn(self): self.click(self.search_submit_btn)
这里注意下元素定位写法,=>和base_page.py中find_element()方法元素定位切割有关系,网上有些人写根据逗号切割或者等号切割,在实际使用xpath定位,发现单独逗号或者单独等号切割都不精确,造成元素定位失败。
4. 测试类的写法举例
import time import unittest from framework.browser_engine import BrowserEngine from pageobjects.baidu_homepage import HomePage class BaiduSearch(unittest.TestCase): def setUp(self): """ 测试固件的setUp()代码,主要是测试的前提工作 :return: """ browse = BrowserEngine(self) self.driver = browse.open_browser(self) def tearDown(self): """ 测试结束后的操作,这里基本上都是关闭浏览器 :return: """ self.driver.quit() def test_baidu_search(self): """ 这里一定要test开头, 把测试逻辑代码封装到一个test开头的方法里 :return: """ homepage = HomePage(self.driver) homepage.type_search('selenium')# 调用页面对象中的方法 homepage.send_submit_btn() #调用页面对象类中的点击搜索按钮方法 time.sleep(2) homepage.get_window_img() # 调用基类截图方法 try: assert 'selenium' in homepage.get_pape_title() # 调用页面对象继承基类中的获取页面标题方法 print('Test Pass.') except Exception as e: print('Test Fail.',format(e)) if __name__ == '__main__': unittest.main()
注意:
homepage = HomePage(self.driver)
上面这行代码要注意,意思是,到一个页面,第一件事情是初始化这个页面的一个页面对象实例。注意,一定要带self.driver,不然会报错,这个self.driver,可以这样理解:在当前测试类里面,self.driver是来自浏览器引擎类中方法得到的,在初始化一个页面对象的时候,也把这个来自浏览器引擎类的driver给赋值给当前的页面对象,这样,才能执行页面对象或者基类里面的相关driver方法。写多了selenium的自动化脚本,你会明白,最重要的是保持前后driver的唯一性。
测试结果:会在logs文件夹生成一个日志文件,也会在screenshots文件夹生成一个png图片。
参考文章:https://blog.csdn.net/u011541946/article/details/70269965