test_session_integration.py 文件源码

def test_if_two_commits_sets_redis_with_two_models_correctly(
            self, session, redis):
        session.add(await Model10.new(session, id=1))
        session.add(await Model11.new(session, id=1))
        await session.commit()

        assert await redis.hgetall(Model10.__key__) == {
            b'1': ujson.dumps({b'id': 1}).encode()
        }
        assert await redis.hgetall(Model11.__key__) == {
            b'1': ujson.dumps({b'id': 1}).encode()
        }

        session.add(await Model10.new(session, id=2))
        session.add(await Model11.new(session, id=2))
        await session.commit()

        assert await redis.hgetall(Model10.__key__) == {
            b'1': ujson.dumps({b'id': 1}).encode(),
            b'2': ujson.dumps({b'id': 2}).encode()
        }
        assert await redis.hgetall(Model11.__key__) == {
            b'1': ujson.dumps({b'id': 1}).encode(),
            b'2': ujson.dumps({b'id': 2}).encode()
        }