/tests/regressiontests/select_related_onetoone/tests.py

 1from django import db
 2from django.conf import settings
 3from django.test import TestCase
 4
 5from models import (User, UserProfile, UserStat, UserStatResult, StatDetails,
 6    AdvancedUserStat, Image, Product)
 7
 8class ReverseSelectRelatedTestCase(TestCase):
 9    def setUp(self):
10        user = User.objects.create(username="test")
11        userprofile = UserProfile.objects.create(user=user, state="KS",
12                                                 city="Lawrence")
13        results = UserStatResult.objects.create(results='first results')
14        userstat = UserStat.objects.create(user=user, posts=150,
15                                           results=results)
16        details = StatDetails.objects.create(base_stats=userstat, comments=259)
17
18        user2 = User.objects.create(username="bob")
19        results2 = UserStatResult.objects.create(results='moar results')
20        advstat = AdvancedUserStat.objects.create(user=user2, posts=200, karma=5,
21                                                  results=results2)
22        StatDetails.objects.create(base_stats=advstat, comments=250)
23
24    def test_basic(self):
25        def test():
26            u = User.objects.select_related("userprofile").get(username="test")
27            self.assertEqual(u.userprofile.state, "KS")
28        self.assertNumQueries(1, test)
29
30    def test_follow_next_level(self):
31        def test():
32            u = User.objects.select_related("userstat__results").get(username="test")
33            self.assertEqual(u.userstat.posts, 150)
34            self.assertEqual(u.userstat.results.results, 'first results')
35        self.assertNumQueries(1, test)
36
37    def test_follow_two(self):
38        def test():
39            u = User.objects.select_related("userprofile", "userstat").get(username="test")
40            self.assertEqual(u.userprofile.state, "KS")
41            self.assertEqual(u.userstat.posts, 150)
42        self.assertNumQueries(1, test)
43
44    def test_follow_two_next_level(self):
45        def test():
46            u = User.objects.select_related("userstat__results", "userstat__statdetails").get(username="test")
47            self.assertEqual(u.userstat.results.results, 'first results')
48            self.assertEqual(u.userstat.statdetails.comments, 259)
49        self.assertNumQueries(1, test)
50
51    def test_forward_and_back(self):
52        def test():
53            stat = UserStat.objects.select_related("user__userprofile").get(user__username="test")
54            self.assertEqual(stat.user.userprofile.state, 'KS')
55            self.assertEqual(stat.user.userstat.posts, 150)
56        self.assertNumQueries(1, test)
57
58    def test_back_and_forward(self):
59        def test():
60            u = User.objects.select_related("userstat").get(username="test")
61            self.assertEqual(u.userstat.user.username, 'test')
62        self.assertNumQueries(1, test)
63
64    def test_not_followed_by_default(self):
65        def test():
66            u = User.objects.select_related().get(username="test")
67            self.assertEqual(u.userstat.posts, 150)
68        self.assertNumQueries(2, test)
69
70    def test_follow_from_child_class(self):
71        def test():
72            stat = AdvancedUserStat.objects.select_related('user', 'statdetails').get(posts=200)
73            self.assertEqual(stat.statdetails.comments, 250)
74            self.assertEqual(stat.user.username, 'bob')
75        self.assertNumQueries(1, test)
76
77    def test_follow_inheritance(self):
78        def test():
79            stat = UserStat.objects.select_related('user', 'advanceduserstat').get(posts=200)
80            self.assertEqual(stat.advanceduserstat.posts, 200)
81            self.assertEqual(stat.user.username, 'bob')
82            self.assertEqual(stat.advanceduserstat.user.username, 'bob')
83        self.assertNumQueries(1, test)
84
85    def test_nullable_relation(self):
86        im = Image.objects.create(name="imag1")
87        p1 = Product.objects.create(name="Django Plushie", image=im)
88        p2 = Product.objects.create(name="Talking Django Plushie")
89
90        self.assertEqual(len(Product.objects.select_related("image")), 2)