update Memcached tests

pallets-eco · Jul 22, 2023 · 96103b1 · 96103b1
1 parent 9d41d82
commit 96103b1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/tests/test_interface_uniformity.py b/tests/test_interface_uniformity.py
@@ -13,7 +13,11 @@
 @pytest.fixture(autouse=True)
 def create_cache_list(request, tmpdir):
     mc = MemcachedCache()
-    mc._client.flush_all()
+    if mc.pylibmc_used:
+        with mc._client.reserve(block=mc.blocking) as client:
+            client.flush_all()
+    else:
+        mc._client.flush_all()
     rc = RedisCache(port=6360)
     rc._write_client.flushdb()
     request.cls.cache_list = [FileSystemCache(tmpdir), mc, rc, SimpleCache()]

diff --git a/tests/test_memcached_cache.py b/tests/test_memcached_cache.py
@@ -10,7 +10,11 @@
 def cache_factory(request):
     def _factory(self, *args, **kwargs):
         mc = MemcachedCache(*args, **kwargs)
-        mc._client.flush_all()
+        if mc.pylibmc_used:
+            with mc._client.reserve(block=mc.blocking) as client:
+                client.flush_all()
+        else:
+            mc._client.flush_all()
         return mc
 
     request.cls.cache_factory = _factory