first version

tests/integration/nested_python_module_test.py

@@ -0,0 +1,72 @@
+"""Test the encryption of complex python modules with multiple files and imports."""
+
+import os
+import sys
+import subprocess
+from pathlib import Path
+
+
+def test_encrypt_and_run_nested_module(
+    nested_python_module: Path,
+    tmp_dir: Path,
+    password: str,
+    aad: str,
+) -> None:
+    """Test encrypting and running a module with nested imports.
+
+    Args:
+        nested_python_module (Path): Path to the nested module directory.
+        tmp_dir (Path): Temporary directory for test files.
+        password (str): The test password.
+        aad (str): Additional authenticated data.
+    """
+
+    encrypted_file = tmp_dir / "encrypted_nested.pec"
+
+    # Encrypt the folder containing the complex module
+    encrypt_args = [
+        sys.executable,
+        "-m",
+        "python_encrypt_code",
+        "encrypt",
+        str(nested_python_module),
+        "-o",
+        str(encrypted_file),
+        "-p",
+        password,
+        "-aad",
+        aad,
+    ]
+    subprocess.run(
+        args=encrypt_args,
+        check=True,
+    )
+
+    # Verify that the encrypted file was created
+    assert encrypted_file.exists()
+
+    # Prepare environment for the PasswordProviderExample
+    env = {**os.environ}
+    env["PEC_PASSWORD"] = password
+    env["PEC_AAD"] = aad
+
+    # Run the encrypted module with environment set
+    run_args = [
+        sys.executable,
+        "-m",
+        "python_encrypt_code",
+        "run-insecure",
+        str(encrypted_file),
+        "--script",
+        "main.py",
+    ]
+    result = subprocess.run(
+        args=run_args,
+        check=True,
+        capture_output=True,
+        text=True,
+        env=env,
+    )
+
+    # Verify nested module functionality works
+    assert "Nested Python Module executed successfully." in result.stderr