简介
上一篇文章我们介绍了使用ChatGPT帮我们编写自动化测试脚本，但是上文编写的脚本并不符合我们的PO设计模式，作为现在主流的设计模式，更加方便我们去编写脚本，一旦页面发生变动，我们的代码改动也会变小，所以我们的目标不是使用ChatGPT编写自动化脚本，而是要使用ChatGPT来编写符合PO设计模式的自动化脚本，而且PO设计模式又经常会结合数据驱动，所以本文就来给大家介绍一下使用ChatGPT来生成符合PO模式的数据驱动测试。
演练示例
提问：请使用PO设计模式编写一个网站登录的Web自动化测试脚本，要求使用到Pytest测试框架和Selenium Web自动化框架。
ChatGPT回复如下：
当使用Page Object 设计模式时，可以将网站的不同页面抽象为单独的类，每个类对应一个页面。这里是一个示例，假设有登录页面和用户账户页面：
首先，创建一个 login_page.py 文件：
from selenium import webdriver
from selenium.webdriver.common.by import By
class LoginPage:
def __init__(self, driver):
self.driver = driver
self.url = "https://example.com/login"
self.username_input = (By.ID, "username")
self.password_input = (By.ID, "password")
self.login_button = (By.ID, "login-button")
def load(self):
self.driver.get(self.url)
def set_username(self, username):
self.driver.find_element(*self.username_input).send_keys(username)
def set_password(self, password):
self.driver.find_element(*self.password_input).send_keys(password)
def click_login(self):
self.driver.find_element(*self.login_button).click()
然后，创建一个user_account_page.py文件：
class UserAccountPage:
def __init__(self, driver):
self.driver = driver
def get_title(self):
return self.driver.title
接下来，编写一个测试文件test_login.py，利用Page Object模式执行登录并验证：
import pytest
from selenium import webdriver
from login_page import LoginPage
from user_account_page import UserAccountPage
@pytest.fixture(scope="module")
def browser():
driver = webdriver.Chrome()
yield driver
driver.quit()
def test_login(browser):
login_page = LoginPage(browser)
user_account_page = UserAccountPage(browser)
login_page.load()
login_page.set_username("your_username")
login_page.set_password("your_password")
login_page.click_login()
# 验证登录后页面的标题
assert "User Account" in user_account_page.get_title(), "页面标题未显示用户账户信息"
这个测试文件使用了 LoginPage 和 UserAccountPage 类来执行登录并验证登录后的页面标题。LoginPage 类包含加载页面和执行登录的方法，而 UserAccountPage 类包含获取页面标题的方法。
但是我们也发现，该脚本还未实现数据驱动测试，我们可以继续提问，实现数据驱动测试。
提问：我现在想把用户名密码使用数据驱动+参数化的方式传入，请修改你的 test_login 的代码。
ChatGPT回复如下：
当使用数据驱动和参数化的方式传入用户名和密码时，可以使用pytest的参数化装饰器。在测试文件中添加参数化装饰器，并将测试数据传入测试函数。
import pytest
from selenium import webdriver
from login_page import LoginPage
from user_account_page import UserAccountPage
@pytest.fixture(scope="module")
def browser():
driver = webdriver.Chrome()
yield driver
driver.quit()
# 测试数据（用户名和密码）
testdata = [
("username1", "password1"),
("username2", "password2"),
# 可以添加更多的测试数据
]
@pytest.mark.parametrize("username, password", testdata)
def test_login(browser, username, password):
login_page = LoginPage(browser)
user_account_page = UserAccountPage(browser)
login_page.load()
login_page.set_username(username)
login_page.set_password(password)
login_page.click_login()
# 验证登录后页面的标题
assert "User Account" in user_account_page.get_title(), "页面标题未显示用户账户信息"
在这个示例中，testdata 列表包含了多组用户名和密码，通过 @pytest.mark.parametrize 装饰器将这些数据传递给 test_login 函数。测试将会针对每一组数据执行登录操作，并验证登录后页面的标题。
总结
本文主要介绍了使用ChatGPT实现PO设计模式和数据驱动测试的结合，我们需要注意的是，我们在提问时可以将PO设计模式和数据驱动测试分开，这样方便我们对ChatGPT的回复进行纠正，也把更为复杂的任务简单化了，希望本文可以帮到大家。
