redis client for tests [redis_client_for_tests]

hjwp · hjwp · commit c69661dbf6c6 · 2021-02-24T20:08:31.000Z
diff --git a/tests/e2e/redis_client.py b/tests/e2e/redis_client.py
@@ -0,0 +1,18 @@
+import json
+import redis
+
+from allocation import config
+
+r = redis.Redis(**config.get_redis_host_and_port())
+
+
+def subscribe_to(channel):
+    pubsub = r.pubsub()
+    pubsub.subscribe(channel)
+    confirmation = pubsub.get_message(timeout=3)
+    assert confirmation["type"] == "subscribe"
+    return pubsub
+
+
+def publish_message(channel, message):
+    r.publish(channel, json.dumps(message))
