test_dates.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712
  1. # -*- coding: utf-8 -*-
  2. from __future__ import unicode_literals
  3. import datetime
  4. from django.core.exceptions import ImproperlyConfigured
  5. from django.test import TestCase, override_settings, skipUnlessDBFeature
  6. from django.test.utils import requires_tz_support
  7. from django.utils import timezone
  8. from .models import Artist, Author, Book, BookSigning, Page
  9. def _make_books(n, base_date):
  10. for i in range(n):
  11. Book.objects.create(
  12. name='Book %d' % i,
  13. slug='book-%d' % i,
  14. pages=100 + i,
  15. pubdate=base_date - datetime.timedelta(days=i))
  16. class TestDataMixin(object):
  17. @classmethod
  18. def setUpTestData(cls):
  19. cls.artist1 = Artist.objects.create(name='Rene Magritte')
  20. cls.author1 = Author.objects.create(name='Roberto Bolaño', slug='roberto-bolano')
  21. cls.author2 = Author.objects.create(name='Scott Rosenberg', slug='scott-rosenberg')
  22. cls.book1 = Book.objects.create(name='2066', slug='2066', pages=800, pubdate=datetime.date(2008, 10, 1))
  23. cls.book1.authors.add(cls.author1)
  24. cls.book2 = Book.objects.create(
  25. name='Dreaming in Code', slug='dreaming-in-code', pages=300, pubdate=datetime.date(2006, 5, 1)
  26. )
  27. cls.page1 = Page.objects.create(
  28. content='I was once bitten by a moose.', template='generic_views/page_template.html'
  29. )
  30. @override_settings(ROOT_URLCONF='generic_views.urls')
  31. class ArchiveIndexViewTests(TestDataMixin, TestCase):
  32. def test_archive_view(self):
  33. res = self.client.get('/dates/books/')
  34. self.assertEqual(res.status_code, 200)
  35. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  36. self.assertEqual(list(res.context['latest']), list(Book.objects.all()))
  37. self.assertTemplateUsed(res, 'generic_views/book_archive.html')
  38. def test_archive_view_context_object_name(self):
  39. res = self.client.get('/dates/books/context_object_name/')
  40. self.assertEqual(res.status_code, 200)
  41. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  42. self.assertEqual(list(res.context['thingies']), list(Book.objects.all()))
  43. self.assertNotIn('latest', res.context)
  44. self.assertTemplateUsed(res, 'generic_views/book_archive.html')
  45. def test_empty_archive_view(self):
  46. Book.objects.all().delete()
  47. res = self.client.get('/dates/books/')
  48. self.assertEqual(res.status_code, 404)
  49. def test_allow_empty_archive_view(self):
  50. Book.objects.all().delete()
  51. res = self.client.get('/dates/books/allow_empty/')
  52. self.assertEqual(res.status_code, 200)
  53. self.assertEqual(list(res.context['date_list']), [])
  54. self.assertTemplateUsed(res, 'generic_views/book_archive.html')
  55. def test_archive_view_template(self):
  56. res = self.client.get('/dates/books/template_name/')
  57. self.assertEqual(res.status_code, 200)
  58. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  59. self.assertEqual(list(res.context['latest']), list(Book.objects.all()))
  60. self.assertTemplateUsed(res, 'generic_views/list.html')
  61. def test_archive_view_template_suffix(self):
  62. res = self.client.get('/dates/books/template_name_suffix/')
  63. self.assertEqual(res.status_code, 200)
  64. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  65. self.assertEqual(list(res.context['latest']), list(Book.objects.all()))
  66. self.assertTemplateUsed(res, 'generic_views/book_detail.html')
  67. def test_archive_view_invalid(self):
  68. with self.assertRaises(ImproperlyConfigured):
  69. self.client.get('/dates/books/invalid/')
  70. def test_archive_view_by_month(self):
  71. res = self.client.get('/dates/books/by_month/')
  72. self.assertEqual(res.status_code, 200)
  73. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'month', 'DESC')))
  74. def test_paginated_archive_view(self):
  75. _make_books(20, base_date=datetime.date.today())
  76. res = self.client.get('/dates/books/paginated/')
  77. self.assertEqual(res.status_code, 200)
  78. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  79. self.assertEqual(list(res.context['latest']), list(Book.objects.all()[0:10]))
  80. self.assertTemplateUsed(res, 'generic_views/book_archive.html')
  81. res = self.client.get('/dates/books/paginated/?page=2')
  82. self.assertEqual(res.status_code, 200)
  83. self.assertEqual(res.context['page_obj'].number, 2)
  84. self.assertEqual(list(res.context['latest']), list(Book.objects.all()[10:20]))
  85. def test_paginated_archive_view_does_not_load_entire_table(self):
  86. # Regression test for #18087
  87. _make_books(20, base_date=datetime.date.today())
  88. # 1 query for years list + 1 query for books
  89. with self.assertNumQueries(2):
  90. self.client.get('/dates/books/')
  91. # same as above + 1 query to test if books exist + 1 query to count them
  92. with self.assertNumQueries(4):
  93. self.client.get('/dates/books/paginated/')
  94. def test_no_duplicate_query(self):
  95. # Regression test for #18354
  96. with self.assertNumQueries(2):
  97. self.client.get('/dates/books/reverse/')
  98. def test_datetime_archive_view(self):
  99. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0))
  100. res = self.client.get('/dates/booksignings/')
  101. self.assertEqual(res.status_code, 200)
  102. @requires_tz_support
  103. @skipUnlessDBFeature('has_zoneinfo_database')
  104. @override_settings(USE_TZ=True, TIME_ZONE='Africa/Nairobi')
  105. def test_aware_datetime_archive_view(self):
  106. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0, tzinfo=timezone.utc))
  107. res = self.client.get('/dates/booksignings/')
  108. self.assertEqual(res.status_code, 200)
  109. def test_date_list_order(self):
  110. """date_list should be sorted descending in index"""
  111. _make_books(5, base_date=datetime.date(2011, 12, 25))
  112. res = self.client.get('/dates/books/')
  113. self.assertEqual(res.status_code, 200)
  114. self.assertEqual(list(res.context['date_list']), list(reversed(sorted(res.context['date_list']))))
  115. def test_archive_view_custom_sorting(self):
  116. Book.objects.create(name="Zebras for Dummies", pages=600, pubdate=datetime.date(2007, 5, 1))
  117. res = self.client.get('/dates/books/sortedbyname/')
  118. self.assertEqual(res.status_code, 200)
  119. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  120. self.assertEqual(list(res.context['latest']), list(Book.objects.order_by('name').all()))
  121. self.assertTemplateUsed(res, 'generic_views/book_archive.html')
  122. def test_archive_view_custom_sorting_dec(self):
  123. Book.objects.create(name="Zebras for Dummies", pages=600, pubdate=datetime.date(2007, 5, 1))
  124. res = self.client.get('/dates/books/sortedbynamedec/')
  125. self.assertEqual(res.status_code, 200)
  126. self.assertEqual(list(res.context['date_list']), list(Book.objects.dates('pubdate', 'year', 'DESC')))
  127. self.assertEqual(list(res.context['latest']), list(Book.objects.order_by('-name').all()))
  128. self.assertTemplateUsed(res, 'generic_views/book_archive.html')
  129. @override_settings(ROOT_URLCONF='generic_views.urls')
  130. class YearArchiveViewTests(TestDataMixin, TestCase):
  131. def test_year_view(self):
  132. res = self.client.get('/dates/books/2008/')
  133. self.assertEqual(res.status_code, 200)
  134. self.assertEqual(list(res.context['date_list']), [datetime.date(2008, 10, 1)])
  135. self.assertEqual(res.context['year'], datetime.date(2008, 1, 1))
  136. self.assertTemplateUsed(res, 'generic_views/book_archive_year.html')
  137. # Since allow_empty=False, next/prev years must be valid (#7164)
  138. self.assertIsNone(res.context['next_year'])
  139. self.assertEqual(res.context['previous_year'], datetime.date(2006, 1, 1))
  140. def test_year_view_make_object_list(self):
  141. res = self.client.get('/dates/books/2006/make_object_list/')
  142. self.assertEqual(res.status_code, 200)
  143. self.assertEqual(list(res.context['date_list']), [datetime.date(2006, 5, 1)])
  144. self.assertEqual(list(res.context['book_list']), list(Book.objects.filter(pubdate__year=2006)))
  145. self.assertEqual(list(res.context['object_list']), list(Book.objects.filter(pubdate__year=2006)))
  146. self.assertTemplateUsed(res, 'generic_views/book_archive_year.html')
  147. def test_year_view_empty(self):
  148. res = self.client.get('/dates/books/1999/')
  149. self.assertEqual(res.status_code, 404)
  150. res = self.client.get('/dates/books/1999/allow_empty/')
  151. self.assertEqual(res.status_code, 200)
  152. self.assertEqual(list(res.context['date_list']), [])
  153. self.assertEqual(list(res.context['book_list']), [])
  154. # Since allow_empty=True, next/prev are allowed to be empty years (#7164)
  155. self.assertEqual(res.context['next_year'], datetime.date(2000, 1, 1))
  156. self.assertEqual(res.context['previous_year'], datetime.date(1998, 1, 1))
  157. def test_year_view_allow_future(self):
  158. # Create a new book in the future
  159. year = datetime.date.today().year + 1
  160. Book.objects.create(name="The New New Testement", pages=600, pubdate=datetime.date(year, 1, 1))
  161. res = self.client.get('/dates/books/%s/' % year)
  162. self.assertEqual(res.status_code, 404)
  163. res = self.client.get('/dates/books/%s/allow_empty/' % year)
  164. self.assertEqual(res.status_code, 200)
  165. self.assertEqual(list(res.context['book_list']), [])
  166. res = self.client.get('/dates/books/%s/allow_future/' % year)
  167. self.assertEqual(res.status_code, 200)
  168. self.assertEqual(list(res.context['date_list']), [datetime.date(year, 1, 1)])
  169. def test_year_view_paginated(self):
  170. res = self.client.get('/dates/books/2006/paginated/')
  171. self.assertEqual(res.status_code, 200)
  172. self.assertEqual(list(res.context['book_list']), list(Book.objects.filter(pubdate__year=2006)))
  173. self.assertEqual(list(res.context['object_list']), list(Book.objects.filter(pubdate__year=2006)))
  174. self.assertTemplateUsed(res, 'generic_views/book_archive_year.html')
  175. def test_year_view_custom_sort_order(self):
  176. # Zebras comes after Dreaming by name, but before on '-pubdate' which is the default sorting
  177. Book.objects.create(name="Zebras for Dummies", pages=600, pubdate=datetime.date(2006, 9, 1))
  178. res = self.client.get('/dates/books/2006/sortedbyname/')
  179. self.assertEqual(res.status_code, 200)
  180. self.assertEqual(list(res.context['date_list']), [datetime.date(2006, 5, 1), datetime.date(2006, 9, 1)])
  181. self.assertEqual(
  182. list(res.context['book_list']),
  183. list(Book.objects.filter(pubdate__year=2006).order_by('name'))
  184. )
  185. self.assertEqual(
  186. list(res.context['object_list']),
  187. list(Book.objects.filter(pubdate__year=2006).order_by('name'))
  188. )
  189. self.assertTemplateUsed(res, 'generic_views/book_archive_year.html')
  190. def test_year_view_two_custom_sort_orders(self):
  191. Book.objects.create(name="Zebras for Dummies", pages=300, pubdate=datetime.date(2006, 9, 1))
  192. Book.objects.create(name="Hunting Hippos", pages=400, pubdate=datetime.date(2006, 3, 1))
  193. res = self.client.get('/dates/books/2006/sortedbypageandnamedec/')
  194. self.assertEqual(res.status_code, 200)
  195. self.assertEqual(
  196. list(res.context['date_list']),
  197. [datetime.date(2006, 3, 1), datetime.date(2006, 5, 1), datetime.date(2006, 9, 1)]
  198. )
  199. self.assertEqual(
  200. list(res.context['book_list']),
  201. list(Book.objects.filter(pubdate__year=2006).order_by('pages', '-name'))
  202. )
  203. self.assertEqual(
  204. list(res.context['object_list']),
  205. list(Book.objects.filter(pubdate__year=2006).order_by('pages', '-name'))
  206. )
  207. self.assertTemplateUsed(res, 'generic_views/book_archive_year.html')
  208. def test_year_view_invalid_pattern(self):
  209. res = self.client.get('/dates/books/no_year/')
  210. self.assertEqual(res.status_code, 404)
  211. def test_no_duplicate_query(self):
  212. # Regression test for #18354
  213. with self.assertNumQueries(4):
  214. self.client.get('/dates/books/2008/reverse/')
  215. def test_datetime_year_view(self):
  216. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0))
  217. res = self.client.get('/dates/booksignings/2008/')
  218. self.assertEqual(res.status_code, 200)
  219. @skipUnlessDBFeature('has_zoneinfo_database')
  220. @override_settings(USE_TZ=True, TIME_ZONE='Africa/Nairobi')
  221. def test_aware_datetime_year_view(self):
  222. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0, tzinfo=timezone.utc))
  223. res = self.client.get('/dates/booksignings/2008/')
  224. self.assertEqual(res.status_code, 200)
  225. def test_date_list_order(self):
  226. """date_list should be sorted ascending in year view"""
  227. _make_books(10, base_date=datetime.date(2011, 12, 25))
  228. res = self.client.get('/dates/books/2011/')
  229. self.assertEqual(list(res.context['date_list']), list(sorted(res.context['date_list'])))
  230. @override_settings(ROOT_URLCONF='generic_views.urls')
  231. class MonthArchiveViewTests(TestDataMixin, TestCase):
  232. def test_month_view(self):
  233. res = self.client.get('/dates/books/2008/oct/')
  234. self.assertEqual(res.status_code, 200)
  235. self.assertTemplateUsed(res, 'generic_views/book_archive_month.html')
  236. self.assertEqual(list(res.context['date_list']), [datetime.date(2008, 10, 1)])
  237. self.assertEqual(list(res.context['book_list']),
  238. list(Book.objects.filter(pubdate=datetime.date(2008, 10, 1))))
  239. self.assertEqual(res.context['month'], datetime.date(2008, 10, 1))
  240. # Since allow_empty=False, next/prev months must be valid (#7164)
  241. self.assertIsNone(res.context['next_month'])
  242. self.assertEqual(res.context['previous_month'], datetime.date(2006, 5, 1))
  243. def test_month_view_allow_empty(self):
  244. # allow_empty = False, empty month
  245. res = self.client.get('/dates/books/2000/jan/')
  246. self.assertEqual(res.status_code, 404)
  247. # allow_empty = True, empty month
  248. res = self.client.get('/dates/books/2000/jan/allow_empty/')
  249. self.assertEqual(res.status_code, 200)
  250. self.assertEqual(list(res.context['date_list']), [])
  251. self.assertEqual(list(res.context['book_list']), [])
  252. self.assertEqual(res.context['month'], datetime.date(2000, 1, 1))
  253. # Since allow_empty=True, next/prev are allowed to be empty months (#7164)
  254. self.assertEqual(res.context['next_month'], datetime.date(2000, 2, 1))
  255. self.assertEqual(res.context['previous_month'], datetime.date(1999, 12, 1))
  256. # allow_empty but not allow_future: next_month should be empty (#7164)
  257. url = datetime.date.today().strftime('/dates/books/%Y/%b/allow_empty/').lower()
  258. res = self.client.get(url)
  259. self.assertEqual(res.status_code, 200)
  260. self.assertIsNone(res.context['next_month'])
  261. def test_month_view_allow_future(self):
  262. future = (datetime.date.today() + datetime.timedelta(days=60)).replace(day=1)
  263. urlbit = future.strftime('%Y/%b').lower()
  264. b = Book.objects.create(name="The New New Testement", pages=600, pubdate=future)
  265. # allow_future = False, future month
  266. res = self.client.get('/dates/books/%s/' % urlbit)
  267. self.assertEqual(res.status_code, 404)
  268. # allow_future = True, valid future month
  269. res = self.client.get('/dates/books/%s/allow_future/' % urlbit)
  270. self.assertEqual(res.status_code, 200)
  271. self.assertEqual(res.context['date_list'][0], b.pubdate)
  272. self.assertEqual(list(res.context['book_list']), [b])
  273. self.assertEqual(res.context['month'], future)
  274. # Since allow_future = True but not allow_empty, next/prev are not
  275. # allowed to be empty months (#7164)
  276. self.assertIsNone(res.context['next_month'])
  277. self.assertEqual(res.context['previous_month'], datetime.date(2008, 10, 1))
  278. # allow_future, but not allow_empty, with a current month. So next
  279. # should be in the future (yup, #7164, again)
  280. res = self.client.get('/dates/books/2008/oct/allow_future/')
  281. self.assertEqual(res.status_code, 200)
  282. self.assertEqual(res.context['next_month'], future)
  283. self.assertEqual(res.context['previous_month'], datetime.date(2006, 5, 1))
  284. def test_month_view_paginated(self):
  285. res = self.client.get('/dates/books/2008/oct/paginated/')
  286. self.assertEqual(res.status_code, 200)
  287. self.assertEqual(
  288. list(res.context['book_list']),
  289. list(Book.objects.filter(pubdate__year=2008, pubdate__month=10))
  290. )
  291. self.assertEqual(
  292. list(res.context['object_list']),
  293. list(Book.objects.filter(pubdate__year=2008, pubdate__month=10))
  294. )
  295. self.assertTemplateUsed(res, 'generic_views/book_archive_month.html')
  296. def test_custom_month_format(self):
  297. res = self.client.get('/dates/books/2008/10/')
  298. self.assertEqual(res.status_code, 200)
  299. def test_month_view_invalid_pattern(self):
  300. res = self.client.get('/dates/books/2007/no_month/')
  301. self.assertEqual(res.status_code, 404)
  302. def test_previous_month_without_content(self):
  303. "Content can exist on any day of the previous month. Refs #14711"
  304. self.pubdate_list = [
  305. datetime.date(2010, month, day)
  306. for month, day in ((9, 1), (10, 2), (11, 3))
  307. ]
  308. for pubdate in self.pubdate_list:
  309. name = str(pubdate)
  310. Book.objects.create(name=name, slug=name, pages=100, pubdate=pubdate)
  311. res = self.client.get('/dates/books/2010/nov/allow_empty/')
  312. self.assertEqual(res.status_code, 200)
  313. self.assertEqual(res.context['previous_month'], datetime.date(2010, 10, 1))
  314. # The following test demonstrates the bug
  315. res = self.client.get('/dates/books/2010/nov/')
  316. self.assertEqual(res.status_code, 200)
  317. self.assertEqual(res.context['previous_month'], datetime.date(2010, 10, 1))
  318. # The bug does not occur here because a Book with pubdate of Sep 1 exists
  319. res = self.client.get('/dates/books/2010/oct/')
  320. self.assertEqual(res.status_code, 200)
  321. self.assertEqual(res.context['previous_month'], datetime.date(2010, 9, 1))
  322. def test_datetime_month_view(self):
  323. BookSigning.objects.create(event_date=datetime.datetime(2008, 2, 1, 12, 0))
  324. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0))
  325. BookSigning.objects.create(event_date=datetime.datetime(2008, 6, 3, 12, 0))
  326. res = self.client.get('/dates/booksignings/2008/apr/')
  327. self.assertEqual(res.status_code, 200)
  328. @skipUnlessDBFeature('has_zoneinfo_database')
  329. @override_settings(USE_TZ=True, TIME_ZONE='Africa/Nairobi')
  330. def test_aware_datetime_month_view(self):
  331. BookSigning.objects.create(event_date=datetime.datetime(2008, 2, 1, 12, 0, tzinfo=timezone.utc))
  332. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0, tzinfo=timezone.utc))
  333. BookSigning.objects.create(event_date=datetime.datetime(2008, 6, 3, 12, 0, tzinfo=timezone.utc))
  334. res = self.client.get('/dates/booksignings/2008/apr/')
  335. self.assertEqual(res.status_code, 200)
  336. def test_date_list_order(self):
  337. """date_list should be sorted ascending in month view"""
  338. _make_books(10, base_date=datetime.date(2011, 12, 25))
  339. res = self.client.get('/dates/books/2011/dec/')
  340. self.assertEqual(list(res.context['date_list']), list(sorted(res.context['date_list'])))
  341. @override_settings(ROOT_URLCONF='generic_views.urls')
  342. class WeekArchiveViewTests(TestDataMixin, TestCase):
  343. def test_week_view(self):
  344. res = self.client.get('/dates/books/2008/week/39/')
  345. self.assertEqual(res.status_code, 200)
  346. self.assertTemplateUsed(res, 'generic_views/book_archive_week.html')
  347. self.assertEqual(res.context['book_list'][0], Book.objects.get(pubdate=datetime.date(2008, 10, 1)))
  348. self.assertEqual(res.context['week'], datetime.date(2008, 9, 28))
  349. # Since allow_empty=False, next/prev weeks must be valid
  350. self.assertIsNone(res.context['next_week'])
  351. self.assertEqual(res.context['previous_week'], datetime.date(2006, 4, 30))
  352. def test_week_view_allow_empty(self):
  353. # allow_empty = False, empty week
  354. res = self.client.get('/dates/books/2008/week/12/')
  355. self.assertEqual(res.status_code, 404)
  356. # allow_empty = True, empty month
  357. res = self.client.get('/dates/books/2008/week/12/allow_empty/')
  358. self.assertEqual(res.status_code, 200)
  359. self.assertEqual(list(res.context['book_list']), [])
  360. self.assertEqual(res.context['week'], datetime.date(2008, 3, 23))
  361. # Since allow_empty=True, next/prev are allowed to be empty weeks
  362. self.assertEqual(res.context['next_week'], datetime.date(2008, 3, 30))
  363. self.assertEqual(res.context['previous_week'], datetime.date(2008, 3, 16))
  364. # allow_empty but not allow_future: next_week should be empty
  365. url = datetime.date.today().strftime('/dates/books/%Y/week/%U/allow_empty/').lower()
  366. res = self.client.get(url)
  367. self.assertEqual(res.status_code, 200)
  368. self.assertIsNone(res.context['next_week'])
  369. def test_week_view_allow_future(self):
  370. # January 7th always falls in week 1, given Python's definition of week numbers
  371. future = datetime.date(datetime.date.today().year + 1, 1, 7)
  372. future_sunday = future - datetime.timedelta(days=(future.weekday() + 1) % 7)
  373. b = Book.objects.create(name="The New New Testement", pages=600, pubdate=future)
  374. res = self.client.get('/dates/books/%s/week/1/' % future.year)
  375. self.assertEqual(res.status_code, 404)
  376. res = self.client.get('/dates/books/%s/week/1/allow_future/' % future.year)
  377. self.assertEqual(res.status_code, 200)
  378. self.assertEqual(list(res.context['book_list']), [b])
  379. self.assertEqual(res.context['week'], future_sunday)
  380. # Since allow_future = True but not allow_empty, next/prev are not
  381. # allowed to be empty weeks
  382. self.assertIsNone(res.context['next_week'])
  383. self.assertEqual(res.context['previous_week'], datetime.date(2008, 9, 28))
  384. # allow_future, but not allow_empty, with a current week. So next
  385. # should be in the future
  386. res = self.client.get('/dates/books/2008/week/39/allow_future/')
  387. self.assertEqual(res.status_code, 200)
  388. self.assertEqual(res.context['next_week'], future_sunday)
  389. self.assertEqual(res.context['previous_week'], datetime.date(2006, 4, 30))
  390. def test_week_view_paginated(self):
  391. week_start = datetime.date(2008, 9, 28)
  392. week_end = week_start + datetime.timedelta(days=7)
  393. res = self.client.get('/dates/books/2008/week/39/')
  394. self.assertEqual(res.status_code, 200)
  395. self.assertEqual(
  396. list(res.context['book_list']),
  397. list(Book.objects.filter(pubdate__gte=week_start, pubdate__lt=week_end))
  398. )
  399. self.assertEqual(
  400. list(res.context['object_list']),
  401. list(Book.objects.filter(pubdate__gte=week_start, pubdate__lt=week_end))
  402. )
  403. self.assertTemplateUsed(res, 'generic_views/book_archive_week.html')
  404. def test_week_view_invalid_pattern(self):
  405. res = self.client.get('/dates/books/2007/week/no_week/')
  406. self.assertEqual(res.status_code, 404)
  407. def test_week_start_Monday(self):
  408. # Regression for #14752
  409. res = self.client.get('/dates/books/2008/week/39/')
  410. self.assertEqual(res.status_code, 200)
  411. self.assertEqual(res.context['week'], datetime.date(2008, 9, 28))
  412. res = self.client.get('/dates/books/2008/week/39/monday/')
  413. self.assertEqual(res.status_code, 200)
  414. self.assertEqual(res.context['week'], datetime.date(2008, 9, 29))
  415. def test_datetime_week_view(self):
  416. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0))
  417. res = self.client.get('/dates/booksignings/2008/week/13/')
  418. self.assertEqual(res.status_code, 200)
  419. @override_settings(USE_TZ=True, TIME_ZONE='Africa/Nairobi')
  420. def test_aware_datetime_week_view(self):
  421. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0, tzinfo=timezone.utc))
  422. res = self.client.get('/dates/booksignings/2008/week/13/')
  423. self.assertEqual(res.status_code, 200)
  424. @override_settings(ROOT_URLCONF='generic_views.urls')
  425. class DayArchiveViewTests(TestDataMixin, TestCase):
  426. def test_day_view(self):
  427. res = self.client.get('/dates/books/2008/oct/01/')
  428. self.assertEqual(res.status_code, 200)
  429. self.assertTemplateUsed(res, 'generic_views/book_archive_day.html')
  430. self.assertEqual(list(res.context['book_list']),
  431. list(Book.objects.filter(pubdate=datetime.date(2008, 10, 1))))
  432. self.assertEqual(res.context['day'], datetime.date(2008, 10, 1))
  433. # Since allow_empty=False, next/prev days must be valid.
  434. self.assertIsNone(res.context['next_day'])
  435. self.assertEqual(res.context['previous_day'], datetime.date(2006, 5, 1))
  436. def test_day_view_allow_empty(self):
  437. # allow_empty = False, empty month
  438. res = self.client.get('/dates/books/2000/jan/1/')
  439. self.assertEqual(res.status_code, 404)
  440. # allow_empty = True, empty month
  441. res = self.client.get('/dates/books/2000/jan/1/allow_empty/')
  442. self.assertEqual(res.status_code, 200)
  443. self.assertEqual(list(res.context['book_list']), [])
  444. self.assertEqual(res.context['day'], datetime.date(2000, 1, 1))
  445. # Since it's allow empty, next/prev are allowed to be empty months (#7164)
  446. self.assertEqual(res.context['next_day'], datetime.date(2000, 1, 2))
  447. self.assertEqual(res.context['previous_day'], datetime.date(1999, 12, 31))
  448. # allow_empty but not allow_future: next_month should be empty (#7164)
  449. url = datetime.date.today().strftime('/dates/books/%Y/%b/%d/allow_empty/').lower()
  450. res = self.client.get(url)
  451. self.assertEqual(res.status_code, 200)
  452. self.assertIsNone(res.context['next_day'])
  453. def test_day_view_allow_future(self):
  454. future = (datetime.date.today() + datetime.timedelta(days=60))
  455. urlbit = future.strftime('%Y/%b/%d').lower()
  456. b = Book.objects.create(name="The New New Testement", pages=600, pubdate=future)
  457. # allow_future = False, future month
  458. res = self.client.get('/dates/books/%s/' % urlbit)
  459. self.assertEqual(res.status_code, 404)
  460. # allow_future = True, valid future month
  461. res = self.client.get('/dates/books/%s/allow_future/' % urlbit)
  462. self.assertEqual(res.status_code, 200)
  463. self.assertEqual(list(res.context['book_list']), [b])
  464. self.assertEqual(res.context['day'], future)
  465. # allow_future but not allow_empty, next/prev must be valid
  466. self.assertIsNone(res.context['next_day'])
  467. self.assertEqual(res.context['previous_day'], datetime.date(2008, 10, 1))
  468. # allow_future, but not allow_empty, with a current month.
  469. res = self.client.get('/dates/books/2008/oct/01/allow_future/')
  470. self.assertEqual(res.status_code, 200)
  471. self.assertEqual(res.context['next_day'], future)
  472. self.assertEqual(res.context['previous_day'], datetime.date(2006, 5, 1))
  473. # allow_future for yesterday, next_day is today (#17192)
  474. today = datetime.date.today()
  475. yesterday = today - datetime.timedelta(days=1)
  476. res = self.client.get('/dates/books/%s/allow_empty_and_future/'
  477. % yesterday.strftime('%Y/%b/%d').lower())
  478. self.assertEqual(res.context['next_day'], today)
  479. def test_day_view_paginated(self):
  480. res = self.client.get('/dates/books/2008/oct/1/')
  481. self.assertEqual(res.status_code, 200)
  482. self.assertEqual(
  483. list(res.context['book_list']),
  484. list(Book.objects.filter(pubdate__year=2008, pubdate__month=10, pubdate__day=1))
  485. )
  486. self.assertEqual(
  487. list(res.context['object_list']),
  488. list(Book.objects.filter(pubdate__year=2008, pubdate__month=10, pubdate__day=1))
  489. )
  490. self.assertTemplateUsed(res, 'generic_views/book_archive_day.html')
  491. def test_next_prev_context(self):
  492. res = self.client.get('/dates/books/2008/oct/01/')
  493. self.assertEqual(res.content, b"Archive for Oct. 1, 2008. Previous day is May 1, 2006\n")
  494. def test_custom_month_format(self):
  495. res = self.client.get('/dates/books/2008/10/01/')
  496. self.assertEqual(res.status_code, 200)
  497. def test_day_view_invalid_pattern(self):
  498. res = self.client.get('/dates/books/2007/oct/no_day/')
  499. self.assertEqual(res.status_code, 404)
  500. def test_today_view(self):
  501. res = self.client.get('/dates/books/today/')
  502. self.assertEqual(res.status_code, 404)
  503. res = self.client.get('/dates/books/today/allow_empty/')
  504. self.assertEqual(res.status_code, 200)
  505. self.assertEqual(res.context['day'], datetime.date.today())
  506. def test_datetime_day_view(self):
  507. BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0))
  508. res = self.client.get('/dates/booksignings/2008/apr/2/')
  509. self.assertEqual(res.status_code, 200)
  510. @requires_tz_support
  511. @override_settings(USE_TZ=True, TIME_ZONE='Africa/Nairobi')
  512. def test_aware_datetime_day_view(self):
  513. bs = BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0, tzinfo=timezone.utc))
  514. res = self.client.get('/dates/booksignings/2008/apr/2/')
  515. self.assertEqual(res.status_code, 200)
  516. # 2008-04-02T00:00:00+03:00 (beginning of day) > 2008-04-01T22:00:00+00:00 (book signing event date)
  517. bs.event_date = datetime.datetime(2008, 4, 1, 22, 0, tzinfo=timezone.utc)
  518. bs.save()
  519. res = self.client.get('/dates/booksignings/2008/apr/2/')
  520. self.assertEqual(res.status_code, 200)
  521. # 2008-04-03T00:00:00+03:00 (end of day) > 2008-04-02T22:00:00+00:00 (book signing event date)
  522. bs.event_date = datetime.datetime(2008, 4, 2, 22, 0, tzinfo=timezone.utc)
  523. bs.save()
  524. res = self.client.get('/dates/booksignings/2008/apr/2/')
  525. self.assertEqual(res.status_code, 404)
  526. @override_settings(ROOT_URLCONF='generic_views.urls')
  527. class DateDetailViewTests(TestDataMixin, TestCase):
  528. def test_date_detail_by_pk(self):
  529. res = self.client.get('/dates/books/2008/oct/01/%s/' % self.book1.pk)
  530. self.assertEqual(res.status_code, 200)
  531. self.assertEqual(res.context['object'], self.book1)
  532. self.assertEqual(res.context['book'], self.book1)
  533. self.assertTemplateUsed(res, 'generic_views/book_detail.html')
  534. def test_date_detail_by_slug(self):
  535. res = self.client.get('/dates/books/2006/may/01/byslug/dreaming-in-code/')
  536. self.assertEqual(res.status_code, 200)
  537. self.assertEqual(res.context['book'], Book.objects.get(slug='dreaming-in-code'))
  538. def test_date_detail_custom_month_format(self):
  539. res = self.client.get('/dates/books/2008/10/01/%s/' % self.book1.pk)
  540. self.assertEqual(res.status_code, 200)
  541. self.assertEqual(res.context['book'], self.book1)
  542. def test_date_detail_allow_future(self):
  543. future = (datetime.date.today() + datetime.timedelta(days=60))
  544. urlbit = future.strftime('%Y/%b/%d').lower()
  545. b = Book.objects.create(name="The New New Testement", slug="new-new", pages=600, pubdate=future)
  546. res = self.client.get('/dates/books/%s/new-new/' % urlbit)
  547. self.assertEqual(res.status_code, 404)
  548. res = self.client.get('/dates/books/%s/%s/allow_future/' % (urlbit, b.id))
  549. self.assertEqual(res.status_code, 200)
  550. self.assertEqual(res.context['book'], b)
  551. self.assertTemplateUsed(res, 'generic_views/book_detail.html')
  552. def test_invalid_url(self):
  553. with self.assertRaises(AttributeError):
  554. self.client.get("/dates/books/2008/oct/01/nopk/")
  555. def test_get_object_custom_queryset(self):
  556. """
  557. Ensure that custom querysets are used when provided to
  558. BaseDateDetailView.get_object()
  559. Refs #16918.
  560. """
  561. res = self.client.get(
  562. '/dates/books/get_object_custom_queryset/2006/may/01/%s/' % self.book2.pk)
  563. self.assertEqual(res.status_code, 200)
  564. self.assertEqual(res.context['object'], self.book2)
  565. self.assertEqual(res.context['book'], self.book2)
  566. self.assertTemplateUsed(res, 'generic_views/book_detail.html')
  567. res = self.client.get(
  568. '/dates/books/get_object_custom_queryset/2008/oct/01/9999999/')
  569. self.assertEqual(res.status_code, 404)
  570. def test_get_object_custom_queryset_numqueries(self):
  571. with self.assertNumQueries(1):
  572. self.client.get('/dates/books/get_object_custom_queryset/2006/may/01/2/')
  573. def test_datetime_date_detail(self):
  574. bs = BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0))
  575. res = self.client.get('/dates/booksignings/2008/apr/2/%d/' % bs.pk)
  576. self.assertEqual(res.status_code, 200)
  577. @requires_tz_support
  578. @override_settings(USE_TZ=True, TIME_ZONE='Africa/Nairobi')
  579. def test_aware_datetime_date_detail(self):
  580. bs = BookSigning.objects.create(event_date=datetime.datetime(2008, 4, 2, 12, 0, tzinfo=timezone.utc))
  581. res = self.client.get('/dates/booksignings/2008/apr/2/%d/' % bs.pk)
  582. self.assertEqual(res.status_code, 200)
  583. # 2008-04-02T00:00:00+03:00 (beginning of day) > 2008-04-01T22:00:00+00:00 (book signing event date)
  584. bs.event_date = datetime.datetime(2008, 4, 1, 22, 0, tzinfo=timezone.utc)
  585. bs.save()
  586. res = self.client.get('/dates/booksignings/2008/apr/2/%d/' % bs.pk)
  587. self.assertEqual(res.status_code, 200)
  588. # 2008-04-03T00:00:00+03:00 (end of day) > 2008-04-02T22:00:00+00:00 (book signing event date)
  589. bs.event_date = datetime.datetime(2008, 4, 2, 22, 0, tzinfo=timezone.utc)
  590. bs.save()
  591. res = self.client.get('/dates/booksignings/2008/apr/2/%d/' % bs.pk)
  592. self.assertEqual(res.status_code, 404)