python-3.x - 如何为测试目的对 sqlite 数据库进行猴子补丁？

Question

在 ORM.py 文件中:

from peewee import *
db = SqliteDatabase('database.db')

class Device(Model):
    uid = CharField(unique=True, max_length=17)

    class Meta:
        database = db

现在在 test.py 中，我想对原始数据库进行猴子补丁 database.db与 test.db

from _pytest.monkeypatch import MonkeyPatch

@pytest.fixture(scope="session")
def monkeysession(request):
    mp = MonkeyPatch()
    yield mp
    mp.undo()


@pytest.fixture(scope='session', autouse=True)
def create_db(monkeysession, request):
    monkeysession.setattr(ORM, 'db', SqliteDatabase('test.db'))

但是我的错误(peewee.OperationalError: table "device"already exists)表明猴子补丁失败了

Answer 1

您可能想查看 Using 2.x 中的助手 http://docs.peewee-orm.com/en/latest/peewee/api.html#Using

或 bind()在 3.0a 中(尚未记录，抱歉)。

还有test_utils 2.x 和 3.0a 中的扩展:http://docs.peewee-orm.com/en/latest/peewee/playhouse.html#test_database