如何使用 hypothesis-extra-sqlalchemy 在 Python 中对 SQLAlchemy 进行属性测试

pip install hypothesis-extra-sqlalchemy

from hypothesis import given, settings
from hypothesis.extra.sqlalchemy import column, TableData
from hypothesis.strategies import text
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String)
    email = Column(String)

engine = create_engine('sqlite:///:memory:')
Session = sessionmaker(bind=engine)
Base.metadata.create_all(engine)

user_data = TableData(
    'users',
    column('id', Integer()),
    column('name', text(min_size=1)),
    column('email', text(min_size=5)),
)

@given(user_data.draw())
@settings(deadline=1000)
def test_create_user(user_data):
    session = Session()
    user = User(**user_data)
    session.add(user)
    session.commit()
    assert user.id is not None