|
| 1 | +from PyQt6.QtCore import QObject, QProcess, pyqtSignal, QProcessEnvironment, QTimer |
| 2 | +from models.run_config import RunConfig |
| 3 | +import os |
| 4 | + |
| 5 | +# MAVEN_EXECUTABLE_PATH = "C:\\Users\\meetd\\Downloads\\apache-maven-3.9.10-bin\\apache-maven-3.9.10\\bin\\mvn.cmd" |
| 6 | +MAVEN_EXECUTABLE_PATH = "mvn" |
| 7 | + |
| 8 | +class TestRunner(QObject): |
| 9 | + process_started = pyqtSignal() |
| 10 | + process_finished = pyqtSignal(int) |
| 11 | + output_received = pyqtSignal(str) |
| 12 | + error_received = pyqtSignal(str) |
| 13 | + |
| 14 | + def __init__(self, e2e_dir: str): |
| 15 | + super().__init__() |
| 16 | + self.e2e_dir = e2e_dir |
| 17 | + self.process = None |
| 18 | + self.is_running = False |
| 19 | + self._force_kill_timer = QTimer() |
| 20 | + self._force_kill_timer.setSingleShot(True) |
| 21 | + self._force_kill_timer.timeout.connect(self._force_kill_process) |
| 22 | + |
| 23 | + def run_test(self, config: RunConfig): |
| 24 | + if self.is_running: |
| 25 | + self.output_received.emit("Test already running! Stop it first.") |
| 26 | + return |
| 27 | + |
| 28 | + # Verify directory and pom.xml |
| 29 | + if not os.path.exists(self.e2e_dir): |
| 30 | + self.error_received.emit(f"E2E directory does not exist: {self.e2e_dir}") |
| 31 | + return |
| 32 | + |
| 33 | + pom_file = os.path.join(self.e2e_dir, "pom.xml") |
| 34 | + if not os.path.exists(pom_file): |
| 35 | + self.error_received.emit(f"pom.xml not found in E2E directory: {self.e2e_dir}") |
| 36 | + return |
| 37 | + |
| 38 | + args = self._build_command_args(config) |
| 39 | + |
| 40 | + self.process = QProcess() |
| 41 | + self.process.setWorkingDirectory(self.e2e_dir) |
| 42 | + |
| 43 | + # Use system environment as-is (your PATH already has everything needed) |
| 44 | + env = QProcessEnvironment.systemEnvironment() |
| 45 | + |
| 46 | + # Just set JAVA_HOME since it's missing |
| 47 | + if not env.contains("JAVA_HOME"): |
| 48 | + env.insert("JAVA_HOME", r"C:\Program Files\Java\jdk-21") |
| 49 | + |
| 50 | + self.process.setProcessEnvironment(env) |
| 51 | + |
| 52 | + self.process.readyReadStandardOutput.connect(self._on_stdout) |
| 53 | + self.process.readyReadStandardError.connect(self._on_stderr) |
| 54 | + self.process.finished.connect(self._on_finished) |
| 55 | + self.process.errorOccurred.connect(self._on_error) |
| 56 | + |
| 57 | + self.is_running = True |
| 58 | + self.process_started.emit() |
| 59 | + self.output_received.emit(f"Starting test: {config.project_name}/{config.module_name}") |
| 60 | + |
| 61 | + # Use mvn.cmd explicitly |
| 62 | + self.process.start("mvn.cmd", args) |
| 63 | + |
| 64 | + def _build_command_args(self, config: RunConfig) -> list: |
| 65 | + exec_args = f"{config.project_name} {config.base_url} {config.browser} {config.video_option} {config.module_name} {config.wait_time}" |
| 66 | + |
| 67 | + return [ |
| 68 | + "clean", |
| 69 | + "compile", |
| 70 | + "exec:java", |
| 71 | + f"-Dexec.mainClass=com.startAuto.StartAuto", |
| 72 | + f"-Dexec.args={exec_args}" |
| 73 | + ] |
| 74 | + |
| 75 | + def stop_test(self): |
| 76 | + if not self.is_running or not self.process: |
| 77 | + return |
| 78 | + |
| 79 | + self.output_received.emit("Stopping test execution...") |
| 80 | + self.process.terminate() |
| 81 | + |
| 82 | + self._force_kill_timer.start(3000) |
| 83 | + |
| 84 | + def _force_kill_process(self): |
| 85 | + if self.is_running and self.process and self.process.state() == QProcess.ProcessState.Running: |
| 86 | + self.process.kill() |
| 87 | + self.output_received.emit("Force killed process") |
| 88 | + self.is_running = False |
| 89 | + self.output_received.emit("Test stopped") |
| 90 | + |
| 91 | + def _on_stdout(self): |
| 92 | + if self.process: |
| 93 | + data = self.process.readAllStandardOutput().data().decode() |
| 94 | + self.output_received.emit(data) |
| 95 | + |
| 96 | + def _on_stderr(self): |
| 97 | + if self.process: |
| 98 | + data = self.process.readAllStandardError().data().decode() |
| 99 | + self.error_received.emit(f"❌ {data}") |
| 100 | + |
| 101 | + def _on_finished(self, exit_code): |
| 102 | + self.is_running = False |
| 103 | + self._force_kill_timer.stop() |
| 104 | + if exit_code == 0: |
| 105 | + self.output_received.emit("✅ Test completed successfully") |
| 106 | + else: |
| 107 | + self.output_received.emit(f"❌ Test failed with exit code: {exit_code}") |
| 108 | + self.process_finished.emit(exit_code) |
| 109 | + |
| 110 | + def is_test_running(self): |
| 111 | + return self.is_running |
| 112 | + |
| 113 | + def cleanup(self): |
| 114 | + if self.process: |
| 115 | + self._force_kill_timer.stop() |
| 116 | + |
| 117 | + if self.process.state() == QProcess.ProcessState.Running: |
| 118 | + self.process.kill() |
| 119 | + self.process.waitForFinished(100) |
| 120 | + |
| 121 | + try: |
| 122 | + self.process.readyReadStandardOutput.disconnect() |
| 123 | + self.process.readyReadStandardError.disconnect() |
| 124 | + self.process.finished.disconnect() |
| 125 | + self.process.errorOccurred.disconnect() |
| 126 | + except: |
| 127 | + pass |
| 128 | + |
| 129 | + self.process = None |
| 130 | + |
| 131 | + self.is_running = False |
| 132 | + |
| 133 | + def _on_error(self, error: QProcess.ProcessError): |
| 134 | + error_messages = { |
| 135 | + QProcess.ProcessError.FailedToStart: "❌ Failed to start the process. Check if Maven and Java are properly configured.", |
| 136 | + QProcess.ProcessError.Crashed: "The process crashed.", |
| 137 | + QProcess.ProcessError.Timedout: "The process timed out.", |
| 138 | + QProcess.ProcessError.ReadError: "An error occurred while reading from the process.", |
| 139 | + QProcess.ProcessError.WriteError: "An error occurred while writing to the process.", |
| 140 | + QProcess.ProcessError.UnknownError: "An unknown error occurred." |
| 141 | + } |
| 142 | + self.error_received.emit(f"Process Error: {error_messages.get(error, 'An unknown error occurred.')}") |
| 143 | + |
| 144 | + if error == QProcess.ProcessError.FailedToStart: |
| 145 | + self.error_received.emit("Please ensure:") |
| 146 | + self.error_received.emit("1. Java JDK is installed and JAVA_HOME is set") |
| 147 | + self.error_received.emit("2. Maven is installed and in PATH") |
| 148 | + self.error_received.emit("4. The E2E directory contains a valid Maven project") |
0 commit comments