Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Migrated many-to-many doctests. Thanks to George Sakkis for the patch.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@14285 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information...
commit c0cb75ce6410b2be7745001f911a0505c5358d66 1 parent 1a270bd
@freakboy3742 freakboy3742 authored
View
248 tests/modeltests/many_to_many/models.py
@@ -27,251 +27,3 @@ def __unicode__(self):
class Meta:
ordering = ('headline',)
-
-__test__ = {'API_TESTS':"""
-# Create a couple of Publications.
->>> p1 = Publication(id=None, title='The Python Journal')
->>> p1.save()
->>> p2 = Publication(id=None, title='Science News')
->>> p2.save()
->>> p3 = Publication(id=None, title='Science Weekly')
->>> p3.save()
-
-# Create an Article.
->>> a1 = Article(id=None, headline='Django lets you build Web apps easily')
-
-# You can't associate it with a Publication until it's been saved.
->>> a1.publications.add(p1)
-Traceback (most recent call last):
-...
-ValueError: 'Article' instance needs to have a primary key value before a many-to-many relationship can be used.
-
-# Save it!
->>> a1.save()
-
-# Associate the Article with a Publication.
->>> a1.publications.add(p1)
-
-# Create another Article, and set it to appear in both Publications.
->>> a2 = Article(id=None, headline='NASA uses Python')
->>> a2.save()
->>> a2.publications.add(p1, p2)
->>> a2.publications.add(p3)
-
-# Adding a second time is OK
->>> a2.publications.add(p3)
-
-# Adding an object of the wrong type raises TypeError
->>> a2.publications.add(a1)
-Traceback (most recent call last):
-...
-TypeError: 'Publication' instance expected
-
-# Add a Publication directly via publications.add by using keyword arguments.
->>> new_publication = a2.publications.create(title='Highlights for Children')
-
-# Article objects have access to their related Publication objects.
->>> a1.publications.all()
-[<Publication: The Python Journal>]
->>> a2.publications.all()
-[<Publication: Highlights for Children>, <Publication: Science News>, <Publication: Science Weekly>, <Publication: The Python Journal>]
-
-# Publication objects have access to their related Article objects.
->>> p2.article_set.all()
-[<Article: NASA uses Python>]
->>> p1.article_set.all()
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
->>> Publication.objects.get(id=4).article_set.all()
-[<Article: NASA uses Python>]
-
-# We can perform kwarg queries across m2m relationships
->>> Article.objects.filter(publications__id__exact=1)
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
->>> Article.objects.filter(publications__pk=1)
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
->>> Article.objects.filter(publications=1)
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
->>> Article.objects.filter(publications=p1)
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
-
->>> Article.objects.filter(publications__title__startswith="Science")
-[<Article: NASA uses Python>, <Article: NASA uses Python>]
-
->>> Article.objects.filter(publications__title__startswith="Science").distinct()
-[<Article: NASA uses Python>]
-
-# The count() function respects distinct() as well.
->>> Article.objects.filter(publications__title__startswith="Science").count()
-2
-
->>> Article.objects.filter(publications__title__startswith="Science").distinct().count()
-1
-
->>> Article.objects.filter(publications__in=[1,2]).distinct()
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
->>> Article.objects.filter(publications__in=[1,p2]).distinct()
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
->>> Article.objects.filter(publications__in=[p1,p2]).distinct()
-[<Article: Django lets you build Web apps easily>, <Article: NASA uses Python>]
-
-# Reverse m2m queries are supported (i.e., starting at the table that doesn't
-# have a ManyToManyField).
->>> Publication.objects.filter(id__exact=1)
-[<Publication: The Python Journal>]
->>> Publication.objects.filter(pk=1)
-[<Publication: The Python Journal>]
-
->>> Publication.objects.filter(article__headline__startswith="NASA")
-[<Publication: Highlights for Children>, <Publication: Science News>, <Publication: Science Weekly>, <Publication: The Python Journal>]
-
->>> Publication.objects.filter(article__id__exact=1)
-[<Publication: The Python Journal>]
->>> Publication.objects.filter(article__pk=1)
-[<Publication: The Python Journal>]
->>> Publication.objects.filter(article=1)
-[<Publication: The Python Journal>]
->>> Publication.objects.filter(article=a1)
-[<Publication: The Python Journal>]
-
->>> Publication.objects.filter(article__in=[1,2]).distinct()
-[<Publication: Highlights for Children>, <Publication: Science News>, <Publication: Science Weekly>, <Publication: The Python Journal>]
->>> Publication.objects.filter(article__in=[1,a2]).distinct()
-[<Publication: Highlights for Children>, <Publication: Science News>, <Publication: Science Weekly>, <Publication: The Python Journal>]
->>> Publication.objects.filter(article__in=[a1,a2]).distinct()
-[<Publication: Highlights for Children>, <Publication: Science News>, <Publication: Science Weekly>, <Publication: The Python Journal>]
-
-# Excluding a related item works as you would expect, too (although the SQL
-# involved is a little complex).
->>> Article.objects.exclude(publications=p2)
-[<Article: Django lets you build Web apps easily>]
-
-# If we delete a Publication, its Articles won't be able to access it.
->>> p1.delete()
->>> Publication.objects.all()
-[<Publication: Highlights for Children>, <Publication: Science News>, <Publication: Science Weekly>]
->>> a1 = Article.objects.get(pk=1)
->>> a1.publications.all()
-[]
-
-# If we delete an Article, its Publications won't be able to access it.
->>> a2.delete()
->>> Article.objects.all()
-[<Article: Django lets you build Web apps easily>]
->>> p2.article_set.all()
-[]
-
-# Adding via the 'other' end of an m2m
->>> a4 = Article(headline='NASA finds intelligent life on Earth')
->>> a4.save()
->>> p2.article_set.add(a4)
->>> p2.article_set.all()
-[<Article: NASA finds intelligent life on Earth>]
->>> a4.publications.all()
-[<Publication: Science News>]
-
-# Adding via the other end using keywords
->>> new_article = p2.article_set.create(headline='Oxygen-free diet works wonders')
->>> p2.article_set.all()
-[<Article: NASA finds intelligent life on Earth>, <Article: Oxygen-free diet works wonders>]
->>> a5 = p2.article_set.all()[1]
->>> a5.publications.all()
-[<Publication: Science News>]
-
-# Removing publication from an article:
->>> a4.publications.remove(p2)
->>> p2.article_set.all()
-[<Article: Oxygen-free diet works wonders>]
->>> a4.publications.all()
-[]
-
-# And from the other end
->>> p2.article_set.remove(a5)
->>> p2.article_set.all()
-[]
->>> a5.publications.all()
-[]
-
-# Relation sets can be assigned. Assignment clears any existing set members
->>> p2.article_set = [a4, a5]
->>> p2.article_set.all()
-[<Article: NASA finds intelligent life on Earth>, <Article: Oxygen-free diet works wonders>]
->>> a4.publications.all()
-[<Publication: Science News>]
->>> a4.publications = [p3]
->>> p2.article_set.all()
-[<Article: Oxygen-free diet works wonders>]
->>> a4.publications.all()
-[<Publication: Science Weekly>]
-
-# Relation sets can be cleared:
->>> p2.article_set.clear()
->>> p2.article_set.all()
-[]
->>> a4.publications.all()
-[<Publication: Science Weekly>]
-
-# And you can clear from the other end
->>> p2.article_set.add(a4, a5)
->>> p2.article_set.all()
-[<Article: NASA finds intelligent life on Earth>, <Article: Oxygen-free diet works wonders>]
->>> a4.publications.all()
-[<Publication: Science News>, <Publication: Science Weekly>]
->>> a4.publications.clear()
->>> a4.publications.all()
-[]
->>> p2.article_set.all()
-[<Article: Oxygen-free diet works wonders>]
-
-# Relation sets can also be set using primary key values
->>> p2.article_set = [a4.id, a5.id]
->>> p2.article_set.all()
-[<Article: NASA finds intelligent life on Earth>, <Article: Oxygen-free diet works wonders>]
->>> a4.publications.all()
-[<Publication: Science News>]
->>> a4.publications = [p3.id]
->>> p2.article_set.all()
-[<Article: Oxygen-free diet works wonders>]
->>> a4.publications.all()
-[<Publication: Science Weekly>]
-
-# Recreate the article and Publication we have deleted.
->>> p1 = Publication(id=None, title='The Python Journal')
->>> p1.save()
->>> a2 = Article(id=None, headline='NASA uses Python')
->>> a2.save()
->>> a2.publications.add(p1, p2, p3)
-
-# Bulk delete some Publications - references to deleted publications should go
->>> Publication.objects.filter(title__startswith='Science').delete()
->>> Publication.objects.all()
-[<Publication: Highlights for Children>, <Publication: The Python Journal>]
->>> Article.objects.all()
-[<Article: Django lets you build Web apps easily>, <Article: NASA finds intelligent life on Earth>, <Article: NASA uses Python>, <Article: Oxygen-free diet works wonders>]
->>> a2.publications.all()
-[<Publication: The Python Journal>]
-
-# Bulk delete some articles - references to deleted objects should go
->>> q = Article.objects.filter(headline__startswith='Django')
->>> print q
-[<Article: Django lets you build Web apps easily>]
->>> q.delete()
-
-# After the delete, the QuerySet cache needs to be cleared, and the referenced objects should be gone
->>> print q
-[]
->>> p1.article_set.all()
-[<Article: NASA uses Python>]
-
-# An alternate to calling clear() is to assign the empty set
->>> p1.article_set = []
->>> p1.article_set.all()
-[]
-
->>> a2.publications = [p1, new_publication]
->>> a2.publications.all()
-[<Publication: Highlights for Children>, <Publication: The Python Journal>]
->>> a2.publications = []
->>> a2.publications.all()
-[]
-
-"""}
View
384 tests/modeltests/many_to_many/tests.py
@@ -0,0 +1,384 @@
+from django.test import TestCase
+from models import Article, Publication
+
+class ManyToManyTests(TestCase):
+
+ def setUp(self):
+ # Create a couple of Publications.
+ self.p1 = Publication.objects.create(id=None, title='The Python Journal')
+ self.p2 = Publication.objects.create(id=None, title='Science News')
+ self.p3 = Publication.objects.create(id=None, title='Science Weekly')
+ self.p4 = Publication.objects.create(title='Highlights for Children')
+
+ self.a1 = Article.objects.create(id=None, headline='Django lets you build Web apps easily')
+ self.a1.publications.add(self.p1)
+
+ self.a2 = Article.objects.create(id=None, headline='NASA uses Python')
+ self.a2.publications.add(self.p1, self.p2, self.p3, self.p4)
+
+ self.a3 = Article.objects.create(headline='NASA finds intelligent life on Earth')
+ self.a3.publications.add(self.p2)
+
+ self.a4 = Article.objects.create(headline='Oxygen-free diet works wonders')
+ self.a4.publications.add(self.p2)
+
+ def test_add(self):
+ # Create an Article.
+ a5 = Article(id=None, headline='Django lets you reate Web apps easily')
+ # You can't associate it with a Publication until it's been saved.
+ self.assertRaises(ValueError, getattr, a5, 'publications')
+ # Save it!
+ a5.save()
+ # Associate the Article with a Publication.
+ a5.publications.add(self.p1)
+ self.assertQuerysetEqual(a5.publications.all(),
+ ['<Publication: The Python Journal>'])
+ # Create another Article, and set it to appear in both Publications.
+ a6 = Article(id=None, headline='ESA uses Python')
+ a6.save()
+ a6.publications.add(self.p1, self.p2)
+ a6.publications.add(self.p3)
+ # Adding a second time is OK
+ a6.publications.add(self.p3)
+ self.assertQuerysetEqual(a6.publications.all(),
+ [
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+
+ # Adding an object of the wrong type raises TypeError
+ self.assertRaises(TypeError, a6.publications.add, a5)
+ # Add a Publication directly via publications.add by using keyword arguments.
+ p4 = a6.publications.create(title='Highlights for Adults')
+ self.assertQuerysetEqual(a6.publications.all(),
+ [
+ '<Publication: Highlights for Adults>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+
+ def test_reverse_add(self):
+ # Adding via the 'other' end of an m2m
+ a5 = Article(headline='NASA finds intelligent life on Mars')
+ a5.save()
+ self.p2.article_set.add(a5)
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA finds intelligent life on Mars>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(a5.publications.all(),
+ ['<Publication: Science News>'])
+
+ # Adding via the other end using keywords
+ new_article = self.p2.article_set.create(headline='Carbon-free diet works wonders')
+ self.assertQuerysetEqual(
+ self.p2.article_set.all(),
+ [
+ '<Article: Carbon-free diet works wonders>',
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA finds intelligent life on Mars>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ a6 = self.p2.article_set.all()[3]
+ self.assertQuerysetEqual(a6.publications.all(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+
+ def test_related_sets(self):
+ # Article objects have access to their related Publication objects.
+ self.assertQuerysetEqual(self.a1.publications.all(),
+ ['<Publication: The Python Journal>'])
+ self.assertQuerysetEqual(self.a2.publications.all(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+ # Publication objects have access to their related Article objects.
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(self.p1.article_set.all(),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(Publication.objects.get(id=self.p4.id).article_set.all(),
+ ['<Article: NASA uses Python>'])
+
+ def test_selects(self):
+ # We can perform kwarg queries across m2m relationships
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__id__exact=self.p1.id),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__pk=self.p1.id),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications=self.p1.id),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications=self.p1),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__title__startswith="Science"),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__title__startswith="Science").distinct(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+
+ # The count() function respects distinct() as well.
+ self.assertEqual(Article.objects.filter(publications__title__startswith="Science").count(), 4)
+ self.assertEqual(Article.objects.filter(publications__title__startswith="Science").distinct().count(), 3)
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__in=[self.p1.id,self.p2.id]).distinct(),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__in=[self.p1.id,self.p2]).distinct(),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(
+ Article.objects.filter(publications__in=[self.p1,self.p2]).distinct(),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+
+ # Excluding a related item works as you would expect, too (although the SQL
+ # involved is a little complex).
+ self.assertQuerysetEqual(Article.objects.exclude(publications=self.p2),
+ ['<Article: Django lets you build Web apps easily>'])
+
+ def test_reverse_selects(self):
+ # Reverse m2m queries are supported (i.e., starting at the table that
+ # doesn't have a ManyToManyField).
+ self.assertQuerysetEqual(Publication.objects.filter(id__exact=self.p1.id),
+ ['<Publication: The Python Journal>'])
+ self.assertQuerysetEqual(Publication.objects.filter(pk=self.p1.id),
+ ['<Publication: The Python Journal>'])
+ self.assertQuerysetEqual(
+ Publication.objects.filter(article__headline__startswith="NASA"),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+ self.assertQuerysetEqual(Publication.objects.filter(article__id__exact=self.a1.id),
+ ['<Publication: The Python Journal>'])
+ self.assertQuerysetEqual(Publication.objects.filter(article__pk=self.a1.id),
+ ['<Publication: The Python Journal>'])
+ self.assertQuerysetEqual(Publication.objects.filter(article=self.a1.id),
+ ['<Publication: The Python Journal>'])
+ self.assertQuerysetEqual(Publication.objects.filter(article=self.a1),
+ ['<Publication: The Python Journal>'])
+
+ self.assertQuerysetEqual(
+ Publication.objects.filter(article__in=[self.a1.id,self.a2.id]).distinct(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+ self.assertQuerysetEqual(
+ Publication.objects.filter(article__in=[self.a1.id,self.a2]).distinct(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+ self.assertQuerysetEqual(
+ Publication.objects.filter(article__in=[self.a1,self.a2]).distinct(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ '<Publication: The Python Journal>',
+ ])
+
+ def test_delete(self):
+ # If we delete a Publication, its Articles won't be able to access it.
+ self.p1.delete()
+ self.assertQuerysetEqual(Publication.objects.all(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: Science News>',
+ '<Publication: Science Weekly>',
+ ])
+ self.assertQuerysetEqual(self.a1.publications.all(), [])
+ # If we delete an Article, its Publications won't be able to access it.
+ self.a2.delete()
+ self.assertQuerysetEqual(Article.objects.all(),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+
+ def test_bulk_delete(self):
+ # Bulk delete some Publications - references to deleted publications should go
+ Publication.objects.filter(title__startswith='Science').delete()
+ self.assertQuerysetEqual(Publication.objects.all(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: The Python Journal>',
+ ])
+ self.assertQuerysetEqual(Article.objects.all(),
+ [
+ '<Article: Django lets you build Web apps easily>',
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(self.a2.publications.all(),
+ [
+ '<Publication: Highlights for Children>',
+ '<Publication: The Python Journal>',
+ ])
+
+ # Bulk delete some articles - references to deleted objects should go
+ q = Article.objects.filter(headline__startswith='Django')
+ self.assertQuerysetEqual(q, ['<Article: Django lets you build Web apps easily>'])
+ q.delete()
+ # After the delete, the QuerySet cache needs to be cleared,
+ # and the referenced objects should be gone
+ self.assertQuerysetEqual(q, [])
+ self.assertQuerysetEqual(self.p1.article_set.all(),
+ ['<Article: NASA uses Python>'])
+
+ def test_remove(self):
+ # Removing publication from an article:
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.a4.publications.remove(self.p2)
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(self.a4.publications.all(), [])
+ # And from the other end
+ self.p2.article_set.remove(self.a3)
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA uses Python>',
+ ])
+ self.assertQuerysetEqual(self.a3.publications.all(), [])
+
+ def test_assign(self):
+ # Relation sets can be assigned. Assignment clears any existing set members
+ self.p2.article_set = [self.a4, self.a3]
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(self.a4.publications.all(),
+ ['<Publication: Science News>'])
+ self.a4.publications = [self.p3.id]
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ ['<Article: NASA finds intelligent life on Earth>'])
+ self.assertQuerysetEqual(self.a4.publications.all(),
+ ['<Publication: Science Weekly>'])
+
+ # An alternate to calling clear() is to assign the empty set
+ self.p2.article_set = []
+ self.assertQuerysetEqual(self.p2.article_set.all(), [])
+ self.a4.publications = []
+ self.assertQuerysetEqual(self.a4.publications.all(), [])
+
+ def test_assign_ids(self):
+ # Relation sets can also be set using primary key values
+ self.p2.article_set = [self.a4.id, self.a3.id]
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(self.a4.publications.all(),
+ ['<Publication: Science News>'])
+ self.a4.publications = [self.p3.id]
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ ['<Article: NASA finds intelligent life on Earth>'])
+ self.assertQuerysetEqual(self.a4.publications.all(),
+ ['<Publication: Science Weekly>'])
+
+ def test_clear(self):
+ # Relation sets can be cleared:
+ self.p2.article_set.clear()
+ self.assertQuerysetEqual(self.p2.article_set.all(), [])
+ self.assertQuerysetEqual(self.a4.publications.all(), [])
+
+ # And you can clear from the other end
+ self.p2.article_set.add(self.a3, self.a4)
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ [
+ '<Article: NASA finds intelligent life on Earth>',
+ '<Article: Oxygen-free diet works wonders>',
+ ])
+ self.assertQuerysetEqual(self.a4.publications.all(),
+ [
+ '<Publication: Science News>',
+ ])
+ self.a4.publications.clear()
+ self.assertQuerysetEqual(self.a4.publications.all(), [])
+ self.assertQuerysetEqual(self.p2.article_set.all(),
+ ['<Article: NASA finds intelligent life on Earth>'])

0 comments on commit c0cb75c

Please sign in to comment.
Something went wrong with that request. Please try again.