fix

Author: zixingdeng

pytest/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "testsolar-pytestx"
-version = "0.1.62"
+version = "0.1.63"
 description = "Pytest tool for TestSolar"
 authors = [
     {name = "asiazhang2002", email = "asiazhang2002@gmail.com"},

pytest/src/testsolar_pytestx/executor.py

@@ -48,6 +48,15 @@ class RunMode(Enum):
     BATCH = "batch"
 
 
+def should_enable_header_injection() -> bool:
+    """判断是否需要启用请求头注入功能"""
+    return os.environ.get("ENABLE_API_COLLECTING", "") == "1"
+
+
+if should_enable_header_injection():
+    initialize_header_injection()
+
+
 class PytestExecutor:
     def __init__(
         self,
@@ -62,13 +71,6 @@ def __init__(
         self.comment_fields = comment_fields
         self.data_drive_key = data_drive_key
 
-        # 初始化API请求头注入功能，传入判定方法
-        initialize_header_injection()
-
-    def _should_enable_header_injection(self) -> bool:
-        """判断是否需要启用请求头注入功能"""
-        return os.environ.get("ENABLE_API_COLLECTING", "") == "1"
-
     def pytest_runtest_logstart(self, nodeid: str, location: Any) -> None:
         """
         Called at the start of running the runtest protocol for a single item.
@@ -79,7 +81,7 @@ def pytest_runtest_logstart(self, nodeid: str, location: Any) -> None:
         testcase_name = normalize_testcase_name(nodeid, self.data_drive_key)
 
         # 设置当前测试用例的nodeid到上下文
-        if self._should_enable_header_injection():
+        if should_enable_header_injection():
             testcase_class_name = testcase_name.split("?", 1)[-1]
             set_current_test_nodeid(testcase_class_name)
 
@@ -228,7 +230,7 @@ def pytest_runtest_logfinish(self, nodeid: str, location: Any) -> None:
             self.testdata.pop(testcase_name, None)
 
         # 清除当前测试用例的nodeid
-        if self._should_enable_header_injection():
+        if should_enable_header_injection():
             set_current_test_nodeid(None)
 
         logger.info(f"E {nodeid} runtest_logfinish")

pytest/testtool.yaml

@@ -2,7 +2,7 @@ schemaVersion: 1.0
 name: pytest
 nameZh: pytest自动化测试
 lang: python
-version: '0.1.62'
+version: '0.1.63'
 langType: INTERPRETED
 description: |-
   pytest是一个成熟的全功能Python测试工具，可以帮助您编写更好的程序。此测试工具允许您在TestSolar上运行pytest。