diff options
Diffstat (limited to 'tests/backends/sqlite')
| -rw-r--r-- | tests/backends/sqlite/test_introspection.py | 19 |
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): |
