summaryrefslogtreecommitdiff
path: root/parts/django/tests/modeltests/raw_query/tests.py
diff options
context:
space:
mode:
authorNishanth Amuluru2011-01-08 11:20:57 +0530
committerNishanth Amuluru2011-01-08 11:20:57 +0530
commit65411d01d448ff0cd4abd14eee14cf60b5f8fc20 (patch)
treeb4c404363c4c63a61d6e2f8bd26c5b057c1fb09d /parts/django/tests/modeltests/raw_query/tests.py
parent2e35094d43b4cc6974172e1febf76abb50f086ec (diff)
downloadpytask-65411d01d448ff0cd4abd14eee14cf60b5f8fc20.tar.gz
pytask-65411d01d448ff0cd4abd14eee14cf60b5f8fc20.tar.bz2
pytask-65411d01d448ff0cd4abd14eee14cf60b5f8fc20.zip
Added buildout stuff and made changes accordingly
--HG-- rename : profile/management/__init__.py => eggs/djangorecipe-0.20-py2.6.egg/EGG-INFO/dependency_links.txt rename : profile/management/__init__.py => eggs/djangorecipe-0.20-py2.6.egg/EGG-INFO/not-zip-safe rename : profile/management/__init__.py => eggs/infrae.subversion-1.4.5-py2.6.egg/EGG-INFO/dependency_links.txt rename : profile/management/__init__.py => eggs/infrae.subversion-1.4.5-py2.6.egg/EGG-INFO/not-zip-safe rename : profile/management/__init__.py => eggs/mercurial-1.7.3-py2.6-linux-x86_64.egg/EGG-INFO/dependency_links.txt rename : profile/management/__init__.py => eggs/mercurial-1.7.3-py2.6-linux-x86_64.egg/EGG-INFO/not-zip-safe rename : profile/management/__init__.py => eggs/py-1.4.0-py2.6.egg/EGG-INFO/dependency_links.txt rename : profile/management/__init__.py => eggs/py-1.4.0-py2.6.egg/EGG-INFO/not-zip-safe rename : profile/management/__init__.py => eggs/zc.buildout-1.5.2-py2.6.egg/EGG-INFO/dependency_links.txt rename : profile/management/__init__.py => eggs/zc.buildout-1.5.2-py2.6.egg/EGG-INFO/not-zip-safe rename : profile/management/__init__.py => eggs/zc.recipe.egg-1.3.2-py2.6.egg/EGG-INFO/dependency_links.txt rename : profile/management/__init__.py => eggs/zc.recipe.egg-1.3.2-py2.6.egg/EGG-INFO/not-zip-safe rename : profile/management/__init__.py => parts/django/Django.egg-info/dependency_links.txt rename : taskapp/models.py => parts/django/django/conf/app_template/models.py rename : taskapp/tests.py => parts/django/django/conf/app_template/tests.py rename : taskapp/views.py => parts/django/django/conf/app_template/views.py rename : taskapp/views.py => parts/django/django/contrib/gis/tests/geo3d/views.py rename : profile/management/__init__.py => parts/django/tests/modeltests/delete/__init__.py rename : profile/management/__init__.py => parts/django/tests/modeltests/files/__init__.py rename : profile/management/__init__.py => parts/django/tests/modeltests/invalid_models/__init__.py rename : profile/management/__init__.py => parts/django/tests/modeltests/m2m_signals/__init__.py rename : profile/management/__init__.py => parts/django/tests/modeltests/model_package/__init__.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/bash_completion/__init__.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/bash_completion/management/__init__.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/bash_completion/management/commands/__init__.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/bash_completion/models.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/delete_regress/__init__.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/file_storage/__init__.py rename : profile/management/__init__.py => parts/django/tests/regressiontests/max_lengths/__init__.py rename : profile/forms.py => pytask/profile/forms.py rename : profile/management/__init__.py => pytask/profile/management/__init__.py rename : profile/management/commands/seed_db.py => pytask/profile/management/commands/seed_db.py rename : profile/models.py => pytask/profile/models.py rename : profile/templatetags/user_tags.py => pytask/profile/templatetags/user_tags.py rename : taskapp/tests.py => pytask/profile/tests.py rename : profile/urls.py => pytask/profile/urls.py rename : profile/utils.py => pytask/profile/utils.py rename : profile/views.py => pytask/profile/views.py rename : static/css/base.css => pytask/static/css/base.css rename : taskapp/tests.py => pytask/taskapp/tests.py rename : taskapp/views.py => pytask/taskapp/views.py rename : templates/base.html => pytask/templates/base.html rename : templates/profile/browse_notifications.html => pytask/templates/profile/browse_notifications.html rename : templates/profile/edit.html => pytask/templates/profile/edit.html rename : templates/profile/view.html => pytask/templates/profile/view.html rename : templates/profile/view_notification.html => pytask/templates/profile/view_notification.html rename : templates/registration/activate.html => pytask/templates/registration/activate.html rename : templates/registration/activation_email.txt => pytask/templates/registration/activation_email.txt rename : templates/registration/activation_email_subject.txt => pytask/templates/registration/activation_email_subject.txt rename : templates/registration/logged_out.html => pytask/templates/registration/logged_out.html rename : templates/registration/login.html => pytask/templates/registration/login.html rename : templates/registration/logout.html => pytask/templates/registration/logout.html rename : templates/registration/password_change_done.html => pytask/templates/registration/password_change_done.html rename : templates/registration/password_change_form.html => pytask/templates/registration/password_change_form.html rename : templates/registration/password_reset_complete.html => pytask/templates/registration/password_reset_complete.html rename : templates/registration/password_reset_confirm.html => pytask/templates/registration/password_reset_confirm.html rename : templates/registration/password_reset_done.html => pytask/templates/registration/password_reset_done.html rename : templates/registration/password_reset_email.html => pytask/templates/registration/password_reset_email.html rename : templates/registration/password_reset_form.html => pytask/templates/registration/password_reset_form.html rename : templates/registration/registration_complete.html => pytask/templates/registration/registration_complete.html rename : templates/registration/registration_form.html => pytask/templates/registration/registration_form.html rename : utils.py => pytask/utils.py
Diffstat (limited to 'parts/django/tests/modeltests/raw_query/tests.py')
-rw-r--r--parts/django/tests/modeltests/raw_query/tests.py236
1 files changed, 236 insertions, 0 deletions
diff --git a/parts/django/tests/modeltests/raw_query/tests.py b/parts/django/tests/modeltests/raw_query/tests.py
new file mode 100644
index 0000000..a1e7edb
--- /dev/null
+++ b/parts/django/tests/modeltests/raw_query/tests.py
@@ -0,0 +1,236 @@
+from datetime import date
+
+from django.conf import settings
+from django.db import connection
+from django.db.models.sql.query import InvalidQuery
+from django.test import TestCase
+
+from models import Author, Book, Coffee, Reviewer, FriendlyAuthor
+
+
+class RawQueryTests(TestCase):
+ fixtures = ['raw_query_books.json']
+
+ def assertSuccessfulRawQuery(self, model, query, expected_results,
+ expected_annotations=(), params=[], translations=None):
+ """
+ Execute the passed query against the passed model and check the output
+ """
+ results = list(model.objects.raw(query, params=params, translations=translations))
+ self.assertProcessed(model, results, expected_results, expected_annotations)
+ self.assertAnnotations(results, expected_annotations)
+
+ def assertProcessed(self, model, results, orig, expected_annotations=()):
+ """
+ Compare the results of a raw query against expected results
+ """
+ self.assertEqual(len(results), len(orig))
+ for index, item in enumerate(results):
+ orig_item = orig[index]
+ for annotation in expected_annotations:
+ setattr(orig_item, *annotation)
+
+ for field in model._meta.fields:
+ # Check that all values on the model are equal
+ self.assertEquals(getattr(item,field.attname),
+ getattr(orig_item,field.attname))
+ # This includes checking that they are the same type
+ self.assertEquals(type(getattr(item,field.attname)),
+ type(getattr(orig_item,field.attname)))
+
+ def assertNoAnnotations(self, results):
+ """
+ Check that the results of a raw query contain no annotations
+ """
+ self.assertAnnotations(results, ())
+
+ def assertAnnotations(self, results, expected_annotations):
+ """
+ Check that the passed raw query results contain the expected
+ annotations
+ """
+ if expected_annotations:
+ for index, result in enumerate(results):
+ annotation, value = expected_annotations[index]
+ self.assertTrue(hasattr(result, annotation))
+ self.assertEqual(getattr(result, annotation), value)
+
+ def assert_num_queries(self, n, func, *args, **kwargs):
+ old_DEBUG = settings.DEBUG
+ settings.DEBUG = True
+ starting_queries = len(connection.queries)
+ try:
+ func(*args, **kwargs)
+ finally:
+ settings.DEBUG = old_DEBUG
+ self.assertEqual(starting_queries + n, len(connection.queries))
+
+ def testSimpleRawQuery(self):
+ """
+ Basic test of raw query with a simple database query
+ """
+ query = "SELECT * FROM raw_query_author"
+ authors = Author.objects.all()
+ self.assertSuccessfulRawQuery(Author, query, authors)
+
+ def testRawQueryLazy(self):
+ """
+ Raw queries are lazy: they aren't actually executed until they're
+ iterated over.
+ """
+ q = Author.objects.raw('SELECT * FROM raw_query_author')
+ self.assert_(q.query.cursor is None)
+ list(q)
+ self.assert_(q.query.cursor is not None)
+
+ def testFkeyRawQuery(self):
+ """
+ Test of a simple raw query against a model containing a foreign key
+ """
+ query = "SELECT * FROM raw_query_book"
+ books = Book.objects.all()
+ self.assertSuccessfulRawQuery(Book, query, books)
+
+ def testDBColumnHandler(self):
+ """
+ Test of a simple raw query against a model containing a field with
+ db_column defined.
+ """
+ query = "SELECT * FROM raw_query_coffee"
+ coffees = Coffee.objects.all()
+ self.assertSuccessfulRawQuery(Coffee, query, coffees)
+
+ def testOrderHandler(self):
+ """
+ Test of raw raw query's tolerance for columns being returned in any
+ order
+ """
+ selects = (
+ ('dob, last_name, first_name, id'),
+ ('last_name, dob, first_name, id'),
+ ('first_name, last_name, dob, id'),
+ )
+
+ for select in selects:
+ query = "SELECT %s FROM raw_query_author" % select
+ authors = Author.objects.all()
+ self.assertSuccessfulRawQuery(Author, query, authors)
+
+ def testTranslations(self):
+ """
+ Test of raw query's optional ability to translate unexpected result
+ column names to specific model fields
+ """
+ query = "SELECT first_name AS first, last_name AS last, dob, id FROM raw_query_author"
+ translations = {'first': 'first_name', 'last': 'last_name'}
+ authors = Author.objects.all()
+ self.assertSuccessfulRawQuery(Author, query, authors, translations=translations)
+
+ def testParams(self):
+ """
+ Test passing optional query parameters
+ """
+ query = "SELECT * FROM raw_query_author WHERE first_name = %s"
+ author = Author.objects.all()[2]
+ params = [author.first_name]
+ results = list(Author.objects.raw(query, params=params))
+ self.assertProcessed(Author, results, [author])
+ self.assertNoAnnotations(results)
+ self.assertEqual(len(results), 1)
+
+ def testManyToMany(self):
+ """
+ Test of a simple raw query against a model containing a m2m field
+ """
+ query = "SELECT * FROM raw_query_reviewer"
+ reviewers = Reviewer.objects.all()
+ self.assertSuccessfulRawQuery(Reviewer, query, reviewers)
+
+ def testExtraConversions(self):
+ """
+ Test to insure that extra translations are ignored.
+ """
+ query = "SELECT * FROM raw_query_author"
+ translations = {'something': 'else'}
+ authors = Author.objects.all()
+ self.assertSuccessfulRawQuery(Author, query, authors, translations=translations)
+
+ def testMissingFields(self):
+ query = "SELECT id, first_name, dob FROM raw_query_author"
+ for author in Author.objects.raw(query):
+ self.assertNotEqual(author.first_name, None)
+ # last_name isn't given, but it will be retrieved on demand
+ self.assertNotEqual(author.last_name, None)
+
+ def testMissingFieldsWithoutPK(self):
+ query = "SELECT first_name, dob FROM raw_query_author"
+ try:
+ list(Author.objects.raw(query))
+ self.fail('Query without primary key should fail')
+ except InvalidQuery:
+ pass
+
+ def testAnnotations(self):
+ query = "SELECT a.*, count(b.id) as book_count FROM raw_query_author a LEFT JOIN raw_query_book b ON a.id = b.author_id GROUP BY a.id, a.first_name, a.last_name, a.dob ORDER BY a.id"
+ expected_annotations = (
+ ('book_count', 3),
+ ('book_count', 0),
+ ('book_count', 1),
+ ('book_count', 0),
+ )
+ authors = Author.objects.all()
+ self.assertSuccessfulRawQuery(Author, query, authors, expected_annotations)
+
+ def testInvalidQuery(self):
+ query = "UPDATE raw_query_author SET first_name='thing' WHERE first_name='Joe'"
+ self.assertRaises(InvalidQuery, Author.objects.raw, query)
+
+ def testWhiteSpaceQuery(self):
+ query = " SELECT * FROM raw_query_author"
+ authors = Author.objects.all()
+ self.assertSuccessfulRawQuery(Author, query, authors)
+
+ def testMultipleIterations(self):
+ query = "SELECT * FROM raw_query_author"
+ normal_authors = Author.objects.all()
+ raw_authors = Author.objects.raw(query)
+
+ # First Iteration
+ first_iterations = 0
+ for index, raw_author in enumerate(raw_authors):
+ self.assertEqual(normal_authors[index], raw_author)
+ first_iterations += 1
+
+ # Second Iteration
+ second_iterations = 0
+ for index, raw_author in enumerate(raw_authors):
+ self.assertEqual(normal_authors[index], raw_author)
+ second_iterations += 1
+
+ self.assertEqual(first_iterations, second_iterations)
+
+ def testGetItem(self):
+ # Indexing on RawQuerySets
+ query = "SELECT * FROM raw_query_author ORDER BY id ASC"
+ third_author = Author.objects.raw(query)[2]
+ self.assertEqual(third_author.first_name, 'Bob')
+
+ first_two = Author.objects.raw(query)[0:2]
+ self.assertEquals(len(first_two), 2)
+
+ self.assertRaises(TypeError, lambda: Author.objects.raw(query)['test'])
+
+ def test_inheritance(self):
+ # date is the end of the Cuban Missile Crisis, I have no idea when
+ # Wesley was bron
+ f = FriendlyAuthor.objects.create(first_name="Wesley", last_name="Chun",
+ dob=date(1962, 10, 28))
+ query = "SELECT * FROM raw_query_friendlyauthor"
+ self.assertEqual(
+ [o.pk for o in FriendlyAuthor.objects.raw(query)], [f.pk]
+ )
+
+ def test_query_count(self):
+ self.assert_num_queries(1,
+ list, Author.objects.raw("SELECT * FROM raw_query_author")
+ )