diff options
Diffstat (limited to 'tests/custom_columns/tests.py')
| -rw-r--r-- | tests/custom_columns/tests.py | 63 |
1 files changed, 35 insertions, 28 deletions
diff --git a/tests/custom_columns/tests.py b/tests/custom_columns/tests.py index 20e649de03..32a48b4840 100644 --- a/tests/custom_columns/tests.py +++ b/tests/custom_columns/tests.py @@ -5,22 +5,25 @@ from .models import Article, Author class CustomColumnsTests(TestCase): - @classmethod def setUpTestData(cls): cls.a1 = Author.objects.create(first_name="John", last_name="Smith") cls.a2 = Author.objects.create(first_name="Peter", last_name="Jones") cls.authors = [cls.a1, cls.a2] - cls.article = Article.objects.create(headline="Django lets you build web apps easily", primary_author=cls.a1) + cls.article = Article.objects.create( + headline="Django lets you build web apps easily", primary_author=cls.a1 + ) cls.article.authors.set(cls.authors) def test_query_all_available_authors(self): self.assertQuerysetEqual( - Author.objects.all(), [ - "Peter Jones", "John Smith", + Author.objects.all(), + [ + "Peter Jones", + "John Smith", ], - str + str, ) def test_get_first_name(self): @@ -31,10 +34,11 @@ class CustomColumnsTests(TestCase): def test_filter_first_name(self): self.assertQuerysetEqual( - Author.objects.filter(first_name__exact="John"), [ + Author.objects.filter(first_name__exact="John"), + [ "John Smith", ], - str + str, ) def test_field_error(self): @@ -54,43 +58,42 @@ class CustomColumnsTests(TestCase): def test_get_all_authors_for_an_article(self): self.assertQuerysetEqual( - self.article.authors.all(), [ + self.article.authors.all(), + [ "Peter Jones", "John Smith", ], - str + str, ) def test_get_all_articles_for_an_author(self): self.assertQuerysetEqual( - self.a1.article_set.all(), [ + self.a1.article_set.all(), + [ "Django lets you build web apps easily", ], - lambda a: a.headline + lambda a: a.headline, ) def test_get_author_m2m_relation(self): self.assertQuerysetEqual( - self.article.authors.filter(last_name='Jones'), [ - "Peter Jones" - ], - str + self.article.authors.filter(last_name="Jones"), ["Peter Jones"], str ) def test_author_querying(self): self.assertSequenceEqual( - Author.objects.all().order_by('last_name'), + Author.objects.all().order_by("last_name"), [self.a2, self.a1], ) def test_author_filtering(self): self.assertSequenceEqual( - Author.objects.filter(first_name__exact='John'), + Author.objects.filter(first_name__exact="John"), [self.a1], ) def test_author_get(self): - self.assertEqual(self.a1, Author.objects.get(first_name__exact='John')) + self.assertEqual(self.a1, Author.objects.get(first_name__exact="John")) def test_filter_on_nonexistent_field(self): msg = ( @@ -98,25 +101,29 @@ class CustomColumnsTests(TestCase): "Author_ID, article, first_name, last_name, primary_set" ) with self.assertRaisesMessage(FieldError, msg): - Author.objects.filter(firstname__exact='John') + 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) - with self.assertRaisesMessage(AttributeError, "'Author' object has no attribute 'firstname'"): - getattr(a, 'firstname') + a = Author.objects.get(last_name__exact="Smith") + self.assertEqual("John", a.first_name) + self.assertEqual("Smith", a.last_name) + with self.assertRaisesMessage( + AttributeError, "'Author' object has no attribute 'firstname'" + ): + getattr(a, "firstname") - with self.assertRaisesMessage(AttributeError, "'Author' object has no attribute 'last'"): - getattr(a, 'last') + with self.assertRaisesMessage( + AttributeError, "'Author' object has no attribute 'last'" + ): + getattr(a, "last") def test_m2m_table(self): self.assertSequenceEqual( - self.article.authors.all().order_by('last_name'), + self.article.authors.all().order_by("last_name"), [self.a2, self.a1], ) self.assertSequenceEqual(self.a1.article_set.all(), [self.article]) self.assertSequenceEqual( - self.article.authors.filter(last_name='Jones'), + self.article.authors.filter(last_name="Jones"), [self.a2], ) |
