fix: when / present in virtual host name and passing as uri (#1979)

* fix: when extra / present in virtual host name

e.g name of vh is  "/test"

there is no problenm if passing vh separately, result url will "amqp://root:123@localhost:5672//test"

but if pass vh in uri, result uri will 
"amqp://root:123@localhost:5672/test"
cause lstrip("/") calling and slash missing in result url, 
so replace with count 1 is ok

* add test for rabbit broker build_url

* refactor building vh and add tests for this

* Update test_url_builder.py

* Update test_url_builder.py

---------

Co-authored-by: Pastukhov Nikita <[email protected]>

docs/docs/SUMMARY.md

@@ -960,6 +960,7 @@ search:
                     - [build_message](api/faststream/rabbit/testing/build_message.md)
                 - utils
                     - [build_url](api/faststream/rabbit/utils/build_url.md)
+                    - [build_virtual_host](api/faststream/rabbit/utils/build_virtual_host.md)
                     - [is_routing_exchange](api/faststream/rabbit/utils/is_routing_exchange.md)
             - redis
                 - [ListSub](api/faststream/redis/ListSub.md)

docs/docs/en/api/faststream/rabbit/utils/build_virtual_host.md

@@ -0,0 +1,11 @@
+---
+# 0.5 - API
+# 2 - Release
+# 3 - Contributing
+# 5 - Template Page
+# 10 - Default
+search:
+  boost: 0.5
+---
+
+::: faststream.rabbit.utils.build_virtual_host

faststream/rabbit/utils.py

@@ -12,6 +12,17 @@
     from faststream.rabbit.schemas import RabbitExchange
 
 
+def build_virtual_host(
+    url: Union[str, "URL", None], virtualhost: Optional[str], path: str
+) -> str:
+    if (not url and not virtualhost) or virtualhost == "/":
+        return ""
+    elif virtualhost:
+        return virtualhost.replace("/", "", 1)
+    else:
+        return path.replace("/", "", 1)
+
+
 def build_url(
     url: Union[str, "URL", None] = None,
     *,
@@ -36,7 +47,7 @@ def build_url(
         port=port or original_url.port or default_port,
         login=login or original_url.user or "guest",
         password=password or original_url.password or "guest",
-        virtualhost=virtualhost or original_url.path.lstrip("/"),
+        virtualhost=build_virtual_host(url, virtualhost, original_url.path),
         ssl=use_ssl,
         ssl_options=ssl_options,
         client_properties=client_properties,

tests/brokers/rabbit/test_connect.py

@@ -15,7 +15,7 @@ def get_broker_args(self, settings):
         return {"url": settings.url}
 
     @pytest.mark.asyncio
-    async def test_init_connect_by_raw_data(self, settings):
+    async def test_connect_handover_config_to_init(self, settings):
         broker = self.broker(
             host=settings.host,
             port=settings.port,
@@ -28,7 +28,7 @@ async def test_init_connect_by_raw_data(self, settings):
         await broker.close()
 
     @pytest.mark.asyncio
-    async def test_connection_by_params(self, settings):
+    async def test_connect_handover_config_to_connect(self, settings):
         broker = self.broker()
         assert await broker.connect(
             host=settings.host,
@@ -41,20 +41,7 @@ async def test_connection_by_params(self, settings):
         await broker.close()
 
     @pytest.mark.asyncio
-    async def test_connect_merge_kwargs_with_priority(self, settings):
-        broker = self.broker(host="fake-host", port=5677)  # kwargs will be ignored
-        assert await broker.connect(
-            host=settings.host,
-            port=settings.port,
-            security=SASLPlaintext(
-                username=settings.login,
-                password=settings.password,
-            ),
-        )
-        await broker.close()
-
-    @pytest.mark.asyncio
-    async def test_connect_merge_args_and_kwargs_native(self, settings):
+    async def test_connect_handover_config_to_connect_override_init(self, settings):
         broker = self.broker("fake-url")  # will be ignored
         assert await broker.connect(url=settings.url)
         await broker.close()

tests/brokers/rabbit/test_url_builder.py

@@ -0,0 +1,38 @@
+from typing import Any, Dict
+
+import pytest
+from yarl import URL
+
+from faststream.rabbit.utils import build_url
+
+
+@pytest.mark.parametrize(
+    ("url_kwargs", "expected_url"),
+    [
+        pytest.param(
+            {},
+            URL("amqp://guest:guest@localhost:5672/"),  # pragma: allowlist secret
+            id="blank params use defaults",
+        ),
+        pytest.param(
+            {"ssl": True},
+            URL("amqps://guest:guest@localhost:5672/"),  # pragma: allowlist secret
+            id="ssl affects protocol",
+        ),
+        pytest.param(
+            {"url": "fake", "virtualhost": "/", "host": "host"},
+            URL("amqp://guest:guest@host:5672/"),  # pragma: allowlist secret
+            id="kwargs overrides url",
+        ),
+        pytest.param(
+            {"virtualhost": "//test"},  # pragma: allowlist secret
+            URL(
+                "amqp://guest:guest@localhost:5672//test"  # pragma: allowlist secret
+            ),
+            id="exotic virtualhost",
+        ),
+    ],
+)
+def test_unpack_args(url_kwargs: Dict[str, Any], expected_url: URL) -> None:
+    url = build_url(**url_kwargs)
+    assert url == expected_url