WebsocketTest 1.0.15__tar.gz → 1.0.17__tar.gz

{websockettest-1.0.15 → websockettest-1.0.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: WebsocketTest
-Version: 1.0.15
+Version: 1.0.17
 Summary: websocket api autotest
 Requires-Dist: allure_python_commons==2.13.5
 Requires-Dist: numpy==2.2.4

websockettest-1.0.17/README.md

@@ -0,0 +1,57 @@
+# WebSocket 接口自动化测试工具 (WebSocketTest)
+
+#### 介绍
+
+这是一个基于 WebSocket 协议的接口自动化测试工具。它可以用于自动化测试 WebSocket 接口，确保接口的稳定性和可靠性。
+
+#### 系统要求
+
+**Python 3.10+**：确保你的系统上已经安装了 Python 3.10（推荐使用最新稳定版）。
+
+
+#### 安装步骤
+
+**1.安装 Python：**
+确保你的系统上已经安装了 Python 3.10 或更高版本。你可以从 [Python 官方网站 ](https://www.python.org/downloads/?spm=5176.28103460.0.0.40f75d27PnqPkU)下载并安装。
+**2.安装 WebSocketTest 工具：**
+
+```
+pip install WebsocketTest
+```
+
+**3.安装allure 工具：**
+```
+install-allure
+```
+**4.创建测试项目：**
+```
+ws startproject yourtestProject
+cd yourtestProject
+```
+
+**5.运行测试：**
+  在命令行中运行以下命令：
+
+```
+ 
+<!-- 安徽5.0链路aqua测试： -->
+ws test --env uat --app 0f0826ab  --service aqua --project vwa
+ws test --env live --app 0f0826ab  --service aqua --project vwa
+<!-- 安徽5.4链路sds_gateway测试： -->
+ws test --env uat --app 3d7d3ea4  --service gateway_5.4 --project vwa
+ws test --env live --app 3d7d3ea4  --service gateway_5.4 --project vwa
+<!-- 安徽5.0链路sds_gateway测试： -->
+ws test --env uat --app 0f0826ab  --service gateway_5.0 --project vwa
+ws test --env live --app 0f0826ab  --service gateway_5.0 --project vwa
+<!-- 奥迪aqua测试： -->
+ws test --env uat --app 576d9f07  --service aqua --project svm
+ws test --env live --app 576d9f07  --service aqua --project svm
+<!-- 上汽5.0链路sds_gateway测试： -->
+ws test --env uat --app 66ba7ded  --service gateway_5.0 --project svw
+ws test --env live --app 66ba7ded  --service gateway_5.0 --project svw
+
+<!-- 指定Gatewaycase -->
+ws test --env uat --app 3d7d3ea4  --service gateway_5.4 --project vwa --testcase testcase/test_all.py::TestGateway::test[case_suite0]
+<!-- 指定Aquacase -->
+ws test --env uat --app 0f0826ab  --service aqua --project vwa testcase/test_all.py::TestAqua::test[case_suite0]
+```

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest/allure_installer.py

@@ -4,6 +4,7 @@ import zipfile
 import subprocess
 import winreg
 from pathlib import Path
+import shutil
 
 # 配置参数
 ALLURE_VERSION = "2.30.0"
@@ -13,43 +14,71 @@ INSTALL_DIR = os.path.expanduser("~/.allure")  # 安装到用户目录
 ALLURE_BIN_DIR = os.path.join(INSTALL_DIR, f"allure-{ALLURE_VERSION}", "bin")
 
 def is_allure_installed():
-    """检查Allure是否在PATH中且可执行"""
+    """检查Allure是否在PATH中且版本匹配
+    Returns:
+        Tuple[bool, str]: (是否安装正确, 版本信息或错误消息)
+    """
+    # 1. 首先检查allure命令是否存在
+    allure_cmd = "allure"
+    if not shutil.which(allure_cmd):
+        return False
+    
+    # 2. 获取版本信息
     try:
-        # 同时检查版本避免找到无效安装
-        result = subprocess.run(["allure", "--version"], 
-                              check=True,
-                              stdout=subprocess.PIPE,
-                              stderr=subprocess.PIPE,
-                              text=True)
+        result = subprocess.run(
+            [allure_cmd, "--version"],
+            check=True,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            encoding="utf-8",
+            timeout=5  # 添加超时防止卡住
+        )
         installed_version = result.stdout.strip()
+        
+        # 3. 验证版本
         if ALLURE_VERSION not in installed_version:
-            print(f"⚠️ 发现不匹配的Allure版本: {installed_version} (需要 {ALLURE_VERSION})")
+            print(f"Allure版本不匹配: 已安装 {installed_version} (需要 {ALLURE_VERSION})")
             return False
+        
         return True
-    except (subprocess.CalledProcessError, FileNotFoundError):
-        return False
+        
+    except subprocess.CalledProcessError as e:
+        return False, f"Allure版本检查失败: {e.stderr}"
+    except Exception as e:
+        return False, f"检查Allure时发生意外错误: {str(e)}"
 
 def get_local_zip_path():
     """获取ZIP文件的绝对路径（兼容开发模式和正式安装）"""
-    # 尝试从包内获取
+    # 正式安装模式
     try:
-        import importlib.resources as pkg_resources
-        with pkg_resources.path(PACKAGE_NAME, LOCAL_ZIP_RELATIVE_PATH) as zip_path:
-            return str(zip_path)
-    except:
+        from importlib.resources import files
+        # Python 3.9+ 标准方式
+        resource = files(PACKAGE_NAME).joinpath(LOCAL_ZIP_RELATIVE_PATH)
+        if Path(str(resource)).exists():
+            return str(resource)
+    except Exception as e:
         pass
-    
-    # 回退方案：从当前工作目录查找
-    base_dirs = [
-        os.getcwd(),
-        os.path.dirname(os.path.abspath(__file__)),
-        sys.prefix
+
+    # 备选搜索路径（按优先级排序）
+    search_paths = [
+        # 1. 当前脚本所在目录的子目录（开发模式）
+        Path(__file__).parent / LOCAL_ZIP_RELATIVE_PATH,
+        
+        # # 2. 当前工作目录的包子目录
+        Path.cwd() / PACKAGE_NAME / LOCAL_ZIP_RELATIVE_PATH,
+        
+        # # 3. Python安装目录下的包子目录
+        Path(sys.prefix) /"Lib/site-packages" / PACKAGE_NAME / LOCAL_ZIP_RELATIVE_PATH,
+        
+        # # 4. 直接尝试相对路径（适用于打包后的exe等情况）
+        # Path(LOCAL_ZIP_RELATIVE_PATH)
     ]
-    
-    for base in base_dirs:
-        zip_path = os.path.join(base, PACKAGE_NAME, LOCAL_ZIP_RELATIVE_PATH)
-        if os.path.exists(zip_path):
-            return zip_path
+    # 回退方案：从当前工作目录查找
+
+    for _path in search_paths:
+        if os.path.exists(_path):
+            return _path
     
     raise FileNotFoundError(f"Allure ZIP文件未在任何位置找到: {LOCAL_ZIP_RELATIVE_PATH}")
 
@@ -75,13 +104,13 @@ def install_allure():
                         f.write(zip_ref.read(file))
 
         # 设置权限（特别是Linux/macOS）
-        if sys.platform != "win32":
-            os.chmod(os.path.join(ALLURE_BIN_DIR, "allure"), 0o755)
+        # if sys.platform != "win32":
+        #     os.chmod(os.path.join(ALLURE_BIN_DIR, "allure"), 0o755)
 
         # 更新PATH
-        if add_to_user_path(ALLURE_BIN_DIR):
-            # 立即在当前进程生效
-            os.environ["PATH"] = f"{ALLURE_BIN_DIR}{os.pathsep}{os.environ.get('PATH', '')}"
+        # if add_to_user_path(ALLURE_BIN_DIR):
+        #     # 立即在当前进程生效
+        #     os.environ["PATH"] = f"{ALLURE_BIN_DIR}{os.pathsep}{os.environ.get('PATH', '')}"
         
         return True
     except Exception as e:
@@ -114,17 +143,8 @@ def add_to_user_path(path):
                 shell=True,
                 check=True
             )
-        else:
-            # Linux/macOS: 修改shell配置文件
-            shell_config = os.path.expanduser("~/.bashrc")
-            if not os.path.exists(shell_config):
-                shell_config = os.path.expanduser("~/.zshrc")
-            
-            with open(shell_config, "a") as f:
-                f.write(f"\nexport PATH=\"$PATH:{path}\"\n")
-        
-        print(f"✅ 已添加PATH: {path}")
-        return True
+            print(f"✅ 已添加PATH: {path}")
+            return True
     except Exception as e:
         print(f"⚠️ 添加PATH失败: {e}\n请手动添加 {path} 到环境变量", file=sys.stderr)
         return False
@@ -144,6 +164,7 @@ def ensure_allure():
             2. 尝试手动运行: {}
             """.format(os.path.join(ALLURE_BIN_DIR, "allure" + (".bat" if sys.platform == "win32" else ""))))
             return False
+        print(f"✅ Allure {ALLURE_VERSION} 安装安装成功")
         return True
     else:
         print(f"""

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest/common/WSBaseApi.py

@@ -7,6 +7,7 @@ from pathlib import Path
 from WebsocketTest.common.Assertion import Assert
 import pytest
 import traceback
+from WebsocketTest.conftest import SHEET_NAME
 class WSBaseApi:
     def __init__(self, **kwargs):
         self.request = {}
@@ -29,11 +30,11 @@ class WSBaseApi:
         asyncio.run(self.WebSocketApi())
 
 
-
 class BaseApiTest:
     # 定义需要子类实现的属性
     TEST_TYPE = "API自动化测试"  # 默认值，可被子类覆盖
     CASE_PATH = Path.cwd().resolve().joinpath("data/case_data.xlsx")
+    
 
     @cached_property
     def API_TEST_RUNNER_CLASS(self):
@@ -51,7 +52,7 @@ class BaseApiTest:
             ) from e
     
 
-    @pytest.mark.parametrize('case_suite', gen_case_suite(CASE_PATH))
+    @pytest.mark.parametrize('case_suite', gen_case_suite(CASE_PATH,sheet_name=SHEET_NAME))
     def test(self, case_suite, setup_env):
         """测试用例执行模板"""
         try:
@@ -70,7 +71,7 @@ class BaseApiTest:
             
         except Exception as e:
             self._record_error(e)
-            raise
+            pytest.fail(f"Case failed: {str(e)}")
     
     def _record_allure_report(self, runner):
         """记录Allure测试报告"""

websockettest-1.0.17/WebsocketTest/common/test.py

@@ -0,0 +1,36 @@
+import websockets
+from functools import cached_property
+import asyncio
+from WebsocketTest.common.utils import *
+import allure
+from pathlib import Path
+from WebsocketTest.common.Assertion import Assert
+import pytest
+import traceback
+
+
+CASE_PATH = Path.cwd().resolve().joinpath("data/case_data.xlsx")
+@pytest.fixture
+def sheet_name(request):
+    """动态获取 sheet_name"""
+    service = request.config.getoption("--service")
+    project = request.config.getoption("--project")
+    return f"{project}_{service}"
+
+@pytest.fixture
+def case_suite(sheet_name):
+    """生成测试数据"""
+    return gen_case_suite(CASE_PATH, sheet_name)
+
+class BaseApiTest:
+    def test(self, case_suite, setup_env):
+        """测试用例执行模板"""
+        try:
+            params = merge_dicts(case_suite, setup_env)
+            runner = self.API_TEST_RUNNER_CLASS(**params)
+            runner.run()
+            self._record_allure_report(runner)
+            self._execute_assertions(runner, case_suite)
+        except Exception as e:
+            self._record_error(e)
+            raise

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest/common/utils.py

@@ -189,8 +189,8 @@ def append_msg_to_excel(file_path, **kwargs):
         current_function = inspect.currentframe().f_code.co_name
         print(f"An error occurred in file '{current_file}' and function '{current_function}': {e}")
 
-def read_excel(file_path):
-    df = pd.read_excel(file_path, header=0, dtype={'debug': str,'debugX': str})
+def read_excel(file_path,sheet_name):
+    df = pd.read_excel(file_path,sheet_name, header=0, dtype={'debug': str,'debugX': str})
     # 如果确实需要空字符串，先转换类型
     pd.set_option('future.no_silent_downcasting', True)
     df = df.astype(object).fillna("")  # 转换为object类型
@@ -234,9 +234,9 @@ def merge_dicts(a: dict, b: dict) -> dict:
         for k in set(a) | set(b)
     }
 
