test_redis.py 文件源码

def test_hmset_with_objects_and_without_ids_with_set_map_len_greater_than_chunks(self, model):
        session = mock.MagicMock()
        session.redis_bind.hkeys.return_value = ['2'.encode(), '1'.encode()]
        model.CHUNKS = 1
        expected_map1 = {
            '1'.encode(): msgpack.dumps({'id': 1})
        }
        expected_map2 = {
            '2'.encode(): msgpack.dumps({'id': 2})
        }

        assert model.update(session, [{'id': 1}, {'id': 2}]) == [{'id': 1}, {'id': 2}]
        assert (session.redis_bind.hmset.call_args_list == [
            mock.call('test', expected_map1),
            mock.call('test', expected_map2)
        ] or session.redis_bind.hmset.call_args_list == [
            mock.call('test', expected_map2),
            mock.call('test', expected_map1)
        ])