def redis_connection(): conn = MagicMock() conn.__enter__ = MagicMock(return_value=conn) conn.__exit__ = MagicMock() conn.get = CoroutineMock() conn.mget = CoroutineMock() conn.set = CoroutineMock() conn.setex = CoroutineMock() conn.mset = CoroutineMock() conn.incrby = CoroutineMock() conn.exists = CoroutineMock() conn.persist = CoroutineMock() conn.expire = CoroutineMock() conn.delete = CoroutineMock() conn.flushdb = CoroutineMock() conn.eval = CoroutineMock() conn.keys = CoroutineMock() conn.multi_exec = MagicMock(return_value=conn) conn.execute = CoroutineMock() return conn