diff --git a/tests/unit/test_qdrant_client.py b/tests/unit/test_qdrant_client.py index f2ca9a3..f3e76c2 100644 --- a/tests/unit/test_qdrant_client.py +++ b/tests/unit/test_qdrant_client.py @@ -14,7 +14,7 @@ def test_get_qdrant_client_no_api_key(monkeypatch): mock_client_cls.return_value = MagicMock() get_qdrant_client() - mock_client_cls.assert_called_once_with(url="http://localhost:6333") + mock_client_cls.assert_called_once_with(url="http://localhost:6333", timeout=60) def test_get_qdrant_client_with_api_key(monkeypatch): @@ -28,6 +28,7 @@ def test_get_qdrant_client_with_api_key(monkeypatch): mock_client_cls.assert_called_once_with( url="https://xyz.qdrant.io:6333", + timeout=60, api_key="secret-cloud-key", )