mirror of
https://github.com/django/django.git
synced 2024-12-23 01:25:58 +00:00
daf7d482db
Once the deprecation period ends CheckConstraint.check() can become the documented method that performs system checks for BaseConstraint subclasses.
547 lines
18 KiB
Python
547 lines
18 KiB
Python
from unittest import mock
|
|
|
|
from django.core import checks
|
|
from django.core.checks import Error, Warning
|
|
from django.db import models
|
|
from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature
|
|
from django.test.utils import (
|
|
isolate_apps,
|
|
modify_settings,
|
|
override_settings,
|
|
override_system_checks,
|
|
)
|
|
|
|
|
|
class EmptyRouter:
|
|
pass
|
|
|
|
|
|
@isolate_apps("check_framework", attr_name="apps")
|
|
@override_system_checks([checks.model_checks.check_all_models])
|
|
class DuplicateDBTableTests(SimpleTestCase):
|
|
def test_collision_in_same_app(self):
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"db_table 'test_table' is used by multiple models: "
|
|
"check_framework.Model1, check_framework.Model2.",
|
|
obj="test_table",
|
|
id="models.E028",
|
|
)
|
|
],
|
|
)
|
|
|
|
@override_settings(
|
|
DATABASE_ROUTERS=["check_framework.test_model_checks.EmptyRouter"]
|
|
)
|
|
def test_collision_in_same_app_database_routers_installed(self):
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Warning(
|
|
"db_table 'test_table' is used by multiple models: "
|
|
"check_framework.Model1, check_framework.Model2.",
|
|
hint=(
|
|
"You have configured settings.DATABASE_ROUTERS. Verify "
|
|
"that check_framework.Model1, check_framework.Model2 are "
|
|
"correctly routed to separate databases."
|
|
),
|
|
obj="test_table",
|
|
id="models.W035",
|
|
)
|
|
],
|
|
)
|
|
|
|
@modify_settings(INSTALLED_APPS={"append": "basic"})
|
|
@isolate_apps("basic", "check_framework", kwarg_name="apps")
|
|
def test_collision_across_apps(self, apps):
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
app_label = "basic"
|
|
db_table = "test_table"
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework"
|
|
db_table = "test_table"
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"db_table 'test_table' is used by multiple models: "
|
|
"basic.Model1, check_framework.Model2.",
|
|
obj="test_table",
|
|
id="models.E028",
|
|
)
|
|
],
|
|
)
|
|
|
|
@modify_settings(INSTALLED_APPS={"append": "basic"})
|
|
@override_settings(
|
|
DATABASE_ROUTERS=["check_framework.test_model_checks.EmptyRouter"]
|
|
)
|
|
@isolate_apps("basic", "check_framework", kwarg_name="apps")
|
|
def test_collision_across_apps_database_routers_installed(self, apps):
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
app_label = "basic"
|
|
db_table = "test_table"
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework"
|
|
db_table = "test_table"
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=apps.get_app_configs()),
|
|
[
|
|
Warning(
|
|
"db_table 'test_table' is used by multiple models: "
|
|
"basic.Model1, check_framework.Model2.",
|
|
hint=(
|
|
"You have configured settings.DATABASE_ROUTERS. Verify "
|
|
"that basic.Model1, check_framework.Model2 are correctly "
|
|
"routed to separate databases."
|
|
),
|
|
obj="test_table",
|
|
id="models.W035",
|
|
)
|
|
],
|
|
)
|
|
|
|
def test_no_collision_for_unmanaged_models(self):
|
|
class Unmanaged(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
managed = False
|
|
|
|
class Managed(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
def test_no_collision_for_proxy_models(self):
|
|
class Model(models.Model):
|
|
class Meta:
|
|
db_table = "test_table"
|
|
|
|
class ProxyModel(Model):
|
|
class Meta:
|
|
proxy = True
|
|
|
|
self.assertEqual(Model._meta.db_table, ProxyModel._meta.db_table)
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
|
|
@isolate_apps("check_framework", attr_name="apps")
|
|
@override_system_checks([checks.model_checks.check_all_models])
|
|
class IndexNameTests(SimpleTestCase):
|
|
def test_collision_in_same_model(self):
|
|
index = models.Index(fields=["id"], name="foo")
|
|
|
|
class Model(models.Model):
|
|
class Meta:
|
|
indexes = [index, index]
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"index name 'foo' is not unique for model check_framework.Model.",
|
|
id="models.E029",
|
|
),
|
|
],
|
|
)
|
|
|
|
def test_collision_in_different_models(self):
|
|
index = models.Index(fields=["id"], name="foo")
|
|
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
indexes = [index]
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
indexes = [index]
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"index name 'foo' is not unique among models: "
|
|
"check_framework.Model1, check_framework.Model2.",
|
|
id="models.E030",
|
|
),
|
|
],
|
|
)
|
|
|
|
def test_collision_abstract_model(self):
|
|
class AbstractModel(models.Model):
|
|
class Meta:
|
|
indexes = [models.Index(fields=["id"], name="foo")]
|
|
abstract = True
|
|
|
|
class Model1(AbstractModel):
|
|
pass
|
|
|
|
class Model2(AbstractModel):
|
|
pass
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"index name 'foo' is not unique among models: "
|
|
"check_framework.Model1, check_framework.Model2.",
|
|
id="models.E030",
|
|
),
|
|
],
|
|
)
|
|
|
|
def test_no_collision_abstract_model_interpolation(self):
|
|
class AbstractModel(models.Model):
|
|
name = models.CharField(max_length=20)
|
|
|
|
class Meta:
|
|
indexes = [
|
|
models.Index(fields=["name"], name="%(app_label)s_%(class)s_foo")
|
|
]
|
|
abstract = True
|
|
|
|
class Model1(AbstractModel):
|
|
pass
|
|
|
|
class Model2(AbstractModel):
|
|
pass
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
@modify_settings(INSTALLED_APPS={"append": "basic"})
|
|
@isolate_apps("basic", "check_framework", kwarg_name="apps")
|
|
def test_collision_across_apps(self, apps):
|
|
index = models.Index(fields=["id"], name="foo")
|
|
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
app_label = "basic"
|
|
indexes = [index]
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework"
|
|
indexes = [index]
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"index name 'foo' is not unique among models: basic.Model1, "
|
|
"check_framework.Model2.",
|
|
id="models.E030",
|
|
),
|
|
],
|
|
)
|
|
|
|
@modify_settings(INSTALLED_APPS={"append": "basic"})
|
|
@isolate_apps("basic", "check_framework", kwarg_name="apps")
|
|
def test_no_collision_across_apps_interpolation(self, apps):
|
|
index = models.Index(fields=["id"], name="%(app_label)s_%(class)s_foo")
|
|
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
app_label = "basic"
|
|
constraints = [index]
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework"
|
|
constraints = [index]
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=apps.get_app_configs()), [])
|
|
|
|
|
|
@isolate_apps("check_framework", attr_name="apps")
|
|
@override_system_checks([checks.model_checks.check_all_models])
|
|
@skipUnlessDBFeature("supports_table_check_constraints")
|
|
class ConstraintNameTests(TestCase):
|
|
def test_collision_in_same_model(self):
|
|
class Model(models.Model):
|
|
class Meta:
|
|
constraints = [
|
|
models.CheckConstraint(condition=models.Q(id__gt=0), name="foo"),
|
|
models.CheckConstraint(condition=models.Q(id__lt=100), name="foo"),
|
|
]
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"constraint name 'foo' is not unique for model "
|
|
"check_framework.Model.",
|
|
id="models.E031",
|
|
),
|
|
],
|
|
)
|
|
|
|
def test_collision_in_different_models(self):
|
|
constraint = models.CheckConstraint(condition=models.Q(id__gt=0), name="foo")
|
|
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
constraints = [constraint]
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
constraints = [constraint]
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"constraint name 'foo' is not unique among models: "
|
|
"check_framework.Model1, check_framework.Model2.",
|
|
id="models.E032",
|
|
),
|
|
],
|
|
)
|
|
|
|
def test_collision_abstract_model(self):
|
|
class AbstractModel(models.Model):
|
|
class Meta:
|
|
constraints = [
|
|
models.CheckConstraint(condition=models.Q(id__gt=0), name="foo")
|
|
]
|
|
abstract = True
|
|
|
|
class Model1(AbstractModel):
|
|
pass
|
|
|
|
class Model2(AbstractModel):
|
|
pass
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"constraint name 'foo' is not unique among models: "
|
|
"check_framework.Model1, check_framework.Model2.",
|
|
id="models.E032",
|
|
),
|
|
],
|
|
)
|
|
|
|
def test_no_collision_abstract_model_interpolation(self):
|
|
class AbstractModel(models.Model):
|
|
class Meta:
|
|
constraints = [
|
|
models.CheckConstraint(
|
|
condition=models.Q(id__gt=0), name="%(app_label)s_%(class)s_foo"
|
|
),
|
|
]
|
|
abstract = True
|
|
|
|
class Model1(AbstractModel):
|
|
pass
|
|
|
|
class Model2(AbstractModel):
|
|
pass
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
@modify_settings(INSTALLED_APPS={"append": "basic"})
|
|
@isolate_apps("basic", "check_framework", kwarg_name="apps")
|
|
def test_collision_across_apps(self, apps):
|
|
constraint = models.CheckConstraint(condition=models.Q(id__gt=0), name="foo")
|
|
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
app_label = "basic"
|
|
constraints = [constraint]
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework"
|
|
constraints = [constraint]
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=apps.get_app_configs()),
|
|
[
|
|
Error(
|
|
"constraint name 'foo' is not unique among models: "
|
|
"basic.Model1, check_framework.Model2.",
|
|
id="models.E032",
|
|
),
|
|
],
|
|
)
|
|
|
|
@modify_settings(INSTALLED_APPS={"append": "basic"})
|
|
@isolate_apps("basic", "check_framework", kwarg_name="apps")
|
|
def test_no_collision_across_apps_interpolation(self, apps):
|
|
constraint = models.CheckConstraint(
|
|
condition=models.Q(id__gt=0), name="%(app_label)s_%(class)s_foo"
|
|
)
|
|
|
|
class Model1(models.Model):
|
|
class Meta:
|
|
app_label = "basic"
|
|
constraints = [constraint]
|
|
|
|
class Model2(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework"
|
|
constraints = [constraint]
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=apps.get_app_configs()), [])
|
|
|
|
|
|
def mocked_is_overridden(self, setting):
|
|
# Force treating DEFAULT_AUTO_FIELD = 'django.db.models.AutoField' as a not
|
|
# overridden setting.
|
|
return (
|
|
setting != "DEFAULT_AUTO_FIELD"
|
|
or self.DEFAULT_AUTO_FIELD != "django.db.models.AutoField"
|
|
)
|
|
|
|
|
|
@mock.patch("django.conf.UserSettingsHolder.is_overridden", mocked_is_overridden)
|
|
@override_settings(DEFAULT_AUTO_FIELD="django.db.models.AutoField")
|
|
@isolate_apps("check_framework.apps.CheckDefaultPKConfig", attr_name="apps")
|
|
@override_system_checks([checks.model_checks.check_all_models])
|
|
class ModelDefaultAutoFieldTests(SimpleTestCase):
|
|
msg = (
|
|
"Auto-created primary key used when not defining a primary key type, "
|
|
"by default 'django.db.models.AutoField'."
|
|
)
|
|
hint = (
|
|
"Configure the DEFAULT_AUTO_FIELD setting or the "
|
|
"CheckDefaultPKConfig.default_auto_field attribute to point to a "
|
|
"subclass of AutoField, e.g. 'django.db.models.BigAutoField'."
|
|
)
|
|
|
|
def test_auto_created_pk(self):
|
|
class Model(models.Model):
|
|
pass
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Warning(self.msg, hint=self.hint, obj=Model, id="models.W042"),
|
|
],
|
|
)
|
|
|
|
def test_explicit_inherited_pk(self):
|
|
class Parent(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
|
|
class Child(Parent):
|
|
pass
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
def test_skipped_on_model_with_invalid_app_label(self):
|
|
class Model(models.Model):
|
|
class Meta:
|
|
app_label = "invalid_app_label"
|
|
|
|
self.assertEqual(Model.check(), [])
|
|
|
|
def test_skipped_on_abstract_model(self):
|
|
class Abstract(models.Model):
|
|
class Meta:
|
|
abstract = True
|
|
|
|
# Call .check() because abstract models are not registered.
|
|
self.assertEqual(Abstract.check(), [])
|
|
|
|
def test_explicit_inherited_parent_link(self):
|
|
class Parent(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
|
|
class Child(Parent):
|
|
parent_ptr = models.OneToOneField(Parent, models.CASCADE, parent_link=True)
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
def test_auto_created_inherited_pk(self):
|
|
class Parent(models.Model):
|
|
pass
|
|
|
|
class Child(Parent):
|
|
pass
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Warning(self.msg, hint=self.hint, obj=Parent, id="models.W042"),
|
|
],
|
|
)
|
|
|
|
def test_auto_created_inherited_parent_link(self):
|
|
class Parent(models.Model):
|
|
pass
|
|
|
|
class Child(Parent):
|
|
parent_ptr = models.OneToOneField(Parent, models.CASCADE, parent_link=True)
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Warning(self.msg, hint=self.hint, obj=Parent, id="models.W042"),
|
|
],
|
|
)
|
|
|
|
def test_auto_created_pk_inherited_abstract_parent(self):
|
|
class Parent(models.Model):
|
|
class Meta:
|
|
abstract = True
|
|
|
|
class Child(Parent):
|
|
pass
|
|
|
|
self.assertEqual(
|
|
checks.run_checks(app_configs=self.apps.get_app_configs()),
|
|
[
|
|
Warning(self.msg, hint=self.hint, obj=Child, id="models.W042"),
|
|
],
|
|
)
|
|
|
|
@override_settings(DEFAULT_AUTO_FIELD="django.db.models.BigAutoField")
|
|
def test_default_auto_field_setting(self):
|
|
class Model(models.Model):
|
|
pass
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
def test_explicit_pk(self):
|
|
class Model(models.Model):
|
|
id = models.BigAutoField(primary_key=True)
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=self.apps.get_app_configs()), [])
|
|
|
|
@isolate_apps("check_framework.apps.CheckPKConfig", kwarg_name="apps")
|
|
def test_app_default_auto_field(self, apps):
|
|
class ModelWithPkViaAppConfig(models.Model):
|
|
class Meta:
|
|
app_label = "check_framework.apps.CheckPKConfig"
|
|
|
|
self.assertEqual(checks.run_checks(app_configs=apps.get_app_configs()), [])
|