parts/django/tests/modeltests/fixtures_model_package/tests.py
changeset 69 c6bca38c1cbf
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/parts/django/tests/modeltests/fixtures_model_package/tests.py	Sat Jan 08 11:20:57 2011 +0530
@@ -0,0 +1,71 @@
+from django.core import management
+from django.test import TestCase
+
+from models import Article
+
+
+class SampleTestCase(TestCase):
+    fixtures = ['fixture1.json', 'fixture2.json']
+
+    def testClassFixtures(self):
+        "Test cases can load fixture objects into models defined in packages"
+        self.assertEqual(Article.objects.count(), 4)
+        self.assertQuerysetEqual(
+            Article.objects.all(),[
+                "Django conquers world!",
+                "Copyright is fine the way it is",
+                "Poker has no place on ESPN",
+                "Python program becomes self aware"
+            ],
+            lambda a: a.headline
+        )
+
+
+class FixtureTestCase(TestCase):
+    def test_initial_data(self):
+        "Fixtures can load initial data into models defined in packages"
+        #Syncdb introduces 1 initial data object from initial_data.json
+        self.assertQuerysetEqual(
+            Article.objects.all(), [
+                "Python program becomes self aware"
+            ],
+            lambda a: a.headline
+        )
+
+    def test_loaddata(self):
+        "Fixtures can load data into models defined in packages"
+        # Load fixture 1. Single JSON file, with two objects
+        management.call_command("loaddata", "fixture1.json", verbosity=0, commit=False)
+        self.assertQuerysetEqual(
+            Article.objects.all(), [
+                "Time to reform copyright",
+                "Poker has no place on ESPN",
+                "Python program becomes self aware",
+            ],
+            lambda a: a.headline,
+        )
+
+        # Load fixture 2. JSON file imported by default. Overwrites some
+        # existing objects
+        management.call_command("loaddata", "fixture2.json", verbosity=0, commit=False)
+        self.assertQuerysetEqual(
+            Article.objects.all(), [
+                "Django conquers world!",
+                "Copyright is fine the way it is",
+                "Poker has no place on ESPN",
+                "Python program becomes self aware",
+            ],
+            lambda a: a.headline,
+        )
+
+        # Load a fixture that doesn't exist
+        management.call_command("loaddata", "unknown.json", verbosity=0, commit=False)
+        self.assertQuerysetEqual(
+            Article.objects.all(), [
+                "Django conquers world!",
+                "Copyright is fine the way it is",
+                "Poker has no place on ESPN",
+                "Python program becomes self aware",
+            ],
+            lambda a: a.headline,
+        )