summaryrefslogtreecommitdiff
path: root/tests/backends/sqlite
diff options
context:
space:
mode:
Diffstat (limited to 'tests/backends/sqlite')
-rw-r--r--tests/backends/sqlite/test_introspection.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/tests/backends/sqlite/test_introspection.py b/tests/backends/sqlite/test_introspection.py
index e378e0ee56..9331b5bb1a 100644
--- a/tests/backends/sqlite/test_introspection.py
+++ b/tests/backends/sqlite/test_introspection.py
@@ -28,6 +28,25 @@ class IntrospectionTests(TestCase):
finally:
cursor.execute('DROP TABLE test_primary')
+ def test_get_primary_key_column_pk_constraint(self):
+ sql = """
+ CREATE TABLE test_primary(
+ id INTEGER NOT NULL,
+ created DATE,
+ PRIMARY KEY(id)
+ )
+ """
+ with connection.cursor() as cursor:
+ try:
+ cursor.execute(sql)
+ field = connection.introspection.get_primary_key_column(
+ cursor,
+ 'test_primary',
+ )
+ self.assertEqual(field, 'id')
+ finally:
+ cursor.execute('DROP TABLE test_primary')
+
@unittest.skipUnless(connection.vendor == 'sqlite', 'SQLite tests')
class ParsingTests(TestCase):