环境变量

环境变量的作用

在自动化测试中，有时需要借助环境变量实现某些特定的目的，常见的场景包括：

• 切换测试环境
• 切换测试配置
• 存储敏感数据（从信息安全的角度出发）

设置环境变量

在终端中预设环境变量

使用环境变量之前，需要先在系统中设置环境变量名称和值，传统的方式为使用 export 命令（Windows系统中使用 set 命令）：

$ export UserName=admin
$ echo $UserName
admin
$ export Password=123456
$ echo $Password
123456

然后，在程序中就可以对系统中的环境变量进行读取。

$ python
>>> import os
>>> os.environ["UserName"]
'admin'

通过 .env 文件设置环境变量

除了这种方式，HttpRunner 还借鉴了 pipenv 加载 .env 的方式。

默认情况下，在自动化测试项目的根目录中，创建 .env 文件，并将敏感数据信息放置到其中，存储采用 name=value 的格式：

$ cat .env
UserName=admin
Password=123456
PROJECT_KEY=ABCDEFGH

同时，.env 文件不应该添加到代码仓库中，建议将 .env 加入到 .gitignore 中。

HttpRunner 运行时，会自动将 .env 文件中的内容加载到运行时（RunTime）的环境变量中，然后在运行时中就可以对环境变量进行读取了。

若需加载不位于自动化项目根目录中的 .env，或者其它名称的 .env 文件（例如 production.env），可以采用 --dot-env-path 参数指定文件路径：

$ hrun /path/to/testcase.yml --dot-env-path /path/to/.env --log-level debug
INFO     Loading environment variables from /path/to/.env
DEBUG    Loaded variable: UserName
DEBUG    Loaded variable: Password
DEBUG    Loaded variable: PROJECT_KEY
...

引用环境变量

在 HttpRunner 中内置了函数 environ（简称 ENV），可用于在 YAML/JSON 脚本中直接引用环境变量。

- test:
    name: login
    request:
        url: http://host/api/login
        method: POST
        headers:
            Content-Type: application/json
        json:
            username: ${ENV(UserName)}
            password: ${ENV(Password)}
        validate:
            - eq: [status_code, 200]

若还需对读取的环境变量做进一步处理，则可以在 debugtalk.py 通过 Python 内置的函数 os.environ 对环境变量进行引用，然后再实现处理逻辑。

例如，若发起请求的密码需要先与密钥进行拼接并生成 MD5，那么就可以在 debugtalk.py 文件中实现如下函数：

import os

def get_encrypt_password():
    raw_passwd = os.environ["Password"]
    PROJECT_KEY = os.environ["PROJECT_KEY"])
    password = (raw_passwd + PROJECT_KEY).encode('ascii')
    return hmac.new(password, hashlib.sha1).hexdigest()

然后，在 YAML/JSON 格式的测试用例中，就可以通过${func()}的方式引用环境变量的值了。

- test:
    name: login
    request:
        url: http://host/api/login
        method: POST
        headers:
            Content-Type: application/json
        json:
            username: ${ENV(UserName)}
            password: ${get_encrypt_password()}
        validate:
            - eq: [status_code, 200]