| 一改测试步骤代码就全写？为什么不试试用 Yaml实现数据驱动？_单元测试案例执行的数据源是yaml里配置的还是代码里写的-优快云博客

在有些场景中，需要上传文件，而 Selenium 无法定位到弹出的文件框，以及网页弹出的提醒。这些都是需要特殊的方式来处理。

理念与同“UI自动化测试框架”中的“测试步骤的数据驱动”相同，接口中的测试步骤的数据驱动就是将接口的参数（比如 method、url、param等）封装到 yaml 文件中管理。当测试步骤发生改变，只需要修改 yaml 文件中的配置即可。
数据驱动就是数据的改变从而驱动自动化测试的执行，最终引起测试结果的改变。简单来说，就是参数化的应用。数据量小的测试用例可以使用代码的参数化来实现数据驱动，数据量大的情况下建议使用一种结构化的文件（例如yaml，json等）来对数据进行存储，然后在测试用例中读取这些数据。

原理与前面章节“UI自动化测试框架”中的“测试数据的数据驱动”大同小异。依然使用 @pytest.mark.parametrize 装饰器来进行参数化，使用参数化来实现数据驱动。
通过参数化的方式，分别判断id为2，3的部门的parentid为1：

import pytest

class TestDepartment:
    department = Department()
    @pytest.mark.parametrize("id", [2, 3])
        def test_department_list(self, id):
                r = self.department.list(id)
                        assert self.department.jsonpath(expr="$..parentid")[0] == 1

上面的代码首先使用 @pytest.mark.parametrize 装饰器，传递了两组数据，测试结果显示有两条测试用例被执行，而不是一条测试用例。也就是 pytest 会将两组测试数据自动生成两个对应的测试用例并执行，生成两条测试结果。

当测试数据量大的情况下，可以考虑把数据存储在结构化的文件中。从文件中读取出代码中所需要格式的数据，传递到测试用例中执行。本次实战以YAML进行演示。YAML以使用动态字段进行结构化，它以数据为中心，比 excel、csv、Json、XML 等更适合做数据驱动。

将上面参数化的两组数据存储到 yaml 文件中，创建一个data/department_list.yml文件，代码如下：

-2
-3

上面的代码定义了一个 yaml 格式的数据文件department_list.yml，文件中定义了一个列表，列表中有两个数据，最后生成的是这样的数据格式：[1,2]。将测试用例中参数化的数据改造成从 department_list.yml 文件中读取，代码如下：

class TestDepartment:
    department = Department()
    @pytest.mark.parametrize("id", \
        yaml.safe_load(open("../data/department_list.yml")))
            def test_department_list(self, id):
                    r = self.department.list(id)
                            assert self.department.jsonpath(expr="$..parentid")[0] == 1

上面的代码，只需要使用yaml.safe_load()方法，读取department_list.yml文件中的数据，分别传入到用例 test_department_list() 方法中完成输入与结果的验证。

实际工作中，对于环境的切换和配置，为了便于维护，通常不会使用硬编码的形式完成。在“多环境下的接口测试”章节中已经介绍了，如何将环境的切换作为一个可配置的选项。本章节会把这部分内容进行重构，使用数据驱动的方式完成多环境的配置。

根据“多环境下的接口测试”章节，将此章节中的环境配置部分改为数据驱动的模式
代码如下：

#把host修改为ip，并附加host header
env={
    "docker.testing-studio.com": {
            "dev": "127.0.0.1",
                    "test": "1.1.1.2"
                        },
                            "default": "dev"
                            }
                            data["url"]=str(data["url"]).replace(
                                "docker.testing-studio.com",
                                    env["docker.testing-studio.com"][env["default"]]
                                    )
                                    data["headers"]["Host"]="docker.testing-studio.com"

依然以yaml为示例，将所有的环境配置信息放到 env.yml 文件中。如果怕出错，可以先使用yaml.safe_dump(env)将dict格式的代码转换为yaml。
如下所示，打印出来的，就是成功转换yaml格式的配置信息：

