pytest-api-framework-alpha 0.2.3__tar.gz → 0.2.5__tar.gz

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-api-framework-alpha
-Version: 0.2.3
+Version: 0.2.5
 Author: alpha
 Author-email: 
 Requires-Python: >=3.6
@@ -23,6 +23,7 @@ Requires-Dist: requests==2.25.1
 Requires-Dist: requests-toolbelt==1.0.0
 Requires-Dist: retry==0.9.2
 Requires-Dist: dill==0.3.8
+Requires-Dist: simplejson==3.20.1
 Dynamic: author
 Dynamic: requires-dist
 Dynamic: requires-python

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/framework/conftest.py

@@ -17,6 +17,7 @@ import retry
 import allure
 import pytest
 from box import Box
+import simplejson as json
 
 from framework.exit_code import ExitCode
 from framework.db.mysql_db import MysqlDB
@@ -399,7 +400,7 @@ def pytest_runtest_setup(item):
         test_object.http = _FRAMEWORK_CONTEXT.get(key="_http")
         data = item.callspec.params.get("data")
         test_object.data = Box(data)
-        test_object.scenario = Box(convert_numbers_to_decimal(data.get("_scenario").get("data")))
+        test_object.scenario = Box(json.loads(json.dumps(convert_numbers_to_decimal(data.get("_scenario").get("data")))))
         test_object.belong_app = data.get("_belong_app")
         test_before_scenario = getattr(test_object, "test_setup", None)
         if test_before_scenario:
@@ -456,7 +457,7 @@ def pytest_runtest_call(item):
     http = item.funcargs.get("http")
     item.funcargs["data"] = item.instance.data = Box(origin_data)
     item.funcargs["scenario"] = item.instance.scenario = Box(
-        convert_numbers_to_decimal(origin_data.get("_scenario").get("data")))
+        json.loads(json.dumps(convert_numbers_to_decimal(origin_data.get("_scenario").get("data")))))
     _belong_app = origin_data.get("_belong_app")
     item.funcargs["belong_app"] = item.instance.belong_app = _belong_app
     item.funcargs["config"] = item.instance.config = CONFIG

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/framework/http_client.py

@@ -4,7 +4,8 @@ from urllib.parse import urljoin
 
 import allure
 import requests
-import json as built_json
+import simplejson as built_json
+
 from box import Box, BoxList
 from jsonpath import jsonpath as jp
 from requests.models import Response
@@ -22,9 +23,9 @@ class ResponseUtil(object):
     def __init__(self, response: Response):
         self.response = response
         if isinstance(self.response.json(), dict):
-            self.box = Box(convert_numbers_to_decimal(self.response.json()))
+            self.box = Box(built_json.loads(built_json.dumps(convert_numbers_to_decimal(self.response.json()))))
         elif isinstance(self.response.json(), list):
-            self.box = BoxList(convert_numbers_to_decimal(self.response.json()))
+            self.box = BoxList(built_json.loads(built_json.dumps(convert_numbers_to_decimal(self.response.json()))))
         else:
             self.box = self.response.text
 
@@ -111,31 +112,36 @@ class HttpClient(object):
 
             with allure.step(f"请求url: {request_obj.get('url')}"):
                 logger.info(f"请求url: {request_obj.get('url')}")
+
             with allure.step(f"请求method: {request_obj.get('method')}"):
                 logger.info(f"请求method: {request_obj.get('method')}")
-            with allure.step(f"请求headers: {built_json.dumps(request_obj.get('headers'))}"):
+
+            with allure.step(f"请求headers: {HttpClient.json(request_obj.get('headers'))}"):
                 if CONSOLE_DETAILED_LOG:
+                    logger.info(f"请求headers: {HttpClient.json(request_obj.get('headers'))}")
 
-                    logger.info(f"请求headers: {built_json.dumps(request_obj.get('headers'))}")
             if request_obj.get('params'):
-                with allure.step(f"请求参数params: {built_json.dumps(request_obj.get('params'))}"):
-                    logger.info(f"请求参数params: {built_json.dumps(request_obj.get('params'))}")
+                with allure.step(f"请求参数params: {HttpClient.json(request_obj.get('params'))}"):
+                    logger.info(f"请求参数params: {HttpClient.json(request_obj.get('params'))}")
+
             if request_obj.get('data'):
-                with allure.step(f"请求参数data: {built_json.dumps(request_obj.get('data'))}"):
-                    logger.info(f"请求参数data: {built_json.dumps(request_obj.get('data'))}")
+                with allure.step(f"请求参数data: {request_obj.get('data')}"):
+                    logger.info(f"请求参数data: {request_obj.get('data')}")
+
             if request_obj.get('json'):