-def gen_case_suite(file_path):
+def gen_case_suite(file_path,sheet_name):
     # 生成测试用例套件
-    data_dict = read_excel(file_path)
+    data_dict = read_excel(file_path,sheet_name)
     case_suite = []
     for case in data_dict:
         case_suite.append(case)

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest/conftest.py

@@ -2,6 +2,9 @@ import pytest
 import yaml
 from pathlib import Path
 
+SHEET_NAME = "0"  # 默认值
+
+
 # 设置环境信息的 fixture
 @pytest.fixture(scope="session")
 def setup_env(request):
@@ -30,7 +33,6 @@ def setup_env(request):
             'service': service,
             'project': project})
     return environments
-
 # 添加命令行选项
 def pytest_addoption(parser):
     parser.addoption(
@@ -56,4 +58,12 @@ def pytest_addoption(parser):
         action="store",
         default="vwa",
         help="Project name is required"
-    )
+    )
+
+def pytest_configure(config):
+    project = config.getoption("--project")
+    service = config.getoption("--service")
+    """在测试初始化阶段解析参数并全局缓存"""
+    global SHEET_NAME
+    SHEET_NAME = f"{project}-{service}"
+    

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest/run_tests.py

@@ -35,7 +35,7 @@ class AllureManager:
                 start_new_session=True
             )
             