def test_send(self):
    env={
            "docker.testing-studio.com": {
                        "dev": "127.0.0.1",
                                    "test": "1.1.1.2"
                                            },
                                                    "default": "dev"
                                                        }
                                                            yaml2 = yaml.safe_dump(env)
                                                                print("")
                                                                    print(yaml2)

将打印出来的内容粘贴到 env.yml 文件中： env.yml

docker.testing-studio.com:
  dev: "127.0.0.1"
    test: "1.1.1.2"
      level: 4
      default:
        "dev"

将环境准备中的代码稍作修改，把env变量从一个典型dict改为，使用yaml.safe_load读取 env.yml：

# 把host修改为ip，并附加host header
env = yaml.safe_load(open("./env.yml"))
data["url"] = str(data["url"]).\
    replace("docker.testing-studio.com",
        env["docker.testing-studio.com"][env["default"]])
        data["headers"]["Host"] = "docker.testing-studio.com"

如此一来，就可以实现使用数据驱动的方式，通过修改 env.yml 文件来直接修改配置信息。

⬇️ 点击“阅读原文”，提升测试核心竞争力！
原文链接

获取更多相关资料，请添加微信 ceshiren6
获取更多技术文章分享
input 标签使用自动化上传，先定位到上传按钮，然后 send_keys 把路径作为值给传进去.
如图所示，是企业微信文件上传的页面定位到标签为 input，type 为 file 的元素信息，然后使用 send_keys 把文件路径作为值给传进去。

driver.find_element(By.CSS_SELECTOR, "#js_upload_input")\
    .send_keys("./hogwarts.png")

在页面操作中有时会遇到 JavaScript 所生成的 alert、confirm 以及 prompt 弹框，可以使用switch_to.alert()方法定位到。然后使用 text、accept、dismiss、send_keys 等方法进行操作。

switch_to.alert()：获取当前页面上的警告框。
- text：返回 alert、confirm、prompt 中的文字信息。
- accept()：接受现有警告框，即点击确定。
- dismiss()：解散现有警告框，即点击取消。
- send_keys(keysToSend)：发送文本至警告框。keysToSend:将文本发送至警告框。
  输入一段文本点击比如提交按钮，会弹出确认内容的弹框，这种场景可以使用下面的方式处理：

"""Alert弹窗获取文本与确认操作"""
driver.get("http://sahitest.com/demo/alertTest.htm")
driver.find_element_by_name("b1").click()
#添加显示等待，等待弹框的出现
WebDriverWait(driver, 5, 0.5).until(EC.alert_is_present())
#切换到弹框
alert = driver.switch_to.alert
#打印弹框的文本
print(alert.text)
#点击确定
alert.accept()
#不点击确定，解除弹框
alert.dismiss()

""" Prompt 弹窗获取文本、输入内容、确认操作 """
driver.get("http://sahitest.com/demo/promptTest.htm")
driver.find_element_by_name("b1").click()
#添加显示等待，等待弹框的出现
WebDriverWait(driver, 5).until(EC.alert_is_present())
#切换到弹框
alert = driver.switch_to.alert
#向弹框输入一段文本
alert.send_keys('Selenium Alert弹出窗口输入信息')
#点击确定
alert.accept()

""" Comfirm弹窗获取文本、确认、取消操作 """
driver.get("http://sahitest.com/demo/confirmTest.htm")
driver.find_element_by_name("b1").click()
# 等待弹出窗口出现
WebDriverWait(driver, 5).until(EC.alert_is_present())
#切换到弹框
alert = driver.switch_to.alert
#点击确定
alert.accept()
#点击取消
alert.dismiss()

Web自动化中文件上传与弹框处理就先讲到这里啦，大家还想看什么内容的文章也可以留言告诉我们哦！

⬇️ 点击“阅读原文”，提升测试核心竞争力！

原文链接

获取更多相关资料，请添加微信 ceshiren6
获取更多技术文章分享