def test_map_to_sorted_set_by_score_column_number(self): self.db = mock_db(('id', 'name', 'salary'), \ [(1, 'A', 5), (2, 'B', 6), None]) self.mapper = Db2RedisMapper(self.db, self.redis) self.mapper.map(query=self.query) \ .to_sorted_set(key_pattern='zset:{id}', \ transform=take_nth(1), score=2) self.mapper.run() self.common_asserts() self.redis.pipeline().zadd.assert_any_call('zset:1', 5, 'A') self.redis.pipeline().zadd.assert_any_call('zset:2', 6, 'B')
def setUp(self): self.redis = Mock(name='redis-client-mock') self.db = mock_db(('id', 'name'), [(1, 'A'), (2, 'B'), None]) self.mapper = Db2RedisMapper(self.db, self.redis) self.query = 'select id, name from Table'