test-pioneer-dev 0.0.38__py3-none-any.whl

test_pioneer/__init__.py

@@ -0,0 +1,7 @@
+from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import EDITOR_EXTEND_TAB
+from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import start_editor
+
+__all__ = [
+    "start_editor", "AutomationEditor", "EDITOR_EXTEND_TAB"
+]

test_pioneer/extend/mail_thunder_extend/mail_thunder_setting.py

@@ -0,0 +1,36 @@
+import sys
+from email.mime.multipart import MIMEMultipart
+
+from test_pioneer.utils.exception.exception_tags import send_html_exception_tag
+from test_pioneer.utils.exception.exceptions import ITESendHtmlReportException
+
+
+def send_after_test(html_report_path: str = None) -> None:
+    try:
+        from je_mail_thunder import SMTPWrapper
+        mail_thunder_smtp: SMTPWrapper = SMTPWrapper()
+        if html_report_path is None and mail_thunder_smtp.login_state is True:
+            user: str = mail_thunder_smtp.user
+            with open("default_name.html", "r+") as file:
+                html_string: str = file.read()
+            message = mail_thunder_smtp.create_message_with_attach(
+                html_string,
+                {"Subject": "Test Report", "To": user, "From": user},
+                "default_name.html", use_html=True)
+            mail_thunder_smtp.send_message(message)
+            mail_thunder_smtp.quit()
+        elif mail_thunder_smtp.login_state is True:
+            user: str = mail_thunder_smtp.user
+            with open(html_report_path, "r+") as file:
+                html_string: str = file.read()
+            message: MIMEMultipart = mail_thunder_smtp.create_message_with_attach(
+                html_string,
+                {"Subject": "Test Report", "To": user, "From": user},
+                html_report_path, use_html=True)
+            mail_thunder_smtp.send_message(message)
+            mail_thunder_smtp.quit()
+        else:
+            raise ITESendHtmlReportException
+    except ITESendHtmlReportException as error:
+        print(repr(error), file=sys.stderr)
+        print(send_html_exception_tag, file=sys.stderr)

