Fix DeprecationWarning for sampling from a set

bmerry · bmerry · commit d6938e2b7b7f · 2021-10-11T10:36:29.000+02:00
random.sample on a set internally casts it to a list, but since Python
3.9 that has been raising a DeprecationWarning. I've explicitly cast to
list myself now.
diff --git a/fakeredis/_server.py b/fakeredis/_server.py
@@ -1937,7 +1937,7 @@ def spop(self, key, count=None):
         if count is None:
             if not key.value:
                 return None
-            item = random.sample(key.value, 1)[0]
+            item = random.sample(list(key.value), 1)[0]
             key.value.remove(item)
             key.updated()
             return item
@@ -1956,10 +1956,10 @@ def srandmember(self, key, count=None):
             if not key.value:
                 return None
             else:
-                return random.sample(key.value, 1)[0]
+                return random.sample(list(key.value), 1)[0]
         elif count >= 0:
             count = min(count, len(key.value))
-            return random.sample(key.value, count)
+            return random.sample(list(key.value), count)
         else:
             items = list(key.value)
             return [random.choice(items) for _ in range(-count)]
