Author: russellm
Date: 2010-10-11 19:56:19 -0500 (Mon, 11 Oct 2010)
New Revision: 14167
Added:
django/trunk/tests/modeltests/one_to_one/tests.py
Modified:
django/trunk/tests/modeltests/one_to_one/models.py
Log:
Migrated one_to_one doctests. Thanks to George Sakkis for the patch. (We have
always been at war with doctests)
Modified: django/trunk/tests/modeltests/one_to_one/models.py
===================================================================
--- django/trunk/tests/modeltests/one_to_one/models.py 2010-10-12 00:55:55 UTC
(rev 14166)
+++ django/trunk/tests/modeltests/one_to_one/models.py 2010-10-12 00:56:19 UTC
(rev 14167)
@@ -45,150 +45,3 @@
def __unicode__(self):
return u"Multimodel %s" % self.name
-
-__test__ = {'API_TESTS':"""
-# Create a couple of Places.
->>> p1 = Place(name='Demon Dogs', address='944 W. Fullerton')
->>> p1.save()
->>> p2 = Place(name='Ace Hardware', address='1013 N. Ashland')
->>> p2.save()
-
-# Create a Restaurant. Pass the ID of the "parent" object as this object's ID.
->>> r = Restaurant(place=p1, serves_hot_dogs=True, serves_pizza=False)
->>> r.save()
-
-# A Restaurant can access its place.
->>> r.place
-<Place: Demon Dogs the place>
-
-# A Place can access its restaurant, if available.
->>> p1.restaurant
-<Restaurant: Demon Dogs the restaurant>
-
-# p2 doesn't have an associated restaurant.
->>> p2.restaurant
-Traceback (most recent call last):
- ...
-DoesNotExist: Restaurant matching query does not exist.
-
-# Set the place using assignment notation. Because place is the primary key on
-# Restaurant, the save will create a new restaurant
->>> r.place = p2
->>> r.save()
->>> p2.restaurant
-<Restaurant: Ace Hardware the restaurant>
->>> r.place
-<Place: Ace Hardware the place>
->>> p2.id
-2
-
-# Set the place back again, using assignment in the reverse direction.
->>> p1.restaurant = r
->>> p1.restaurant
-<Restaurant: Demon Dogs the restaurant>
-
->>> r = Restaurant.objects.get(pk=1)
->>> r.place
-<Place: Demon Dogs the place>
-
-# Restaurant.objects.all() just returns the Restaurants, not the Places.
-# Note that there are two restaurants - Ace Hardware the Restaurant was created
-# in the call to r.place = p2.
->>> Restaurant.objects.all()
-[<Restaurant: Demon Dogs the restaurant>, <Restaurant: Ace Hardware the
restaurant>]
-
-# Place.objects.all() returns all Places, regardless of whether they have
-# Restaurants.
->>> Place.objects.order_by('name')
-[<Place: Ace Hardware the place>, <Place: Demon Dogs the place>]
-
->>> Restaurant.objects.get(place__id__exact=1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(pk=1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(place__exact=1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(place__exact=p1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(place=1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(place=p1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(place__pk=1)
-<Restaurant: Demon Dogs the restaurant>
->>> Restaurant.objects.get(place__name__startswith="Demon")
-<Restaurant: Demon Dogs the restaurant>
-
->>> Place.objects.get(id__exact=1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(pk=1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant__place__exact=1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant__place__exact=p1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant__pk=1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant=1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant=r)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant__exact=1)
-<Place: Demon Dogs the place>
->>> Place.objects.get(restaurant__exact=r)
-<Place: Demon Dogs the place>
-
-# Add a Waiter to the Restaurant.
->>> w = r.waiter_set.create(name='Joe')
->>> w.save()
->>> w
-<Waiter: Joe the waiter at Demon Dogs the restaurant>
-
-# Query the waiters
->>> Waiter.objects.filter(restaurant__place__pk=1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(restaurant__place__exact=1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(restaurant__place__exact=p1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(restaurant__pk=1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(id__exact=1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(pk=1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(restaurant=1)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
->>> Waiter.objects.filter(restaurant=r)
-[<Waiter: Joe the waiter at Demon Dogs the restaurant>]
-
-# Delete the restaurant; the waiter should also be removed
->>> r = Restaurant.objects.get(pk=1)
->>> r.delete()
-
-# One-to-one fields still work if you create your own primary key
->>> o1 = ManualPrimaryKey(primary_key="abc123", name="primary")
->>> o1.save()
->>> o2 = RelatedModel(link=o1, name="secondary")
->>> o2.save()
-
-# You can have multiple one-to-one fields on a model, too.
->>> x1 = MultiModel(link1=p1, link2=o1, name="x1")
->>> x1.save()
->>> o1.multimodel
-<MultiModel: Multimodel x1>
-
-# This will fail because each one-to-one field must be unique (and link2=o1 was
-# used for x1, above).
->>> sid = transaction.savepoint()
->>> try:
-... MultiModel(link1=p2, link2=o1, name="x1").save()
-... except Exception, e:
-... if isinstance(e, IntegrityError):
-... print "Pass"
-... else:
-... print "Fail with %s" % type(e)
-Pass
->>> transaction.savepoint_rollback(sid)
-
-"""}
Added: django/trunk/tests/modeltests/one_to_one/tests.py
===================================================================
--- django/trunk/tests/modeltests/one_to_one/tests.py
(rev 0)
+++ django/trunk/tests/modeltests/one_to_one/tests.py 2010-10-12 00:56:19 UTC
(rev 14167)
@@ -0,0 +1,119 @@
+from django.test import TestCase
+from django.db import transaction, IntegrityError
+from models import Place, Restaurant, Waiter, ManualPrimaryKey, RelatedModel,
MultiModel
+
+class OneToOneTests(TestCase):
+
+ def setUp(self):
+ self.p1 = Place(name='Demon Dogs', address='944 W. Fullerton')
+ self.p1.save()
+ self.p2 = Place(name='Ace Hardware', address='1013 N. Ashland')
+ self.p2.save()
+ self.r = Restaurant(place=self.p1, serves_hot_dogs=True,
serves_pizza=False)
+ self.r.save()
+
+ def test_getter(self):
+ # A Restaurant can access its place.
+ self.assertEqual(repr(self.r.place), '<Place: Demon Dogs the place>')
+ # A Place can access its restaurant, if available.
+ self.assertEqual(repr(self.p1.restaurant), '<Restaurant: Demon Dogs
the restaurant>')
+ # p2 doesn't have an associated restaurant.
+ self.assertRaises(Restaurant.DoesNotExist, getattr, self.p2,
'restaurant')
+
+ def test_setter(self):
+ # Set the place using assignment notation. Because place is the primary
+ # key on Restaurant, the save will create a new restaurant
+ self.r.place = self.p2
+ self.r.save()
+ self.assertEqual(repr(self.p2.restaurant), '<Restaurant: Ace Hardware
the restaurant>')
+ self.assertEqual(repr(self.r.place), '<Place: Ace Hardware the place>')
+ self.assertEqual(self.p2.pk, self.r.pk)
+ # Set the place back again, using assignment in the reverse direction.
+ self.p1.restaurant = self.r
+ self.assertEqual(repr(self.p1.restaurant), '<Restaurant: Demon Dogs
the restaurant>')
+ r = Restaurant.objects.get(pk=self.p1.id)
+ self.assertEqual(repr(r.place), '<Place: Demon Dogs the place>')
+
+ def test_manager_all(self):
+ # Restaurant.objects.all() just returns the Restaurants, not the
Places.
+ self.assertQuerysetEqual(Restaurant.objects.all(), [
+ '<Restaurant: Demon Dogs the restaurant>',
+ ])
+ # Place.objects.all() returns all Places, regardless of whether they
+ # have Restaurants.
+ self.assertQuerysetEqual(Place.objects.order_by('name'), [
+ '<Place: Ace Hardware the place>',
+ '<Place: Demon Dogs the place>',
+ ])
+
+ def test_manager_get(self):
+ def assert_get_restaurant(**params):
+ self.assertEqual(repr(Restaurant.objects.get(**params)),
+ '<Restaurant: Demon Dogs the restaurant>')
+ assert_get_restaurant(place__id__exact=self.p1.pk)
+ assert_get_restaurant(place__id=self.p1.pk)
+ assert_get_restaurant(place__exact=self.p1.pk)
+ assert_get_restaurant(place__exact=self.p1)
+ assert_get_restaurant(place=self.p1.pk)
+ assert_get_restaurant(place=self.p1)
+ assert_get_restaurant(pk=self.p1.pk)
+ assert_get_restaurant(place__pk__exact=self.p1.pk)
+ assert_get_restaurant(place__pk=self.p1.pk)
+ assert_get_restaurant(place__name__startswith="Demon")
+
+ def assert_get_place(**params):
+ self.assertEqual(repr(Place.objects.get(**params)),
+ '<Place: Demon Dogs the place>')
+ assert_get_place(restaurant__place__exact=self.p1.pk)
+ assert_get_place(restaurant__place__exact=self.p1)
+ assert_get_place(restaurant__place__pk=self.p1.pk)
+ assert_get_place(restaurant__exact=self.p1.pk)
+ assert_get_place(restaurant__exact=self.r)
+ assert_get_place(restaurant__pk=self.p1.pk)
+ assert_get_place(restaurant=self.p1.pk)
+ assert_get_place(restaurant=self.r)
+ assert_get_place(id__exact=self.p1.pk)
+ assert_get_place(pk=self.p1.pk)
+
+ def test_foreign_key(self):
+ # Add a Waiter to the Restaurant.
+ w = self.r.waiter_set.create(name='Joe')
+ w.save()
+ self.assertEqual(repr(w), '<Waiter: Joe the waiter at Demon Dogs the
restaurant>')
+ # Query the waiters
+ def assert_filter_waiters(**params):
+ self.assertQuerysetEqual(Waiter.objects.filter(**params), [
+ '<Waiter: Joe the waiter at Demon Dogs the restaurant>'
+ ])
+ assert_filter_waiters(restaurant__place__exact=self.p1.pk)
+ assert_filter_waiters(restaurant__place__exact=self.p1)
+ assert_filter_waiters(restaurant__place__pk=self.p1.pk)
+ assert_filter_waiters(restaurant__exact=self.p1.pk)
+ assert_filter_waiters(restaurant__exact=self.p1)
+ assert_filter_waiters(restaurant__pk=self.p1.pk)
+ assert_filter_waiters(restaurant=self.p1.pk)
+ assert_filter_waiters(restaurant=self.r)
+ assert_filter_waiters(id__exact=self.p1.pk)
+ assert_filter_waiters(pk=self.p1.pk)
+ # Delete the restaurant; the waiter should also be removed
+ r = Restaurant.objects.get(pk=self.p1.pk)
+ r.delete()
+ self.assertEqual(Waiter.objects.count(), 0)
+
+ def test_multiple_o2o(self):
+ # One-to-one fields still work if you create your own primary key
+ o1 = ManualPrimaryKey(primary_key="abc123", name="primary")
+ o1.save()
+ o2 = RelatedModel(link=o1, name="secondary")
+ o2.save()
+
+ # You can have multiple one-to-one fields on a model, too.
+ x1 = MultiModel(link1=self.p1, link2=o1, name="x1")
+ x1.save()
+ self.assertEqual(repr(o1.multimodel), '<MultiModel: Multimodel x1>')
+ # This will fail because each one-to-one field must be unique (and
+ # link2=o1 was used for x1, above).
+ sid = transaction.savepoint()
+ mm = MultiModel(link1=self.p2, link2=o1, name="x1")
+ self.assertRaises(IntegrityError, mm.save)
+ transaction.savepoint_rollback(sid)
--
You received this message because you are subscribed to the Google Groups
"Django updates" group.
To post to this group, send email to django-upda...@googlegroups.com.
To unsubscribe from this group, send email to
django-updates+unsubscr...@googlegroups.com.
For more options, visit this group at
http://groups.google.com/group/django-updates?hl=en.
