From 891bc66d55606e93f414e70cd017336d7718fc6a Mon Sep 17 00:00:00 2001
From: Russell Keith-Magee <russell@keith-magee.com>
Date: Tue, 28 Sep 2010 07:06:15 +0000
Subject: [PATCH] Migrated reverse_single_related doctests. Thanks to Stephan
 Jaekel.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13926 bcc190cf-cafb-0310-a4f2-bffc1f526a37
---
 .../reverse_single_related/models.py          | 42 -------------------
 .../reverse_single_related/tests.py           | 36 ++++++++++++++++
 2 files changed, 36 insertions(+), 42 deletions(-)
 create mode 100644 tests/regressiontests/reverse_single_related/tests.py

diff --git a/tests/regressiontests/reverse_single_related/models.py b/tests/regressiontests/reverse_single_related/models.py
index b2b75392aa..a2d8fb0fac 100644
--- a/tests/regressiontests/reverse_single_related/models.py
+++ b/tests/regressiontests/reverse_single_related/models.py
@@ -1,11 +1,5 @@
-"""
-Regression tests for an object that cannot access a single related object due
-to a restrictive default manager.
-"""
-
 from django.db import models
 
-
 class SourceManager(models.Manager):
     def get_query_set(self):
         return super(SourceManager, self).get_query_set().filter(is_public=True)
@@ -16,39 +10,3 @@ class Source(models.Model):
 
 class Item(models.Model):
     source = models.ForeignKey(Source)
-
-
-__test__ = {'API_TESTS':"""
-
->>> public_source = Source.objects.create(is_public=True)
->>> public_item = Item.objects.create(source=public_source)
-
->>> private_source = Source.objects.create(is_public=False)
->>> private_item = Item.objects.create(source=private_source)
-
-# Only one source is available via all() due to the custom default manager.
-
->>> Source.objects.all()
-[<Source: Source object>]
-
->>> public_item.source
-<Source: Source object>
-
-# Make sure that an item can still access its related source even if the default
-# manager doesn't normally allow it.
-
->>> private_item.source
-<Source: Source object>
-
-# If the manager is marked "use_for_related_fields", it'll get used instead
-# of the "bare" queryset. Usually you'd define this as a property on the class,
-# but this approximates that in a way that's easier in tests.
-
->>> Source.objects.use_for_related_fields = True
->>> private_item = Item.objects.get(pk=private_item.pk)
->>> private_item.source
-Traceback (most recent call last):
-    ...
-DoesNotExist: Source matching query does not exist.
-
-"""}
diff --git a/tests/regressiontests/reverse_single_related/tests.py b/tests/regressiontests/reverse_single_related/tests.py
new file mode 100644
index 0000000000..14f3a6665e
--- /dev/null
+++ b/tests/regressiontests/reverse_single_related/tests.py
@@ -0,0 +1,36 @@
+from django.test import TestCase
+
+from regressiontests.reverse_single_related.models import *
+
+class ReverseSingleRelatedTests(TestCase):
+    """
+    Regression tests for an object that cannot access a single related
+    object due to a restrictive default manager.
+    """
+
+    def test_reverse_single_related(self):
+
+        public_source = Source.objects.create(is_public=True)
+        public_item = Item.objects.create(source=public_source)
+
+        private_source = Source.objects.create(is_public=False)
+        private_item = Item.objects.create(source=private_source)
+
+        # Only one source is available via all() due to the custom default manager.
+        self.assertQuerysetEqual(
+                Source.objects.all(),
+                ["<Source: Source object>"]
+        )
+
+        self.assertEquals(public_item.source, public_source)
+
+        # Make sure that an item can still access its related source even if the default
+        # manager doesn't normally allow it.
+        self.assertEquals(private_item.source, private_source)
+
+        # If the manager is marked "use_for_related_fields", it'll get used instead
+        # of the "bare" queryset. Usually you'd define this as a property on the class,
+        # but this approximates that in a way that's easier in tests.
+        Source.objects.use_for_related_fields = True
+        private_item = Item.objects.get(pk=private_item.pk)
+        self.assertRaises(Source.DoesNotExist, lambda: private_item.source)