From 8cc506840e271cefc53a1bdd97559a7e7b8e504c Mon Sep 17 00:00:00 2001
From: Jacob Kaplan-Moss <jacob@jacobian.org>
Date: Thu, 2 Apr 2009 04:32:48 +0000
Subject: [PATCH] Added tests for database introspection. Refs #9779.
git-svn-id: http://code.djangoproject.com/svn/django/trunk@10362 bcc190cf-cafb-0310-a4f2-bffc1f526a37
---
.../regressiontests/introspection/__init__.py | 0
tests/regressiontests/introspection/models.py | 20 ++++
tests/regressiontests/introspection/tests.py | 99 +++++++++++++++++++
3 files changed, 119 insertions(+)
create mode 100644 tests/regressiontests/introspection/__init__.py
create mode 100644 tests/regressiontests/introspection/models.py
create mode 100644 tests/regressiontests/introspection/tests.py
diff --git a/tests/regressiontests/introspection/__init__.py b/tests/regressiontests/introspection/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
diff --git a/tests/regressiontests/introspection/models.py b/tests/regressiontests/introspection/models.py
new file mode 100644
index 0000000000..3227067bd8
--- /dev/null
+++ b/tests/regressiontests/introspection/models.py
@@ -0,0 +1,20 @@
+from django.db import models
+
+class Reporter(models.Model):
+ first_name = models.CharField(max_length=30)
+ last_name = models.CharField(max_length=30)
+ email = models.EmailField()
+
+ def __unicode__(self):
+ return u"%s %s" % (self.first_name, self.last_name)
+
+class Article(models.Model):
+ headline = models.CharField(max_length=100)
+ pub_date = models.DateField()
+ reporter = models.ForeignKey(Reporter)
+
+ def __unicode__(self):
+ return self.headline
+
+ class Meta:
+ ordering = ('headline',)
\ No newline at end of file
diff --git a/tests/regressiontests/introspection/tests.py b/tests/regressiontests/introspection/tests.py
new file mode 100644
index 0000000000..658470d1ee
--- /dev/null
+++ b/tests/regressiontests/introspection/tests.py
@@ -0,0 +1,99 @@
+from django.db import connection
+from django.test import TestCase
+from django.utils import functional
+
+from models import Reporter, Article
+
+#
+# The introspection module is optional, so methods tested here might raise
+# NotImplementedError. This is perfectly acceptable behavior for the backend
+# in question, but the tests need to handle this without failing. Ideally we'd
+# skip these tests, but until #4788 is done we'll just ignore them.
+#
+# The easiest way to accomplish this is to decorate every test case with a
+# wrapper that ignores the exception.
+#
+# The metaclass is just for fun.
+#
+
+def ignore_not_implemented(func):
+ def _inner(*args, **kwargs):
+ try:
+ return func(*args, **kwargs)
+ except NotImplementedError:
+ return None
+ functional.update_wrapper(_inner, func)
+ return _inner
+
+class IgnoreNotimplementedError(type):
+ def __new__(cls, name, bases, attrs):
+ for k,v in attrs.items():
+ if k.startswith('test'):
+ attrs[k] = ignore_not_implemented(v)
+ return type.__new__(cls, name, bases, attrs)
+
+class IntrospectionTests(TestCase):
+ __metaclass__ = IgnoreNotimplementedError
+
+ def test_table_names(self):
+ tl = connection.introspection.table_names()
+ self.assert_(Reporter._meta.db_table in tl,
+ "'%s' isn't in table_list()." % Reporter._meta.db_table)
+ self.assert_(Article._meta.db_table in tl,
+ "'%s' isn't in table_list()." % Article._meta.db_table)
+
+ def test_django_table_names(self):
+ cursor = connection.cursor()
+ cursor.execute('CREATE TABLE django_introspection_testcase_table (id INTEGER);');
+ tl = connection.introspection.django_table_names()
+ self.assert_('django_introspection_testcase_table' not in tl,
+ "django_table_names() returned a non-Django table")
+ cursor.execute("DROP TABLE django_introspection_testcase_table;")
+
+ def test_installed_models(self):
+ tables = [Article._meta.db_table, Reporter._meta.db_table]
+ models = connection.introspection.installed_models(tables)
+ self.assertEqual(models, set([Article, Reporter]))
+
+ def test_sequence_list(self):
+ sequences = connection.introspection.sequence_list()
+ expected = {'table': Reporter._meta.db_table, 'column': 'id'}
+ self.assert_(expected in sequences,
+ 'Reporter sequence not found in sequence_list()')
+
+ def test_get_table_description_names(self):
+ cursor = connection.cursor()
+ desc = connection.introspection.get_table_description(cursor, Reporter._meta.db_table)
+ self.assertEqual([r[0] for r in desc],
+ [f.column for f in Reporter._meta.fields])
+
+ def test_get_table_description_types(self):
+ cursor = connection.cursor()
+ desc = connection.introspection.get_table_description(cursor, Reporter._meta.db_table)
+
+ # Convert the datatype into a string
+ def datatype(dbtype):
+ dt = connection.introspection.data_types_reverse[dbtype]
+ if type(dt) is tuple:
+ return dt[0]
+ else:
+ return dt
+
+ self.assertEqual([datatype(r[1]) for r in desc],
+ ['IntegerField', 'CharField', 'CharField', 'CharField'])
+
+ def test_get_relations(self):
+ cursor = connection.cursor()
+ relations = connection.introspection.get_relations(cursor, Article._meta.db_table)
+
+ # Older versions of MySQL don't have the chops to report on this stuff,
+ # so just skip it if no relations come back. If they do, though, we
+ # should test that the response is correct.
+ if relations:
+ # That's {field_index: (field_index_other_table, other_table)}
+ self.assertEqual(relations, {3: (0, Reporter._meta.db_table)})
+
+ def test_get_indexes(self):
+ cursor = connection.cursor()
+ indexes = connection.introspection.get_indexes(cursor, Article._meta.db_table)
+ self.assertEqual(indexes['reporter_id'], {'unique': False, 'primary_key': False})
\ No newline at end of file