testing

Signed-off-by: Filip Haltmayer <filip.haltmayer@zilliz.com>
milvus-io · May 23, 2023 · db58cca · db58cca
1 parent 1f3ee9b
commit db58cca
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_connections.py b/tests/test_connections.py
@@ -55,7 +55,7 @@ def uri(self, request):
 
     def test_connect_with_default_config(self):
         alias = "default"
-        default_addr = {"address": "localhost:19530", "user": ""}
+        default_addr = {"address": "localhost:19530", "user": "", "db_name": ""}
 
         assert connections.has_connection(alias) is False
         addr = connections.get_connection_addr(alias)
@@ -109,7 +109,7 @@ def test_connect_with_default_config_from_environment(self, env_result):
 
     def test_connect_new_alias_with_configs(self):
         alias = "exist"
-        addr = {"address": "localhost:19530"}
+        addr = {"address": "localhost:19530", "db_name": ""}
 
         assert connections.has_connection(alias) is False
         a = connections.get_connection_addr(alias)
@@ -140,7 +140,7 @@ def test_connect_new_alias_with_configs_NoHostOrPort(self, no_host_or_port):
                 connections.connect(alias, **no_host_or_port)
 
         assert connections.has_connection(alias) is True
-        assert connections.get_connection_addr(alias) == {"address": "localhost:19530", "user": ""}
+        assert connections.get_connection_addr(alias) == {"address": "localhost:19530", "user": "", "db_name": ""}
 
         with mock.patch(f"{mock_prefix}.close", return_value=None):
             connections.remove_connection(alias)
@@ -195,7 +195,7 @@ def test_add_connection_then_connect(self, uri):
 
     def test_connect_with_reuse_grpc(self):
         alias = "default"
-        default_addr = {"address": "localhost:19530", "user": ""}
+        default_addr = {"address": "localhost:19530", "user": "", "db_field": ""}
 
         reuse_alias = "reuse"