vhost test

Author: effedib

tests/http_requests.py

@@ -15,6 +15,30 @@ def get_connections_names() -> list:
     return connection_names
 
 
+# not used
+def get_vhosts() -> list:
+    request = "http://localhost:15672/api/vhosts"
+    responses = requests.get(request, auth=HTTPBasicAuth("guest", "guest"))
+    responses.raise_for_status()
+    vhosts = responses.json()
+    vhosts_names = []
+    for vhost in vhosts:
+        vhosts_names.append(vhost["name"])
+    return vhosts_names
+
+
+def create_vhost(vhost_name: str) -> None:
+    request = "http://localhost:15672/api/vhosts/{}".format(vhost_name)
+    responses = requests.put(request, auth=HTTPBasicAuth("guest", "guest"))
+    responses.raise_for_status()
+
+
+def delete_vhost(vhost_name: str) -> None:
+    request = "http://localhost:15672/api/vhosts/{}/".format(vhost_name)
+    responses = requests.delete(request, auth=HTTPBasicAuth("guest", "guest"))
+    responses.raise_for_status()
+
+
 def delete_connections(connection_names: []) -> None:
     for connection_name in connection_names:
         request = (

tests/test_connection.py

@@ -17,6 +17,7 @@
 )
 
 from .http_requests import delete_all_connections
+from .http_requests import create_vhost, delete_vhost
 from .utils import token
 
 
@@ -126,7 +127,7 @@ def test_connection_oauth_refresh_token(environment_auth: Environment) -> None:
 
 
 def test_connection_oauth_refresh_token_with_disconnection(
-    environment_auth: Environment,
+    environment_auth: Environment
 ) -> None:
 
     connection = environment_auth.connection()
@@ -233,3 +234,17 @@ def test_reconnection_parameters() -> None:
         exception = True
 
     assert exception is True
+
+
+def test_connection_vhost() -> None:
+    vhost = "tmpVhost" + str(time.time())
+    create_vhost(vhost)
+    uri = "amqp://guest:guest@localhost:5672/{}".format(vhost)
+    environment = Environment(uri=uri)
+    connection = environment.connection()
+    connection.dial()
+    is_correct_vhost = connection._conn.conn.hostname == 'vhost:{}'.format(vhost)
+    environment.close()
+    delete_vhost(vhost)
+
+    assert is_correct_vhost is True