From c47fa3b4814240bb47342a4446d40ea83bd3ed19 Mon Sep 17 00:00:00 2001 From: Aymeric Augustin Date: Mon, 28 Jan 2013 10:21:07 +0100 Subject: Fixed #19676 -- Supported 'self' foreign keys in inspectdb. Thanks Georgy Kutsurua for the report and Simon Charette for the patch. --- tests/regressiontests/inspectdb/models.py | 1 + tests/regressiontests/inspectdb/tests.py | 2 ++ 2 files changed, 3 insertions(+) (limited to 'tests') diff --git a/tests/regressiontests/inspectdb/models.py b/tests/regressiontests/inspectdb/models.py index 4a6621402f..a8adf862b1 100644 --- a/tests/regressiontests/inspectdb/models.py +++ b/tests/regressiontests/inspectdb/models.py @@ -3,6 +3,7 @@ from django.db import models class People(models.Model): name = models.CharField(max_length=255) + parent = models.ForeignKey('self') class Message(models.Model): from_field = models.ForeignKey(People, db_column='from_id') diff --git a/tests/regressiontests/inspectdb/tests.py b/tests/regressiontests/inspectdb/tests.py index 028d263337..f9cbbce5c4 100644 --- a/tests/regressiontests/inspectdb/tests.py +++ b/tests/regressiontests/inspectdb/tests.py @@ -31,6 +31,8 @@ class InspectDBTestCase(TestCase): stdout=out) output = out.getvalue() error_message = "inspectdb generated an attribute name which is a python keyword" + # Recursive foreign keys should be set to 'self' + self.assertIn("parent = models.ForeignKey('self')", output) self.assertNotIn("from = models.ForeignKey(InspectdbPeople)", output, msg=error_message) # As InspectdbPeople model is defined after InspectdbMessage, it should be quoted self.assertIn("from_field = models.ForeignKey('InspectdbPeople', db_column='from_id')", -- cgit v1.3