Pony ORM --测试中的拆卸

Question

我使用Pony ORM来管理我正在开发的python包中的sqlite数据库。

我想使用pytest进行测试。

我的包提供了一个"agent“对象，用于连接到服务器API并检索”事件“。在代理初始化时，设置小马orm并将其绑定到sqlite db，在内存中(用于测试)或作为文件。

    def setup_db(filepath=None):
        if filepath:
            db.bind(provider="sqlite", filename=filepath, create_db=True)
        else:
            db.bind(provider="sqlite", filename=":memory:", create_db=True)
        db.provider.converter_classes.append((Enum, EnumConverter))
        db.generate_mapping(create_tables=True)

使用pony orm将事件的状态存储在sqlite db中。

我希望创建一个新的代理对象，每个测试都有一个干净的数据库，所以我在conftest.py文件中使用了pytest fixture。

    @pytest.fixture
    def agent():
        agent=Agent(parm1="param1",...)
        return agent

我无法正确地从数据库“解除绑定”，并在我的第二次测试中得到以下错误：

pony.orm.core.BindingError: Database object was already bound to SQLite provider

我想要一些关于最好的继续进行的方法的建议。谢谢。

Answer 1

我认为对于您的情况，您应该为实体创建一些工厂，并为每个设置创建新的db对象。

def define_entities(db):
    class Student(db.Entity):
        ...

    class Group(db.Entity):
        ...

这样你就可以像这样做了

def setup_db(filepath=None):
    db = Database()
    if filepath:
        db.bind(provider="sqlite", filename=filepath, create_db=True)
    else:
        db.bind(provider="sqlite", filename=":memory:", create_db=True)
    define_entities(db)
    db.provider.converter_classes.append((Enum, EnumConverter))
    db.generate_mapping(create_tables=True)

Answer 2

看一下Pony代码，似乎清除Database实例的provider属性就足够了，这样它就可以再次绑定。

如果您放弃Agent而不是从fixture返回它，那么放在yield语句之后的所有内容都将作为fixture拆卸代码运行。

Answer 3

previous answer from zgoda几乎可以正常工作。除了db.provider之外，还需要清除db.schema。我的建议是创建另一个函数：

def unbind_db():
    db.provider = db.schema = None

你的fixture应该是这样的：

@fixture
def database() -> None:
    setup_db()
    ...
    try:
       yield
    finally:
       unbind_db()