Additional timeout tests added

Author: sanders41

meilisearch/tests/client/test_client.py

@@ -1,11 +1,9 @@
 # pylint: disable=invalid-name
 
-import pytest
-
 import meilisearch
-from meilisearch.errors import MeiliSearchTimeoutError
 from meilisearch.tests import BASE_URL, MASTER_KEY
 
+
 def test_get_client():
     """Tests getting a client instance."""
     client = meilisearch.Client(BASE_URL, MASTER_KEY)
@@ -15,19 +13,16 @@ def test_get_client():
 
 
 def test_client_timeout_set():
-    client = meilisearch.Client(BASE_URL, MASTER_KEY, timeout=5)
+    timeout = 5
+    client = meilisearch.Client(BASE_URL, MASTER_KEY, timeout=timeout)
     response = client.health()
+    assert client.config.timeout == timeout
     assert response.status_code >= 200 and response.status_code < 400
 
 
 def test_client_timeout_not_set():
+    default_timeout = 10
     client = meilisearch.Client(BASE_URL, MASTER_KEY)
     response = client.health()
+    assert client.config.timeout == default_timeout
     assert response.status_code >= 200 and response.status_code < 400
-
-
-def test_client_timeout_error():
-    client = meilisearch.Client("http://wrongurl:1234", MASTER_KEY, timeout=1)
-
-    with pytest.raises(MeiliSearchTimeoutError):
-        client.health()

meilisearch/tests/errors/test_timeout_error_meilisearch.py

@@ -0,0 +1,33 @@
+import pytest
+import meilisearch
+from meilisearch.errors import MeiliSearchTimeoutError
+from meilisearch.tests import BASE_URL, MASTER_KEY
+
+
+@pytest.mark.usefixtures("indexes_sample")
+def test_client_timeout_error(small_movies):
+    client = meilisearch.Client(BASE_URL, MASTER_KEY, timeout=1e-99)
+
+    with pytest.raises(MeiliSearchTimeoutError):
+        index = client.index("indexUID")
+        index.add_documents(small_movies)
+
+
+def test_client_timeout_set():
+    timeout = 1
+    client = meilisearch.Client("http://wrongurl:1234", MASTER_KEY, timeout=timeout)
+
+    with pytest.raises(Exception):
+        client.health()
+
+    assert client.config.timeout == timeout
+
+
+def test_client_timeout_not_set():
+    timeout = 10
+    client = meilisearch.Client("http://wrongurl:1234", MASTER_KEY)
+
+    with pytest.raises(Exception):
+        client.health()
+
+    assert client.config.timeout == timeout