test_pioneer/extend/process_executor/api_testka/api_testka_process.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from test_pioneer.extend.process_executor.process_executor_utils import build_process
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+import sys
+
+from test_pioneer.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
+
+
+def call_api_testka(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_api_testka", exec_str, False, program_buffer)
+
+
+def call_api_testka_with_send(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_api_testka", exec_str, True, program_buffer)
+
+
+def call_api_testka_multi_file(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None \
+                and isinstance(need_to_execute_list, list) and len(need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_api_testka(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)
+
+
+def call_api_testka_multi_file_and_send(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None \
+                and isinstance(need_to_execute_list, list) and len(need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_api_testka_with_send(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)

test_pioneer/extend/process_executor/auto_control/auto_control_process.py

@@ -0,0 +1,62 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from test_pioneer.extend.process_executor.process_executor_utils import build_process
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+import sys
+
+from test_pioneer.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
+
+
+def call_auto_control(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_auto_control", exec_str, False, program_buffer)
+
+
+def call_auto_control_with_send(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_auto_control", exec_str, True, program_buffer)
+
+
+def call_auto_control_multi_file(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+    if need_to_execute_list is not None \
+            and isinstance(need_to_execute_list, list) and len(need_to_execute_list) > 0:
+        for execute_file in need_to_execute_list:
+            with open(execute_file, "r+") as test_script_json:
+                call_auto_control(
+                    main_window,
+                    test_script_json.read(),
+                    program_buffer
+                )
+
+
+def call_auto_control_multi_file_and_send(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None \
+                and isinstance(need_to_execute_list, list) and len(need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_auto_control_with_send(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)

test_pioneer/extend/process_executor/file_automation/file_automation_process.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from test_pioneer.extend.process_executor.process_executor_utils import build_process
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+import sys
+
+from test_pioneer.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
+
+
+def call_file_automation_test(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "automation_file", exec_str, False, program_buffer)
+
+
+def call_file_automation_test_with_send(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "automation_file", exec_str, True, program_buffer)
+
+
+def call_file_automation_test_multi_file(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None \
+                and isinstance(need_to_execute_list, list) and len(need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_file_automation_test(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)
+
+
+def call_file_automation_test_multi_file_and_send(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None \
+                and isinstance(need_to_execute_list, list) and len(need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_file_automation_test_with_send(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)

test_pioneer/extend/process_executor/load_density/load_density_process.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from test_pioneer.extend.process_executor.process_executor_utils import build_process
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+import sys
+
+from test_pioneer.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
+
+
+def call_load_density(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_load_density", exec_str, False, program_buffer)
+
+
+def call_load_density_with_send(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_load_density", exec_str, True, program_buffer)
+
+
+def call_load_density_multi_file(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None and isinstance(need_to_execute_list, list) and len(
+                need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_load_density(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)
+
+
+def call_load_density_multi_file_and_send(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None and isinstance(need_to_execute_list, list) and len(
+                need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_load_density_with_send(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)

test_pioneer/extend/process_executor/mail_thunder/mail_thunder_process.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from test_pioneer.extend.process_executor.process_executor_utils import build_process
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+
+
+def call_mail_thunder(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_mail_thunder", exec_str, False, program_buffer)

test_pioneer/extend/process_executor/process_executor_utils.py

@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+import json
+import sys
+from typing import TYPE_CHECKING, Union
+
+from je_editor import EditorWidget
+
+from test_pioneer.test_pioneer_editor_ui.show_code_window.code_window import CodeWindow
+from test_pioneer.extend.mail_thunder_extend.mail_thunder_setting import send_after_test
+from test_pioneer.extend.process_executor.task_process_manager import TaskProcessManager
+from test_pioneer.utils.exception.exception_tags import wrong_test_data_format_exception_tag
+from test_pioneer.utils.exception.exceptions import ITETestExecutorException
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+
+
+def build_process(
+        main_window: AutomationEditor,
+        package: str,
+        exec_str: Union[str, None] = None,
+        send_mail: bool = False,
+        program_buffer: int = 1024000,
+):
+    try:
+        widget = main_window.tab_widget.currentWidget()
+        if isinstance(widget, EditorWidget) and exec_str is None:
+            test_format_code = widget.code_edit.toPlainText()
+        else:
+            test_format_code = exec_str
+        start_process(main_window, package, test_format_code, send_mail, program_buffer)
+    except json.decoder.JSONDecodeError as error:
+        print(
+            repr(error) +
+            "\n"
+            + wrong_test_data_format_exception_tag,
+            file=sys.stderr
+        )
+    except ITETestExecutorException as error:
+        print(repr(error), file=sys.stderr)
+
+
+def start_process(
+        main_window: AutomationEditor,
+        package: str,
+        test_format_code: str,
+        send_mail: bool = False,
+        program_buffer: int = 1024000
+):
+    # Code window init
+    code_window = CodeWindow()
+    main_window.current_run_code_window.append(code_window)
+    main_window.clear_code_result()
+    # Process init
+    if send_mail:
+        process = TaskProcessManager(
+            main_window=code_window,
+            program_buffer_size=program_buffer,
+            program_encoding=main_window.encoding
+        )
+    else:
+        process = TaskProcessManager(
+            code_window,
+            task_done_trigger_function=send_after_test,
+            program_buffer_size=program_buffer,
+            program_encoding=main_window.encoding
+        )
+    process.start_test_process(
+        package,
+        exec_str=test_format_code,
+    )

test_pioneer/extend/process_executor/task_process_manager.py

@@ -0,0 +1,164 @@
+import json
+import queue
+import subprocess
+import sys
+import threading
+import typing
+from pathlib import Path
+from queue import Queue
+from threading import Thread
+
+from PySide6.QtCore import QTimer
+from je_editor.pyside_ui.main_ui.save_settings.user_color_setting_file import actually_color_dict
+from je_editor.utils.venv_check.check_venv import check_and_choose_venv
+
+from test_pioneer.test_pioneer_editor_ui.show_code_window.code_window import CodeWindow
+
+
+class TaskProcessManager(object):
+    def __init__(
+            self,
+            main_window: CodeWindow,
+            task_done_trigger_function: typing.Callable = None,
+            error_trigger_function: typing.Callable = None,
+            program_buffer_size: int = 1024,
+            program_encoding: str = "utf-8"
+    ):
+        super().__init__()
+        self.compiler_path = None
+        # ite_instance param
+        self.read_program_error_output_from_thread: [threading.Thread, None] = None
+        self.read_program_output_from_thread: [threading.Thread, None] = None
+        self.main_window: CodeWindow = main_window
+        self.timer: QTimer = QTimer(self.main_window)
+        self.still_run_program: bool = True
+        self.program_encoding: str = program_encoding
+        self.run_output_queue: Queue = Queue()
+        self.run_error_queue: Queue = Queue()
+        self.process: [subprocess.Popen, None] = None
+
+        self.task_done_trigger_function: typing.Callable = task_done_trigger_function
+        self.error_trigger_function: typing.Callable = error_trigger_function
+        self.program_buffer_size = program_buffer_size
+
+    def renew_path(self) -> None:
+        if self.main_window.python_compiler is None:
+            # Renew compiler path
+            if sys.platform in ["win32", "cygwin", "msys"]:
+                venv_path = Path(str(Path.cwd()) + "/venv/Scripts")
+            else:
+                venv_path = Path(str(Path.cwd()) + "/venv/bin")
+            self.compiler_path = check_and_choose_venv(venv_path)
+        else:
+            self.compiler_path = self.main_window.python_compiler
+
+    def start_test_process(self, package: str, exec_str: str):
+        self.renew_path()
+        if sys.platform in ["win32", "cygwin", "msys"]:
+            exec_str = json.dumps(exec_str)
+            args = [
+                self.compiler_path,
+                "-m",
+                package,
+                "--execute_str",
+                exec_str
+            ]
+        else:
+            args = " ".join([f"{self.compiler_path}", f"-m {package}", "--execute_str", f"{exec_str}"])
+        self.process: subprocess.Popen = subprocess.Popen(
+            args,
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            shell=True,
+            encoding=self.program_encoding
+        )
+        self.still_run_program = True
+        # program output message queue thread
+        self.read_program_output_from_thread = Thread(
+            target=self.read_program_output_from_process,
+            daemon=True
+        )
+        self.read_program_output_from_thread.start()
+        # program error message queue thread
+        self.read_program_error_output_from_thread = Thread(
+            target=self.read_program_error_output_from_process,
+            daemon=True
+        )
+        self.read_program_error_output_from_thread.start()
+        # start Pyside update
+        # start timer
+        self.main_window.setWindowTitle(package)
+        self.main_window.show()
+        self.timer = QTimer()
+        self.timer.setInterval(100)
+        self.timer.timeout.connect(self.pull_text)
+        self.timer.start()
+
+    # Pyside UI update method
+    def pull_text(self):
+        try:
+            self.main_window.code_result.setTextColor(actually_color_dict.get("normal_output_color"))
+            if not self.run_output_queue.empty():
+                output_message = self.run_output_queue.get_nowait()
+                output_message = str(output_message).strip()
+                if output_message:
+                    self.main_window.code_result.append(output_message)
+            self.main_window.code_result.setTextColor(actually_color_dict.get("error_output_color"))
+            if not self.run_error_queue.empty():
+                error_message = self.run_error_queue.get_nowait()
+                error_message = str(error_message).strip()
+                if error_message:
+                    self.main_window.code_result.append(error_message)
+            self.main_window.code_result.setTextColor(actually_color_dict.get("normal_output_color"))
+        except queue.Empty:
+            pass
+        if self.process is not None:
+            if self.process.returncode == 0:
+                if self.timer.isActive():
+                    self.timer.stop()
+                self.exit_program()
+            elif self.process.returncode is not None:
+                if self.timer.isActive():
+                    self.timer.stop()
+                self.exit_program()
+            if self.still_run_program:
+                # poll return code
+                self.process.poll()
+        else:
+            if self.timer.isActive():
+                self.timer.stop()
+
+    # exit program change run flag to false and clean read thread and queue and process
+    def exit_program(self):
+        self.still_run_program = False
+        if self.read_program_output_from_thread is not None:
+            self.read_program_output_from_thread = None
+        if self.read_program_error_output_from_thread is not None:
+            self.read_program_error_output_from_thread = None
+        self.print_and_clear_queue()
+        if self.process is not None:
+            self.process.terminate()
+            self.main_window.code_result.append(f"Task exit with code {self.process.returncode}")
+            self.process = None
+
+    def print_and_clear_queue(self):
+        self.run_output_queue = queue.Queue()
+        self.run_error_queue = queue.Queue()
+
+    def read_program_output_from_process(self):
+        while self.still_run_program:
+            self.process: subprocess.Popen
+            program_output_data = self.process.stdout.read(self.program_buffer_size)
+            if self.process:
+                self.process.stdout.flush()
+            if program_output_data.strip() != "":
+                self.run_output_queue.put(program_output_data)
+
+    def read_program_error_output_from_process(self):
+        while self.still_run_program:
+            program_error_output_data = self.process.stderr.read(self.program_buffer_size)
+            if self.process:
+                self.process.stderr.flush()
+            if program_error_output_data.strip() != "":
+                self.run_error_queue.put(program_error_output_data)

test_pioneer/extend/process_executor/web_runner/web_runner_process.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Union
+
+from test_pioneer.extend.process_executor.process_executor_utils import build_process
+
+if TYPE_CHECKING:
+    from test_pioneer.test_pioneer_editor_ui.editor_main.main_ui import AutomationEditor
+import sys
+
+from test_pioneer.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
+
+
+def call_web_runner_test(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_web_runner", exec_str, False, program_buffer)
+
+
+def call_web_runner_test_with_send(
+        main_window: AutomationEditor,
+        exec_str: Union[str, None] = None,
+        program_buffer: int = 1024000
+):
+    build_process(main_window, "je_web_runner", exec_str, True, program_buffer)
+
+
+def call_web_runner_test_multi_file(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None and isinstance(need_to_execute_list, list) and len(
+                need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_web_runner_test(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)
+
+
+def call_web_runner_test_multi_file_and_send(
+        main_window: AutomationEditor,
+        program_buffer: int = 1024000
+):
+    try:
+
+        need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
+        if need_to_execute_list is not None and isinstance(need_to_execute_list, list) and len(
+                need_to_execute_list) > 0:
+            for execute_file in need_to_execute_list:
+                with open(execute_file, "r+") as test_script_json:
+                    call_web_runner_test_with_send(
+                        main_window,
+                        test_script_json.read(),
+                        program_buffer
+                    )
+    except Exception as error:
+        print(repr(error), file=sys.stderr)