-            time.sleep(3)  # 等待服务启动
+            time.sleep(1)  # 等待服务启动
             return True
         except Exception as e:
             logger.error(f"Failed to start Allure server: {e}")
@@ -81,11 +81,11 @@ class TestRunner:
             "--service", self.args.service,
             "--project", self.args.project,
             "--alluredir", self.allure_results
+            ,"testcase\\test_all.py"
         ]
            # 添加可选测试路径（如果存在）
         if hasattr(self.args, 'testcase') and self.args.testcase:
-            cmd.insert(1, self.args.testcase)
-        print(f"run_pytest_tests Executing: {' '.join(cmd)}")
+            cmd.insert(1, self.args.testcase) 
         try:
             # logger.info(f"run_pytest_tests Executing: {' '.join(cmd)}")
             subprocess.run(cmd, check=True, timeout=3600)
@@ -115,8 +115,7 @@ class TestRunner:
                 webbrowser.open(f"http://localhost:{self.args.port}")
         else:
             logger.info("Starting new Allure server...")
-            if self.allure_manager.start_server(self.args.report_dir):
-                webbrowser.open(f"http://localhost:{self.args.port}")
+            self.allure_manager.start_server(self.args.report_dir)
     def run(self):
         # 1. 运行测试
         self.run_pytest_tests()

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest/templates/basic//346/211/247/350/241/214/345/221/275/344/273/244.txt

