update

Author: Ying Chen

src/promptflow/tests/sdk_cli_test/e2etests/test_executable.py

@@ -7,8 +7,12 @@
 import requests
 import platform
 import sys
+import os
+import ast
+import importlib
 
 from .test_cli import run_pf_command
+from promptflow._utils.context_utils import _change_working_dir
 
 FLOWS_DIR = "./tests/test_configs/flows"
 RUNS_DIR = "./tests/test_configs/runs"
@@ -20,10 +24,6 @@
 @pytest.mark.cli_test
 @pytest.mark.e2etest
 class TestExecutable:
-    # @pytest.mark.skipif(
-    #     sys.platform == "win32" or sys.platform == "darwin",
-    #     reason="Raise Exception: Process terminated with exit code 4294967295",
-    # )
     def test_flow_build_executable(self):
         source = f"{FLOWS_DIR}/web_classification/flow.dag.yaml"
         target = "promptflow._sdk.operations._flow_operations.FlowOperations._run_pyinstaller"
@@ -41,29 +41,32 @@ def test_flow_build_executable(self):
                     "--format",
                     "executable",
                 )
-                # Start the Python script as a subprocess
-                app_file = Path(temp_dir, "app.py").as_posix()
-                if not Path(app_file).exists():
-                    raise Exception(f"File {app_file} does not exist.")
-                process = subprocess.Popen([sys.executable, app_file], stderr=subprocess.PIPE, shell=platform.system() == 'Windows')
-                time.sleep(5)
-                try:
-                    error_message = process.stderr.read().decode("utf-8")
-                    if process.poll() is not None:
-                        print(f"Error output from child process: {error_message}")
-                        raise Exception(f"Streamlit server did not start successfully. "
-                                        f"error code: {process.returncode} message:{error_message}")
-                    else:
-                        try:
-                            response = requests.get("http://localhost:8501")
-                            if response.status_code == 200:
-                                print("Streamlit server started successfully.")
-                            else:
-                                raise Exception(f"Streamlit server did not start successfully. "
-                                                f"error code: {process.returncode} message:{error_message}")
-                        except requests.exceptions.ConnectionError:
-                            raise Exception(f"Could not connect to Streamlit server. error code: "
-                                            f"{process.returncode} message:{error_message}")
-                finally:
-                    process.terminate()
-                    process.wait()
+                with _change_working_dir(temp_dir):
+                    for filename in os.listdir(temp_dir):
+                        file_path = Path(temp_dir, filename)
+                        if os.path.isfile(file_path) and filename.endswith('.py'):
+                            with open(file_path, 'r') as file:
+                                try:
+                                    tree = ast.parse(file.read())
+                                except SyntaxError as e:
+                                    raise SyntaxError(f"Syntax error in file {file_path}: {e}")
+
+                                for node in ast.walk(tree):
+                                    if isinstance(node, (ast.Import, ast.ImportFrom)):
+                                        for alias in node.names:
+                                            module_name = alias.name
+                                            if isinstance(node, ast.ImportFrom):
+                                                module_name = node.module
+                                            try:
+                                                module = importlib.import_module(module_name)
+                                                if isinstance(node, ast.ImportFrom):
+                                                    getattr(module, alias.name)
+                                            except ImportError:
+                                                raise ImportError(f"Module {module_name} in file {file_path} does not exist")
+                                            except AttributeError:
+                                                module_name = f"{node.module}.{alias.name}"
+                                                try:
+                                                    importlib.import_module(module_name)
+                                                except ImportError:
+                                                    raise ImportError(
+                                                        f"Cannot import {alias.name} from module {node.module} in file {file_path}")