summaryrefslogtreecommitdiff
path: root/tests/introspection
diff options
context:
space:
mode:
authorClaude Paroz <claude@2xlibre.net>2016-08-20 10:28:42 +0200
committerClaude Paroz <claude@2xlibre.net>2016-08-20 15:57:17 +0200
commit00bb47b58f974277330f9845aee0702f7a46d736 (patch)
treec152b55f5a127430c7a647dd7d5b7ce072923b1f /tests/introspection
parent9f27735612fc775380d7801c68af0ea1c97cf5a3 (diff)
downloaddjango-00bb47b58f974277330f9845aee0702f7a46d736.tar.gz
Fixed #27096 -- Fixed primary key introspection for sqlite3 backend
Diffstat (limited to 'tests/introspection')
-rw-r--r--tests/introspection/models.py2
-rw-r--r--tests/introspection/tests.py4
2 files changed, 4 insertions, 2 deletions
diff --git a/tests/introspection/models.py b/tests/introspection/models.py
index f989709e79..6d6650bc24 100644
--- a/tests/introspection/models.py
+++ b/tests/introspection/models.py
@@ -15,7 +15,7 @@ class City(models.Model):
@python_2_unicode_compatible
class District(models.Model):
- city = models.ForeignKey(City, models.CASCADE)
+ city = models.ForeignKey(City, models.CASCADE, primary_key=True)
name = models.CharField(max_length=50)
def __str__(self):
diff --git a/tests/introspection/tests.py b/tests/introspection/tests.py
index f01f81e718..5e2d6f1c99 100644
--- a/tests/introspection/tests.py
+++ b/tests/introspection/tests.py
@@ -6,7 +6,7 @@ from django.db import connection
from django.db.utils import DatabaseError
from django.test import TransactionTestCase, mock, skipUnlessDBFeature
-from .models import Article, ArticleReporter, City, Reporter
+from .models import Article, ArticleReporter, City, District, Reporter
class IntrospectionTests(TransactionTestCase):
@@ -165,7 +165,9 @@ class IntrospectionTests(TransactionTestCase):
def test_get_primary_key_column(self):
with connection.cursor() as cursor:
primary_key_column = connection.introspection.get_primary_key_column(cursor, Article._meta.db_table)
+ pk_fk_column = connection.introspection.get_primary_key_column(cursor, District._meta.db_table)
self.assertEqual(primary_key_column, 'id')
+ self.assertEqual(pk_fk_column, 'city_id')
def test_get_indexes(self):
with connection.cursor() as cursor: