tests.py 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770
  1. import collections
  2. from datetime import datetime
  3. from operator import attrgetter
  4. from django.core.exceptions import FieldError
  5. from django.test import TestCase, skipUnlessDBFeature
  6. from .models import Article, Author, Game, Player, Season, Tag
  7. class LookupTests(TestCase):
  8. def setUp(self):
  9. # Create a few Authors.
  10. self.au1 = Author.objects.create(name='Author 1')
  11. self.au2 = Author.objects.create(name='Author 2')
  12. # Create a couple of Articles.
  13. self.a1 = Article.objects.create(headline='Article 1', pub_date=datetime(2005, 7, 26), author=self.au1)
  14. self.a2 = Article.objects.create(headline='Article 2', pub_date=datetime(2005, 7, 27), author=self.au1)
  15. self.a3 = Article.objects.create(headline='Article 3', pub_date=datetime(2005, 7, 27), author=self.au1)
  16. self.a4 = Article.objects.create(headline='Article 4', pub_date=datetime(2005, 7, 28), author=self.au1)
  17. self.a5 = Article.objects.create(headline='Article 5', pub_date=datetime(2005, 8, 1, 9, 0), author=self.au2)
  18. self.a6 = Article.objects.create(headline='Article 6', pub_date=datetime(2005, 8, 1, 8, 0), author=self.au2)
  19. self.a7 = Article.objects.create(headline='Article 7', pub_date=datetime(2005, 7, 27), author=self.au2)
  20. # Create a few Tags.
  21. self.t1 = Tag.objects.create(name='Tag 1')
  22. self.t1.articles.add(self.a1, self.a2, self.a3)
  23. self.t2 = Tag.objects.create(name='Tag 2')
  24. self.t2.articles.add(self.a3, self.a4, self.a5)
  25. self.t3 = Tag.objects.create(name='Tag 3')
  26. self.t3.articles.add(self.a5, self.a6, self.a7)
  27. def test_exists(self):
  28. # We can use .exists() to check that there are some
  29. self.assertTrue(Article.objects.exists())
  30. for a in Article.objects.all():
  31. a.delete()
  32. # There should be none now!
  33. self.assertFalse(Article.objects.exists())
  34. def test_lookup_int_as_str(self):
  35. # Integer value can be queried using string
  36. self.assertQuerysetEqual(Article.objects.filter(id__iexact=str(self.a1.id)),
  37. ['<Article: Article 1>'])
  38. @skipUnlessDBFeature('supports_date_lookup_using_string')
  39. def test_lookup_date_as_str(self):
  40. # A date lookup can be performed using a string search
  41. self.assertQuerysetEqual(
  42. Article.objects.filter(pub_date__startswith='2005'),
  43. [
  44. '<Article: Article 5>',
  45. '<Article: Article 6>',
  46. '<Article: Article 4>',
  47. '<Article: Article 2>',
  48. '<Article: Article 3>',
  49. '<Article: Article 7>',
  50. '<Article: Article 1>',
  51. ]
  52. )
  53. def test_iterator(self):
  54. # Each QuerySet gets iterator(), which is a generator that "lazily"
  55. # returns results using database-level iteration.
  56. self.assertIsInstance(Article.objects.iterator(), collections.Iterator)
  57. self.assertQuerysetEqual(
  58. Article.objects.iterator(),
  59. [
  60. 'Article 5',
  61. 'Article 6',
  62. 'Article 4',
  63. 'Article 2',
  64. 'Article 3',
  65. 'Article 7',
  66. 'Article 1',
  67. ],
  68. transform=attrgetter('headline')
  69. )
  70. # iterator() can be used on any QuerySet.
  71. self.assertQuerysetEqual(
  72. Article.objects.filter(headline__endswith='4').iterator(),
  73. ['Article 4'],
  74. transform=attrgetter('headline'))
  75. def test_count(self):
  76. # count() returns the number of objects matching search criteria.
  77. self.assertEqual(Article.objects.count(), 7)
  78. self.assertEqual(Article.objects.filter(pub_date__exact=datetime(2005, 7, 27)).count(), 3)
  79. self.assertEqual(Article.objects.filter(headline__startswith='Blah blah').count(), 0)
  80. # count() should respect sliced query sets.
  81. articles = Article.objects.all()
  82. self.assertEqual(articles.count(), 7)
  83. self.assertEqual(articles[:4].count(), 4)
  84. self.assertEqual(articles[1:100].count(), 6)
  85. self.assertEqual(articles[10:100].count(), 0)
  86. # Date and date/time lookups can also be done with strings.
  87. self.assertEqual(Article.objects.filter(pub_date__exact='2005-07-27 00:00:00').count(), 3)
  88. def test_in_bulk(self):
  89. # in_bulk() takes a list of IDs and returns a dictionary mapping IDs to objects.
  90. arts = Article.objects.in_bulk([self.a1.id, self.a2.id])
  91. self.assertEqual(arts[self.a1.id], self.a1)
  92. self.assertEqual(arts[self.a2.id], self.a2)
  93. self.assertEqual(
  94. Article.objects.in_bulk(),
  95. {
  96. self.a1.id: self.a1,
  97. self.a2.id: self.a2,
  98. self.a3.id: self.a3,
  99. self.a4.id: self.a4,
  100. self.a5.id: self.a5,
  101. self.a6.id: self.a6,
  102. self.a7.id: self.a7,
  103. }
  104. )
  105. self.assertEqual(Article.objects.in_bulk([self.a3.id]), {self.a3.id: self.a3})
  106. self.assertEqual(Article.objects.in_bulk({self.a3.id}), {self.a3.id: self.a3})
  107. self.assertEqual(Article.objects.in_bulk(frozenset([self.a3.id])), {self.a3.id: self.a3})
  108. self.assertEqual(Article.objects.in_bulk((self.a3.id,)), {self.a3.id: self.a3})
  109. self.assertEqual(Article.objects.in_bulk([1000]), {})
  110. self.assertEqual(Article.objects.in_bulk([]), {})
  111. self.assertEqual(Article.objects.in_bulk(iter([self.a1.id])), {self.a1.id: self.a1})
  112. self.assertEqual(Article.objects.in_bulk(iter([])), {})
  113. with self.assertRaises(TypeError):
  114. Article.objects.in_bulk(headline__startswith='Blah')
  115. def test_values(self):
  116. # values() returns a list of dictionaries instead of object instances --
  117. # and you can specify which fields you want to retrieve.
  118. self.assertSequenceEqual(
  119. Article.objects.values('headline'),
  120. [
  121. {'headline': 'Article 5'},
  122. {'headline': 'Article 6'},
  123. {'headline': 'Article 4'},
  124. {'headline': 'Article 2'},
  125. {'headline': 'Article 3'},
  126. {'headline': 'Article 7'},
  127. {'headline': 'Article 1'},
  128. ],
  129. )
  130. self.assertSequenceEqual(
  131. Article.objects.filter(pub_date__exact=datetime(2005, 7, 27)).values('id'),
  132. [{'id': self.a2.id}, {'id': self.a3.id}, {'id': self.a7.id}],
  133. )
  134. self.assertSequenceEqual(
  135. Article.objects.values('id', 'headline'),
  136. [
  137. {'id': self.a5.id, 'headline': 'Article 5'},
  138. {'id': self.a6.id, 'headline': 'Article 6'},
  139. {'id': self.a4.id, 'headline': 'Article 4'},
  140. {'id': self.a2.id, 'headline': 'Article 2'},
  141. {'id': self.a3.id, 'headline': 'Article 3'},
  142. {'id': self.a7.id, 'headline': 'Article 7'},
  143. {'id': self.a1.id, 'headline': 'Article 1'},
  144. ],
  145. )
  146. # You can use values() with iterator() for memory savings,
  147. # because iterator() uses database-level iteration.
  148. self.assertSequenceEqual(
  149. list(Article.objects.values('id', 'headline').iterator()),
  150. [
  151. {'headline': 'Article 5', 'id': self.a5.id},
  152. {'headline': 'Article 6', 'id': self.a6.id},
  153. {'headline': 'Article 4', 'id': self.a4.id},
  154. {'headline': 'Article 2', 'id': self.a2.id},
  155. {'headline': 'Article 3', 'id': self.a3.id},
  156. {'headline': 'Article 7', 'id': self.a7.id},
  157. {'headline': 'Article 1', 'id': self.a1.id},
  158. ],
  159. )
  160. # The values() method works with "extra" fields specified in extra(select).
  161. self.assertSequenceEqual(
  162. Article.objects.extra(select={'id_plus_one': 'id + 1'}).values('id', 'id_plus_one'),
  163. [
  164. {'id': self.a5.id, 'id_plus_one': self.a5.id + 1},
  165. {'id': self.a6.id, 'id_plus_one': self.a6.id + 1},
  166. {'id': self.a4.id, 'id_plus_one': self.a4.id + 1},
  167. {'id': self.a2.id, 'id_plus_one': self.a2.id + 1},
  168. {'id': self.a3.id, 'id_plus_one': self.a3.id + 1},
  169. {'id': self.a7.id, 'id_plus_one': self.a7.id + 1},
  170. {'id': self.a1.id, 'id_plus_one': self.a1.id + 1},
  171. ],
  172. )
  173. data = {
  174. 'id_plus_one': 'id+1',
  175. 'id_plus_two': 'id+2',
  176. 'id_plus_three': 'id+3',
  177. 'id_plus_four': 'id+4',
  178. 'id_plus_five': 'id+5',
  179. 'id_plus_six': 'id+6',
  180. 'id_plus_seven': 'id+7',
  181. 'id_plus_eight': 'id+8',
  182. }
  183. self.assertSequenceEqual(
  184. Article.objects.filter(id=self.a1.id).extra(select=data).values(*data.keys()),
  185. [{
  186. 'id_plus_one': self.a1.id + 1,
  187. 'id_plus_two': self.a1.id + 2,
  188. 'id_plus_three': self.a1.id + 3,
  189. 'id_plus_four': self.a1.id + 4,
  190. 'id_plus_five': self.a1.id + 5,
  191. 'id_plus_six': self.a1.id + 6,
  192. 'id_plus_seven': self.a1.id + 7,
  193. 'id_plus_eight': self.a1.id + 8,
  194. }],
  195. )
  196. # You can specify fields from forward and reverse relations, just like filter().
  197. self.assertSequenceEqual(
  198. Article.objects.values('headline', 'author__name'),
  199. [
  200. {'headline': self.a5.headline, 'author__name': self.au2.name},
  201. {'headline': self.a6.headline, 'author__name': self.au2.name},
  202. {'headline': self.a4.headline, 'author__name': self.au1.name},
  203. {'headline': self.a2.headline, 'author__name': self.au1.name},
  204. {'headline': self.a3.headline, 'author__name': self.au1.name},
  205. {'headline': self.a7.headline, 'author__name': self.au2.name},
  206. {'headline': self.a1.headline, 'author__name': self.au1.name},
  207. ],
  208. )
  209. self.assertSequenceEqual(
  210. Author.objects.values('name', 'article__headline').order_by('name', 'article__headline'),
  211. [
  212. {'name': self.au1.name, 'article__headline': self.a1.headline},
  213. {'name': self.au1.name, 'article__headline': self.a2.headline},
  214. {'name': self.au1.name, 'article__headline': self.a3.headline},
  215. {'name': self.au1.name, 'article__headline': self.a4.headline},
  216. {'name': self.au2.name, 'article__headline': self.a5.headline},
  217. {'name': self.au2.name, 'article__headline': self.a6.headline},
  218. {'name': self.au2.name, 'article__headline': self.a7.headline},
  219. ],
  220. )
  221. self.assertSequenceEqual(
  222. (
  223. Author.objects
  224. .values('name', 'article__headline', 'article__tag__name')
  225. .order_by('name', 'article__headline', 'article__tag__name')
  226. ),
  227. [
  228. {'name': self.au1.name, 'article__headline': self.a1.headline, 'article__tag__name': self.t1.name},
  229. {'name': self.au1.name, 'article__headline': self.a2.headline, 'article__tag__name': self.t1.name},
  230. {'name': self.au1.name, 'article__headline': self.a3.headline, 'article__tag__name': self.t1.name},
  231. {'name': self.au1.name, 'article__headline': self.a3.headline, 'article__tag__name': self.t2.name},
  232. {'name': self.au1.name, 'article__headline': self.a4.headline, 'article__tag__name': self.t2.name},
  233. {'name': self.au2.name, 'article__headline': self.a5.headline, 'article__tag__name': self.t2.name},
  234. {'name': self.au2.name, 'article__headline': self.a5.headline, 'article__tag__name': self.t3.name},
  235. {'name': self.au2.name, 'article__headline': self.a6.headline, 'article__tag__name': self.t3.name},
  236. {'name': self.au2.name, 'article__headline': self.a7.headline, 'article__tag__name': self.t3.name},
  237. ],
  238. )
  239. # However, an exception FieldDoesNotExist will be thrown if you specify
  240. # a nonexistent field name in values() (a field that is neither in the
  241. # model nor in extra(select)).
  242. with self.assertRaises(FieldError):
  243. Article.objects.extra(select={'id_plus_one': 'id + 1'}).values('id', 'id_plus_two')
  244. # If you don't specify field names to values(), all are returned.
  245. self.assertSequenceEqual(
  246. Article.objects.filter(id=self.a5.id).values(),
  247. [{
  248. 'id': self.a5.id,
  249. 'author_id': self.au2.id,
  250. 'headline': 'Article 5',
  251. 'pub_date': datetime(2005, 8, 1, 9, 0)
  252. }],
  253. )
  254. def test_values_list(self):
  255. # values_list() is similar to values(), except that the results are
  256. # returned as a list of tuples, rather than a list of dictionaries.
  257. # Within each tuple, the order of the elements is the same as the order
  258. # of fields in the values_list() call.
  259. self.assertSequenceEqual(
  260. Article.objects.values_list('headline'),
  261. [
  262. ('Article 5',),
  263. ('Article 6',),
  264. ('Article 4',),
  265. ('Article 2',),
  266. ('Article 3',),
  267. ('Article 7',),
  268. ('Article 1',),
  269. ],
  270. )
  271. self.assertSequenceEqual(
  272. Article.objects.values_list('id').order_by('id'),
  273. [(self.a1.id,), (self.a2.id,), (self.a3.id,), (self.a4.id,), (self.a5.id,), (self.a6.id,), (self.a7.id,)],
  274. )
  275. self.assertSequenceEqual(
  276. Article.objects.values_list('id', flat=True).order_by('id'),
  277. [self.a1.id, self.a2.id, self.a3.id, self.a4.id, self.a5.id, self.a6.id, self.a7.id],
  278. )
  279. self.assertSequenceEqual(
  280. Article.objects.extra(select={'id_plus_one': 'id+1'}).order_by('id').values_list('id'),
  281. [(self.a1.id,), (self.a2.id,), (self.a3.id,), (self.a4.id,), (self.a5.id,), (self.a6.id,), (self.a7.id,)],
  282. )
  283. self.assertSequenceEqual(
  284. Article.objects.extra(select={'id_plus_one': 'id+1'}).order_by('id').values_list('id_plus_one', 'id'),
  285. [
  286. (self.a1.id + 1, self.a1.id),
  287. (self.a2.id + 1, self.a2.id),
  288. (self.a3.id + 1, self.a3.id),
  289. (self.a4.id + 1, self.a4.id),
  290. (self.a5.id + 1, self.a5.id),
  291. (self.a6.id + 1, self.a6.id),
  292. (self.a7.id + 1, self.a7.id)
  293. ],
  294. )
  295. self.assertSequenceEqual(
  296. Article.objects.extra(select={'id_plus_one': 'id+1'}).order_by('id').values_list('id', 'id_plus_one'),
  297. [
  298. (self.a1.id, self.a1.id + 1),
  299. (self.a2.id, self.a2.id + 1),
  300. (self.a3.id, self.a3.id + 1),
  301. (self.a4.id, self.a4.id + 1),
  302. (self.a5.id, self.a5.id + 1),
  303. (self.a6.id, self.a6.id + 1),
  304. (self.a7.id, self.a7.id + 1)
  305. ],
  306. )
  307. args = ('name', 'article__headline', 'article__tag__name')
  308. self.assertSequenceEqual(
  309. Author.objects.values_list(*args).order_by(*args),
  310. [
  311. (self.au1.name, self.a1.headline, self.t1.name),
  312. (self.au1.name, self.a2.headline, self.t1.name),
  313. (self.au1.name, self.a3.headline, self.t1.name),
  314. (self.au1.name, self.a3.headline, self.t2.name),
  315. (self.au1.name, self.a4.headline, self.t2.name),
  316. (self.au2.name, self.a5.headline, self.t2.name),
  317. (self.au2.name, self.a5.headline, self.t3.name),
  318. (self.au2.name, self.a6.headline, self.t3.name),
  319. (self.au2.name, self.a7.headline, self.t3.name),
  320. ],
  321. )
  322. with self.assertRaises(TypeError):
  323. Article.objects.values_list('id', 'headline', flat=True)
  324. def test_get_next_previous_by(self):
  325. # Every DateField and DateTimeField creates get_next_by_FOO() and
  326. # get_previous_by_FOO() methods. In the case of identical date values,
  327. # these methods will use the ID as a fallback check. This guarantees
  328. # that no records are skipped or duplicated.
  329. self.assertEqual(repr(self.a1.get_next_by_pub_date()), '<Article: Article 2>')
  330. self.assertEqual(repr(self.a2.get_next_by_pub_date()), '<Article: Article 3>')
  331. self.assertEqual(repr(self.a2.get_next_by_pub_date(headline__endswith='6')), '<Article: Article 6>')
  332. self.assertEqual(repr(self.a3.get_next_by_pub_date()), '<Article: Article 7>')
  333. self.assertEqual(repr(self.a4.get_next_by_pub_date()), '<Article: Article 6>')
  334. with self.assertRaises(Article.DoesNotExist):
  335. self.a5.get_next_by_pub_date()
  336. self.assertEqual(repr(self.a6.get_next_by_pub_date()), '<Article: Article 5>')
  337. self.assertEqual(repr(self.a7.get_next_by_pub_date()), '<Article: Article 4>')
  338. self.assertEqual(repr(self.a7.get_previous_by_pub_date()), '<Article: Article 3>')
  339. self.assertEqual(repr(self.a6.get_previous_by_pub_date()), '<Article: Article 4>')
  340. self.assertEqual(repr(self.a5.get_previous_by_pub_date()), '<Article: Article 6>')
  341. self.assertEqual(repr(self.a4.get_previous_by_pub_date()), '<Article: Article 7>')
  342. self.assertEqual(repr(self.a3.get_previous_by_pub_date()), '<Article: Article 2>')
  343. self.assertEqual(repr(self.a2.get_previous_by_pub_date()), '<Article: Article 1>')
  344. def test_escaping(self):
  345. # Underscores, percent signs and backslashes have special meaning in the
  346. # underlying SQL code, but Django handles the quoting of them automatically.
  347. Article.objects.create(headline='Article_ with underscore', pub_date=datetime(2005, 11, 20))
  348. self.assertQuerysetEqual(
  349. Article.objects.filter(headline__startswith='Article'),
  350. [
  351. '<Article: Article_ with underscore>',
  352. '<Article: Article 5>',
  353. '<Article: Article 6>',
  354. '<Article: Article 4>',
  355. '<Article: Article 2>',
  356. '<Article: Article 3>',
  357. '<Article: Article 7>',
  358. '<Article: Article 1>',
  359. ]
  360. )
  361. self.assertQuerysetEqual(
  362. Article.objects.filter(headline__startswith='Article_'),
  363. ['<Article: Article_ with underscore>']
  364. )
  365. Article.objects.create(headline='Article% with percent sign', pub_date=datetime(2005, 11, 21))
  366. self.assertQuerysetEqual(
  367. Article.objects.filter(headline__startswith='Article'),
  368. [
  369. '<Article: Article% with percent sign>',
  370. '<Article: Article_ with underscore>',
  371. '<Article: Article 5>',
  372. '<Article: Article 6>',
  373. '<Article: Article 4>',
  374. '<Article: Article 2>',
  375. '<Article: Article 3>',
  376. '<Article: Article 7>',
  377. '<Article: Article 1>',
  378. ]
  379. )
  380. self.assertQuerysetEqual(
  381. Article.objects.filter(headline__startswith='Article%'),
  382. ['<Article: Article% with percent sign>']
  383. )
  384. Article.objects.create(headline='Article with \\ backslash', pub_date=datetime(2005, 11, 22))
  385. self.assertQuerysetEqual(
  386. Article.objects.filter(headline__contains='\\'),
  387. [r'<Article: Article with \ backslash>']
  388. )
  389. def test_exclude(self):
  390. Article.objects.create(headline='Article_ with underscore', pub_date=datetime(2005, 11, 20))
  391. Article.objects.create(headline='Article% with percent sign', pub_date=datetime(2005, 11, 21))
  392. Article.objects.create(headline='Article with \\ backslash', pub_date=datetime(2005, 11, 22))
  393. # exclude() is the opposite of filter() when doing lookups:
  394. self.assertQuerysetEqual(
  395. Article.objects.filter(headline__contains='Article').exclude(headline__contains='with'),
  396. [
  397. '<Article: Article 5>',
  398. '<Article: Article 6>',
  399. '<Article: Article 4>',
  400. '<Article: Article 2>',
  401. '<Article: Article 3>',
  402. '<Article: Article 7>',
  403. '<Article: Article 1>',
  404. ]
  405. )
  406. self.assertQuerysetEqual(
  407. Article.objects.exclude(headline__startswith="Article_"),
  408. [
  409. '<Article: Article with \\ backslash>',
  410. '<Article: Article% with percent sign>',
  411. '<Article: Article 5>',
  412. '<Article: Article 6>',
  413. '<Article: Article 4>',
  414. '<Article: Article 2>',
  415. '<Article: Article 3>',
  416. '<Article: Article 7>',
  417. '<Article: Article 1>',
  418. ]
  419. )
  420. self.assertQuerysetEqual(
  421. Article.objects.exclude(headline="Article 7"),
  422. [
  423. '<Article: Article with \\ backslash>',
  424. '<Article: Article% with percent sign>',
  425. '<Article: Article_ with underscore>',
  426. '<Article: Article 5>',
  427. '<Article: Article 6>',
  428. '<Article: Article 4>',
  429. '<Article: Article 2>',
  430. '<Article: Article 3>',
  431. '<Article: Article 1>',
  432. ]
  433. )
  434. def test_none(self):
  435. # none() returns a QuerySet that behaves like any other QuerySet object
  436. self.assertQuerysetEqual(Article.objects.none(), [])
  437. self.assertQuerysetEqual(Article.objects.none().filter(headline__startswith='Article'), [])
  438. self.assertQuerysetEqual(Article.objects.filter(headline__startswith='Article').none(), [])
  439. self.assertEqual(Article.objects.none().count(), 0)
  440. self.assertEqual(Article.objects.none().update(headline="This should not take effect"), 0)
  441. self.assertQuerysetEqual([article for article in Article.objects.none().iterator()], [])
  442. def test_in(self):
  443. # using __in with an empty list should return an empty query set
  444. self.assertQuerysetEqual(Article.objects.filter(id__in=[]), [])
  445. self.assertQuerysetEqual(
  446. Article.objects.exclude(id__in=[]),
  447. [
  448. '<Article: Article 5>',
  449. '<Article: Article 6>',
  450. '<Article: Article 4>',
  451. '<Article: Article 2>',
  452. '<Article: Article 3>',
  453. '<Article: Article 7>',
  454. '<Article: Article 1>',
  455. ]
  456. )
  457. def test_in_different_database(self):
  458. with self.assertRaisesMessage(
  459. ValueError,
  460. "Subqueries aren't allowed across different databases. Force the "
  461. "inner query to be evaluated using `list(inner_query)`."
  462. ):
  463. list(Article.objects.filter(id__in=Article.objects.using('other').all()))
  464. def test_error_messages(self):
  465. # Programming errors are pointed out with nice error messages
  466. with self.assertRaisesMessage(
  467. FieldError,
  468. "Cannot resolve keyword 'pub_date_year' into field. Choices are: "
  469. "author, author_id, headline, id, pub_date, tag"
  470. ):
  471. Article.objects.filter(pub_date_year='2005').count()
  472. with self.assertRaisesMessage(
  473. FieldError,
  474. "Unsupported lookup 'starts' for CharField or join on the field "
  475. "not permitted."
  476. ):
  477. Article.objects.filter(headline__starts='Article')
  478. def test_relation_nested_lookup_error(self):
  479. # An invalid nested lookup on a related field raises a useful error.
  480. msg = 'Related Field got invalid lookup: editor'
  481. with self.assertRaisesMessage(FieldError, msg):
  482. Article.objects.filter(author__editor__name='James')
  483. msg = 'Related Field got invalid lookup: foo'
  484. with self.assertRaisesMessage(FieldError, msg):
  485. Tag.objects.filter(articles__foo='bar')
  486. def test_regex(self):
  487. # Create some articles with a bit more interesting headlines for testing field lookups:
  488. for a in Article.objects.all():
  489. a.delete()
  490. now = datetime.now()
  491. Article.objects.create(pub_date=now, headline='f')
  492. Article.objects.create(pub_date=now, headline='fo')
  493. Article.objects.create(pub_date=now, headline='foo')
  494. Article.objects.create(pub_date=now, headline='fooo')
  495. Article.objects.create(pub_date=now, headline='hey-Foo')
  496. Article.objects.create(pub_date=now, headline='bar')
  497. Article.objects.create(pub_date=now, headline='AbBa')
  498. Article.objects.create(pub_date=now, headline='baz')
  499. Article.objects.create(pub_date=now, headline='baxZ')
  500. # zero-or-more
  501. self.assertQuerysetEqual(
  502. Article.objects.filter(headline__regex=r'fo*'),
  503. ['<Article: f>', '<Article: fo>', '<Article: foo>', '<Article: fooo>']
  504. )
  505. self.assertQuerysetEqual(
  506. Article.objects.filter(headline__iregex=r'fo*'),
  507. [
  508. '<Article: f>',
  509. '<Article: fo>',
  510. '<Article: foo>',
  511. '<Article: fooo>',
  512. '<Article: hey-Foo>',
  513. ]
  514. )
  515. # one-or-more
  516. self.assertQuerysetEqual(
  517. Article.objects.filter(headline__regex=r'fo+'),
  518. ['<Article: fo>', '<Article: foo>', '<Article: fooo>']
  519. )
  520. # wildcard
  521. self.assertQuerysetEqual(
  522. Article.objects.filter(headline__regex=r'fooo?'),
  523. ['<Article: foo>', '<Article: fooo>']
  524. )
  525. # leading anchor
  526. self.assertQuerysetEqual(
  527. Article.objects.filter(headline__regex=r'^b'),
  528. ['<Article: bar>', '<Article: baxZ>', '<Article: baz>']
  529. )
  530. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'^a'), ['<Article: AbBa>'])
  531. # trailing anchor
  532. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'z$'), ['<Article: baz>'])
  533. self.assertQuerysetEqual(
  534. Article.objects.filter(headline__iregex=r'z$'),
  535. ['<Article: baxZ>', '<Article: baz>']
  536. )
  537. # character sets
  538. self.assertQuerysetEqual(
  539. Article.objects.filter(headline__regex=r'ba[rz]'),
  540. ['<Article: bar>', '<Article: baz>']
  541. )
  542. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'ba.[RxZ]'), ['<Article: baxZ>'])
  543. self.assertQuerysetEqual(
  544. Article.objects.filter(headline__iregex=r'ba[RxZ]'),
  545. ['<Article: bar>', '<Article: baxZ>', '<Article: baz>']
  546. )
  547. # and more articles:
  548. Article.objects.create(pub_date=now, headline='foobar')
  549. Article.objects.create(pub_date=now, headline='foobaz')
  550. Article.objects.create(pub_date=now, headline='ooF')
  551. Article.objects.create(pub_date=now, headline='foobarbaz')
  552. Article.objects.create(pub_date=now, headline='zoocarfaz')
  553. Article.objects.create(pub_date=now, headline='barfoobaz')
  554. Article.objects.create(pub_date=now, headline='bazbaRFOO')
  555. # alternation
  556. self.assertQuerysetEqual(
  557. Article.objects.filter(headline__regex=r'oo(f|b)'),
  558. [
  559. '<Article: barfoobaz>',
  560. '<Article: foobar>',
  561. '<Article: foobarbaz>',
  562. '<Article: foobaz>',
  563. ]
  564. )
  565. self.assertQuerysetEqual(
  566. Article.objects.filter(headline__iregex=r'oo(f|b)'),
  567. [
  568. '<Article: barfoobaz>',
  569. '<Article: foobar>',
  570. '<Article: foobarbaz>',
  571. '<Article: foobaz>',
  572. '<Article: ooF>',
  573. ]
  574. )
  575. self.assertQuerysetEqual(
  576. Article.objects.filter(headline__regex=r'^foo(f|b)'),
  577. ['<Article: foobar>', '<Article: foobarbaz>', '<Article: foobaz>']
  578. )
  579. # greedy matching
  580. self.assertQuerysetEqual(
  581. Article.objects.filter(headline__regex=r'b.*az'),
  582. [
  583. '<Article: barfoobaz>',
  584. '<Article: baz>',
  585. '<Article: bazbaRFOO>',
  586. '<Article: foobarbaz>',
  587. '<Article: foobaz>',
  588. ]
  589. )
  590. self.assertQuerysetEqual(
  591. Article.objects.filter(headline__iregex=r'b.*ar'),
  592. [
  593. '<Article: bar>',
  594. '<Article: barfoobaz>',
  595. '<Article: bazbaRFOO>',
  596. '<Article: foobar>',
  597. '<Article: foobarbaz>',
  598. ]
  599. )
  600. @skipUnlessDBFeature('supports_regex_backreferencing')
  601. def test_regex_backreferencing(self):
  602. # grouping and backreferences
  603. now = datetime.now()
  604. Article.objects.create(pub_date=now, headline='foobar')
  605. Article.objects.create(pub_date=now, headline='foobaz')
  606. Article.objects.create(pub_date=now, headline='ooF')
  607. Article.objects.create(pub_date=now, headline='foobarbaz')
  608. Article.objects.create(pub_date=now, headline='zoocarfaz')
  609. Article.objects.create(pub_date=now, headline='barfoobaz')
  610. Article.objects.create(pub_date=now, headline='bazbaRFOO')
  611. self.assertQuerysetEqual(
  612. Article.objects.filter(headline__regex=r'b(.).*b\1'),
  613. ['<Article: barfoobaz>', '<Article: bazbaRFOO>', '<Article: foobarbaz>']
  614. )
  615. def test_regex_null(self):
  616. """
  617. A regex lookup does not fail on null/None values
  618. """
  619. Season.objects.create(year=2012, gt=None)
  620. self.assertQuerysetEqual(Season.objects.filter(gt__regex=r'^$'), [])
  621. def test_regex_non_string(self):
  622. """
  623. A regex lookup does not fail on non-string fields
  624. """
  625. Season.objects.create(year=2013, gt=444)
  626. self.assertQuerysetEqual(Season.objects.filter(gt__regex=r'^444$'), ['<Season: 2013>'])
  627. def test_regex_non_ascii(self):
  628. """
  629. A regex lookup does not trip on non-ASCII characters.
  630. """
  631. Player.objects.create(name='\u2660')
  632. Player.objects.get(name__regex='\u2660')
  633. def test_nonfield_lookups(self):
  634. """
  635. A lookup query containing non-fields raises the proper exception.
  636. """
  637. with self.assertRaises(FieldError):
  638. Article.objects.filter(headline__blahblah=99)
  639. with self.assertRaises(FieldError):
  640. Article.objects.filter(headline__blahblah__exact=99)
  641. with self.assertRaises(FieldError):
  642. Article.objects.filter(blahblah=99)
  643. def test_lookup_collision(self):
  644. """
  645. Genuine field names don't collide with built-in lookup types
  646. ('year', 'gt', 'range', 'in' etc.) (#11670).
  647. """
  648. # 'gt' is used as a code number for the year, e.g. 111=>2009.
  649. season_2009 = Season.objects.create(year=2009, gt=111)
  650. season_2009.games.create(home="Houston Astros", away="St. Louis Cardinals")
  651. season_2010 = Season.objects.create(year=2010, gt=222)
  652. season_2010.games.create(home="Houston Astros", away="Chicago Cubs")
  653. season_2010.games.create(home="Houston Astros", away="Milwaukee Brewers")
  654. season_2010.games.create(home="Houston Astros", away="St. Louis Cardinals")
  655. season_2011 = Season.objects.create(year=2011, gt=333)
  656. season_2011.games.create(home="Houston Astros", away="St. Louis Cardinals")
  657. season_2011.games.create(home="Houston Astros", away="Milwaukee Brewers")
  658. hunter_pence = Player.objects.create(name="Hunter Pence")
  659. hunter_pence.games.set(Game.objects.filter(season__year__in=[2009, 2010]))
  660. pudge = Player.objects.create(name="Ivan Rodriquez")
  661. pudge.games.set(Game.objects.filter(season__year=2009))
  662. pedro_feliz = Player.objects.create(name="Pedro Feliz")
  663. pedro_feliz.games.set(Game.objects.filter(season__year__in=[2011]))
  664. johnson = Player.objects.create(name="Johnson")
  665. johnson.games.set(Game.objects.filter(season__year__in=[2011]))
  666. # Games in 2010
  667. self.assertEqual(Game.objects.filter(season__year=2010).count(), 3)
  668. self.assertEqual(Game.objects.filter(season__year__exact=2010).count(), 3)
  669. self.assertEqual(Game.objects.filter(season__gt=222).count(), 3)
  670. self.assertEqual(Game.objects.filter(season__gt__exact=222).count(), 3)
  671. # Games in 2011
  672. self.assertEqual(Game.objects.filter(season__year=2011).count(), 2)
  673. self.assertEqual(Game.objects.filter(season__year__exact=2011).count(), 2)
  674. self.assertEqual(Game.objects.filter(season__gt=333).count(), 2)
  675. self.assertEqual(Game.objects.filter(season__gt__exact=333).count(), 2)
  676. self.assertEqual(Game.objects.filter(season__year__gt=2010).count(), 2)
  677. self.assertEqual(Game.objects.filter(season__gt__gt=222).count(), 2)
  678. # Games played in 2010 and 2011
  679. self.assertEqual(Game.objects.filter(season__year__in=[2010, 2011]).count(), 5)
  680. self.assertEqual(Game.objects.filter(season__year__gt=2009).count(), 5)
  681. self.assertEqual(Game.objects.filter(season__gt__in=[222, 333]).count(), 5)
  682. self.assertEqual(Game.objects.filter(season__gt__gt=111).count(), 5)
  683. # Players who played in 2009
  684. self.assertEqual(Player.objects.filter(games__season__year=2009).distinct().count(), 2)
  685. self.assertEqual(Player.objects.filter(games__season__year__exact=2009).distinct().count(), 2)
  686. self.assertEqual(Player.objects.filter(games__season__gt=111).distinct().count(), 2)
  687. self.assertEqual(Player.objects.filter(games__season__gt__exact=111).distinct().count(), 2)
  688. # Players who played in 2010
  689. self.assertEqual(Player.objects.filter(games__season__year=2010).distinct().count(), 1)
  690. self.assertEqual(Player.objects.filter(games__season__year__exact=2010).distinct().count(), 1)
  691. self.assertEqual(Player.objects.filter(games__season__gt=222).distinct().count(), 1)
  692. self.assertEqual(Player.objects.filter(games__season__gt__exact=222).distinct().count(), 1)
  693. # Players who played in 2011
  694. self.assertEqual(Player.objects.filter(games__season__year=2011).distinct().count(), 2)
  695. self.assertEqual(Player.objects.filter(games__season__year__exact=2011).distinct().count(), 2)
  696. self.assertEqual(Player.objects.filter(games__season__gt=333).distinct().count(), 2)
  697. self.assertEqual(Player.objects.filter(games__season__year__gt=2010).distinct().count(), 2)
  698. self.assertEqual(Player.objects.filter(games__season__gt__gt=222).distinct().count(), 2)
  699. def test_chain_date_time_lookups(self):
  700. self.assertQuerysetEqual(
  701. Article.objects.filter(pub_date__month__gt=7),
  702. ['<Article: Article 5>', '<Article: Article 6>'],
  703. ordered=False
  704. )
  705. self.assertQuerysetEqual(
  706. Article.objects.filter(pub_date__day__gte=27),
  707. ['<Article: Article 2>', '<Article: Article 3>',
  708. '<Article: Article 4>', '<Article: Article 7>'],
  709. ordered=False
  710. )
  711. self.assertQuerysetEqual(
  712. Article.objects.filter(pub_date__hour__lt=8),
  713. ['<Article: Article 1>', '<Article: Article 2>',
  714. '<Article: Article 3>', '<Article: Article 4>',
  715. '<Article: Article 7>'],
  716. ordered=False
  717. )
  718. self.assertQuerysetEqual(
  719. Article.objects.filter(pub_date__minute__lte=0),
  720. ['<Article: Article 1>', '<Article: Article 2>',
  721. '<Article: Article 3>', '<Article: Article 4>',
  722. '<Article: Article 5>', '<Article: Article 6>',
  723. '<Article: Article 7>'],
  724. ordered=False
  725. )