feat: adds additional plugin slot value

feat: adds additional slot value

Author: cofin

litestar_saq/config.py

@@ -2,7 +2,7 @@
 
 from dataclasses import dataclass, field
 from pathlib import Path
-from typing import TYPE_CHECKING, Callable, Collection, Dict, Mapping, TypeVar, cast
+from typing import TYPE_CHECKING, Callable, Collection, Dict, Mapping, TypeVar, Union, cast
 
 from litestar.exceptions import ImproperlyConfiguredException
 from litestar.serialization import decode_json, encode_json
@@ -23,8 +23,8 @@
     from saq.types import Function
 
 T = TypeVar("T")
-TaskQueue = Queue | SaqQueue
-DumpType = SaqDumpType | Callable[[Dict], bytes]
+TaskQueue = Union[Queue, SaqQueue]
+DumpType = Union[SaqDumpType, Callable[[Dict], bytes]]
 
 
 def serializer(value: Any) -> str:

litestar_saq/controllers.py

@@ -23,7 +23,10 @@ async def job_info(queue: TaskQueue, job_id: str) -> Job:
 
 
 @lru_cache(typed=True)
-def build_controller(url_base: str = "/saq", controller_guards: list[Guard] | None = None) -> type[Controller]:  # noqa: C901
+def build_controller(  # noqa: C901
+    url_base: str = "/saq",
+    controller_guards: list[Guard] | None = None,
+) -> type[Controller]:
     from litestar import Controller, MediaType, get, post
     from litestar.exceptions import NotFoundException
     from litestar.status_codes import HTTP_202_ACCEPTED

litestar_saq/plugin.py

@@ -23,7 +23,7 @@
 class SAQPlugin(InitPluginProtocol, CLIPluginProtocol):
     """SAQ plugin."""
 
-    __slots__ = ("_config",)
+    __slots__ = ("_config", "_worker_instances")
 
     def __init__(self, config: SAQConfig) -> None:
         """Initialize ``SAQPlugin``.

pyproject.toml

@@ -30,7 +30,7 @@ license = {text = "MIT"}
 name = "litestar-saq"
 readme = "README.md"
 requires-python = ">=3.8"
-version = "0.1.11"
+version = "0.1.12"
 
 [project.optional-dependencies]
 hiredis = ["hiredis"]
@@ -201,6 +201,7 @@ classmethod-decorators = [
   "TRY",
   "UP006",
   "SLF001",
+  "ERA001",
 ]
 
 [tool.ruff.isort]

tests/conftest.py

@@ -0,0 +1,52 @@
+from unittest.mock import Mock
+
+import pytest
+from litestar.cli._utils import LitestarGroup as Group
+
+from litestar_saq.config import SAQConfig, TaskQueues
+from litestar_saq.plugin import SAQPlugin
+
+# Assuming SAQConfig, Worker, Group, AppConfig, TaskQueues, Queue are available and can be imported
+# Assuming there are meaningful __eq__ methods for comparisons where needed
+
+
+# Test on_cli_init method
+@pytest.mark.parametrize(
+    "cli_group",
+    [
+        # ID: Test-CLI-Init-1
+        Mock(Group),
+    ],
+    ids=["happy-path"],
+)
+def test_on_cli_init(cli_group: Group) -> None:
+    # Arrange
+    config = Mock(SAQConfig)
+    plugin = SAQPlugin(config)
+
+    # Act
+    plugin.on_cli_init(cli_group)
+
+    # Assert
+    cli_group.add_command.assert_called_once()
+
+
+# Test get_queues method
+@pytest.mark.parametrize(
+    "queues",
+    [
+        # ID: Test-Get-Queues-1
+        Mock(),
+    ],
+    ids=["happy-path"],
+)
+def test_get_queues(queues: TaskQueues) -> None:
+    # Arrange
+    config = Mock(SAQConfig, get_queues=Mock(return_value=queues))
+    plugin = SAQPlugin(config)
+
+    # Act
+    result = plugin.get_queues()
+
+    # Assert
+    assert result == queues