WebsocketTest 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

WebsocketTest/__init__.py

@@ -0,0 +1,21 @@
+__version__ = "1.1.0"
+__description__ = "One-stop solution for AUTO testing."
+
+# import firstly for monkey patch if needed
+# from pastor.ext.locust import main_locusts
+# from pastor.parser import parse_parameters as Parameters
+# from pastor.runner import Pastor
+# from pastor.testcase import Config, Step, RunRequest, RunTestCase, RunLocation
+
+__all__ = [
+    "__version__",
+    "__description__"
+    # ,
+    # "Pastor",
+    # "Config",
+    # "Step",
+    # "RunLocation",
+    # "RunRequest",
+    # "RunTestCase",
+    # "Parameters",
+]

WebsocketTest/cli.py

@@ -0,0 +1,56 @@
+import argparse
+import enum
+import sys
+import pytest
+
+def main_run(pytest_args: list) -> enum.IntEnum:
+    """Run pytest with given arguments
+    
+    Args:
+        pytest_args: List of arguments to pass to pytest
+        
+    Returns:
+        pytest.ExitCode enum value
+    """
+    print("Running pytest with args:", pytest_args)
+    return pytest.main(pytest_args)
+
+def main():
+    """API test: parse command line options and run commands."""
+    # 主解析器
+    parser = argparse.ArgumentParser(description="Run tests with Allure reporting")
+    
+    # 添加必选参数
+    parser.add_argument("--env", required=True, help="Test environment")
+    parser.add_argument("--app", required=True, help="Application ID")
+    parser.add_argument("--service", required=True, help="Service name")
+    parser.add_argument("--project", required=True, help="Project name")
+    
+    # 解析参数，分离已知参数和要传递给pytest的参数
+    args, pytest_args = parser.parse_known_args()
+    
+    # 打印调试信息
+    print("Tool arguments:")
+    print(vars(args))
+    print("Pytest arguments:", pytest_args)
+    
+    # 将工具参数转换为pytest可用的格式
+    pytest_args.extend([
+        f"--env={args.env}",
+        f"--app={args.app}",
+        f"--service={args.service}",
+        f"--project={args.project}"
+    ])
+    
+    # 运行测试
+    return_code = main_run(pytest_args)
+    
+    # 返回退出码
+    sys.exit(return_code)
+def main_run_alias():
+    """ command alias
+        prun = pastor run
+    """
+    main()
+if __name__ == "__main__":
+    main()

WebsocketTest/conftest.py

@@ -0,0 +1,63 @@
+import pytest
+import yaml
+from pathlib import Path
+
+# 设置环境信息的 fixture
+@pytest.fixture(scope="session")
+def setup_env(request):
+    # 获取命令行参数
+    service = request.config.getoption("service")
+    project = request.config.getoption("project")
+    env = request.config.getoption("env")
+    app_id = request.config.getoption("app")
+     # 获取当前工作目录，并解析为绝对路径
+    current_working_directory = Path.cwd().resolve()
+    config_dir = current_working_directory.joinpath('config', project)
+    
+    # Load base configuration
+    with open(config_dir.joinpath(f'{env}.yml')) as base_file:
+        config = yaml.safe_load(base_file)
+    
+    # Load environment-specific configuration and merge into the base config
+    with open(config_dir.joinpath(service, f'{env}.yml')) as env_file:
+        env_config = yaml.safe_load(env_file)
+    # 合并配置
+    environments = config.get('environments', {})
+    environments.update(env_config.get('environments', {}))
+    # 添加运行时参数
+    environments.update({'appId': app_id,
+            'env_name': env,
+            'service': service,
+            'project': project})
+    return environments
+
+# 添加命令行选项
+def pytest_addoption(parser):
+    parser.addoption(
+        "--env",
+        action="store",
+        default="uat",
+        help="Environment to use (default: uat)"
+    )
+    parser.addoption(
+        "--app",
+        action="store",
+        default="0f0826ab",
+        help="App to use (default: 0f0826ab)"
+    )
+    parser.addoption(
+        "--service",
+        action="store",
+        default="0f0826ab",
+        help="Service to use (default: aqua)"
+    )
+    parser.addoption(
+        "--project",
+        action="store",
+        default="vwa",
+        help="Project name is required"
+    )
+
+
+
+

WebsocketTest/run_tests.py

