test_session_integration.py 文件源码

def test_if_two_commits_sets_redis_with_two_models_correctly(
            self, session, model1, model2, redis):
        session.add(model1(session, id=1))
        session.add(model2(session, id=1))
        session.commit()
        session.add(model1(session, id=2))
        session.add(model2(session, id=2))
        session.commit()

        expected = [
            mock.call('test1', {b'1': msgpack.dumps({'id': 1})}),
            mock.call('test2', {b'1': msgpack.dumps({'id': 1})}),
            mock.call('test1', {b'2': msgpack.dumps({'id': 2})}),
            mock.call('test2', {b'2': msgpack.dumps({'id': 2})})
        ]

        assert len(expected) == len(redis.hmset.call_args_list)

        for call_ in redis.hmset.call_args_list:
            assert call_ in expected