-                with allure.step(f"请求参数json: {built_json.dumps(request_obj.get('json'))}"):
-                    logger.info(f"请求参数json: {built_json.dumps(request_obj.get('json'))}")
+                with allure.step(f"请求参数json: {HttpClient.json(request_obj.get('json'))}"):
+                    logger.info(f"请求参数json: {HttpClient.json(request_obj.get('json'))}")
 
         with allure.step("响应结果"):
             with allure.step(f"响应status code: {self.response.status_code}"):
                 logger.info(f"响应status code: {self.response.status_code}")
 
-            with allure.step(f"响应headers: {built_json.dumps(dict(self.response.headers))}"):
+            with allure.step(f"响应headers: {HttpClient.json(dict(self.response.headers))}"):
                 if CONSOLE_DETAILED_LOG:
-                    logger.info(f"响应headers: {built_json.dumps(dict(self.response.headers))}")
-            with allure.step(f"响应body: {built_json.dumps(self.response.json())}"):
-                logger.info(f"响应body: {built_json.dumps(self.response.json())}")
+                    logger.info(f"响应headers: {HttpClient.json(dict(self.response.headers))}")
+
+            with allure.step(f"响应body: {HttpClient.json(self.response.json())}"):
+                logger.info(f"响应body: {HttpClient.json(self.response.json())}")
 
         # 断言
         validates = data.get("validate")
@@ -175,3 +181,10 @@ class HttpClient(object):
 
     def update_headers(self, headers):
         self.headers.update(headers)
+
+    @staticmethod
+    def json(dic):
+        try:
+            return built_json.dumps(dic)
+        except:
+            return dic

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/framework/utils/common.py

@@ -265,9 +265,3 @@ def convert_numbers_to_decimal(obj: Any) -> Any:
 def remove_spaces(s: str) -> str:
     """去掉字符串中的所有空格"""
     return s.replace(" ", "")
-
-
-if __name__ == '__main__':
-    a = decimal.Decimal("123")
-    print(a)
-    assert a == 123

pytest_api_framework_alpha-0.2.5/framework/utils/mock_util.py

