def setUp(self):
connection = MagicMock(spec=Connection)
cursor = MagicMock(spec=Cursor)
connection.cursor.return_value = cursor
cursor.__aenter__ = CoroutineMock(return_value=cursor)
cursor.execute = CoroutineMock()
self.accessor = DataAccessor(connection)
self.connection = connection
self.cursor = cursor
def setUp(self):
connection = MagicMock(spec=Connection)
cursor = MagicMock(spec=Cursor)
connection.cursor.return_value = cursor
cursor.__aenter__ = CoroutineMock(return_value=cursor)
votes = [('user_a', 'item_a', 0), ('user_a', 'item_b', 1),
('user_b', 'item_a', 1), ('user_b', 'item_b', 1)]
vote_scale = (0, 1)
cursor.fetchall = CoroutineMock(return_value=votes)
cursor.fetchone = CoroutineMock(return_value=vote_scale)
cursor.execute = CoroutineMock()
self.accessor = DataAccessor(connection)
self.connection = connection
self.cursor = cursor
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
