Ticket #18081: 18081-test.diff

tests/proxy_models/tests.py

@@ -8 +8 @@
 from django.contrib.contenttypes.models import ContentType
 from django.core import checks, exceptions, management
 from django.core.urlresolvers import reverse
-from django.db import DEFAULT_DB_ALIAS, models
+from django.db import DEFAULT_DB_ALIAS, IntegrityError, models, transaction
 from django.db.models import signals
-from django.test import TestCase, override_settings
+from django.test import TestCase, TransactionTestCase, override_settings
 
 from .admin import admin as force_admin_model_registration  # NOQA
 from .models import (
@@ -447 +447 @@
         delete_str = response.context['deleted_objects'][0]
         self.assertEqual(delete_str, proxy_str)
         self.client.logout()
+
+
+class TransactionalProxyModelTests(TransactionTestCase):
+    available_apps = ['proxy_models']
+
+    def test_proxy_fk(self):
+        """
+        Test that the DB contains proper foreign keys for proxy model references.
+        """
+        @transaction.atomic
+        def create_failing_pk():
+            t = TrackerUser.objects.create(status='bar')
+            proxy_tracker = ProxyTrackerUser.objects.get(pk=t.pk)
+            Improvement.objects.create(
+                summary='foof', version='foof',
+                reporter_id=1, associated_bug_id=1,
+                assignee=proxy_tracker,
+           )
+        self.assertRaises(IntegrityError, create_failing_pk)