chore: Pass test suite

packages/ref-celery/pyproject.toml

@@ -34,7 +34,7 @@ ref-celery = "ref_celery.cli:app"
 
 [tool.uv]
 dev-dependencies = [
-
+    "pytest-mock>=3.14.0",
 ]
 
 [tool.uv.sources]

packages/ref-celery/src/ref_celery/cli.py

@@ -34,13 +34,15 @@ def start_worker(
     try:
         imp = importlib.import_module(package.replace("-", "_"))  # type: ignore
     except ModuleNotFoundError:
-        raise ValueError(f"Package '{package}' not found")
+        typer.echo(f"Package '{package}' not found")
+        raise typer.Abort()
 
     # Get the provider from the package
     try:
         provider = imp.provider
     except AttributeError:
-        raise ValueError("The package must define a 'provider' variable")
+        typer.echo("The package must define a 'provider' variable")
+        raise typer.Abort()
 
     # Wrap each metrics in the provider with a celery tasks
     register_celery_tasks(celery_app, provider)
@@ -49,5 +51,5 @@ def start_worker(
     celery_app.worker_main(argv=argv)
 
 
-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: no cover
     app()

packages/ref-celery/tests/test_cli.py

@@ -0,0 +1,33 @@
+from ref_celery.cli import app
+from typer.testing import CliRunner
+
+runner = CliRunner()
+
+
+def test_cli_help():
+    result = runner.invoke(app, ["--help"])
+    assert result.exit_code == 0
+
+
+def test_cli_spawns_worker(mocker):
+    mock_app = mocker.patch("ref_celery.cli.create_celery_app")
+    result = runner.invoke(app, ["--package", "ref-metrics-example"])
+    assert result.exit_code == 0
+
+    mock_app().worker_main.assert_called_once()
+
+
+def test_cli_wrong_package():
+    result = runner.invoke(app, ["--package", "missing"])
+    assert result.exit_code == 1
+
+    print(result.output)
+    assert "Package 'missing' not found" in result.output
+
+
+def test_cli_missing_provider():
+    result = runner.invoke(app, ["--package", "pandas"])
+    assert result.exit_code == 1
+
+    print(result.output)
+    assert "The package must define a 'provider' variable" in result.output

packages/ref-celery/tests/test_tasks.py

@@ -0,0 +1 @@
+# TODO: add tests for executing tasks