[英]Django unit test failing for multiple Postgres schemas
我的Postgres DB具有3个架构:默认,cedirData和webData。
对于那些指向与默认模式不同的模式的模型,我将其指定如下:
class Person(models.Model):
first_name = models.CharField(max_length=200, null=False, blank=False)
last_name = models.CharField(max_length=200, null=False, blank=False)
class Meta:
db_table = 'cedirData\".\"persons'
该应用程序运行正常,但是当我尝试运行测试时:
$ ./manage.py test
我得到以下内容:
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/core/management/commands/migrate.py", line 160, in handle
executor.migrate(targets, plan, fake=options.get("fake", False))
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/migrations/executor.py", line 63, in migrate
self.apply_migration(migration, fake=fake)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/migrations/executor.py", line 97, in apply_migration
migration.apply(project_state, schema_editor)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/migrations/migration.py", line 107, in apply
operation.database_forwards(self.app_label, schema_editor, project_state, new_state)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/migrations/operations/models.py", line 36, in database_forwards
schema_editor.create_model(model)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/backends/schema.py", line 270, in create_model
self.execute(sql, params)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/backends/schema.py", line 98, in execute
cursor.execute(sql, params)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/backends/utils.py", line 65, in execute
return self.cursor.execute(sql, params)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/utils.py", line 94, in __exit__
six.reraise(dj_exc_type, dj_exc_value, traceback)
File "/home/wbrunetti/.virtualenvs/cedir/local/lib/python2.7/site-packages/django/db/backends/utils.py", line 65, in execute
return self.cursor.execute(sql, params)
django.db.utils.ProgrammingError: schema "cedirData" does not exist
看起来它可能与迁移有关。 由于数据库表已经存在,所以我创建了初始迁移并运行了--fake:
$ python manage.py makemigrations
$ ./manage.py migrate --fake
仅使用默认架构创建测试数据库。
我正在使用Django 1.7和Python 2.7.6。
任何想法或想法都会有所帮助。
谢谢!
在许多其他数据库引擎中未使用架构。 通过在模型中指定架构,您已经在代码中引入了对postgres的依赖。
您可以采取两种方法来解决问题:
首先,您可以向postgres用户添加默认搜索路径。 这种方法的缺点是不能再将模式用于命名空间,但是优点是,如果您的数据库曾经更改为其他引擎,则代码将正常运行。 可以通过选择一些标准的表命名方式来实现表的命名空间,类似于Django默认使用的方式(例如,appName_className)
有两种方法可以实现此目的。 使用这种方法的postgres命令是:
ALTER USER (your user) SET search_path = "$user",(schema1),(schema2),(schema3),(...)
仅限django的方式是:
# Warning! This is untested, I just glanced at the docs and it looks right.
DATABASES = {
'default': {
'ENGINE': 'django.db.backends.postgresql_psycopg2',
# some configuration here
'OPTIONS': {
'options': '-c search_path=schema1,schema2,schema3'
}
}
}
您还需要更改:
db_table = 'cedirData\".\"persons'
至:
db_table = 'persons'
作为奖励,您现在可以使用:
manage.py inspectdb > models.py
这是一个不错的功能,这样您就不必手动复制现有数据库。
但是,如果在数据库中大量使用了模式命名空间,而其他应用程序则依赖该命名空间,则该解决方案将无济于事。 一种不同的方法是编写一个自定义的testrunner,以在测试数据库中创建那些模式。 这比上面的方法要复杂得多,并且可能有点混乱。 我真的不建议这样做,但是如果您有兴趣,我可以尝试提供帮助。
较不混乱,但更“棘手”的方法是在运行测试时简单地覆盖meta。 这也将是一个测试运行者。
from django.test.simple import DjangoTestSuiteRunner
from django.db.models.loading import get_models
class SchemaModelTestRunner(DjangoTestSuiteRunner):
"""Docstring"""
def setup_test_environment(self, *args, **kwargs):
self.original_db_tables = {}
self.schema_models = [m for m in get_models()
if '"."' in m._meta.db_table]
for m in self.schema_models:
schema, table = m._meta.db_table.split('"."')
self.original_db_tables[m] = m._meta.db_table
m._meta.db_table = 'schema_'+schema+'_table_'+table
super(SchemaModelTestRunner, self).setup_test_environment(*args,
**kwargs)
def teardown_test_environment(self, *args, **kwargs):
super(SchemaModelTestRunner, self).teardown_test_environment(*args,
**kwargs)
# reset models
for m in self.schema_models:
m._meta.db_table = self.original_db_tables[m]
您还需要在settings.py文件中将其定义为testrunner。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.