summaryrefslogtreecommitdiff
path: root/tests/foreign_object/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/foreign_object/tests.py')
-rw-r--r--tests/foreign_object/tests.py53
1 files changed, 52 insertions, 1 deletions
diff --git a/tests/foreign_object/tests.py b/tests/foreign_object/tests.py
index 06e2551e4f..5188645325 100644
--- a/tests/foreign_object/tests.py
+++ b/tests/foreign_object/tests.py
@@ -2,7 +2,9 @@ import datetime
from operator import attrgetter
from django.core.exceptions import FieldError
-from django.test import TestCase, skipUnlessDBFeature
+from django.db import models
+from django.db.models.fields.related import ForeignObject
+from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature
from django.utils import translation
from .models import (
@@ -391,3 +393,52 @@ class MultiColumnFKTests(TestCase):
""" See: https://code.djangoproject.com/ticket/21566 """
objs = [Person(name="abcd_%s" % i, person_country=self.usa) for i in range(0, 5)]
Person.objects.bulk_create(objs, 10)
+
+
+class TestModelCheckTests(SimpleTestCase):
+
+ def test_check_composite_foreign_object(self):
+ class Parent(models.Model):
+ a = models.PositiveIntegerField()
+ b = models.PositiveIntegerField()
+
+ class Meta:
+ unique_together = (('a', 'b'),)
+
+ class Child(models.Model):
+ a = models.PositiveIntegerField()
+ b = models.PositiveIntegerField()
+ value = models.CharField(max_length=255)
+ parent = ForeignObject(
+ Parent,
+ on_delete=models.SET_NULL,
+ from_fields=('a', 'b'),
+ to_fields=('a', 'b'),
+ related_name='children',
+ )
+
+ self.assertEqual(Child._meta.get_field('parent').check(from_model=Child), [])
+
+ def test_check_subset_composite_foreign_object(self):
+ class Parent(models.Model):
+ a = models.PositiveIntegerField()
+ b = models.PositiveIntegerField()
+ c = models.PositiveIntegerField()
+
+ class Meta:
+ unique_together = (('a', 'b'),)
+
+ class Child(models.Model):
+ a = models.PositiveIntegerField()
+ b = models.PositiveIntegerField()
+ c = models.PositiveIntegerField()
+ d = models.CharField(max_length=255)
+ parent = ForeignObject(
+ Parent,
+ on_delete=models.SET_NULL,
+ from_fields=('a', 'b', 'c'),
+ to_fields=('a', 'b', 'c'),
+ related_name='children',
+ )
+
+ self.assertEqual(Child._meta.get_field('parent').check(from_model=Child), [])