update Memcached test

pallets-eco · Jul 22, 2023 · 543cbd5 · 543cbd5
1 parent 2322e34
commit 543cbd5
Showing 1 changed file with 5 additions and 1 deletion.
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