Hotfix: Allow overriding node celery config with kwargs

src/pytest_celery/api/base.py

@@ -74,7 +74,21 @@ def ready(self) -> bool:
 
     def config(self, *args: tuple, **kwargs: dict) -> dict:
         """Compile the configurations required for Celery from this node."""
-        return self.container.celeryconfig
+        config = self.container.celeryconfig
+
+        if not args and not kwargs:
+            return config
+
+        for key, value in kwargs.items():
+            config[key] = value
+
+            if key == "vhost":
+                vhost = str(value)
+                config["url"] = config["url"][:-1] + f"{vhost}"
+                config["host_url"] = config["host_url"][:-1] + f"{vhost}"
+                config[key] = vhost
+
+        return config
 
     def logs(self) -> str:
         """Get the logs of the underlying container."""

tests/unit/api/test_base.py

@@ -56,6 +56,16 @@ def test_ready(self, node: CeleryTestNode):
     def test_config(self, node: CeleryTestNode):
         assert node.config()
 
+    def test_config_kwarg_vhost(self, node: CeleryTestNode):
+        config = node.config()
+        assert "vhost" not in config
+        assert config["url"].endswith("test_vhost") is False
+        assert config["host_url"].endswith("test_vhost") is False
+        config = node.config(vhost="test_vhost")
+        assert config["vhost"] == "test_vhost"
+        assert config["url"].endswith("test_vhost")
+        assert config["host_url"].endswith("test_vhost")
+
     def test_logs(self, node: CeleryTestNode):
         assert node.logs()
 

tests/unit/conftest.py

@@ -20,8 +20,8 @@ def mocked_container(spec: type) -> Mock:
     mocked_container = Mock(spec=spec)
     mocked_container.id = "mocked_test_container_id"
     mocked_container.celeryconfig = {
-        "url": "mocked_url",
-        "host_url": "mocked_host_url",
+        "url": "mocked_url/",
+        "host_url": "mocked_host_url/",
     }
     return mocked_container