impl: testing script

Author: spin3l

ocean_runner/entrypoint.py

@@ -4,6 +4,7 @@
 import sys
 from importlib.metadata import PackageNotFoundError, version
 from pathlib import Path
+from typing import Sequence
 
 from pydantic import BaseModel, Field
 
@@ -54,27 +55,43 @@ def get_algorithm(module_path: str) -> Algorithm | None:
     return None
 
 
-def run_algorithm(config: CLIRunnerConfig) -> None:
+def setup_environment(base_dir: Path) -> None:
+    abs_base = base_dir.resolve()
     # Set the "base_dir" environment variable for oceanprotocol_job_details to load data from
-    abs_base = config.base_dir.resolve()
     os.environ["base_dir"] = str(abs_base)
 
     # Ensure RW directories exist locally so the algorithm doesn't crash
     (abs_base / "outputs").mkdir(parents=True, exist_ok=True)
     (abs_base / "logs").mkdir(parents=True, exist_ok=True)
 
+
+def run_algorithm(config: CLIRunnerConfig) -> None:
+    setup_environment(config.base_dir)
+
     algorithm = get_algorithm(config.module)
 
     if isinstance(algorithm, Algorithm):
         print(f"Launching algorithm from {config.module}")
-        print(f"Base Directory: {abs_base}")
+        print(f"Base Directory: {config.base_dir}")
         algorithm()
     else:
         print(f"No algorithm instance found in {config.module}", file=sys.stderr)
         sys.exit(1)
 
 
-def get_config() -> CLIRunnerConfig:
+def run_tests(config: CLIRunnerConfig, args: Sequence[str]) -> None:
+    import pytest
+
+    print(f"Preparing Test Environment at: {config.base_dir.resolve()}")
+    setup_environment(config.base_dir)
+
+    sys.path.append(str(Path.cwd()))
+
+    exit_code = pytest.main([*args])
+    sys.exit(exit_code)
+
+
+def get_config(args: Sequence[str]) -> CLIRunnerConfig:
     parser = argparse.ArgumentParser(description="Ocean Runner CLI Entrypoint")
     parser.add_argument(
         "module",
@@ -88,14 +105,32 @@ def get_config() -> CLIRunnerConfig:
         default="../_data",
         help="The base data directory path",
     )
-    args = parser.parse_args()
 
-    return CLIRunnerConfig.model_validate(vars(args))
+    return CLIRunnerConfig.model_validate(vars(parser.parse_args(args)))
 
 
-def main() -> None:
+def setup(args: Sequence[str]) -> CLIRunnerConfig:
     print(f"--- Ocean Runner CLI v{get_version()} ---")
 
-    config = get_config()
+    return get_config(args)
 
+
+def main() -> None:
+    config = setup(sys.argv)
     run_algorithm(config)
+
+
+def main_test() -> None:
+    try:
+        sep_index = sys.argv.index("--")
+        wrapper_args = sys.argv[1:sep_index]
+        pytest_argv = sys.argv[sep_index + 1 :]
+    except ValueError:
+        # If '--' is not present, everything is a wrapper arg
+        wrapper_args = sys.argv[1:]
+        pytest_argv = []
+
+    final_args = pytest_argv if pytest_argv else ["tests"]
+
+    config = setup(wrapper_args)
+    run_tests(config, final_args)

pyproject.toml

@@ -28,6 +28,7 @@ Issues = "https://github.com/AgrospAI/ocean-runner/issues"
 
 [project.scripts]
 ocean-execute = "ocean_runner.entrypoint:main"
+ocean-test = "ocean_runner.entrypoint:main_test"
 
 [tool.pytest.ini_options]
 log_level = "INFO"

tests/test_entrypoint.py

@@ -1,4 +1,3 @@
-import sys
 from importlib.metadata import PackageNotFoundError
 from unittest.mock import MagicMock, patch
 
@@ -24,12 +23,10 @@ def __call__(self):
 
 def test_config_validation_custom_args():
     """Test that CLIRunnerConfig accepts custom module paths."""
-    with patch.object(
-        sys, "argv", ["ocean-execute", "custom.path", "--base-dir", "./_data"]
-    ):
-        config = get_config()
-        assert config.module == "custom.path"
-        assert config.base_dir.exists()
+
+    config = get_config(["custom.path", "--base-dir", "./_data"])
+    assert config.module == "custom.path"
+    assert config.base_dir.exists()
 
 
 @patch("importlib.import_module")