@@ -9,29 +9,33 @@ ws test --env live --app 3d7d3ea4  --service gateway_5.4 --project vwa
 ws test --env uat --app 0f0826ab  --service gateway_5.0 --project vwa
 ws test --env live --app 0f0826ab  --service gateway_5.0 --project vwa
 奥迪aqua测试：
-ws test --env uat --app 576d9f07  --service aqua --project aodi
-ws test --env live --app 576d9f07  --service aqua --project aodi
+ws test --env uat --app 576d9f07  --service aqua --project svm
+ws test --env live --app 576d9f07  --service aqua --project svm
 上汽5.0链路sds_gateway测试：
 ws test --env uat --app 66ba7ded  --service gateway_5.0 --project svw
 ws test --env live --app 66ba7ded  --service gateway_5.0 --project svw
 
-  ws startproject d
-  python run_tests.py --env live --app 0f0826ab  --service gateway_5.0 --project vwa
-   pip install -e . 
-    ws test --env uat --app 3d7d3ea4  --service gateway_5.4 --project vwa --testcase testcase/test_all.py::TestGateway::test[case_suite0]
+ws startproject d
+python run_tests.py --env live --app 0f0826ab  --service gateway_5.0 --project vwa
+pip install -e . 
+指定Gatewaycase
+ws test --env uat --app 3d7d3ea4  --service gateway_5.4 --project vwa --testcase testcase/test_all.py::TestGateway::test[case_suite0]
+指定Aquacase
+testcase/test_all.py::TestAqua::test[case_suite0]
 
 关闭allure进程
 netstat -ano | findstr "8883"
