如何在Behave中给before_all函数传入指定代理变量？

1. 调整命令行传参方式 #

Behave的--tags是用来过滤执行的测试场景，不适合传递配置参数。推荐用Behave内置的用户自定义参数（-D/--define）来传递代理标识，执行命令示例：

behave .\features\automated_test.feature -D proxy=first
# 切换第二个代理的命令
behave .\features\automated_test.feature -D proxy=second

2. 修改environment.py代码 #

重构代理配置逻辑，从命令行参数中读取代理标识，动态加载对应配置：

from selenium import webdriver

# 统一管理代理配置，后续新增代理直接加条目即可
PROXY_CONFIGS = {
    "first": {
        "server": "firstproxy:3128",
        "bypass_list": ";".join(["github.com", "stackoverflow.com"])
    },
    "second": {
        "server": "secondproxy:3128",
        "bypass_list": ";".join(["microsoft.com", "google.com"])
    }
}

def before_all(context):
    # 从命令行获取代理标识，默认用"first"（也可删掉默认值，强制用户指定）
    proxy_key = context.config.userdata.get("proxy", "first")
    
    # 校验代理是否存在，避免无效参数报错
    if proxy_key not in PROXY_CONFIGS:
        raise ValueError(f"无效代理标识：{proxy_key}，可选值：{list(PROXY_CONFIGS.keys())}")
    
    proxy_info = PROXY_CONFIGS[proxy_key]
    
    chrome_options = webdriver.ChromeOptions()
    chrome_options.add_argument(f'--proxy-server={proxy_info["server"]}')
    chrome_options.add_argument(f"--proxy-bypass-list={proxy_info['bypass_list']}")
    
    context.driver = webdriver.Chrome(chrome_options)
    context.driver.maximize_window()

def after_all(context):
    context.driver.quit()  # 用quit()比close()更彻底，确保浏览器进程完全退出

3. 关键说明 #

• 用字典PROXY_CONFIGS统一管理代理信息，后续新增代理无需修改逻辑代码，直接加字典条目就行
• 通过context.config.userdata.get()读取命令行参数，支持设置默认代理
• 增加参数校验，避免传入无效代理标识导致模糊报错
• 替换close()为quit()，彻底关闭浏览器进程，防止残留进程占用资源

（不推荐）如果坚持用--tags传递参数 #

如果一定要用--tags来传递，需要从context中读取运行的tags并提取代理标识，但这种方式会过滤测试场景——只有带对应tag的场景才会执行。示例代码：

from selenium import webdriver

PROXY_CONFIGS = {
    "proxy1": {
        "server": "firstproxy:3128",
        "bypass_list": ";".join(["github.com", "stackoverflow.com"])
    },
    "proxy2": {
        "server": "secondproxy:3128",
        "bypass_list": ";".join(["microsoft.com", "google.com"])
    }
}

def before_all(context):
    # 获取当前运行的所有tags
    run_tags = context.config.tags.ands
    # 筛选出代理相关的tag（假设格式为proxy1、proxy2）
    proxy_tags = [tag for tag in run_tags if tag.startswith("proxy")]
    
    if not proxy_tags:
        raise ValueError("请通过--tags指定代理标识，如--tags=proxy1")
    if len(proxy_tags) > 1:
        raise ValueError("只能指定一个代理标识")
    
    proxy_key = proxy_tags[0]
    if proxy_key not in PROXY_CONFIGS:
        raise ValueError(f"无效代理标识：{proxy_key}，可选值：{list(PROXY_CONFIGS.keys())}")
    
    proxy_info = PROXY_CONFIGS[proxy_key]
    
    chrome_options = webdriver.ChromeOptions()
    chrome_options.add_argument(f'--proxy-server={proxy_info["server"]}')
    chrome_options.add_argument(f"--proxy-bypass-list={proxy_info['bypass_list']}")
    
    context.driver = webdriver.Chrome(chrome_options)
    context.driver.maximize_window()

def after_all(context):
    context.driver.quit()

执行命令：

behave .\features\automated_test.feature --tags=proxy1

注意：这种方式要求你的feature文件中，需要执行的场景必须带有@proxy1或@proxy2标签，否则会被过滤跳过。

内容的提问来源于stack exchange，提问作者Matt