Catch a known DeprecationWarning when calling .close() (redis#3335)

Co-authored-by: Vladyslav Vildanov <117659936+vladvildanov@users.noreply.github.com>
kurtmckee · Sep 2, 2024 · b4fcbaa · b4fcbaa
1 parent 93b9d85
commit b4fcbaa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_asyncio/test_cluster.py b/tests/test_asyncio/test_cluster.py
@@ -313,7 +313,8 @@ async def mock_aclose():
             called += 1
 
         with mock.patch.object(cluster, "aclose", mock_aclose):
-            await cluster.close()
+            with pytest.warns(DeprecationWarning, match=r"Use aclose\(\) instead"):
+                await cluster.close()
             assert called == 1
         await cluster.aclose()