如何使用sqllite在pytest中启用外键检查

2022-03-01 00:00:00 python django pytest sqlite

我有一个Django项目,其测试在我调用py.test时运行,但我最近注意到它没有检查外键约束。如何让它检查外键约束?

显然,foreign key constraints weren't even possible until sqlite 3,但是我真的不知道我正在运行什么版本,因为我没有sqlite的CLI,但是它只是被Django自动包括在内?(我使用的是Django 1.9.10),但是sqlite3是在2009年发布的,所以这不是问题所在,对吧?

也许it must still be enabled by the application at runtime, using the PRAGMA foreign_keys command.,但我不知道如何使我的测试做到这一点?

[更新] 因此,看起来sqlite开箱即用并未检查它们。

class Referenced(models.Model):
    pass

class Referencer(models.Model):
    fk = models.ForeignKey(Referenced)

>>> Referencer.objects.create(fk_id=-1)
<Referencer>
>>> Referencer.objects.all()[0].fk
DoesNotExist

解决方案

我最终创建了一个包来解决此问题。

pip install sqlite_checkforeignkeys

然后配置数据库:

DATABASES = {
    'default': {
        'ENGINE': 'sqlite_checkforeignkeys_engine',
        'TEST': {'NAME': ':memory:'},
    }
}

它涉及重写现有的SQLite引擎以强制执行外键

有关详细信息,请访问https://github.com/hulu/sqlite_checkforeignkeys

相关文章