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):

        # Explicitly enable debug for these tests - we need to count

        # the queries that have been issued.

        self.old_debug = settings.DEBUG

        settings.DEBUG = True

        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)

        db.reset_queries()

    def assertQueries(self, queries):

        self.assertEqual(len(db.connection.queries), queries)

    def tearDown(self):

        settings.DEBUG = self.old_debug

    def test_basic(self):

        u = User.objects.select_related("userprofile").get(username="test")

        self.assertEqual(u.userprofile.state, "KS")

        self.assertQueries(1)

    def test_follow_next_level(self):

        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.assertQueries(1)

    def test_follow_two(self):

        u = User.objects.select_related("userprofile", "userstat").get(username="test")

        self.assertEqual(u.userprofile.state, "KS")

        self.assertEqual(u.userstat.posts, 150)

        self.assertQueries(1)

    def test_follow_two_next_level(self):

        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.assertQueries(1)

    def test_forward_and_back(self):

        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.assertQueries(1)

    def test_back_and_forward(self):

        u = User.objects.select_related("userstat").get(username="test")

        self.assertEqual(u.userstat.user.username, 'test')

        self.assertQueries(1)

    def test_not_followed_by_default(self):

        u = User.objects.select_related().get(username="test")

        self.assertEqual(u.userstat.posts, 150)

        self.assertQueries(2)

    def test_follow_from_child_class(self):

        stat = AdvancedUserStat.objects.select_related('user', 'statdetails').get(posts=200)

        self.assertEqual(stat.statdetails.comments, 250)

        self.assertEqual(stat.user.username, 'bob')

        self.assertQueries(1)

    def test_follow_inheritance(self):

        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.assertQueries(1)

    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)