(2023-05-13) 1. 封装Python脚本：使用pymysql+sshtunnel，支持通过SSH隧道方式链接mysql数据库 2. 调整用例参数，支持数据库断言。可以通过assert_sql用例参数进行数据库断言

2023-05-18 08:52:20 +08:00
parent 950675b751
commit 476420d3f4
9 changed files with 345 additions and 94 deletions
@@ -1,47 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Version: Python 3.9
-# @Time    : 2023/1/9 16:41
-# @Author  : chenyinhua
-# @File    : test_login.py
-# @Software: PyCharm
-# @Desc: python脚本编写的测试用例文件
-
-
-import pytest
-import os
-from common_utils.yaml_handle import HandleYaml
-from config.project_path import DATA_DIR
-from case_utils.requests_handle import BaseRequest
-from case_utils.assert_handle import assert_result
-from loguru import logger
-from case_utils.request_data_handle import RequestPreDataHandle
-from config.global_vars import GLOBAL_VARS
-
-# 读取用例数据
-cases = HandleYaml(filename=os.path.join(DATA_DIR, "test_login.yaml")).read_yaml
-
-
-@pytest.mark.test_login
-class TestLogin:
-    """
-    登录模块
-    """
-
-    @pytest.mark.parametrize("case", cases)
-    def test_login(self, case):
-        logger.info("-----------------------------START-开始执行用例-----------------------------")
-        logger.debug(f"当前执行的用例数据:{case}")
-        # 将用例标题title作为全局变量，方便后续写入到测试报告report.description中
-        GLOBAL_VARS["title"] = case.get("title", "")
-        if case.get("run", None):
-            # 获取处理完成的用例数据
-            case_data = RequestPreDataHandle(case).request_data_handle()
-            # 发送请求
-            response = BaseRequest.send_request(case_data)
-            # 进行断言
-            assert_result(response, case_data["expected"])
-        else:
-            reason = f"标记了该用例为false，不执行\\n"
-            logger.warning(f"{reason}")
-            pytest.skip(reason)
-        logger.info("-----------------------------END-用例执行结束-----------------------------")
@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+# @Version: Python 3.9
+# @Time    : 2023/1/9 16:41
+# @Author  : chenyinhua
+# @File    : test_login_demo.py
+# @Software: PyCharm
+# @Desc: python脚本编写的测试用例文件
+
+
+import pytest
+import os
+from common_utils.yaml_handle import HandleYaml
+from config.settings import DATA_DIR
+from common_utils.base_request import BaseRequest
+from case_utils.assert_handle import assert_response, assert_sql
+from loguru import logger
+from case_utils.request_data_handle import RequestPreDataHandle, after_extract, case_data_replace, eval_data_process
+from pytest_html import extras  # 往pytest-html报告中填写额外的内容
+from common_utils.func_handle import add_docstring
+
+# 读取用例数据
+cases = HandleYaml(filename=os.path.join(DATA_DIR, "test_login_demo.yaml")).read_yaml
+
+
+@pytest.mark.test_login_demo
+@pytest.mark.parametrize("case", cases)
+def test_login_demo(case, extra, request):
+    logger.info("-----------------------------START-开始执行用例-----------------------------")
+    logger.debug(f"当前执行的用例数据:{case}")
+    try:
+        # 获取命令行参数，判断当前处于哪个环境
+        env = request.config.getoption("--env")
+        # 给当前测试方法添加文档注释
+        add_docstring(case.get("title", ""))(test_login)
+        if case.get("run", None):
+            # 处理请求前的用例数据
+            case_data = RequestPreDataHandle(case).request_data_handle()
+            # 将用例数据显示在pytest-html报告中
+            extra.append(extras.text(str(case_data), name="用例数据"))
+            # 发送请求
+            response = BaseRequest.send_request(case_data)
+            # 将响应数据显示在pytest-html报告中
+            extra.append(extras.text(str(response.text), name="响应数据"))
+            # 请求后，提取后置参数作为全局变量
+            after_extract(response, case_data["extract"])
+            # 从全局变量中获取最新值，替换数据库断言中的参数
+            case_data["assert_sql"] = eval_data_process(case_data_replace(case_data["assert_sql"]))
+            # 进行响应断言
+            assert_response(response, case_data["assert_response"])
+            # 进行数据库断言
+            assert_sql(env, case_data["assert_sql"])
+        else:
+            reason = f"标记了该用例为false，不执行\\n"
+            logger.warning(f"{reason}")
+            pytest.skip(reason)
+    except Exception as e:
+        logger.error(f"用例执行过程中报错：{e}")
+        raise e
+    finally:
+        logger.info("-----------------------------END-用例执行结束-----------------------------")