引言

前面《BDD - Python Behave 配置文件 behave.ini》有介紹默認配置文件 behave.ini 一些通用配置。《BDD - Python Behave 用戶自定義命令行選項 -D》通過命令行或默認配置文件 behave.ini 自定義參數。在實際自動化測試項目中，不同環境測試配置也不一樣，例如：服務器地址，數據庫名，API URL 等等，Behave 測試框架怎么自定義配置文件并應用到不同的測試環境中，今天就來了解一下，這也是框架設計的一部分，非常重要。

想了解更多 Behave 相關的文章，歡迎閱讀《Python BDD Behave 系列》，持續更新中。

默認 behave.ini 配置文件

在項目根目錄下，創建 behave.ini 文件，在 behave.userdata 處用戶自定義參數。這里的用戶參數通常是整個項目的全局參數，不區分環境。

[behave.userdata]
env = dev

腳本可以通過 context.config.userdata[‘env’] 便可訪問到

自定義配置文件

官網 User Data，用戶可以自定義配置文件 json 格式或 ini 格式，只是需要在 before_all() hook 中加載解析后才能應用的測試用例中。

json 格式的配置文件

新建一個 userconfig.json 文件，內容如下：

{
    "user": "user_name",
    "pw": "password"
}

在 environment.py 文件中，before_all hook 中加載該配置文件
configfile_json = userdata.get(“configfile”, “BDD/config/userconfig.json”)
獲取配置文件，后面參數是配置文件的相對路徑。

from behave import *
import os
import json
from behave.configuration import ConfigParser

def before_all(context):
    userdata = context.config.userdata
    # get json config
    configfile_json = userdata.get("configfile", "BDD/config/userconfig.json")
    if os.path.exists(configfile_json):
        config = json.load(open(configfile_json))
        userdata.update(config)

這樣就可以通過 context.config.userdata[‘user’] 和 context.config.userdata[‘pw’] 來訪問了

ini 格式的配置文件

新建 dev_userconfig.ini 文件用來配置 dev 環境，內容如下：

[behave.userdata.more]
db = dev_db

新建 prod_userconfig.ini 文件用來配置 prod 環境，內容如下：

[behave.userdata.more]
db = prod_db

在 environment.py 文件中，before_all hook 中加載該配置文件
可以根據默認配置文件 behave.ini 中 或通過命令行參數自定義的 env 變量來加載相應的配置文件

env = userdata.get("env")    
configfile = userdata.get("configfile", f"BDD/config/{env}_userconfig.ini")
section = userdata.get("config_section", "behave.userdata.more")

以上是用來獲取配置文件并指定 section 部分，然后通過 ConfigParser 解析加載配置項。

from behave import *
import os
import json
from behave.configuration import ConfigParser

def before_all(context):
    context.before_all = "before all"
    userdata = context.config.userdata
  
    # get ini config
    env = userdata.get("env")    
    configfile = userdata.get("configfile", f"BDD/config/{env}_userconfig.ini")
    section = userdata.get("config_section", "behave.userdata.more")
    parser = ConfigParser()
    parser.read(configfile)
    if parser.has_section(section):
        userdata.update(parser.items(section)) 

這樣就可以通過 context.config.userdata[‘db’] 來訪問了

實例應用

綜合上面的講解，來個完整的案例吧，基本上可以解決不同環境的配置問題。

項目結構

代碼

BDD/Features/user_data.feature 文件

# user_data.feature 
Feature: Context User Data Example

	@user_data
	Scenario: User data scenario
		Given user data is set
		Then get the user data

BDD/steps/user_data_steps.py 文件

# user_data_steps.py

from behave import *

@given('user data is set')
def step_user_data_is_set(context):
    pass

@then('get the user data')
def then_get_user_data(context):
    print(f"env:{context.config.userdata['env']}")
    print(f"user:{context.config.userdata['user']}")
    print(f"pw:{context.config.userdata['pw']}")
    print(f"pw:{context.config.userdata['db']}")

BDD/environment.py 文件

# environment.py

from behave import *
import os
import json
from behave.configuration import ConfigParser

def before_all(context):

    userdata = context.config.userdata

    # get json config
    configfile_json = userdata.get("configfile", "BDD/config/userconfig.json")
    if os.path.exists(configfile_json):
        config = json.load(open(configfile_json))
        userdata.update(config)
    
    # get ini config
    env = userdata.get("env")    
    configfile = userdata.get("configfile", f"BDD/config/{env}_userconfig.ini")
    section = userdata.get("config_section", "behave.userdata.more")
    parser = ConfigParser()
    parser.read(configfile)
    if parser.has_section(section):
        userdata.update(parser.items(section))     

    print("userdata", userdata)  

默認配置文件 behave.ini

# behave.ini
[behave]
paths=BDD/Features/user_data
dry_run = false
format = my_html
outfiles = output/my_report.html
stdout_capture = False

[behave.formatters]
my_html = behave_html_formatter:HTMLFormatter

[behave.userdata]
env = dev

自定義配置文件

BDD/config/userconfig.json 文件，內容如下：

{
    "user": "user_name",
    "pw": "password"
}

BDD/config/dev_userconfig.ini 文件用來配置 dev 環境，內容如下：

[behave.userdata.more]
db = dev_db

BDD/config/prod_userconfig.ini 文件用來配置 prod 環境，內容如下：

[behave.userdata.more]
db = prod_db

vscode 環境下 behave 設置 .vscode/settings.json 文件

{
    "behave-vsc.featuresPath": "BDD/Features"
}

執行

當前是項目根目錄 C:\Automation\Test

用默認 behave.ini 自定義的變量值

執行命令：behave

默認取 env = dev，所以會加載到 dev_userconfig.ini，
所以 db = dev_db

PS C:\Automation\Test> behave
userdata {'env': 'dev', 'user': 'user_name', 'pw': 'password', 'db': 'dev_db'}
Feature: Context User Data Example # BDD/Features/user_data/user_data.feature:2

  @user_data
  Scenario: User data scenario  # BDD/Features/user_data/user_data.feature:5   
    Given user data is set      # BDD/steps/user_data_steps.py:5
    Then get the user data      # BDD/steps/user_data_steps.py:9
env:dev
user:user_name
pw:password
pw:dev_db

1 feature passed, 0 failed, 0 skipped
1 scenario passed, 0 failed, 0 skipped
2 steps passed, 0 failed, 0 skipped, 0 undefined
Took 0m0.000s

命令行自定義變量值

執行命令：behave -D env=prod
會覆蓋默認配置文件 behave.ini 中 env=dev 的值，用最新的 env=prod，會加載到 prod_userconfig.ini 文件，db=prod_db

PS C:\Automation\Test> behave -D env=prod
userdata {'env': 'prod', 'user': 'user_name', 'pw': 'password', 'db': 'prod_db'}
Feature: Context User Data Example # BDD/Features/user_data/user_data.feature:2

  @user_data
  Scenario: User data scenario  # BDD/Features/user_data/user_data.feature:5
    Given user data is set      # BDD/steps/user_data_steps.py:5
    Then get the user data      # BDD/steps/user_data_steps.py:9
env:prod
user:user_name
pw:password
pw:prod_db

1 feature passed, 0 failed, 0 skipped
1 scenario passed, 0 failed, 0 skipped
2 steps passed, 0 failed, 0 skipped, 0 undefined
Took 0m0.001s