-taskkill /PID 25804 /F
+taskkill  /F /PID 25804
 
-卸载
-pip uninstall WebsocketTest
-安装
-pip install WebsocketTest==1.0.14
 打包
 python setup.py sdist bdist_wheel
 上传
 twine upload --repository pypi setup_temp/dist/*
+卸载
+pip uninstall WebsocketTest
+安装
+pip install WebsocketTest==1.0.14
+
 
 
 安装虚拟环境

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: WebsocketTest
-Version: 1.0.15
+Version: 1.0.17
 Summary: websocket api autotest
 Requires-Dist: allure_python_commons==2.13.5
 Requires-Dist: numpy==2.2.4

{websockettest-1.0.15 → websockettest-1.0.17}/WebsocketTest.egg-info/SOURCES.txt

@@ -24,11 +24,13 @@ WebsocketTest/common/WSBaseApi.py
 WebsocketTest/common/__init__.py
 WebsocketTest/common/assertUtils.py
 WebsocketTest/common/logger.py
+WebsocketTest/common/test.py
 WebsocketTest/common/utils.py
 WebsocketTest/libs/allure-2.30.0.zip
 WebsocketTest/templates/basic/conftest.py
 WebsocketTest/templates/basic/pytest.ini
 WebsocketTest/templates/basic/执行命令.txt
+WebsocketTest/templates/basic/__pycache__/conftest.cpython-310-pytest-8.2.2.pyc
 WebsocketTest/templates/basic/config/svw/live.yml
 WebsocketTest/templates/basic/config/svw/uat.yml
 WebsocketTest/templates/basic/config/svw/aqua/uat.yml

{websockettest-1.0.15 → websockettest-1.0.17}/setup.py

@@ -31,7 +31,7 @@ setup(
         'bdist_wheel': CustomBdistWheel
     },
     name="WebsocketTest",
-    version="1.0.15",
+    version="1.0.17",
     packages=find_packages(exclude=[
         "WebsocketTest.allure_report", 
         "WebsocketTest.logs", 

websockettest-1.0.15/README.md

@@ -1,64 +0,0 @@
-# WebSocket 接口自动化测试工具 (WS_API_TEST)
-
-#### 介绍
-
-这是一个基于 WebSocket 协议的接口自动化测试工具。它可以用于自动化测试 WebSocket 接口，确保接口的稳定性和可靠性。
-
-#### 系统要求
-
-**Python 3.10+**：确保你的系统上已经安装了 Python 3.10（推荐使用最新稳定版）。
-**项目依赖**：项目需要一些第三方库，可以通过 requirements.txt 文件安装。
-
-#### 安装步骤
-
-**1.安装 Python：**
-确保你的系统上已经安装了 Python 3.10 或更高版本。你可以从 [Python 官方网站 ](https://www.python.org/downloads/?spm=5176.28103460.0.0.40f75d27PnqPkU)下载并安装。
-**2.克隆项目：**
-使用 Git 克隆项目到你的本地机器。
-
-```
-git clone https://code.iflytek.com/ZNQC_AUTO_AI/python_scripts.git
-cd python_scripts
-git checkout WS_API_TEST
-```
-
-**3.安装项目依赖：**
-使用 pip 安装项目所需的依赖库。
-
-```
-pip install -r requirements.txt
-```
-**4.运行项目：**
-你可以通过以下两种方式之一来运行项目：
-
-* **使用命令行**：
-  在命令行中运行以下命令：
-
-```
-python run_tests.py --env uat --app 3d7d3ea4  --service gateway_5.4 --project vwa
-```
-
-* **使用批处理脚本**：
-  双击 run_tests.bat 文件来运行项目。
-
-#### 项目结构
-├─allure_report
-├─allure_results
-├─build
-├─common
-├─config
-├─data
-├─dist
-├─logs
-├─testcase
-
-* **文件说明**:
-**allure_report**：存放 Allure 生成的 HTML 报告，用于展示测试结果。
-**allure_results**：存放 Allure 生成的测试结果数据文件，如 result-12345.xml。
-**build**：存放构建过程中生成的临时文件
-**common**：存放通用的工具类
-**config**：存放环境配置文件
-**data**：存放测试数据文件
-**dist**：存放最终发布的测试包
-**logs**：存放运行日志文件
-**testcase**：存放测试用例脚本