summaryrefslogtreecommitdiff
path: root/parts/django/tests/modeltests/one_to_one/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/one_to_one/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/one_to_one/tests.py')
-rw-r--r--parts/django/tests/modeltests/one_to_one/tests.py119
1 files changed, 119 insertions, 0 deletions
diff --git a/parts/django/tests/modeltests/one_to_one/tests.py b/parts/django/tests/modeltests/one_to_one/tests.py
new file mode 100644
index 0000000..c3e1704
--- /dev/null
+++ b/parts/django/tests/modeltests/one_to_one/tests.py
@@ -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)