test_redis.py 文件源码

def test_hmset_with_objects_and_with_ids_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}]) == [
            {'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)
        ])