@@ -0,0 +1,159 @@
+import requests
+from enum import Enum, unique
+from typing import Dict, Optional
+
+from config.settings import AVAILABLE_ENVS, ENV_CONFIG
+
+
+@unique
+class PaymentType(Enum):
+    """支付类型枚举"""
+    DEPOSIT = 0  # 入金
+    WITHDRAWAL = 1  # 出金
+    CHECKOUT = 2  # 结账
+
+
+@unique
+class RiskLevel(Enum):
+    """风险等级枚举"""
+    DISABLE_MOCK = 0  # 禁用模拟
+    LOW = 1  # 低风险
+    MEDIUM_LOW = 2  # 中低风险
+    MEDIUM_HIGH = 3  # 中高风险
+    HIGH = 4  # 高风险
+    SEVERE = 5  # 严重风险
+
+
+@unique
+class IsDelayed(Enum):
+    """是否延迟枚举"""
+    NO_DELAY = 0  # 不延迟
+    DELAY = 1  # 延迟
+
+
+def get_environment_config(env: str) -> Dict:
+    """
+    获取指定环境的完整配置
+
+    参数:
+        env: 环境名称（如"dev", "test"等）
+
+    返回:
+        对应环境的配置字典
+
+    异常:
+        ValueError: 当环境名称不存在时抛出
+    """
+    if env not in AVAILABLE_ENVS:
+        raise ValueError(f"不支持的环境: {env}，可用环境: {AVAILABLE_ENVS}")
+    return ENV_CONFIG[env]
+
+
+def get_full_api_url(env,method):
+    """拼接完整的API URL"""
+    config = get_environment_config(env)
+    return f"{config['base_url']}{config[method]}"
+
+
+def set_customized_kytmock(env: str,request_id: str, payment_type: PaymentType,risk_level: RiskLevel,is_delayed: IsDelayed = IsDelayed.NO_DELAY):
+    """
+    调用指定环境的HTTP接口
+
+    参数:
+        env: 环境名称（如"dev", "test"）
+        request_id: 请求ID
+        payment_type: 支付类型枚举
+        risk_level: 风险等级枚举
+        is_delayed: 是否延迟枚举，默认不延迟
+        custom_url: 自定义URL（可选，用于临时测试特殊地址）
+
+    返回:
+        接口响应的JSON字典，失败则返回None
+    """
+    # 获取环境配置
+    env_config = get_environment_config(env)
+    method = "set_mock"
+    url = get_full_api_url(env,method)
+
+    # 参数验证
+    if not isinstance(request_id, str) or not request_id:
+        raise ValueError("request_id必须是非空字符串")
+
+    for param, name in [
+        (payment_type, "payment_type"),
+        (risk_level, "risk_level"),
+        (is_delayed, "is_delayed")
+    ]:
+        if not isinstance(param, (PaymentType, RiskLevel, IsDelayed)):
+            raise ValueError(f"{name}必须是对应的枚举类型")
+
+    # 构建请求参数
+    payload = {
+        "request_id": request_id,
+        "payment_type": payment_type.value,
+        "risk_level": risk_level.value,
+        "is_delayed": is_delayed.value
+    }
+
+    try:
+        # 使用环境配置中的超时时间
+        response = requests.post(
+            url,
+            json=payload,
+            timeout=env_config["timeout"]
+        )
+        response.raise_for_status()
+
+        return response.json()
+    except requests.exceptions.HTTPError as e:
+        print(f"HTTP错误 [{env}]: {e}")
+    except requests.exceptions.RequestException as e:
+        print(f"请求异常 [{env}]: {e}")
+    except ValueError:
+        print(f"响应解析失败 [{env}]")
+
+    return None
+
+
+def get_customized_kytmock(env,request_id):
+    # 获取环境配置
+    env_config = get_environment_config(env)
+
+    # 参数验证
+    if not isinstance(request_id, str) or not request_id:
+        raise ValueError("request_id必须是非空字符串")
+    method = f"get_mock"
+    url = f"{get_full_api_url(env, method)}{request_id}"
+    try:
+        response = requests.get(url)
+        return response.json()
+    except requests.exceptions.HTTPError as e:
+        print(f"HTTP错误 [{env}]: {e}")
+    except requests.exceptions.RequestException as e:
+        print(f"请求异常 [{env}]: {e}")
+    except ValueError:
+        print(f"响应解析失败 [{env}]")
+
+    return None
+
+if __name__ == "__main__":
+    # # 1. 调用测试环境的入金接口
+    # test_deposit = set_customized_kytmock(
+    #     env="dev",
+    #     request_id="A0101585",
+    #     payment_type=PaymentType.DEPOSIT,
+    #     risk_level=RiskLevel.LOW
+    # )
+    # print(f"测试环境入金响应: {test_deposit}")
+    #
+    # # 2. 调用测试环境的出金接口
+    # dev_withdrawal = set_customized_kytmock(
+    #     env="dev",
+    #     request_id="0xf3C3e95597544cD0525393CE49B685994b8e50A7",
+    #     payment_type=PaymentType.WITHDRAWAL,
+    #     risk_level=RiskLevel.LOW
+    # )
+    # print(f"开发环境出金响应: {dev_withdrawal}")
+
+    get_content = get_customized_kytmock("dev","A0101585")
+    print(f"查询结果：{get_content}")

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/pytest_api_framework_alpha.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-api-framework-alpha
-Version: 0.2.3
+Version: 0.2.5
 Author: alpha
 Author-email: 
 Requires-Python: >=3.6
@@ -23,6 +23,7 @@ Requires-Dist: requests==2.25.1
 Requires-Dist: requests-toolbelt==1.0.0
 Requires-Dist: retry==0.9.2
 Requires-Dist: dill==0.3.8
+Requires-Dist: simplejson==3.20.1
 Dynamic: author
 Dynamic: requires-dist
 Dynamic: requires-python

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/pytest_api_framework_alpha.egg-info/SOURCES.txt

@@ -19,6 +19,7 @@ framework/utils/__init__.py
 framework/utils/common.py
 framework/utils/encrypt.py
 framework/utils/log_util.py
+framework/utils/mock_util.py
 framework/utils/teams_util.py
 framework/utils/yaml_util.py
 pytest_api_framework_alpha.egg-info/PKG-INFO

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/pytest_api_framework_alpha.egg-info/requires.txt

@@ -17,3 +17,4 @@ requests==2.25.1
 requests-toolbelt==1.0.0
 retry==0.9.2
 dill==0.3.8
+simplejson==3.20.1

{pytest_api_framework_alpha-0.2.3 → pytest_api_framework_alpha-0.2.5}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pytest-api-framework-alpha",  # 包名（必须唯一）
-    version="0.2.3",
+    version="0.2.5",
     packages=find_packages(),
     author="alpha",
     author_email="",
@@ -27,6 +27,7 @@ setup(
         "requests==2.25.1",
         "requests-toolbelt==1.0.0",
         "retry==0.9.2",
-        "dill==0.3.8"
+        "dill==0.3.8",
+        "simplejson==3.20.1"
     ]
 )