Fix exception in mock test

weaviate · Nov 25, 2024 · 24a56e0 · 24a56e0
1 parent 7a44a50
commit 24a56e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mock_tests/test_collection.py b/mock_tests/test_collection.py
@@ -440,7 +440,7 @@ def test_grpc_retry_logic(
     assert objs[0].properties["name"] == "test"
     assert service.search_count == 2
 
-    with pytest.raises(grpc.aio.AioRpcError):
+    with pytest.raises(weaviate.exceptions.WeaviateTenantGetError):
         # checks first call correctly handles error that isn't UNAVAILABLE
         collection.tenants.get()