diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_runner/models.py | 7 | ||||
| -rw-r--r-- | tests/test_runner/test_parallel.py | 5 |
2 files changed, 12 insertions, 0 deletions
diff --git a/tests/test_runner/models.py b/tests/test_runner/models.py index 80bf8dd8c7..54ec5e2b3d 100644 --- a/tests/test_runner/models.py +++ b/tests/test_runner/models.py @@ -6,6 +6,13 @@ class Person(models.Model): last_name = models.CharField(max_length=20) friends = models.ManyToManyField("self") + system_check_run_count = 0 + + @classmethod + def check(cls, *args, **kwargs): + cls.system_check_run_count += 1 + return super().check(**kwargs) + # A set of models that use a non-abstract inherited 'through' model. class ThroughBase(models.Model): diff --git a/tests/test_runner/test_parallel.py b/tests/test_runner/test_parallel.py index 3af0fbf2f9..f344f1a2db 100644 --- a/tests/test_runner/test_parallel.py +++ b/tests/test_runner/test_parallel.py @@ -8,6 +8,8 @@ from unittest.suite import TestSuite, _ErrorHolder from django.test import SimpleTestCase from django.test.runner import ParallelTestSuite, RemoteTestResult +from . import models + try: import tblib.pickling_support except ImportError: @@ -48,6 +50,9 @@ class ParallelTestRunnerTest(SimpleTestCase): with self.subTest(index=i): self.assertEqual(i, i) + def test_system_checks(self): + self.assertEqual(models.Person.system_check_run_count, 1) + class SampleFailingSubtest(SimpleTestCase): # This method name doesn't begin with "test" to prevent test discovery |
