Refactored model lookup tests; refs #18586. (d1e87aeb) · Commits · Dom Sekotill / django

AUTHORS

+1 −0

Original line number	Diff line number	Diff line
		@@ -107,6 +107,7 @@ answer newbie questions, and generally made Django that much better:
		Brian Ray <http://brianray.chipy.org/>
		Brian Rosner <brosner@gmail.com>
		Bruce Kroeze <http://coderseye.com/>
		Bruno Alla <alla.brunoo@gmail.com>
		Bruno Renié <buburno@gmail.com>
		brut.alll@gmail.com
		Bryan Chow <bryan at verdjn dot com>

tests/basic/tests.py

+134 −144

Original line number	Diff line number	Diff line
		@@ -178,150 +178,6 @@ class ModelTest(TestCase):
		self.assertTrue(Article.objects.get(id__exact=a1.id) != Article.objects.get(id__exact=a2.id))
		self.assertFalse(Article.objects.get(id__exact=a2.id) == Article.objects.get(id__exact=a1.id))

		def test_lookup(self):
		# No articles are in the system yet.
		self.assertQuerysetEqual(Article.objects.all(), [])

		# Create an Article.
		a = Article(
		id=None,
		headline='Area man programs in Python',
		pub_date=datetime(2005, 7, 28),
		)

		# Save it into the database. You have to call save() explicitly.
		a.save()

		# Now it has an ID.
		self.assertTrue(a.id is not None)

		# Models have a pk property that is an alias for the primary key
		# attribute (by default, the 'id' attribute).
		self.assertEqual(a.pk, a.id)

		# Access database columns via Python attributes.
		self.assertEqual(a.headline, 'Area man programs in Python')
		self.assertEqual(a.pub_date, datetime(2005, 7, 28, 0, 0))

		# Change values by changing the attributes, then calling save().
		a.headline = 'Area woman programs in Python'
		a.save()

		# Article.objects.all() returns all the articles in the database.
		self.assertQuerysetEqual(Article.objects.all(),
		['<Article: Area woman programs in Python>'])

		# Django provides a rich database lookup API.
		self.assertEqual(Article.objects.get(id__exact=a.id), a)
		self.assertEqual(Article.objects.get(headline__startswith='Area woman'), a)
		self.assertEqual(Article.objects.get(pub_date__year=2005), a)
		self.assertEqual(Article.objects.get(pub_date__year=2005, pub_date__month=7), a)
		self.assertEqual(Article.objects.get(pub_date__year=2005, pub_date__month=7, pub_date__day=28), a)
		self.assertEqual(Article.objects.get(pub_date__week_day=5), a)

		# The "__exact" lookup type can be omitted, as a shortcut.
		self.assertEqual(Article.objects.get(id=a.id), a)
		self.assertEqual(Article.objects.get(headline='Area woman programs in Python'), a)

		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__year=2005),
		['<Article: Area woman programs in Python>'],
		)
		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__year=2004),
		[],
		)
		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__year=2005, pub_date__month=7),
		['<Article: Area woman programs in Python>'],
		)

		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__week_day=5),
		['<Article: Area woman programs in Python>'],
		)
		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__week_day=6),
		[],
		)

		# Django raises an Article.DoesNotExist exception for get() if the
		# parameters don't match any object.
		six.assertRaisesRegex(
		self,
		ObjectDoesNotExist,
		"Article matching query does not exist.",
		Article.objects.get,
		id__exact=2000,
		)
		# To avoid dict-ordering related errors check only one lookup
		# in single assert.
		self.assertRaises(
		ObjectDoesNotExist,
		Article.objects.get,
		pub_date__year=2005,
		pub_date__month=8,
		)

		six.assertRaisesRegex(
		self,
		ObjectDoesNotExist,
		"Article matching query does not exist.",
		Article.objects.get,
		pub_date__week_day=6,
		)

		# Lookup by a primary key is the most common case, so Django
		# provides a shortcut for primary-key exact lookups.
		# The following is identical to articles.get(id=a.id).
		self.assertEqual(Article.objects.get(pk=a.id), a)

		# pk can be used as a shortcut for the primary key name in any query.
		self.assertQuerysetEqual(Article.objects.filter(pk__in=[a.id]),
		["<Article: Area woman programs in Python>"])

		# Model instances of the same type and same ID are considered equal.
		a = Article.objects.get(pk=a.id)
		b = Article.objects.get(pk=a.id)
		self.assertEqual(a, b)

		# Create a very similar object
		a = Article(
		id=None,
		headline='Area man programs in Python',
		pub_date=datetime(2005, 7, 28),
		)
		a.save()

		self.assertEqual(Article.objects.count(), 2)

		# Django raises an Article.MultipleObjectsReturned exception if the
		# lookup matches more than one object
		six.assertRaisesRegex(
		self,
		MultipleObjectsReturned,
		"get\(\) returned more than one Article -- it returned 2!",
		Article.objects.get,
		headline__startswith='Area',
		)

		six.assertRaisesRegex(
		self,
		MultipleObjectsReturned,
		"get\(\) returned more than one Article -- it returned 2!",
		Article.objects.get,
		pub_date__year=2005,
		)

		six.assertRaisesRegex(
		self,
		MultipleObjectsReturned,
		"get\(\) returned more than one Article -- it returned 2!",
		Article.objects.get,
		pub_date__year=2005,
		pub_date__month=7,
		)

		def test_multiple_objects_max_num_fetched(self):
		"""
		#6785 - get() should fetch a limited number of results.
		@@ -584,6 +440,140 @@ class ModelTest(TestCase):
		hash(Article())


		class ModelLookupTest(TestCase):
		def setUp(self):
		# Create an Article.
		self.a = Article(
		id=None,
		headline='Area woman programs in Python',
		pub_date=datetime(2005, 7, 28),
		)
		# Save it into the database. You have to call save() explicitly.
		self.a.save()

		def test_all_lookup(self):
		# Change values by changing the attributes, then calling save().
		self.a.headline = 'Area man programs in Python'
		self.a.save()

		# Article.objects.all() returns all the articles in the database.
		self.assertQuerysetEqual(Article.objects.all(),
		['<Article: Area man programs in Python>'])

		def test_rich_lookup(self):
		# Django provides a rich database lookup API.
		self.assertEqual(Article.objects.get(id__exact=self.a.id), self.a)
		self.assertEqual(Article.objects.get(headline__startswith='Area woman'), self.a)
		self.assertEqual(Article.objects.get(pub_date__year=2005), self.a)
		self.assertEqual(Article.objects.get(pub_date__year=2005, pub_date__month=7), self.a)
		self.assertEqual(Article.objects.get(pub_date__year=2005, pub_date__month=7, pub_date__day=28), self.a)
		self.assertEqual(Article.objects.get(pub_date__week_day=5), self.a)

		def test_equal_lookup(self):
		# The "__exact" lookup type can be omitted, as a shortcut.
		self.assertEqual(Article.objects.get(id=self.a.id), self.a)
		self.assertEqual(Article.objects.get(headline='Area woman programs in Python'), self.a)

		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__year=2005),
		['<Article: Area woman programs in Python>'],
		)
		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__year=2004),
		[],
		)
		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__year=2005, pub_date__month=7),
		['<Article: Area woman programs in Python>'],
		)

		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__week_day=5),
		['<Article: Area woman programs in Python>'],
		)
		self.assertQuerysetEqual(
		Article.objects.filter(pub_date__week_day=6),
		[],
		)

		def test_does_not_exist(self):
		# Django raises an Article.DoesNotExist exception for get() if the
		# parameters don't match any object.
		six.assertRaisesRegex(
		self,
		ObjectDoesNotExist,
		"Article matching query does not exist.",
		Article.objects.get,
		id__exact=2000,
		)
		# To avoid dict-ordering related errors check only one lookup
		# in single assert.
		self.assertRaises(
		ObjectDoesNotExist,
		Article.objects.get,
		pub_date__year=2005,
		pub_date__month=8,
		)
		six.assertRaisesRegex(
		self,
		ObjectDoesNotExist,
		"Article matching query does not exist.",
		Article.objects.get,
		pub_date__week_day=6,
		)

		def test_lookup_by_primary_key(self):
		# Lookup by a primary key is the most common case, so Django
		# provides a shortcut for primary-key exact lookups.
		# The following is identical to articles.get(id=a.id).
		self.assertEqual(Article.objects.get(pk=self.a.id), self.a)

		# pk can be used as a shortcut for the primary key name in any query.
		self.assertQuerysetEqual(Article.objects.filter(pk__in=[self.a.id]),
		["<Article: Area woman programs in Python>"])

		# Model instances of the same type and same ID are considered equal.
		a = Article.objects.get(pk=self.a.id)
		b = Article.objects.get(pk=self.a.id)
		self.assertEqual(a, b)

		def test_too_many(self):
		# Create a very similar object
		a = Article(
		id=None,
		headline='Area man programs in Python',
		pub_date=datetime(2005, 7, 28),
		)
		a.save()

		self.assertEqual(Article.objects.count(), 2)

		# Django raises an Article.MultipleObjectsReturned exception if the
		# lookup matches more than one object
		six.assertRaisesRegex(
		self,
		MultipleObjectsReturned,
		"get\(\) returned more than one Article -- it returned 2!",
		Article.objects.get,
		headline__startswith='Area',
		)
		six.assertRaisesRegex(
		self,
		MultipleObjectsReturned,
		"get\(\) returned more than one Article -- it returned 2!",
		Article.objects.get,
		pub_date__year=2005,
		)
		six.assertRaisesRegex(
		self,
		MultipleObjectsReturned,
		"get\(\) returned more than one Article -- it returned 2!",
		Article.objects.get,
		pub_date__year=2005,
		pub_date__month=7,
		)


		class ConcurrentSaveTests(TransactionTestCase):

		available_apps = ['basic']