/tests/regressiontests/select_related_onetoone/tests.py
Python | 90 lines | 76 code | 14 blank | 0 comment | 0 complexity | 46bf9c55392355bb871763ab04452209 MD5 | raw file
Possible License(s): BSD-3-Clause
- from django import db
- from django.conf import settings
- from django.test import TestCase
- from models import (User, UserProfile, UserStat, UserStatResult, StatDetails,
- AdvancedUserStat, Image, Product)
- class ReverseSelectRelatedTestCase(TestCase):
- def setUp(self):
- user = User.objects.create(username="test")
- userprofile = UserProfile.objects.create(user=user, state="KS",
- city="Lawrence")
- results = UserStatResult.objects.create(results='first results')
- userstat = UserStat.objects.create(user=user, posts=150,
- results=results)
- details = StatDetails.objects.create(base_stats=userstat, comments=259)
- user2 = User.objects.create(username="bob")
- results2 = UserStatResult.objects.create(results='moar results')
- advstat = AdvancedUserStat.objects.create(user=user2, posts=200, karma=5,
- results=results2)
- StatDetails.objects.create(base_stats=advstat, comments=250)
- def test_basic(self):
- def test():
- u = User.objects.select_related("userprofile").get(username="test")
- self.assertEqual(u.userprofile.state, "KS")
- self.assertNumQueries(1, test)
- def test_follow_next_level(self):
- def test():
- u = User.objects.select_related("userstat__results").get(username="test")
- self.assertEqual(u.userstat.posts, 150)
- self.assertEqual(u.userstat.results.results, 'first results')
- self.assertNumQueries(1, test)
- def test_follow_two(self):
- def test():
- u = User.objects.select_related("userprofile", "userstat").get(username="test")
- self.assertEqual(u.userprofile.state, "KS")
- self.assertEqual(u.userstat.posts, 150)
- self.assertNumQueries(1, test)
- def test_follow_two_next_level(self):
- def test():
- u = User.objects.select_related("userstat__results", "userstat__statdetails").get(username="test")
- self.assertEqual(u.userstat.results.results, 'first results')
- self.assertEqual(u.userstat.statdetails.comments, 259)
- self.assertNumQueries(1, test)
- def test_forward_and_back(self):
- def test():
- stat = UserStat.objects.select_related("user__userprofile").get(user__username="test")
- self.assertEqual(stat.user.userprofile.state, 'KS')
- self.assertEqual(stat.user.userstat.posts, 150)
- self.assertNumQueries(1, test)
- def test_back_and_forward(self):
- def test():
- u = User.objects.select_related("userstat").get(username="test")
- self.assertEqual(u.userstat.user.username, 'test')
- self.assertNumQueries(1, test)
- def test_not_followed_by_default(self):
- def test():
- u = User.objects.select_related().get(username="test")
- self.assertEqual(u.userstat.posts, 150)
- self.assertNumQueries(2, test)
- def test_follow_from_child_class(self):
- def test():
- stat = AdvancedUserStat.objects.select_related('user', 'statdetails').get(posts=200)
- self.assertEqual(stat.statdetails.comments, 250)
- self.assertEqual(stat.user.username, 'bob')
- self.assertNumQueries(1, test)
- def test_follow_inheritance(self):
- def test():
- stat = UserStat.objects.select_related('user', 'advanceduserstat').get(posts=200)
- self.assertEqual(stat.advanceduserstat.posts, 200)
- self.assertEqual(stat.user.username, 'bob')
- self.assertEqual(stat.advanceduserstat.user.username, 'bob')
- self.assertNumQueries(1, test)
- def test_nullable_relation(self):
- im = Image.objects.create(name="imag1")
- p1 = Product.objects.create(name="Django Plushie", image=im)
- p2 = Product.objects.create(name="Talking Django Plushie")
- self.assertEqual(len(Product.objects.select_related("image")), 2)