72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
|
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)
|
||
|
|
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)
|
||
|
|
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)
|
||
|
|
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,
|
||
|
|
)
|