/tests/modeltests/lookup/tests.py

https://code.google.com/p/mango-py/ · Python · 608 lines · 532 code · 26 blank · 50 comment · 7 complexity · 4dc5e53f61357d6a185aac9b08ca7207 MD5 · raw file

  1. from datetime import datetime
  2. from operator import attrgetter
  3. from django.core.exceptions import FieldError
  4. from django.db import connection
  5. from django.test import TestCase, skipUnlessDBFeature
  6. from models import Author, Article, Tag
  7. class LookupTests(TestCase):
  8. #def setUp(self):
  9. def setUp(self):
  10. # Create a few Authors.
  11. self.au1 = Author(name='Author 1')
  12. self.au1.save()
  13. self.au2 = Author(name='Author 2')
  14. self.au2.save()
  15. # Create a couple of Articles.
  16. self.a1 = Article(headline='Article 1', pub_date=datetime(2005, 7, 26), author=self.au1)
  17. self.a1.save()
  18. self.a2 = Article(headline='Article 2', pub_date=datetime(2005, 7, 27), author=self.au1)
  19. self.a2.save()
  20. self.a3 = Article(headline='Article 3', pub_date=datetime(2005, 7, 27), author=self.au1)
  21. self.a3.save()
  22. self.a4 = Article(headline='Article 4', pub_date=datetime(2005, 7, 28), author=self.au1)
  23. self.a4.save()
  24. self.a5 = Article(headline='Article 5', pub_date=datetime(2005, 8, 1, 9, 0), author=self.au2)
  25. self.a5.save()
  26. self.a6 = Article(headline='Article 6', pub_date=datetime(2005, 8, 1, 8, 0), author=self.au2)
  27. self.a6.save()
  28. self.a7 = Article(headline='Article 7', pub_date=datetime(2005, 7, 27), author=self.au2)
  29. self.a7.save()
  30. # Create a few Tags.
  31. self.t1 = Tag(name='Tag 1')
  32. self.t1.save()
  33. self.t1.articles.add(self.a1, self.a2, self.a3)
  34. self.t2 = Tag(name='Tag 2')
  35. self.t2.save()
  36. self.t2.articles.add(self.a3, self.a4, self.a5)
  37. self.t3 = Tag(name='Tag 3')
  38. self.t3.save()
  39. self.t3.articles.add(self.a5, self.a6, self.a7)
  40. def test_exists(self):
  41. # We can use .exists() to check that there are some
  42. self.assertTrue(Article.objects.exists())
  43. for a in Article.objects.all():
  44. a.delete()
  45. # There should be none now!
  46. self.assertFalse(Article.objects.exists())
  47. def test_lookup_int_as_str(self):
  48. # Integer value can be queried using string
  49. self.assertQuerysetEqual(Article.objects.filter(id__iexact=str(self.a1.id)),
  50. ['<Article: Article 1>'])
  51. @skipUnlessDBFeature('supports_date_lookup_using_string')
  52. def test_lookup_date_as_str(self):
  53. # A date lookup can be performed using a string search
  54. self.assertQuerysetEqual(Article.objects.filter(pub_date__startswith='2005'),
  55. [
  56. '<Article: Article 5>',
  57. '<Article: Article 6>',
  58. '<Article: Article 4>',
  59. '<Article: Article 2>',
  60. '<Article: Article 3>',
  61. '<Article: Article 7>',
  62. '<Article: Article 1>',
  63. ])
  64. def test_iterator(self):
  65. # Each QuerySet gets iterator(), which is a generator that "lazily"
  66. # returns results using database-level iteration.
  67. self.assertQuerysetEqual(Article.objects.iterator(),
  68. [
  69. 'Article 5',
  70. 'Article 6',
  71. 'Article 4',
  72. 'Article 2',
  73. 'Article 3',
  74. 'Article 7',
  75. 'Article 1',
  76. ],
  77. transform=attrgetter('headline'))
  78. # iterator() can be used on any QuerySet.
  79. self.assertQuerysetEqual(
  80. Article.objects.filter(headline__endswith='4').iterator(),
  81. ['Article 4'],
  82. transform=attrgetter('headline'))
  83. def test_count(self):
  84. # count() returns the number of objects matching search criteria.
  85. self.assertEqual(Article.objects.count(), 7)
  86. self.assertEqual(Article.objects.filter(pub_date__exact=datetime(2005, 7, 27)).count(), 3)
  87. self.assertEqual(Article.objects.filter(headline__startswith='Blah blah').count(), 0)
  88. # count() should respect sliced query sets.
  89. articles = Article.objects.all()
  90. self.assertEqual(articles.count(), 7)
  91. self.assertEqual(articles[:4].count(), 4)
  92. self.assertEqual(articles[1:100].count(), 6)
  93. self.assertEqual(articles[10:100].count(), 0)
  94. # Date and date/time lookups can also be done with strings.
  95. self.assertEqual(Article.objects.filter(pub_date__exact='2005-07-27 00:00:00').count(), 3)
  96. def test_in_bulk(self):
  97. # in_bulk() takes a list of IDs and returns a dictionary mapping IDs to objects.
  98. arts = Article.objects.in_bulk([self.a1.id, self.a2.id])
  99. self.assertEqual(arts[self.a1.id], self.a1)
  100. self.assertEqual(arts[self.a2.id], self.a2)
  101. self.assertEqual(Article.objects.in_bulk([self.a3.id]), {self.a3.id: self.a3})
  102. self.assertEqual(Article.objects.in_bulk(set([self.a3.id])), {self.a3.id: self.a3})
  103. self.assertEqual(Article.objects.in_bulk(frozenset([self.a3.id])), {self.a3.id: self.a3})
  104. self.assertEqual(Article.objects.in_bulk((self.a3.id,)), {self.a3.id: self.a3})
  105. self.assertEqual(Article.objects.in_bulk([1000]), {})
  106. self.assertEqual(Article.objects.in_bulk([]), {})
  107. self.assertRaises(AssertionError, Article.objects.in_bulk, 'foo')
  108. self.assertRaises(TypeError, Article.objects.in_bulk)
  109. self.assertRaises(TypeError, Article.objects.in_bulk, headline__startswith='Blah')
  110. def test_values(self):
  111. # values() returns a list of dictionaries instead of object instances --
  112. # and you can specify which fields you want to retrieve.
  113. identity = lambda x:x
  114. self.assertQuerysetEqual(Article.objects.values('headline'),
  115. [
  116. {'headline': u'Article 5'},
  117. {'headline': u'Article 6'},
  118. {'headline': u'Article 4'},
  119. {'headline': u'Article 2'},
  120. {'headline': u'Article 3'},
  121. {'headline': u'Article 7'},
  122. {'headline': u'Article 1'},
  123. ],
  124. transform=identity)
  125. self.assertQuerysetEqual(
  126. Article.objects.filter(pub_date__exact=datetime(2005, 7, 27)).values('id'),
  127. [{'id': self.a2.id}, {'id': self.a3.id}, {'id': self.a7.id}],
  128. transform=identity)
  129. self.assertQuerysetEqual(Article.objects.values('id', 'headline'),
  130. [
  131. {'id': self.a5.id, 'headline': 'Article 5'},
  132. {'id': self.a6.id, 'headline': 'Article 6'},
  133. {'id': self.a4.id, 'headline': 'Article 4'},
  134. {'id': self.a2.id, 'headline': 'Article 2'},
  135. {'id': self.a3.id, 'headline': 'Article 3'},
  136. {'id': self.a7.id, 'headline': 'Article 7'},
  137. {'id': self.a1.id, 'headline': 'Article 1'},
  138. ],
  139. transform=identity)
  140. # You can use values() with iterator() for memory savings,
  141. # because iterator() uses database-level iteration.
  142. self.assertQuerysetEqual(Article.objects.values('id', 'headline').iterator(),
  143. [
  144. {'headline': u'Article 5', 'id': self.a5.id},
  145. {'headline': u'Article 6', 'id': self.a6.id},
  146. {'headline': u'Article 4', 'id': self.a4.id},
  147. {'headline': u'Article 2', 'id': self.a2.id},
  148. {'headline': u'Article 3', 'id': self.a3.id},
  149. {'headline': u'Article 7', 'id': self.a7.id},
  150. {'headline': u'Article 1', 'id': self.a1.id},
  151. ],
  152. transform=identity)
  153. # The values() method works with "extra" fields specified in extra(select).
  154. self.assertQuerysetEqual(
  155. Article.objects.extra(select={'id_plus_one': 'id + 1'}).values('id', 'id_plus_one'),
  156. [
  157. {'id': self.a5.id, 'id_plus_one': self.a5.id + 1},
  158. {'id': self.a6.id, 'id_plus_one': self.a6.id + 1},
  159. {'id': self.a4.id, 'id_plus_one': self.a4.id + 1},
  160. {'id': self.a2.id, 'id_plus_one': self.a2.id + 1},
  161. {'id': self.a3.id, 'id_plus_one': self.a3.id + 1},
  162. {'id': self.a7.id, 'id_plus_one': self.a7.id + 1},
  163. {'id': self.a1.id, 'id_plus_one': self.a1.id + 1},
  164. ],
  165. transform=identity)
  166. data = {
  167. 'id_plus_one': 'id+1',
  168. 'id_plus_two': 'id+2',
  169. 'id_plus_three': 'id+3',
  170. 'id_plus_four': 'id+4',
  171. 'id_plus_five': 'id+5',
  172. 'id_plus_six': 'id+6',
  173. 'id_plus_seven': 'id+7',
  174. 'id_plus_eight': 'id+8',
  175. }
  176. self.assertQuerysetEqual(
  177. Article.objects.filter(id=self.a1.id).extra(select=data).values(*data.keys()),
  178. [{
  179. 'id_plus_one': self.a1.id + 1,
  180. 'id_plus_two': self.a1.id + 2,
  181. 'id_plus_three': self.a1.id + 3,
  182. 'id_plus_four': self.a1.id + 4,
  183. 'id_plus_five': self.a1.id + 5,
  184. 'id_plus_six': self.a1.id + 6,
  185. 'id_plus_seven': self.a1.id + 7,
  186. 'id_plus_eight': self.a1.id + 8,
  187. }], transform=identity)
  188. # You can specify fields from forward and reverse relations, just like filter().
  189. self.assertQuerysetEqual(
  190. Article.objects.values('headline', 'author__name'),
  191. [
  192. {'headline': self.a5.headline, 'author__name': self.au2.name},
  193. {'headline': self.a6.headline, 'author__name': self.au2.name},
  194. {'headline': self.a4.headline, 'author__name': self.au1.name},
  195. {'headline': self.a2.headline, 'author__name': self.au1.name},
  196. {'headline': self.a3.headline, 'author__name': self.au1.name},
  197. {'headline': self.a7.headline, 'author__name': self.au2.name},
  198. {'headline': self.a1.headline, 'author__name': self.au1.name},
  199. ], transform=identity)
  200. self.assertQuerysetEqual(
  201. Author.objects.values('name', 'article__headline').order_by('name', 'article__headline'),
  202. [
  203. {'name': self.au1.name, 'article__headline': self.a1.headline},
  204. {'name': self.au1.name, 'article__headline': self.a2.headline},
  205. {'name': self.au1.name, 'article__headline': self.a3.headline},
  206. {'name': self.au1.name, 'article__headline': self.a4.headline},
  207. {'name': self.au2.name, 'article__headline': self.a5.headline},
  208. {'name': self.au2.name, 'article__headline': self.a6.headline},
  209. {'name': self.au2.name, 'article__headline': self.a7.headline},
  210. ], transform=identity)
  211. self.assertQuerysetEqual(
  212. Author.objects.values('name', 'article__headline', 'article__tag__name').order_by('name', 'article__headline', 'article__tag__name'),
  213. [
  214. {'name': self.au1.name, 'article__headline': self.a1.headline, 'article__tag__name': self.t1.name},
  215. {'name': self.au1.name, 'article__headline': self.a2.headline, 'article__tag__name': self.t1.name},
  216. {'name': self.au1.name, 'article__headline': self.a3.headline, 'article__tag__name': self.t1.name},
  217. {'name': self.au1.name, 'article__headline': self.a3.headline, 'article__tag__name': self.t2.name},
  218. {'name': self.au1.name, 'article__headline': self.a4.headline, 'article__tag__name': self.t2.name},
  219. {'name': self.au2.name, 'article__headline': self.a5.headline, 'article__tag__name': self.t2.name},
  220. {'name': self.au2.name, 'article__headline': self.a5.headline, 'article__tag__name': self.t3.name},
  221. {'name': self.au2.name, 'article__headline': self.a6.headline, 'article__tag__name': self.t3.name},
  222. {'name': self.au2.name, 'article__headline': self.a7.headline, 'article__tag__name': self.t3.name},
  223. ], transform=identity)
  224. # However, an exception FieldDoesNotExist will be thrown if you specify
  225. # a non-existent field name in values() (a field that is neither in the
  226. # model nor in extra(select)).
  227. self.assertRaises(FieldError,
  228. Article.objects.extra(select={'id_plus_one': 'id + 1'}).values,
  229. 'id', 'id_plus_two')
  230. # If you don't specify field names to values(), all are returned.
  231. self.assertQuerysetEqual(Article.objects.filter(id=self.a5.id).values(),
  232. [{
  233. 'id': self.a5.id,
  234. 'author_id': self.au2.id,
  235. 'headline': 'Article 5',
  236. 'pub_date': datetime(2005, 8, 1, 9, 0)
  237. }], transform=identity)
  238. def test_values_list(self):
  239. # values_list() is similar to values(), except that the results are
  240. # returned as a list of tuples, rather than a list of dictionaries.
  241. # Within each tuple, the order of the elemnts is the same as the order
  242. # of fields in the values_list() call.
  243. identity = lambda x:x
  244. self.assertQuerysetEqual(Article.objects.values_list('headline'),
  245. [
  246. (u'Article 5',),
  247. (u'Article 6',),
  248. (u'Article 4',),
  249. (u'Article 2',),
  250. (u'Article 3',),
  251. (u'Article 7',),
  252. (u'Article 1',),
  253. ], transform=identity)
  254. self.assertQuerysetEqual(Article.objects.values_list('id').order_by('id'),
  255. [(self.a1.id,), (self.a2.id,), (self.a3.id,), (self.a4.id,), (self.a5.id,), (self.a6.id,), (self.a7.id,)],
  256. transform=identity)
  257. self.assertQuerysetEqual(
  258. Article.objects.values_list('id', flat=True).order_by('id'),
  259. [self.a1.id, self.a2.id, self.a3.id, self.a4.id, self.a5.id, self.a6.id, self.a7.id],
  260. transform=identity)
  261. self.assertQuerysetEqual(
  262. Article.objects.extra(select={'id_plus_one': 'id+1'})
  263. .order_by('id').values_list('id'),
  264. [(self.a1.id,), (self.a2.id,), (self.a3.id,), (self.a4.id,), (self.a5.id,), (self.a6.id,), (self.a7.id,)],
  265. transform=identity)
  266. self.assertQuerysetEqual(
  267. Article.objects.extra(select={'id_plus_one': 'id+1'})
  268. .order_by('id').values_list('id_plus_one', 'id'),
  269. [
  270. (self.a1.id+1, self.a1.id),
  271. (self.a2.id+1, self.a2.id),
  272. (self.a3.id+1, self.a3.id),
  273. (self.a4.id+1, self.a4.id),
  274. (self.a5.id+1, self.a5.id),
  275. (self.a6.id+1, self.a6.id),
  276. (self.a7.id+1, self.a7.id)
  277. ],
  278. transform=identity)
  279. self.assertQuerysetEqual(
  280. Article.objects.extra(select={'id_plus_one': 'id+1'})
  281. .order_by('id').values_list('id', 'id_plus_one'),
  282. [
  283. (self.a1.id, self.a1.id+1),
  284. (self.a2.id, self.a2.id+1),
  285. (self.a3.id, self.a3.id+1),
  286. (self.a4.id, self.a4.id+1),
  287. (self.a5.id, self.a5.id+1),
  288. (self.a6.id, self.a6.id+1),
  289. (self.a7.id, self.a7.id+1)
  290. ],
  291. transform=identity)
  292. self.assertQuerysetEqual(
  293. Author.objects.values_list('name', 'article__headline', 'article__tag__name').order_by('name', 'article__headline', 'article__tag__name'),
  294. [
  295. (self.au1.name, self.a1.headline, self.t1.name),
  296. (self.au1.name, self.a2.headline, self.t1.name),
  297. (self.au1.name, self.a3.headline, self.t1.name),
  298. (self.au1.name, self.a3.headline, self.t2.name),
  299. (self.au1.name, self.a4.headline, self.t2.name),
  300. (self.au2.name, self.a5.headline, self.t2.name),
  301. (self.au2.name, self.a5.headline, self.t3.name),
  302. (self.au2.name, self.a6.headline, self.t3.name),
  303. (self.au2.name, self.a7.headline, self.t3.name),
  304. ], transform=identity)
  305. self.assertRaises(TypeError, Article.objects.values_list, 'id', 'headline', flat=True)
  306. def test_get_next_previous_by(self):
  307. # Every DateField and DateTimeField creates get_next_by_FOO() and
  308. # get_previous_by_FOO() methods. In the case of identical date values,
  309. # these methods will use the ID as a fallback check. This guarantees
  310. # that no records are skipped or duplicated.
  311. self.assertEqual(repr(self.a1.get_next_by_pub_date()),
  312. '<Article: Article 2>')
  313. self.assertEqual(repr(self.a2.get_next_by_pub_date()),
  314. '<Article: Article 3>')
  315. self.assertEqual(repr(self.a2.get_next_by_pub_date(headline__endswith='6')),
  316. '<Article: Article 6>')
  317. self.assertEqual(repr(self.a3.get_next_by_pub_date()),
  318. '<Article: Article 7>')
  319. self.assertEqual(repr(self.a4.get_next_by_pub_date()),
  320. '<Article: Article 6>')
  321. self.assertRaises(Article.DoesNotExist, self.a5.get_next_by_pub_date)
  322. self.assertEqual(repr(self.a6.get_next_by_pub_date()),
  323. '<Article: Article 5>')
  324. self.assertEqual(repr(self.a7.get_next_by_pub_date()),
  325. '<Article: Article 4>')
  326. self.assertEqual(repr(self.a7.get_previous_by_pub_date()),
  327. '<Article: Article 3>')
  328. self.assertEqual(repr(self.a6.get_previous_by_pub_date()),
  329. '<Article: Article 4>')
  330. self.assertEqual(repr(self.a5.get_previous_by_pub_date()),
  331. '<Article: Article 6>')
  332. self.assertEqual(repr(self.a4.get_previous_by_pub_date()),
  333. '<Article: Article 7>')
  334. self.assertEqual(repr(self.a3.get_previous_by_pub_date()),
  335. '<Article: Article 2>')
  336. self.assertEqual(repr(self.a2.get_previous_by_pub_date()),
  337. '<Article: Article 1>')
  338. def test_escaping(self):
  339. # Underscores, percent signs and backslashes have special meaning in the
  340. # underlying SQL code, but Django handles the quoting of them automatically.
  341. a8 = Article(headline='Article_ with underscore', pub_date=datetime(2005, 11, 20))
  342. a8.save()
  343. self.assertQuerysetEqual(Article.objects.filter(headline__startswith='Article'),
  344. [
  345. '<Article: Article_ with underscore>',
  346. '<Article: Article 5>',
  347. '<Article: Article 6>',
  348. '<Article: Article 4>',
  349. '<Article: Article 2>',
  350. '<Article: Article 3>',
  351. '<Article: Article 7>',
  352. '<Article: Article 1>',
  353. ])
  354. self.assertQuerysetEqual(Article.objects.filter(headline__startswith='Article_'),
  355. ['<Article: Article_ with underscore>'])
  356. a9 = Article(headline='Article% with percent sign', pub_date=datetime(2005, 11, 21))
  357. a9.save()
  358. self.assertQuerysetEqual(Article.objects.filter(headline__startswith='Article'),
  359. [
  360. '<Article: Article% with percent sign>',
  361. '<Article: Article_ with underscore>',
  362. '<Article: Article 5>',
  363. '<Article: Article 6>',
  364. '<Article: Article 4>',
  365. '<Article: Article 2>',
  366. '<Article: Article 3>',
  367. '<Article: Article 7>',
  368. '<Article: Article 1>',
  369. ])
  370. self.assertQuerysetEqual(Article.objects.filter(headline__startswith='Article%'),
  371. ['<Article: Article% with percent sign>'])
  372. a10 = Article(headline='Article with \\ backslash', pub_date=datetime(2005, 11, 22))
  373. a10.save()
  374. self.assertQuerysetEqual(Article.objects.filter(headline__contains='\\'),
  375. ['<Article: Article with \ backslash>'])
  376. def test_exclude(self):
  377. a8 = Article.objects.create(headline='Article_ with underscore', pub_date=datetime(2005, 11, 20))
  378. a9 = Article.objects.create(headline='Article% with percent sign', pub_date=datetime(2005, 11, 21))
  379. a10 = Article.objects.create(headline='Article with \\ backslash', pub_date=datetime(2005, 11, 22))
  380. # exclude() is the opposite of filter() when doing lookups:
  381. self.assertQuerysetEqual(
  382. Article.objects.filter(headline__contains='Article').exclude(headline__contains='with'),
  383. [
  384. '<Article: Article 5>',
  385. '<Article: Article 6>',
  386. '<Article: Article 4>',
  387. '<Article: Article 2>',
  388. '<Article: Article 3>',
  389. '<Article: Article 7>',
  390. '<Article: Article 1>',
  391. ])
  392. self.assertQuerysetEqual(Article.objects.exclude(headline__startswith="Article_"),
  393. [
  394. '<Article: Article with \\ backslash>',
  395. '<Article: Article% with percent sign>',
  396. '<Article: Article 5>',
  397. '<Article: Article 6>',
  398. '<Article: Article 4>',
  399. '<Article: Article 2>',
  400. '<Article: Article 3>',
  401. '<Article: Article 7>',
  402. '<Article: Article 1>',
  403. ])
  404. self.assertQuerysetEqual(Article.objects.exclude(headline="Article 7"),
  405. [
  406. '<Article: Article with \\ backslash>',
  407. '<Article: Article% with percent sign>',
  408. '<Article: Article_ with underscore>',
  409. '<Article: Article 5>',
  410. '<Article: Article 6>',
  411. '<Article: Article 4>',
  412. '<Article: Article 2>',
  413. '<Article: Article 3>',
  414. '<Article: Article 1>',
  415. ])
  416. def test_none(self):
  417. # none() returns an EmptyQuerySet that behaves like any other QuerySet object
  418. self.assertQuerysetEqual(Article.objects.none(), [])
  419. self.assertQuerysetEqual(
  420. Article.objects.none().filter(headline__startswith='Article'), [])
  421. self.assertQuerysetEqual(
  422. Article.objects.filter(headline__startswith='Article').none(), [])
  423. self.assertEqual(Article.objects.none().count(), 0)
  424. self.assertEqual(
  425. Article.objects.none().update(headline="This should not take effect"), 0)
  426. self.assertQuerysetEqual(
  427. [article for article in Article.objects.none().iterator()],
  428. [])
  429. def test_in(self):
  430. # using __in with an empty list should return an empty query set
  431. self.assertQuerysetEqual(Article.objects.filter(id__in=[]), [])
  432. self.assertQuerysetEqual(Article.objects.exclude(id__in=[]),
  433. [
  434. '<Article: Article 5>',
  435. '<Article: Article 6>',
  436. '<Article: Article 4>',
  437. '<Article: Article 2>',
  438. '<Article: Article 3>',
  439. '<Article: Article 7>',
  440. '<Article: Article 1>',
  441. ])
  442. def test_error_messages(self):
  443. # Programming errors are pointed out with nice error messages
  444. try:
  445. Article.objects.filter(pub_date_year='2005').count()
  446. self.fail('FieldError not raised')
  447. except FieldError, ex:
  448. self.assertEqual(str(ex), "Cannot resolve keyword 'pub_date_year' "
  449. "into field. Choices are: author, headline, id, pub_date, tag")
  450. try:
  451. Article.objects.filter(headline__starts='Article')
  452. self.fail('FieldError not raised')
  453. except FieldError, ex:
  454. self.assertEqual(str(ex), "Join on field 'headline' not permitted. "
  455. "Did you misspell 'starts' for the lookup type?")
  456. def test_regex(self):
  457. # Create some articles with a bit more interesting headlines for testing field lookups:
  458. for a in Article.objects.all():
  459. a.delete()
  460. now = datetime.now()
  461. a1 = Article(pub_date=now, headline='f')
  462. a1.save()
  463. a2 = Article(pub_date=now, headline='fo')
  464. a2.save()
  465. a3 = Article(pub_date=now, headline='foo')
  466. a3.save()
  467. a4 = Article(pub_date=now, headline='fooo')
  468. a4.save()
  469. a5 = Article(pub_date=now, headline='hey-Foo')
  470. a5.save()
  471. a6 = Article(pub_date=now, headline='bar')
  472. a6.save()
  473. a7 = Article(pub_date=now, headline='AbBa')
  474. a7.save()
  475. a8 = Article(pub_date=now, headline='baz')
  476. a8.save()
  477. a9 = Article(pub_date=now, headline='baxZ')
  478. a9.save()
  479. # zero-or-more
  480. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'fo*'),
  481. ['<Article: f>', '<Article: fo>', '<Article: foo>', '<Article: fooo>'])
  482. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'fo*'),
  483. [
  484. '<Article: f>',
  485. '<Article: fo>',
  486. '<Article: foo>',
  487. '<Article: fooo>',
  488. '<Article: hey-Foo>',
  489. ])
  490. # one-or-more
  491. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'fo+'),
  492. ['<Article: fo>', '<Article: foo>', '<Article: fooo>'])
  493. # wildcard
  494. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'fooo?'),
  495. ['<Article: foo>', '<Article: fooo>'])
  496. # leading anchor
  497. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'^b'),
  498. ['<Article: bar>', '<Article: baxZ>', '<Article: baz>'])
  499. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'^a'),
  500. ['<Article: AbBa>'])
  501. # trailing anchor
  502. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'z$'),
  503. ['<Article: baz>'])
  504. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'z$'),
  505. ['<Article: baxZ>', '<Article: baz>'])
  506. # character sets
  507. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'ba[rz]'),
  508. ['<Article: bar>', '<Article: baz>'])
  509. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'ba.[RxZ]'),
  510. ['<Article: baxZ>'])
  511. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'ba[RxZ]'),
  512. ['<Article: bar>', '<Article: baxZ>', '<Article: baz>'])
  513. # and more articles:
  514. a10 = Article(pub_date=now, headline='foobar')
  515. a10.save()
  516. a11 = Article(pub_date=now, headline='foobaz')
  517. a11.save()
  518. a12 = Article(pub_date=now, headline='ooF')
  519. a12.save()
  520. a13 = Article(pub_date=now, headline='foobarbaz')
  521. a13.save()
  522. a14 = Article(pub_date=now, headline='zoocarfaz')
  523. a14.save()
  524. a15 = Article(pub_date=now, headline='barfoobaz')
  525. a15.save()
  526. a16 = Article(pub_date=now, headline='bazbaRFOO')
  527. a16.save()
  528. # alternation
  529. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'oo(f|b)'),
  530. [
  531. '<Article: barfoobaz>',
  532. '<Article: foobar>',
  533. '<Article: foobarbaz>',
  534. '<Article: foobaz>',
  535. ])
  536. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'oo(f|b)'),
  537. [
  538. '<Article: barfoobaz>',
  539. '<Article: foobar>',
  540. '<Article: foobarbaz>',
  541. '<Article: foobaz>',
  542. '<Article: ooF>',
  543. ])
  544. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'^foo(f|b)'),
  545. ['<Article: foobar>', '<Article: foobarbaz>', '<Article: foobaz>'])
  546. # greedy matching
  547. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'b.*az'),
  548. [
  549. '<Article: barfoobaz>',
  550. '<Article: baz>',
  551. '<Article: bazbaRFOO>',
  552. '<Article: foobarbaz>',
  553. '<Article: foobaz>',
  554. ])
  555. self.assertQuerysetEqual(Article.objects.filter(headline__iregex=r'b.*ar'),
  556. [
  557. '<Article: bar>',
  558. '<Article: barfoobaz>',
  559. '<Article: bazbaRFOO>',
  560. '<Article: foobar>',
  561. '<Article: foobarbaz>',
  562. ])
  563. @skipUnlessDBFeature('supports_regex_backreferencing')
  564. def test_regex_backreferencing(self):
  565. # grouping and backreferences
  566. now = datetime.now()
  567. a10 = Article(pub_date=now, headline='foobar')
  568. a10.save()
  569. a11 = Article(pub_date=now, headline='foobaz')
  570. a11.save()
  571. a12 = Article(pub_date=now, headline='ooF')
  572. a12.save()
  573. a13 = Article(pub_date=now, headline='foobarbaz')
  574. a13.save()
  575. a14 = Article(pub_date=now, headline='zoocarfaz')
  576. a14.save()
  577. a15 = Article(pub_date=now, headline='barfoobaz')
  578. a15.save()
  579. a16 = Article(pub_date=now, headline='bazbaRFOO')
  580. a16.save()
  581. self.assertQuerysetEqual(Article.objects.filter(headline__regex=r'b(.).*b\1'),
  582. ['<Article: barfoobaz>', '<Article: bazbaRFOO>', '<Article: foobarbaz>'])