引言

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

想了解更多 Behave 相關(guān)的文章，歡迎閱讀《Python BDD Behave 系列》，持續(xù)更新中。

默認(rèn) behave.ini 配置文件

在項(xiàng)目根目錄下，創(chuàng)建 behave.ini 文件，在 behave.userdata 處用戶自定義參數(shù)。這里的用戶參數(shù)通常是整個項(xiàng)目的全局參數(shù)，不區(qū)分環(huán)境。

[behave.userdata]
env = dev

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

自定義配置文件

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

json 格式的配置文件

新建一個 userconfig.json 文件，內(nèi)容如下：

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

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

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 環(huán)境，內(nèi)容如下：

[behave.userdata.more]
db = dev_db

新建 prod_userconfig.ini 文件用來配置 prod 環(huán)境，內(nèi)容如下：

[behave.userdata.more]
db = prod_db

在 environment.py 文件中，before_all hook 中加載該配置文件
可以根據(jù)默認(rèn)配置文件 behave.ini 中 或通過命令行參數(shù)自定義的 env 變量來加載相應(yīng)的配置文件

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

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

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’] 來訪問了

實(shí)例應(yīng)用

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

項(xiàng)目結(jié)構(gòu)

代碼

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)  

默認(rèn)配置文件 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 文件，內(nèi)容如下：

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

BDD/config/dev_userconfig.ini 文件用來配置 dev 環(huán)境，內(nèi)容如下：

[behave.userdata.more]
db = dev_db

BDD/config/prod_userconfig.ini 文件用來配置 prod 環(huán)境，內(nèi)容如下：

[behave.userdata.more]
db = prod_db

vscode 環(huán)境下 behave 設(shè)置 .vscode/settings.json 文件

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

執(zhí)行

當(dāng)前是項(xiàng)目根目錄 C:\Automation\Test

用默認(rèn) behave.ini 自定義的變量值

執(zhí)行命令：behave

默認(rèn)取 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

命令行自定義變量值

執(zhí)行命令：behave -D env=prod
會覆蓋默認(rèn)配置文件 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