diff options
Diffstat (limited to 'tests/custom_columns_regress/tests.py')
-rw-r--r-- | tests/custom_columns_regress/tests.py | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/tests/custom_columns_regress/tests.py b/tests/custom_columns_regress/tests.py new file mode 100644 index 0000000000..7cc66ca2a6 --- /dev/null +++ b/tests/custom_columns_regress/tests.py @@ -0,0 +1,80 @@ +from __future__ import absolute_import + +from django.core.exceptions import FieldError +from django.test import TestCase + +from .models import Author, Article + + +def pks(objects): + """ Return pks to be able to compare lists""" + return [o.pk for o in objects] + +class CustomColumnRegression(TestCase): + + def setUp(self): + self.a1 = Author.objects.create(first_name='John', last_name='Smith') + self.a2 = Author.objects.create(first_name='Peter', last_name='Jones') + self.authors = [self.a1, self.a2] + + def test_basic_creation(self): + art = Article(headline='Django lets you build Web apps easily', primary_author=self.a1) + art.save() + art.authors = [self.a1, self.a2] + + def test_author_querying(self): + self.assertQuerysetEqual( + Author.objects.all().order_by('last_name'), + ['<Author: Peter Jones>', '<Author: John Smith>'] + ) + + def test_author_filtering(self): + self.assertQuerysetEqual( + Author.objects.filter(first_name__exact='John'), + ['<Author: John Smith>'] + ) + + def test_author_get(self): + self.assertEqual(self.a1, Author.objects.get(first_name__exact='John')) + + def test_filter_on_nonexistant_field(self): + self.assertRaisesMessage( + FieldError, + "Cannot resolve keyword 'firstname' into field. Choices are: Author_ID, article, first_name, last_name, primary_set", + Author.objects.filter, + firstname__exact='John' + ) + + def test_author_get_attributes(self): + a = Author.objects.get(last_name__exact='Smith') + self.assertEqual('John', a.first_name) + self.assertEqual('Smith', a.last_name) + self.assertRaisesMessage( + AttributeError, + "'Author' object has no attribute 'firstname'", + getattr, + a, 'firstname' + ) + + self.assertRaisesMessage( + AttributeError, + "'Author' object has no attribute 'last'", + getattr, + a, 'last' + ) + + def test_m2m_table(self): + art = Article.objects.create(headline='Django lets you build Web apps easily', primary_author=self.a1) + art.authors = self.authors + self.assertQuerysetEqual( + art.authors.all().order_by('last_name'), + ['<Author: Peter Jones>', '<Author: John Smith>'] + ) + self.assertQuerysetEqual( + self.a1.article_set.all(), + ['<Article: Django lets you build Web apps easily>'] + ) + self.assertQuerysetEqual( + art.authors.filter(last_name='Jones'), + ['<Author: Peter Jones>'] + ) |