extend from_dict test

deepset-ai · Oct 18, 2024 · ba41fcd · ba41fcd
1 parent 7723e40
commit ba41fcd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/integrations/weaviate/tests/test_document_store.py b/integrations/weaviate/tests/test_document_store.py
@@ -303,6 +303,7 @@ def test_from_dict(self, _mock_weaviate, monkeypatch):
                         "connection": {
                             "session_pool_connections": 20,
                             "session_pool_maxsize": 20,
+                            "session_pool_timeout": 5,
                         },
                         "proxies": {"http": "http://proxy:1234"},
                         "timeout": [10, 60],
@@ -339,6 +340,7 @@ def test_from_dict(self, _mock_weaviate, monkeypatch):
         assert document_store._embedded_options.grpc_port == DEFAULT_GRPC_PORT
         assert document_store._additional_config.connection.session_pool_connections == 20
         assert document_store._additional_config.connection.session_pool_maxsize == 20
+        assert document_store._additional_config.connection.session_pool_timeout == 5
 
     def test_to_data_object(self, document_store, test_files_path):
         doc = Document(content="test doc")