自动化测试往往需要在多个环境中跑测试用例，最开始我用pytest+selenium搭建了一套测试工程，使用jenkins管理定时拉起任务，每次需要在另一个环境跑自动化时就临时改下配置再拉起任务。改的次数多了就烦了，就想办法解决这个问题，实现在运行时根据用户输入的参数来决定使用哪个环境的配置。

要实现运行时根据用户输入来决定使用什么环境配置，需要借助系统环境变量，首先运行程序时提取用户输入参数并将其写入环境变量，然后程序通过读取系统环境变量，根据环境变量的值来决定要运行的环境参数，下面列出python示例代码来演示是如何实现的。

示例代码

main.py为自动化工程运行的入口，为简便起见，这里仅将选择的环境参数打印出来，代码如下：

import sysimport osfrom conf import * class Autotest: def __init__(self): self.running_env = “” self.config = None def get_env_config(self): self.running_env = os.environ[“env”] if “dev” in self.running_env: self.config = DevConfig else: self.config = ProdConfig def show_config(self): print(self.config.desc) print(self.config.demo_url) print(self.config.demo_other) if __name__ == “__main__”: app = Autotest() # 根据运行输入参数设置环境变量 if len(sys.argv) > 1: os.environ[“env”] = sys.argv[1] else: # 用户没有输入默认设置环境为开发环境 os.environ[“env”] = “dev” # 读取环境变量并根据环境变量获取运行配置 app.get_env_config() app.show_config()

conf.py为环境配置文件，和main.py在同一个目录，代码如下：

class Config: desc = “” demo_url = “” demo_other = “继承配置项示例” class DevConfig(Config): desc = “开发环境” demo_url = “https://localhost:8080″# 未设置的参数直接继承Config的配置class ProdConfig(Config): desc = “生产环境” demo_url = “https://localhost:8090” # 未设置的参数直接继承Config的配置