pytest-api-framework-alpha 0.3.2__tar.gz → 0.3.4__tar.gz

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-api-framework-alpha
-Version: 0.3.2
+Version: 0.3.4
 Author: alpha
 Author-email: 
 Requires-Python: >=3.6

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/framework/base_class.py

@@ -18,7 +18,8 @@ from framework.utils.date_util import DateUtil
 from framework.utils.common import snake_to_pascal, SingletonFaker
 from framework.global_attribute import GlobalAttribute, _FRAMEWORK_CONTEXT, CONTEXT
 from framework.exceptions import ValidateException, RenderException, RequestException, GetAccountError, GetAppHttpError
-from framework.utils.mock_util import get_customized_kytmock, set_customized_kytmock, PaymentType, RiskLevel, IsDelayed
+from framework.utils.mock_util import get_customized_kytmock, set_customized_kytmock, PaymentType, RiskLevel, IsDelayed, \
+    mock_mq
 
 from handlers.extend_base_test_case_attr import ExtendBaseTestCase
 from config.settings import UNAUTHORIZED_CODE, FAKER_LANGUAGE
@@ -190,3 +191,12 @@ class BaseTestCase(ExtendBaseTestCase):
 
         """
         return set_customized_kytmock(self.context.get("env"), request_id, payment_type, risk_level, is_delayed)
+
+    def mock_mq_message(self,target_exchange,exchange_type,message,routing_key:str=None):
+        """
+        target_exchange:目标交换机名称
+        exchange_type：交换机类型，枚举：fanout，direct，topic，header
+        message：消息体，json格式
+        routing_key: 路由关键字，默认为null
+        """
+        return  mock_mq(self.context.get("env"),target_exchange,exchange_type,message,routing_key)

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/framework/conftest.py

@@ -365,6 +365,7 @@ def pytest_generate_tests(metafunc):
         if mark_obj.name not in ["skip", "skipif", "xfail", "usefixtures", "filterwarnings",
                                  "parametrize", "order", "timeout", "django_db", "asyncio"] and not mark_obj.args:
             marks.append(mark_obj.name)
+    marks = [item.lower() for item in marks]
     # 校验标签
     is_subset, diff = subset_and_diff(set(marks), set(MARK_LIST))
 
@@ -614,6 +615,8 @@ def sort(case_items):
 def pytest_collection_finish(session):
     """获取最终排序后的 items 列表"""
     logger.info(f"共收集到 {len(session.items)} 个测试用例")
+    if not session.items:
+        pytest.exit("未收集到用例")
     # 过滤掉item名称是test_setup或test_teardown的
     session.items = [item for item in session.items if item.name not in ["test_setup", "test_teardown"]]
 

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/framework/script.py

@@ -16,3 +16,9 @@ class BaseScript(__BaseScript):
 
     def default_app(self, app):
         return app or self.app
+
+    def context_set(self, key, value):
+        self.context.set(app=self.app, key=key, value=value)
+
+    def context_get(self, key):
+        return self.context.get(app=self.app, key=key)

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/framework/utils/mock_util.py

@@ -136,6 +136,45 @@ def get_customized_kytmock(env,request_id):
 
     return None
 
+def mock_mq(env,target_exchange,exchange_type,message,routing_key):
+    # 获取环境配置
+    env_config = get_environment_config(env)
+    method = "mock_mq"
+    url = get_full_api_url(env,method)
+    # 参数验证
+    if not isinstance(target_exchange, str) or not target_exchange:
+        raise ValueError("target_exchange必须是非空字符串")
+    if not isinstance(exchange_type,str ) or not exchange_type:
+        raise ValueError("exchange_type 必须是非空字符串")
+    if not isinstance(message,str ) or not message:
+        raise ValueError("message 必须是非空字符串")
+    # 构建请求参数
+    payload = {
+            "targetExchange": target_exchange,
+            "targetExchangeType": exchange_type,
+            "message": message,
+            "targetRoutingKey": routing_key
+        }
+
+    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
+
 if __name__ == "__main__":
     # # 1. 调用测试环境的入金接口
     # test_deposit = set_customized_kytmock(
@@ -155,5 +194,11 @@ if __name__ == "__main__":
     # )
     # print(f"开发环境出金响应: {dev_withdrawal}")
 
-    get_content = get_customized_kytmock("dev","A0101585")
-    print(f"查询结果：{get_content}")
+    # get_content = get_customized_kytmock("dev","A0101585")
+    # print(f"查询结果：{get_content}")
+    exchange_name = "camp.gateway.backbone.mock.ex.order-status"
+    exchange_type = "fanout"
+    message = "{ \"avgFillPrice\": 440.23, \"filled\": 11.0000000000000000, \"orderId\": 409, \"remaining\": 0.0000000000000000, \"status\": \"Filled\" }"
+    result = mock_mq("dev",exchange_name,exchange_type,message)
+    print(result)
+

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/pytest_api_framework_alpha.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-api-framework-alpha
-Version: 0.3.2
+Version: 0.3.4
 Author: alpha
 Author-email: 
 Requires-Python: >=3.6

{pytest_api_framework_alpha-0.3.2 → pytest_api_framework_alpha-0.3.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pytest-api-framework-alpha",  # 包名（必须唯一）
-    version="0.3.2",
+    version="0.3.4",
     packages=find_packages(),
     author="alpha",
     author_email="",