使用 Pytest 和 Selenium 进行参数化测试：动态获取测试数据

本文旨在解决在使用 pytest 和 Selenium 进行参数化测试时，如何动态地将函数返回值作为测试参数的问题。传统的 @pytest.mark.parametrize 无法直接接受函数调用，因此我们将介绍如何使用 pytest_generate_tests 钩子函数来实现动态参数化，并提供详细的代码示例和使用说明，帮助读者理解和应用该方法。

在使用 Pytest 进行测试时，@pytest.mark.parametrize 是一个强大的工具，可以方便地实现参数化测试。然而，在某些场景下，我们可能需要动态地生成测试数据，例如从页面上获取元素列表，并对每个元素进行测试。直接将函数调用传递给 @pytest.mark.parametrize 的 argvalues 参数是不被允许的。本文将介绍如何使用 pytest_generate_tests 钩子函数来解决这个问题。

使用 pytest_generate_tests 实现动态参数化

pytest_generate_tests 是 Pytest 提供的一个钩子函数，它允许我们在测试用例收集阶段动态地生成测试用例。通过这个钩子函数，我们可以获取测试函数的元数据，并根据需要修改测试函数的参数。

以下是一个示例，演示如何使用 pytest_generate_tests 钩子函数来动态地生成测试用例，从而解决 Selenium 测试中动态获取图片列表并进行参数化测试的问题：

def pytest_generate_tests(metafunc):     if "image" in metafunc.fixturenames:         driver = metafunc.config.pluginmanager.getplugin("selenium").webdriver         driver.get('https://openweathermap.org/') # 需要在获取图片列表前打开网页         images_list = driver.execute_script("return document.getElementsByTagName('img')")         metafunc.parametrize("image", images_list)   def test_images(driver, image):     image_check = driver.execute_script("return (arguments[0].width>0 && arguments[0].height>0);", image)     assert image_check is True, "The image is not displayed"

代码解释：

1. pytest_generate_tests(metafunc) 函数: 这是 Pytest 的钩子函数，在测试用例收集阶段被调用。metafunc 对象包含了测试函数的元数据，例如函数名、参数名等。
2. if “image” in metafunc.fixturenames:: 这行代码检查测试函数是否使用了名为 “image” 的 fixture。如果使用了，说明我们需要为这个测试函数生成参数。
3. driver = metafunc.config.pluginmanager.getplugin(“selenium”).webdriver: 这行代码获取 Selenium WebDriver 实例。 注意： 这依赖于你如何设置 Selenium 和 Pytest 的集成。 这里假设你使用了一个名为 “selenium” 的 Pytest 插件，并且该插件提供了一个名为 webdriver 的 fixture。 你可能需要根据你的实际配置进行调整。
4. driver.get(‘https://openweathermap.org/’): 这行代码使用 Selenium 打开需要测试的网页。 重要： 必须在获取图片列表 之前 打开网页。
5. images_list = driver.execute_script(“return document.getElementsByTagName(‘img’)”): 这行代码使用 Selenium 执行 JavaScript 代码，获取页面上所有 <img> 标签的列表。
6. metafunc.parametrize(“image”, images_list): 这行代码使用 metafunc.parametrize 方法，将 images_list 中的每个元素作为参数传递给测试函数 test_images。

test_images(driver, image) 函数:

• 这个函数接收 driver 和 image 作为参数。 driver 是 Selenium WebDriver 实例，image 是从 images_list 中获取的 <img> 元素。
• image_check = driver.execute_script(“return (arguments[0].width>0 && arguments[0].height>0);”, image)： 使用 JavaScript 检查图片元素的宽度和高度是否大于 0，以此判断图片是否正常显示。
• assert image_check is True, “The image is not displayed”： 断言图片是否正常显示。

注意事项

• 确保 Selenium WebDriver 已经正确配置，并且可以在 Pytest 中使用。
• pytest_generate_tests 钩子函数在测试用例收集阶段被调用，因此在这个函数中执行的操作应该尽可能快，避免影响测试的性能。
• 根据实际情况修改代码中的 URL 和 JavaScript 代码，以适应不同的测试场景。
• metafunc.config.pluginmanager.getplugin(“selenium”).webdriver 获取 driver 的方式依赖于 Selenium 和 Pytest 的集成方式。 如果你使用不同的集成方式，你需要相应地修改这部分代码。

总结

pytest_generate_tests 钩子函数为我们提供了一种灵活的方式来动态地生成测试用例，从而解决在使用 Pytest 和 Selenium 进行参数化测试时遇到的问题。通过本文的介绍和示例，相信读者可以掌握如何使用 pytest_generate_tests 钩子函数来实现动态参数化，并将其应用到实际的测试项目中。