diff --git a/src/docforge/_projection_worker_main.py b/src/docforge/_projection_worker_main.py new file mode 100644 index 0000000..801c5c3 --- /dev/null +++ b/src/docforge/_projection_worker_main.py @@ -0,0 +1,8 @@ +"""Private module entry point for the detached projection worker.""" + +from __future__ import annotations + +from .projection_worker import main + +if __name__ == "__main__": + raise SystemExit(main()) diff --git a/src/docforge/projection_worker.py b/src/docforge/projection_worker.py index 558c197..dc6d48a 100644 --- a/src/docforge/projection_worker.py +++ b/src/docforge/projection_worker.py @@ -201,7 +201,7 @@ def _invoke_worker(request: bytes) -> subprocess.CompletedProcess[bytes]: "PYTHONUTF8": "1", } ) - command = [sys.executable, "-I", "-m", "docforge.projection_worker"] + command = [sys.executable, "-I", "-m", "docforge._projection_worker_main"] with tempfile.TemporaryFile() as output: completed = subprocess.run( command, diff --git a/tests/test_projection_worker.py b/tests/test_projection_worker.py index c0214a7..cde73e6 100644 --- a/tests/test_projection_worker.py +++ b/tests/test_projection_worker.py @@ -118,7 +118,7 @@ class ProjectionWorkerTests(unittest.TestCase): command = launched.call_args.args[0] options = launched.call_args.kwargs self.assertEqual( - [sys.executable, "-I", "-m", "docforge.projection_worker"], + [sys.executable, "-I", "-m", "docforge._projection_worker_main"], command, ) self.assertIs(options["shell"], False) @@ -372,7 +372,7 @@ class ProjectionWorkerTests(unittest.TestCase): for request in requests: with self.subTest(length=len(request)): completed = subprocess.run( - [sys.executable, "-m", "docforge.projection_worker"], + [sys.executable, "-I", "-m", "docforge._projection_worker_main"], cwd=ROOT, input=request, capture_output=True, @@ -381,6 +381,7 @@ class ProjectionWorkerTests(unittest.TestCase): ) self.assertEqual(2, completed.returncode) self.assertEqual(b"", completed.stdout) + self.assertEqual(b"", completed.stderr) if __name__ == "__main__":