summaryrefslogtreecommitdiff
path: root/tests/check_framework
diff options
context:
space:
mode:
authorSanyam Khurana <sanyam.khurana01@gmail.com>2018-12-13 01:41:50 +0530
committerTim Graham <timograham@gmail.com>2018-12-24 10:40:11 -0500
commit5d25804eaf81795c7d457e5a2a9f0b9b0989136c (patch)
tree8b7dbceae0f842ec3d22b6059f35931cb935aab8 /tests/check_framework
parente626a3f993fc0581a3e0770a3930ac1fc20196cb (diff)
Fixed #20098 -- Added a check for model Meta.db_table collisions.
Diffstat (limited to 'tests/check_framework')
-rw-r--r--tests/check_framework/test_model_checks.py75
1 files changed, 75 insertions, 0 deletions
diff --git a/tests/check_framework/test_model_checks.py b/tests/check_framework/test_model_checks.py
new file mode 100644
index 0000000000..2e55ad637d
--- /dev/null
+++ b/tests/check_framework/test_model_checks.py
@@ -0,0 +1,75 @@
+from django.core import checks
+from django.core.checks import Error
+from django.db import models
+from django.test import SimpleTestCase
+from django.test.utils import (
+ isolate_apps, modify_settings, override_system_checks,
+)
+
+
+@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',
+ )
+ ])
+
+ @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',
+ )
+ ])
+
+ 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()), [])