@@ -0,0 +1,147 @@
+import subprocess,sys
+import argparse
+import shutil
+from common.logger import *
+import time
+import webbrowser
+from urllib.request import urlopen
+from urllib.error import URLError
+
+logger = logging.getLogger('run_tests')
+
+class AllureManager:
+    def __init__(self, port: int = 8883):
+        self.port = port
+        self.allure_path = shutil.which("allure")
+        if not self.allure_path:
+            raise RuntimeError("Allure command line tool not found in PATH")
+
+    def is_server_running(self) -> bool:
+        """检查Allure服务是否已在运行"""
+        try:
+            with urlopen(f"http://localhost:{self.port}") as response:
+                return response.status == 200
+        except URLError:
+            return False
+
+    def start_server(self, report_dir: str) -> bool:
+        """启动Allure服务"""
+        try:
+            cmd = [self.allure_path, "open", report_dir, "-p", str(self.port)]
+            logger.info(f"start_server Executing: {' '.join(cmd)}")
+            subprocess.Popen(
+                cmd,
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                start_new_session=True
+            )
+            
+            time.sleep(3)  # 等待服务启动
+            return True
+        except Exception as e:
+            logger.error(f"Failed to start Allure server: {e}")
+            return False
+
+    def refresh_browser_tab(self) -> bool:
+        """尝试刷新已打开的Allure标签页"""
+        url = f"http://localhost:{self.port}"
+        
+        # 方法1: 使用JavaScript刷新（需要浏览器支持）
+        try:
+            webbrowser.open_new_tab("javascript:location.reload(true);")
+            return True
+        except Exception as e:
+            logger.warning(f"JavaScript refresh failed: {e}")
+        
+        # 方法2: 使用webbrowser直接打开（会聚焦到已有标签页）
+        try:
+            browser = webbrowser.get() # 获取系统默认浏览器控制器
+            if hasattr(browser, 'open_new_tab'):
+                browser.open_new_tab(url)  # 大多数浏览器会聚焦到已有标签页
+                return True
+        except Exception as e:
+            logger.error(f"Browser refresh failed: {e}")
+        
+        return False
+
+class TestRunner:
+    def __init__(self, args):
+        """直接存储args对象"""
+        self.args = args
+        self.allure_manager = AllureManager(self.args.port)
+      
+    def run_pytest_tests(self) -> bool:
+        """执行pytest测试"""
+        case = self.args.service.split('_')[0]
+        cmd = [
+            "pytest",
+            "-m", case,
+            "--env", self.args.env,
+            "--app", self.args.app,
+            "--service", self.args.service,
+            "--project", self.args.project,
+            "--alluredir", "./allure_results"
+        ]
+        try:
+            logger.info(f"run_pytest_tests Executing: {' '.join(cmd)}")
+            subprocess.run(cmd, check=True, timeout=3600)
+        except subprocess.CalledProcessError as e:
+            logger.error(f"Tests failed with exit code {e.returncode}")
+        except subprocess.TimeoutExpired:
+            logger.error("Test execution timed out after 1 hour")
+
+    def generate_allure_report(self) -> bool:
+        """生成Allure报告"""
+        try:
+            cmd = [self.allure_manager.allure_path, "generate", "./allure_results", "-o", self.args.report_dir, "--clean"]
+            subprocess.run(
+                cmd,
+                check=True,
+                timeout=300
+            )
+            logger.info(f"generate_allure_report Executing: {' '.join(cmd)}")
+        except subprocess.CalledProcessError as e:
+            logger.error(f"Report generation failed: {e}")
+    def _handle_allure_report(self) -> bool:
+        """Handle Allure report serving and browser opening."""
+        if  self.allure_manager.is_server_running():
+            logger.info("Refreshing existing Allure report tab...")
+            if not self.allure_manager.refresh_browser_tab():
+                logger.info("Opening new report...")
+                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}")
+        
+    def run(self):
+        # 1. 运行测试
+        self.run_pytest_tests()
+
+        # 2. 生成报告数据
+        self.generate_allure_report()
+
+        # 3. 启动Allure服务
+        self._handle_allure_report()
+        logger.info(f"http://localhost:{self.args.port}")
+        return 0
+def main():
+    try:
+        parser = argparse.ArgumentParser(description="Run tests with Allure reporting")
+        parser.add_argument("--env", required=True, help="Test environment")
+        parser.add_argument("--app", required=True, help="Application ID")
+        parser.add_argument("--service", required=True, help="Service name")
+        parser.add_argument("--project", required=True, help="Project name")
+        parser.add_argument("--port", type=int, default=8883, help="Allure report port")
+        parser.add_argument("--report-dir", default="allure_report", help="Allure report directory")
+        args = parser.parse_args()
+        test_runner = TestRunner(args)
+        exit(test_runner.run())
+    except KeyboardInterrupt:
+        logger.info("Process interrupted by user")
+        exit(1)
+    except Exception as e:
+        logger.error(f"Unexpected error: {e}", exc_info=True)
+        exit(1) 
+if __name__ == "__main__":
+     main()

