『心善渊』Selenium3.0基础 — 34.POM设计模式（

（二）将普通的Selenium代码封装成POM模式

1、案例说明：

这里只是提供一种封装的思路，小伙伴们可以根据自己的实际情况，按需封装。

以下是简单普通的登录测试用例

# 1. 导入包
from selenium import webdriver
import time

# 2. 打开谷歌浏览器（获取浏览器操作对象）
driver = webdriver.Chrome()

# 3. 打开快递100网站
url = "https://sso.kuaidi100.com/sso/authorize.do"
driver.get(url)
time.sleep(3)

# 4. 登陆网站
driver.find_element_by_id("name").send_keys('xxxxxxxxxxx')
driver.find_element_by_id("password").send_keys('xxxxxx')
driver.find_element_by_id("submit").click()
time.sleep(3)

# 5. 关闭浏览器
driver.quit()

那我们如何进行一个改造升级呢？

2、加入unittest测试框架

# 1. 导入包
from selenium import webdriver
import time
import unittest


# 定义测试类
class TestCaseLogin(unittest.TestCase):
    def setUp(self) -> None:
        """
            前置函数
            用于打开浏览器，连接数据库，初始化数据等操作
        """
        # 2. 打开谷歌浏览器（获取浏览器操作对象）
        self.driver = webdriver.Chrome()

        # 3. 打开快递100网站
        url = "https://sso.kuaidi100.com/sso/authorize.do"
        self.driver.get(url)
        time.sleep(3)

    def tearDown(self) -> None:
        """
            后置函数
            用于关闭浏览器，断开数据库连接，清理测试数据等操作
        """
        # 5. 关闭浏览器
        self.driver.quit()

    def testLogin(self):
        """登陆测试用例"""
        self.driver.find_element_by_id("name").send_keys('xxxxxxxxxxx')
        self.driver.find_element_by_id("password").send_keys('xxxxxx')
        self.driver.find_element_by_id("submit").click()
        time.sleep(3)


if __name__ == '__main__':
    unittest.main()

如果有不清楚unittest测试框架的小伙伴可以查看我以前的unittest测试框架博客有4篇，简单易懂。

3、加入元素显示等待

我们上边的示例中，用的是固定的等待时间，我们需要有话一下代码的效率，加入元素的显示等待。

关于元素显示等待请看：元素等待的使用

Seleniun的EC模块：EC模块的使用

# 1. 导入包
from selenium import webdriver
import time
import unittest
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC


# 定义测试类
class TestCaseLogin(unittest.TestCase):
    def setUp(self) -> None:
        """
            前置函数
            用于打开浏览器，连接数据库，初始化数据等操作
        """
        # 2. 打开谷歌浏览器（获取浏览器操作对象）
        self.driver = webdriver.Chrome()

        # 3. 打开快递100网站
        url = "https://sso.kuaidi100.com/sso/authorize.do"
        self.driver.get(url)
        time.sleep(2)

    def tearDown(self) -> None:
        """
            后置函数
            用于关闭浏览器，断开数据库连接，清理测试数据等操作
        """
        # 5. 关闭浏览器
        time.sleep(2)
        self.driver.quit()

    def testLogin(self):
        """登陆测试用例"""
        # 编写定位器
        name_input_locator = ("id", "name")
        passwd_input_locator = ("id", "password")
        submit_button_locator = ("id", "submit")

        # 等待元素出现在操作元素
        WebDriverWait(self.driver, 5).until(EC.visibility_of_element_located(name_input_locator))
        WebDriverWait(self.driver, 5).until(EC.visibility_of_element_located(passwd_input_locator))
        WebDriverWait(self.driver, 5).until(EC.visibility_of_element_located(submit_button_locator))
        self.driver.find_element_by_id("name").send_keys('xxxxxxxxxxx')
        self.driver.find_element_by_id("password").send_keys('xxxxxx')
        self.driver.find_element_by_id("submit").click()


if __name__ == '__main__':
    unittest.main()