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'