{WebsocketTest-1.0.1.dist-info → WebsocketTest-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: WebsocketTest
-Version: 1.0.1
+Version: 1.0.3
 Summary: websocket api autotest
 Author: chencheng
 Requires-Python: >=3.10

WebsocketTest-1.0.3.dist-info/RECORD

@@ -0,0 +1,21 @@
+WebsocketTest/__init__.py,sha256=u71SAVmbgsyp0K21kilo7pIDgeyxsaHAi93clC0OIPQ,556
+WebsocketTest/cli.py,sha256=Y8lQ_V5blKxNg-Q9PeuPu7zasMNDPmry9g4KXnliDfk,1636
+WebsocketTest/conftest.py,sha256=y2_2961mhCYSRMp5X9xLq3nOmRzaTSgD3GDdrlTqxn4,1885
+WebsocketTest/run_tests.py,sha256=cRKvcQGBZG-1XkeoSMeCU8TI76FunrS2jkx6COx91_o,5776
+WebsocketTest/caseScript/Aqua.py,sha256=zr1mURYGikzzJDsx2lwdH9m_ENlJzH3vDvepsfwhfdg,7405
+WebsocketTest/caseScript/Gateway.py,sha256=D_oVhhKGFA3FdFblg_YmUj2HhlPNI3kht9LqHnSuNZA,11723
+WebsocketTest/caseScript/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+WebsocketTest/common/Assertion.py,sha256=xMesYiZY6FSf7_Zf6BIXGl9a2Prx-blRYGK2zKQy81M,5215
+WebsocketTest/common/WSBaseApi.py,sha256=-zDpMsaF4ugVKUpzDwbs6l9DBvlh2uw1oJGY64fk6fY,3982
+WebsocketTest/common/WebSocketApi.py,sha256=JpgX1qAA8skh_t9hLDmgEx6KugewF284u2cjrQkbLnI,2743
+WebsocketTest/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+WebsocketTest/common/assertUtils.py,sha256=xJvjm0QDIxUHIZm96-2sKkDUWzPQHaOJB9x9eSLlIoU,4699
+WebsocketTest/common/logger.py,sha256=B7jjPd5miAYorveHUEGW8xKxUNlJzKlVug0rKWRK3p0,783
+WebsocketTest/common/utils.py,sha256=5NrU_cXzpzTYSI38LZe4IjJe3HxNB64rtrXZcf3Vixs,8706
+WebsocketTest/testcase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+WebsocketTest/testcase/test_all.py,sha256=mdfivt5vmfcn_4LZVcQ1KTH0n69gKxd_Id-DCapGlu4,198
+WebsocketTest-1.0.3.dist-info/METADATA,sha256=TOfnxrmuLy4hqPkXk_9d5w-94RuyYqC4r3ecSH0X-Sg,378
+WebsocketTest-1.0.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+WebsocketTest-1.0.3.dist-info/entry_points.txt,sha256=7ppZml2d-BVAurDYhw301dOBU8wDgQrtTNlkJNGSc_Q,59
+WebsocketTest-1.0.3.dist-info/top_level.txt,sha256=2iF1gZSbXLjVFOe5ZKQiCLC1FzAwhcQUM88yGi-vrCU,14
+WebsocketTest-1.0.3.dist-info/RECORD,,

WebsocketTest-1.0.3.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+wsrun = WebsocketTest.cli:main_run_alias

WebsocketTest-1.0.3.dist-info/top_level.txt

@@ -0,0 +1 @@
+WebsocketTest

WebsocketTest-1.0.1.dist-info/RECORD

@@ -1,17 +0,0 @@
-caseScript/Aqua.py,sha256=zr1mURYGikzzJDsx2lwdH9m_ENlJzH3vDvepsfwhfdg,7405
-caseScript/Gateway.py,sha256=D_oVhhKGFA3FdFblg_YmUj2HhlPNI3kht9LqHnSuNZA,11723
-caseScript/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-common/Assertion.py,sha256=xMesYiZY6FSf7_Zf6BIXGl9a2Prx-blRYGK2zKQy81M,5215
-common/WSBaseApi.py,sha256=-zDpMsaF4ugVKUpzDwbs6l9DBvlh2uw1oJGY64fk6fY,3982
-common/WebSocketApi.py,sha256=JpgX1qAA8skh_t9hLDmgEx6KugewF284u2cjrQkbLnI,2743
-common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-common/assertUtils.py,sha256=xJvjm0QDIxUHIZm96-2sKkDUWzPQHaOJB9x9eSLlIoU,4699
-common/logger.py,sha256=B7jjPd5miAYorveHUEGW8xKxUNlJzKlVug0rKWRK3p0,783
-common/utils.py,sha256=5NrU_cXzpzTYSI38LZe4IjJe3HxNB64rtrXZcf3Vixs,8706
-testcase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-testcase/test_all.py,sha256=mdfivt5vmfcn_4LZVcQ1KTH0n69gKxd_Id-DCapGlu4,198
-WebsocketTest-1.0.1.dist-info/METADATA,sha256=Vr1be7OlsFVWKE1DMIgKEZBCFAz93S8-GBLrSCney68,378
-WebsocketTest-1.0.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-WebsocketTest-1.0.1.dist-info/entry_points.txt,sha256=9TXEzvfAj1QvYGZKFgi09VFqBnizLmJ8HjVsHhhmenk,55
-WebsocketTest-1.0.1.dist-info/top_level.txt,sha256=KmqByBRxmIHZiIEEGQnX5_C3urP-43-6TCFUhM3aMKk,27
-WebsocketTest-1.0.1.dist-info/RECORD,,

WebsocketTest-1.0.1.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-wsrun = WebsocketTest.run_tests:main

WebsocketTest-1.0.1.dist-info/top_level.txt

@@ -1,3 +0,0 @@
-caseScript
-common
-testcase