From 9907495b3c412ece76f5e5e4c4e5df12b93caf78 Mon Sep 17 00:00:00 2001 From: Jacob Kaplan-Moss Date: Fri, 3 Apr 2009 19:56:30 +0000 Subject: [1.0.X] Fixed #9546: GenericRelations inherited from base models no longer query using the wrong content type. Backport of r10373 from trunk. git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.0.X@10374 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- .../generic_relations_regress/__init__.py | 0 .../generic_relations_regress/models.py | 22 ++++++++++++++++++++++ .../generic_relations_regress/tests.py | 19 +++++++++++++++++++ 3 files changed, 41 insertions(+) create mode 100644 tests/regressiontests/generic_relations_regress/__init__.py create mode 100644 tests/regressiontests/generic_relations_regress/models.py create mode 100644 tests/regressiontests/generic_relations_regress/tests.py (limited to 'tests') diff --git a/tests/regressiontests/generic_relations_regress/__init__.py b/tests/regressiontests/generic_relations_regress/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/tests/regressiontests/generic_relations_regress/models.py b/tests/regressiontests/generic_relations_regress/models.py new file mode 100644 index 0000000000..1a4826aaed --- /dev/null +++ b/tests/regressiontests/generic_relations_regress/models.py @@ -0,0 +1,22 @@ +from django.db import models +from django.contrib.contenttypes import generic +from django.contrib.contenttypes.models import ContentType + +class Link(models.Model): + content_type = models.ForeignKey(ContentType) + object_id = models.PositiveIntegerField() + content_object = generic.GenericForeignKey() + + def __unicode__(self): + return "Link to %s id=%s" % (self.content_type, self.object_id) + +class Place(models.Model): + name = models.CharField(max_length=100) + links = generic.GenericRelation(Link) + + def __unicode__(self): + return "Place: %s" % self.name + +class Restaurant(Place): + def __unicode__(self): + return "Restaurant: %s" % self.name \ No newline at end of file diff --git a/tests/regressiontests/generic_relations_regress/tests.py b/tests/regressiontests/generic_relations_regress/tests.py new file mode 100644 index 0000000000..6f0863d7b4 --- /dev/null +++ b/tests/regressiontests/generic_relations_regress/tests.py @@ -0,0 +1,19 @@ +from django.test import TestCase +from django.contrib.contenttypes.models import ContentType +from models import Link, Place, Restaurant + +class GenericRelationTests(TestCase): + + def test_inherited_models_content_type(self): + """ + Test that GenericRelations on inherited classes use the correct content + type. + """ + + p = Place.objects.create(name="South Park") + r = Restaurant.objects.create(name="Chubby's") + l1 = Link.objects.create(content_object=p) + l2 = Link.objects.create(content_object=r) + self.assertEqual(list(p.links.all()), [l1]) + self.assertEqual(list(r.links.all()), [l2]) + \ No newline at end of file -- cgit v1.3