diff --git a/swh/scrubber/__init__.py b/swh/scrubber/__init__.py --- a/swh/scrubber/__init__.py +++ b/swh/scrubber/__init__.py @@ -12,8 +12,10 @@ def get_scrubber_db(cls: str, **kwargs) -> ScrubberDb: - if cls != "local": - raise ValueError(f"Unknown scrubber db class '{cls}', use 'local' instead.") + if cls not in ("local", "postgresql"): + raise ValueError( + f"Unknown scrubber db class '{cls}', use 'postgresql' instead." + ) from .db import ScrubberDb diff --git a/swh/scrubber/tests/test_cli.py b/swh/scrubber/tests/test_cli.py --- a/swh/scrubber/tests/test_cli.py +++ b/swh/scrubber/tests/test_cli.py @@ -25,7 +25,7 @@ runner = CliRunner() config = { - "scrubber_db": {"cls": "local", "db": scrubber_db.conn.dsn}, + "scrubber_db": {"cls": "postgresql", "db": scrubber_db.conn.dsn}, "graph": {"url": "http://graph.example.org:5009/"}, } if storage: @@ -72,7 +72,7 @@ assert result.exit_code == 0, result.output assert result.output == "" - get_scrubber_db.assert_called_once_with(cls="local", db=scrubber_db.conn.dsn) + get_scrubber_db.assert_called_once_with(cls="postgresql", db=scrubber_db.conn.dsn) StorageChecker.assert_called_once_with( db=scrubber_db, storage=StorageChecker.mock_calls[0][2]["storage"], @@ -103,7 +103,7 @@ assert result.exit_code == 0, result.output assert result.output == "" - get_scrubber_db.assert_called_once_with(cls="local", db=scrubber_db.conn.dsn) + get_scrubber_db.assert_called_once_with(cls="postgresql", db=scrubber_db.conn.dsn) JournalChecker.assert_called_once_with( db=scrubber_db, journal_client={ @@ -129,7 +129,7 @@ assert result.exit_code == 0, result.output assert result.output == "" - get_scrubber_db.assert_called_once_with(cls="local", db=scrubber_db.conn.dsn) + get_scrubber_db.assert_called_once_with(cls="postgresql", db=scrubber_db.conn.dsn) OriginLocator.assert_called_once_with( db=scrubber_db, storage=OriginLocator.mock_calls[0][2]["storage"], @@ -150,7 +150,7 @@ assert result.exit_code == 0, result.output assert result.output == "" - get_scrubber_db.assert_called_once_with(cls="local", db=scrubber_db.conn.dsn) + get_scrubber_db.assert_called_once_with(cls="postgresql", db=scrubber_db.conn.dsn) Fixer.assert_called_once_with( db=scrubber_db, start_object=CoreSWHID.from_string("swh:1:cnt:" + "00" * 20),