vhost not exists test

Author: effedib

tests/test_connection.py

@@ -15,9 +15,15 @@
     ValidationCodeException,
     WinSslConfigurationContext,
 )
+from rabbitmq_amqp_python_client.qpid.proton import (
+    ConnectionException,
+)
 
-from .http_requests import delete_all_connections
-from .http_requests import create_vhost, delete_vhost
+from .http_requests import (
+    create_vhost,
+    delete_all_connections,
+    delete_vhost,
+)
 from .utils import token
 
 
@@ -127,7 +133,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()
@@ -243,8 +249,25 @@ def test_connection_vhost() -> None:
     environment = Environment(uri=uri)
     connection = environment.connection()
     connection.dial()
-    is_correct_vhost = connection._conn.conn.hostname == 'vhost:{}'.format(vhost)
+    is_correct_vhost = connection._conn.conn.hostname == "vhost:{}".format(vhost)
     environment.close()
     delete_vhost(vhost)
 
-    assert is_correct_vhost is True
+    assert is_correct_vhost is True
+
+
+def test_connection_vhost_not_exists() -> None:
+
+    exception = False
+
+    vhost = "tmpVhost" + str(time.time())
+    uri = "amqp://guest:guest@localhost:5672/{}".format(vhost)
+
+    environment = Environment(uri=uri)
+    try:
+        connection = environment.connection()
+        connection.dial()
+    except ConnectionException:
+        exception = True
+
+    assert exception is True