redis client for tests [redis_client_for_tests]

hjwp · hjwp · commit d015d966bc34 · 2019-07-26T15:30:09.000+01:00
diff --git a/tests/e2e/redis_client.py b/tests/e2e/redis_client.py
@@ -0,0 +1,19 @@
+import json
+import redis
+
+from allocation import config
+from .wait_for import wait_for
+
+r = redis.Redis(**config.get_redis_host_and_port())
+
+
+def subscribe_to(channel):
+    pubsub = r.pubsub()
+    pubsub.subscribe(channel)
+    confirmation = wait_for(pubsub.get_message)
+    assert confirmation['type'] == 'subscribe'
+    return pubsub
+
+
+def publish_message(channel, message):
+    r.publish(channel, json.dumps(message))
