tests.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845
  1. import datetime
  2. from xml.dom import minidom
  3. from django.contrib.sites.models import Site
  4. from django.contrib.syndication import views
  5. from django.core.exceptions import ImproperlyConfigured
  6. from django.templatetags.static import static
  7. from django.test import TestCase, override_settings
  8. from django.test.utils import requires_tz_support
  9. from django.urls import reverse, reverse_lazy
  10. from django.utils import timezone
  11. from django.utils.feedgenerator import (
  12. Atom1Feed,
  13. Rss201rev2Feed,
  14. Stylesheet,
  15. SyndicationFeed,
  16. rfc2822_date,
  17. rfc3339_date,
  18. )
  19. from .models import Article, Entry
  20. TZ = timezone.get_default_timezone()
  21. class FeedTestCase(TestCase):
  22. @classmethod
  23. def setUpTestData(cls):
  24. cls.e1 = Entry.objects.create(
  25. title="My first entry",
  26. updated=datetime.datetime(1980, 1, 1, 12, 30),
  27. published=datetime.datetime(1986, 9, 25, 20, 15, 00),
  28. )
  29. cls.e2 = Entry.objects.create(
  30. title="My second entry",
  31. updated=datetime.datetime(2008, 1, 2, 12, 30),
  32. published=datetime.datetime(2006, 3, 17, 18, 0),
  33. )
  34. cls.e3 = Entry.objects.create(
  35. title="My third entry",
  36. updated=datetime.datetime(2008, 1, 2, 13, 30),
  37. published=datetime.datetime(2005, 6, 14, 10, 45),
  38. )
  39. cls.e4 = Entry.objects.create(
  40. title="A & B < C > D",
  41. updated=datetime.datetime(2008, 1, 3, 13, 30),
  42. published=datetime.datetime(2005, 11, 25, 12, 11, 23),
  43. )
  44. cls.e5 = Entry.objects.create(
  45. title="My last entry",
  46. updated=datetime.datetime(2013, 1, 20, 0, 0),
  47. published=datetime.datetime(2013, 3, 25, 20, 0),
  48. )
  49. cls.a1 = Article.objects.create(
  50. title="My first article",
  51. entry=cls.e1,
  52. updated=datetime.datetime(1986, 11, 21, 9, 12, 18),
  53. published=datetime.datetime(1986, 10, 21, 9, 12, 18),
  54. )
  55. def assertChildNodes(self, elem, expected):
  56. actual = {n.nodeName for n in elem.childNodes}
  57. expected = set(expected)
  58. self.assertEqual(actual, expected)
  59. def assertChildNodeContent(self, elem, expected):
  60. for k, v in expected.items():
  61. self.assertEqual(elem.getElementsByTagName(k)[0].firstChild.wholeText, v)
  62. def assertCategories(self, elem, expected):
  63. self.assertEqual(
  64. {
  65. i.firstChild.wholeText
  66. for i in elem.childNodes
  67. if i.nodeName == "category"
  68. },
  69. set(expected),
  70. )
  71. @override_settings(ROOT_URLCONF="syndication_tests.urls")
  72. class SyndicationFeedTest(FeedTestCase):
  73. """
  74. Tests for the high-level syndication feed framework.
  75. """
  76. @classmethod
  77. def setUpClass(cls):
  78. super().setUpClass()
  79. # This cleanup is necessary because contrib.sites cache
  80. # makes tests interfere with each other, see #11505
  81. Site.objects.clear_cache()
  82. def test_rss2_feed(self):
  83. """
  84. Test the structure and content of feeds generated by Rss201rev2Feed.
  85. """
  86. response = self.client.get("/syndication/rss2/")
  87. doc = minidom.parseString(response.content)
  88. # Making sure there's only 1 `rss` element and that the correct
  89. # RSS version was specified.
  90. feed_elem = doc.getElementsByTagName("rss")
  91. self.assertEqual(len(feed_elem), 1)
  92. feed = feed_elem[0]
  93. self.assertEqual(feed.getAttribute("version"), "2.0")
  94. self.assertEqual(
  95. feed.getElementsByTagName("language")[0].firstChild.nodeValue, "en"
  96. )
  97. # Making sure there's only one `channel` element w/in the
  98. # `rss` element.
  99. chan_elem = feed.getElementsByTagName("channel")
  100. self.assertEqual(len(chan_elem), 1)
  101. chan = chan_elem[0]
  102. # Find the last build date
  103. d = Entry.objects.latest("published").published
  104. last_build_date = rfc2822_date(timezone.make_aware(d, TZ))
  105. self.assertChildNodes(
  106. chan,
  107. [
  108. "title",
  109. "link",
  110. "description",
  111. "language",
  112. "lastBuildDate",
  113. "item",
  114. "atom:link",
  115. "ttl",
  116. "copyright",
  117. "category",
  118. ],
  119. )
  120. self.assertChildNodeContent(
  121. chan,
  122. {
  123. "title": "My blog",
  124. "description": "A more thorough description of my blog.",
  125. "link": "http://example.com/blog/",
  126. "language": "en",
  127. "lastBuildDate": last_build_date,
  128. "ttl": "600",
  129. "copyright": "Copyright (c) 2007, Sally Smith",
  130. },
  131. )
  132. self.assertCategories(chan, ["python", "django"])
  133. # Ensure the content of the channel is correct
  134. self.assertChildNodeContent(
  135. chan,
  136. {
  137. "title": "My blog",
  138. "link": "http://example.com/blog/",
  139. },
  140. )
  141. # Check feed_url is passed
  142. self.assertEqual(
  143. chan.getElementsByTagName("atom:link")[0].getAttribute("href"),
  144. "http://example.com/syndication/rss2/",
  145. )
  146. # Find the pubdate of the first feed item
  147. d = Entry.objects.get(pk=self.e1.pk).published
  148. pub_date = rfc2822_date(timezone.make_aware(d, TZ))
  149. items = chan.getElementsByTagName("item")
  150. self.assertEqual(len(items), Entry.objects.count())
  151. self.assertChildNodeContent(
  152. items[0],
  153. {
  154. "title": "My first entry",
  155. "description": "Overridden description: My first entry",
  156. "link": "http://example.com/blog/%s/" % self.e1.pk,
  157. "guid": "http://example.com/blog/%s/" % self.e1.pk,
  158. "pubDate": pub_date,
  159. "author": "test@example.com (Sally Smith)",
  160. "comments": "/blog/%s/comments" % self.e1.pk,
  161. },
  162. )
  163. self.assertCategories(items[0], ["python", "testing"])
  164. for item in items:
  165. self.assertChildNodes(
  166. item,
  167. [
  168. "title",
  169. "link",
  170. "description",
  171. "guid",
  172. "category",
  173. "pubDate",
  174. "author",
  175. "comments",
  176. ],
  177. )
  178. # Assert that <guid> does not have any 'isPermaLink' attribute
  179. self.assertIsNone(
  180. item.getElementsByTagName("guid")[0].attributes.get("isPermaLink")
  181. )
  182. def test_rss2_feed_with_callable_object(self):
  183. response = self.client.get("/syndication/rss2/with-callable-object/")
  184. doc = minidom.parseString(response.content)
  185. chan = doc.getElementsByTagName("rss")[0].getElementsByTagName("channel")[0]
  186. self.assertChildNodeContent(chan, {"ttl": "700"})
  187. def test_rss2_feed_with_decorated_methods(self):
  188. response = self.client.get("/syndication/rss2/with-decorated-methods/")
  189. doc = minidom.parseString(response.content)
  190. chan = doc.getElementsByTagName("rss")[0].getElementsByTagName("channel")[0]
  191. self.assertCategories(chan, ["javascript", "vue"])
  192. self.assertChildNodeContent(
  193. chan,
  194. {
  195. "title": "Overridden title -- decorated by @wraps.",
  196. "description": "Overridden description -- decorated by @wraps.",
  197. "ttl": "800 -- decorated by @wraps.",
  198. "copyright": "Copyright (c) 2022, John Doe -- decorated by @wraps.",
  199. },
  200. )
  201. items = chan.getElementsByTagName("item")
  202. self.assertChildNodeContent(
  203. items[0],
  204. {
  205. "title": (
  206. f"Overridden item title: {self.e1.title} -- decorated by @wraps."
  207. ),
  208. "description": "Overridden item description -- decorated by @wraps.",
  209. },
  210. )
  211. def test_rss2_feed_with_wrong_decorated_methods(self):
  212. msg = (
  213. "Feed method 'item_description' decorated by 'wrapper' needs to use "
  214. "@functools.wraps."
  215. )
  216. with self.assertRaisesMessage(ImproperlyConfigured, msg):
  217. self.client.get("/syndication/rss2/with-wrong-decorated-methods/")
  218. def test_rss2_feed_guid_permalink_false(self):
  219. """
  220. Test if the 'isPermaLink' attribute of <guid> element of an item
  221. in the RSS feed is 'false'.
  222. """
  223. response = self.client.get("/syndication/rss2/guid_ispermalink_false/")
  224. doc = minidom.parseString(response.content)
  225. chan = doc.getElementsByTagName("rss")[0].getElementsByTagName("channel")[0]
  226. items = chan.getElementsByTagName("item")
  227. for item in items:
  228. self.assertEqual(
  229. item.getElementsByTagName("guid")[0]
  230. .attributes.get("isPermaLink")
  231. .value,
  232. "false",
  233. )
  234. def test_rss2_feed_guid_permalink_true(self):
  235. """
  236. Test if the 'isPermaLink' attribute of <guid> element of an item
  237. in the RSS feed is 'true'.
  238. """
  239. response = self.client.get("/syndication/rss2/guid_ispermalink_true/")
  240. doc = minidom.parseString(response.content)
  241. chan = doc.getElementsByTagName("rss")[0].getElementsByTagName("channel")[0]
  242. items = chan.getElementsByTagName("item")
  243. for item in items:
  244. self.assertEqual(
  245. item.getElementsByTagName("guid")[0]
  246. .attributes.get("isPermaLink")
  247. .value,
  248. "true",
  249. )
  250. def test_rss2_single_enclosure(self):
  251. response = self.client.get("/syndication/rss2/single-enclosure/")
  252. doc = minidom.parseString(response.content)
  253. chan = doc.getElementsByTagName("rss")[0].getElementsByTagName("channel")[0]
  254. items = chan.getElementsByTagName("item")
  255. for item in items:
  256. enclosures = item.getElementsByTagName("enclosure")
  257. self.assertEqual(len(enclosures), 1)
  258. def test_rss2_multiple_enclosures(self):
  259. with self.assertRaisesMessage(
  260. ValueError,
  261. "RSS feed items may only have one enclosure, see "
  262. "http://www.rssboard.org/rss-profile#element-channel-item-enclosure",
  263. ):
  264. self.client.get("/syndication/rss2/multiple-enclosure/")
  265. def test_rss091_feed(self):
  266. """
  267. Test the structure and content of feeds generated by RssUserland091Feed.
  268. """
  269. response = self.client.get("/syndication/rss091/")
  270. doc = minidom.parseString(response.content)
  271. # Making sure there's only 1 `rss` element and that the correct
  272. # RSS version was specified.
  273. feed_elem = doc.getElementsByTagName("rss")
  274. self.assertEqual(len(feed_elem), 1)
  275. feed = feed_elem[0]
  276. self.assertEqual(feed.getAttribute("version"), "0.91")
  277. # Making sure there's only one `channel` element w/in the
  278. # `rss` element.
  279. chan_elem = feed.getElementsByTagName("channel")
  280. self.assertEqual(len(chan_elem), 1)
  281. chan = chan_elem[0]
  282. self.assertChildNodes(
  283. chan,
  284. [
  285. "title",
  286. "link",
  287. "description",
  288. "language",
  289. "lastBuildDate",
  290. "item",
  291. "atom:link",
  292. "ttl",
  293. "copyright",
  294. "category",
  295. ],
  296. )
  297. # Ensure the content of the channel is correct
  298. self.assertChildNodeContent(
  299. chan,
  300. {
  301. "title": "My blog",
  302. "link": "http://example.com/blog/",
  303. },
  304. )
  305. self.assertCategories(chan, ["python", "django"])
  306. # Check feed_url is passed
  307. self.assertEqual(
  308. chan.getElementsByTagName("atom:link")[0].getAttribute("href"),
  309. "http://example.com/syndication/rss091/",
  310. )
  311. items = chan.getElementsByTagName("item")
  312. self.assertEqual(len(items), Entry.objects.count())
  313. self.assertChildNodeContent(
  314. items[0],
  315. {
  316. "title": "My first entry",
  317. "description": "Overridden description: My first entry",
  318. "link": "http://example.com/blog/%s/" % self.e1.pk,
  319. },
  320. )
  321. for item in items:
  322. self.assertChildNodes(item, ["title", "link", "description"])
  323. self.assertCategories(item, [])
  324. def test_atom_feed(self):
  325. """
  326. Test the structure and content of feeds generated by Atom1Feed.
  327. """
  328. response = self.client.get("/syndication/atom/")
  329. feed = minidom.parseString(response.content).firstChild
  330. self.assertEqual(feed.nodeName, "feed")
  331. self.assertEqual(feed.getAttribute("xmlns"), "http://www.w3.org/2005/Atom")
  332. self.assertChildNodes(
  333. feed,
  334. [
  335. "title",
  336. "subtitle",
  337. "link",
  338. "id",
  339. "updated",
  340. "entry",
  341. "rights",
  342. "category",
  343. "author",
  344. ],
  345. )
  346. for link in feed.getElementsByTagName("link"):
  347. if link.getAttribute("rel") == "self":
  348. self.assertEqual(
  349. link.getAttribute("href"), "http://example.com/syndication/atom/"
  350. )
  351. entries = feed.getElementsByTagName("entry")
  352. self.assertEqual(len(entries), Entry.objects.count())
  353. for entry in entries:
  354. self.assertChildNodes(
  355. entry,
  356. [
  357. "title",
  358. "link",
  359. "id",
  360. "summary",
  361. "category",
  362. "updated",
  363. "published",
  364. "rights",
  365. "author",
  366. ],
  367. )
  368. summary = entry.getElementsByTagName("summary")[0]
  369. self.assertEqual(summary.getAttribute("type"), "html")
  370. def test_atom_feed_published_and_updated_elements(self):
  371. """
  372. The published and updated elements are not
  373. the same and now adhere to RFC 4287.
  374. """
  375. response = self.client.get("/syndication/atom/")
  376. feed = minidom.parseString(response.content).firstChild
  377. entries = feed.getElementsByTagName("entry")
  378. published = entries[0].getElementsByTagName("published")[0].firstChild.wholeText
  379. updated = entries[0].getElementsByTagName("updated")[0].firstChild.wholeText
  380. self.assertNotEqual(published, updated)
  381. def test_atom_single_enclosure(self):
  382. response = self.client.get("/syndication/atom/single-enclosure/")
  383. feed = minidom.parseString(response.content).firstChild
  384. items = feed.getElementsByTagName("entry")
  385. for item in items:
  386. links = item.getElementsByTagName("link")
  387. links = [link for link in links if link.getAttribute("rel") == "enclosure"]
  388. self.assertEqual(len(links), 1)
  389. def test_atom_multiple_enclosures(self):
  390. response = self.client.get("/syndication/atom/multiple-enclosure/")
  391. feed = minidom.parseString(response.content).firstChild
  392. items = feed.getElementsByTagName("entry")
  393. for item in items:
  394. links = item.getElementsByTagName("link")
  395. links = [link for link in links if link.getAttribute("rel") == "enclosure"]
  396. self.assertEqual(len(links), 2)
  397. def test_latest_post_date(self):
  398. """
  399. Both the published and updated dates are
  400. considered when determining the latest post date.
  401. """
  402. # this feed has a `published` element with the latest date
  403. response = self.client.get("/syndication/atom/")
  404. feed = minidom.parseString(response.content).firstChild
  405. updated = feed.getElementsByTagName("updated")[0].firstChild.wholeText
  406. d = Entry.objects.latest("published").published
  407. latest_published = rfc3339_date(timezone.make_aware(d, TZ))
  408. self.assertEqual(updated, latest_published)
  409. # this feed has an `updated` element with the latest date
  410. response = self.client.get("/syndication/latest/")
  411. feed = minidom.parseString(response.content).firstChild
  412. updated = feed.getElementsByTagName("updated")[0].firstChild.wholeText
  413. d = Entry.objects.exclude(title="My last entry").latest("updated").updated
  414. latest_updated = rfc3339_date(timezone.make_aware(d, TZ))
  415. self.assertEqual(updated, latest_updated)
  416. def test_custom_feed_generator(self):
  417. response = self.client.get("/syndication/custom/")
  418. feed = minidom.parseString(response.content).firstChild
  419. self.assertEqual(feed.nodeName, "feed")
  420. self.assertEqual(feed.getAttribute("django"), "rocks")
  421. self.assertChildNodes(
  422. feed,
  423. [
  424. "title",
  425. "subtitle",
  426. "link",
  427. "id",
  428. "updated",
  429. "entry",
  430. "spam",
  431. "rights",
  432. "category",
  433. "author",
  434. ],
  435. )
  436. entries = feed.getElementsByTagName("entry")
  437. self.assertEqual(len(entries), Entry.objects.count())
  438. for entry in entries:
  439. self.assertEqual(entry.getAttribute("bacon"), "yum")
  440. self.assertChildNodes(
  441. entry,
  442. [
  443. "title",
  444. "link",
  445. "id",
  446. "summary",
  447. "ministry",
  448. "rights",
  449. "author",
  450. "updated",
  451. "published",
  452. "category",
  453. ],
  454. )
  455. summary = entry.getElementsByTagName("summary")[0]
  456. self.assertEqual(summary.getAttribute("type"), "html")
  457. def test_feed_generator_language_attribute(self):
  458. response = self.client.get("/syndication/language/")
  459. feed = minidom.parseString(response.content).firstChild
  460. self.assertEqual(
  461. feed.firstChild.getElementsByTagName("language")[0].firstChild.nodeValue,
  462. "de",
  463. )
  464. def test_title_escaping(self):
  465. """
  466. Titles are escaped correctly in RSS feeds.
  467. """
  468. response = self.client.get("/syndication/rss2/")
  469. doc = minidom.parseString(response.content)
  470. for item in doc.getElementsByTagName("item"):
  471. link = item.getElementsByTagName("link")[0]
  472. if link.firstChild.wholeText == "http://example.com/blog/4/":
  473. title = item.getElementsByTagName("title")[0]
  474. self.assertEqual(title.firstChild.wholeText, "A &amp; B &lt; C &gt; D")
  475. def test_naive_datetime_conversion(self):
  476. """
  477. Datetimes are correctly converted to the local time zone.
  478. """
  479. # Naive date times passed in get converted to the local time zone, so
  480. # check the received zone offset against the local offset.
  481. response = self.client.get("/syndication/naive-dates/")
  482. doc = minidom.parseString(response.content)
  483. updated = doc.getElementsByTagName("updated")[0].firstChild.wholeText
  484. d = Entry.objects.latest("published").published
  485. latest = rfc3339_date(timezone.make_aware(d, TZ))
  486. self.assertEqual(updated, latest)
  487. def test_aware_datetime_conversion(self):
  488. """
  489. Datetimes with timezones don't get trodden on.
  490. """
  491. response = self.client.get("/syndication/aware-dates/")
  492. doc = minidom.parseString(response.content)
  493. published = doc.getElementsByTagName("published")[0].firstChild.wholeText
  494. self.assertEqual(published[-6:], "+00:42")
  495. def test_feed_no_content_self_closing_tag(self):
  496. tests = [
  497. (Atom1Feed, "link"),
  498. (Rss201rev2Feed, "atom:link"),
  499. ]
  500. for feedgenerator, tag in tests:
  501. with self.subTest(feedgenerator=feedgenerator.__name__):
  502. feed = feedgenerator(
  503. title="title",
  504. link="https://example.com",
  505. description="self closing tags test",
  506. feed_url="https://feed.url.com",
  507. )
  508. doc = feed.writeString("utf-8")
  509. self.assertIn(f'<{tag} href="https://feed.url.com" rel="self"/>', doc)
  510. def test_stylesheets_none(self):
  511. feed = Rss201rev2Feed(
  512. title="test",
  513. link="https://example.com",
  514. description="test",
  515. stylesheets=None,
  516. )
  517. self.assertNotIn("xml-stylesheet", feed.writeString("utf-8"))
  518. def test_stylesheets(self):
  519. testdata = [
  520. # Plain strings.
  521. ("/test.xsl", 'href="/test.xsl" type="text/xsl" media="screen"'),
  522. ("/test.xslt", 'href="/test.xslt" type="text/xsl" media="screen"'),
  523. ("/test.css", 'href="/test.css" type="text/css" media="screen"'),
  524. ("/test", 'href="/test" media="screen"'),
  525. (
  526. "https://example.com/test.xsl",
  527. 'href="https://example.com/test.xsl" type="text/xsl" media="screen"',
  528. ),
  529. (
  530. "https://example.com/test.css",
  531. 'href="https://example.com/test.css" type="text/css" media="screen"',
  532. ),
  533. (
  534. "https://example.com/test",
  535. 'href="https://example.com/test" media="screen"',
  536. ),
  537. ("/♥.xsl", 'href="/%E2%99%A5.xsl" type="text/xsl" media="screen"'),
  538. (
  539. static("stylesheet.xsl"),
  540. 'href="/static/stylesheet.xsl" type="text/xsl" media="screen"',
  541. ),
  542. (
  543. static("stylesheet.css"),
  544. 'href="/static/stylesheet.css" type="text/css" media="screen"',
  545. ),
  546. (static("stylesheet"), 'href="/static/stylesheet" media="screen"'),
  547. (
  548. reverse("syndication-xsl-stylesheet"),
  549. 'href="/syndication/stylesheet.xsl" type="text/xsl" media="screen"',
  550. ),
  551. (
  552. reverse_lazy("syndication-xsl-stylesheet"),
  553. 'href="/syndication/stylesheet.xsl" type="text/xsl" media="screen"',
  554. ),
  555. # Stylesheet objects.
  556. (
  557. Stylesheet("/test.xsl"),
  558. 'href="/test.xsl" type="text/xsl" media="screen"',
  559. ),
  560. (Stylesheet("/test.xsl", mimetype=None), 'href="/test.xsl" media="screen"'),
  561. (Stylesheet("/test.xsl", media=None), 'href="/test.xsl" type="text/xsl"'),
  562. (Stylesheet("/test.xsl", mimetype=None, media=None), 'href="/test.xsl"'),
  563. (
  564. Stylesheet("/test.xsl", mimetype="text/xml"),
  565. 'href="/test.xsl" type="text/xml" media="screen"',
  566. ),
  567. ]
  568. for stylesheet, expected in testdata:
  569. feed = Rss201rev2Feed(
  570. title="test",
  571. link="https://example.com",
  572. description="test",
  573. stylesheets=[stylesheet],
  574. )
  575. doc = feed.writeString("utf-8")
  576. with self.subTest(expected=expected):
  577. self.assertIn(f"<?xml-stylesheet {expected}?>", doc)
  578. def test_stylesheets_instructions_are_at_the_top(self):
  579. response = self.client.get("/syndication/stylesheet/")
  580. doc = minidom.parseString(response.content)
  581. self.assertEqual(doc.childNodes[0].nodeName, "xml-stylesheet")
  582. self.assertEqual(
  583. doc.childNodes[0].data,
  584. 'href="/stylesheet1.xsl" type="text/xsl" media="screen"',
  585. )
  586. self.assertEqual(doc.childNodes[1].nodeName, "xml-stylesheet")
  587. self.assertEqual(
  588. doc.childNodes[1].data,
  589. 'href="/stylesheet2.xsl" type="text/xsl" media="screen"',
  590. )
  591. def test_stylesheets_typeerror_if_str_or_stylesheet(self):
  592. for stylesheet, error_message in [
  593. ("/stylesheet.xsl", "stylesheets should be a list, not <class 'str'>"),
  594. (
  595. Stylesheet("/stylesheet.xsl"),
  596. "stylesheets should be a list, "
  597. "not <class 'django.utils.feedgenerator.Stylesheet'>",
  598. ),
  599. ]:
  600. args = ("title", "/link", "description")
  601. with self.subTest(stylesheets=stylesheet):
  602. self.assertRaisesMessage(
  603. TypeError,
  604. error_message,
  605. SyndicationFeed,
  606. *args,
  607. stylesheets=stylesheet,
  608. )
  609. def test_stylesheets_repr(self):
  610. testdata = [
  611. (Stylesheet("/test.xsl", mimetype=None), "('/test.xsl', None, 'screen')"),
  612. (Stylesheet("/test.xsl", media=None), "('/test.xsl', 'text/xsl', None)"),
  613. (
  614. Stylesheet("/test.xsl", mimetype=None, media=None),
  615. "('/test.xsl', None, None)",
  616. ),
  617. (
  618. Stylesheet("/test.xsl", mimetype="text/xml"),
  619. "('/test.xsl', 'text/xml', 'screen')",
  620. ),
  621. ]
  622. for stylesheet, expected in testdata:
  623. self.assertEqual(repr(stylesheet), expected)
  624. @requires_tz_support
  625. def test_feed_last_modified_time_naive_date(self):
  626. """
  627. Tests the Last-Modified header with naive publication dates.
  628. """
  629. response = self.client.get("/syndication/naive-dates/")
  630. self.assertEqual(
  631. response.headers["Last-Modified"], "Tue, 26 Mar 2013 01:00:00 GMT"
  632. )
  633. def test_feed_last_modified_time(self):
  634. """
  635. Tests the Last-Modified header with aware publication dates.
  636. """
  637. response = self.client.get("/syndication/aware-dates/")
  638. self.assertEqual(
  639. response.headers["Last-Modified"], "Mon, 25 Mar 2013 19:18:00 GMT"
  640. )
  641. # No last-modified when feed has no item_pubdate
  642. response = self.client.get("/syndication/no_pubdate/")
  643. self.assertFalse(response.has_header("Last-Modified"))
  644. def test_feed_url(self):
  645. """
  646. The feed_url can be overridden.
  647. """
  648. response = self.client.get("/syndication/feedurl/")
  649. doc = minidom.parseString(response.content)
  650. for link in doc.getElementsByTagName("link"):
  651. if link.getAttribute("rel") == "self":
  652. self.assertEqual(
  653. link.getAttribute("href"), "http://example.com/customfeedurl/"
  654. )
  655. def test_secure_urls(self):
  656. """
  657. Test URLs are prefixed with https:// when feed is requested over HTTPS.
  658. """
  659. response = self.client.get(
  660. "/syndication/rss2/",
  661. **{
  662. "wsgi.url_scheme": "https",
  663. },
  664. )
  665. doc = minidom.parseString(response.content)
  666. chan = doc.getElementsByTagName("channel")[0]
  667. self.assertEqual(
  668. chan.getElementsByTagName("link")[0].firstChild.wholeText[0:5], "https"
  669. )
  670. atom_link = chan.getElementsByTagName("atom:link")[0]
  671. self.assertEqual(atom_link.getAttribute("href")[0:5], "https")
  672. for link in doc.getElementsByTagName("link"):
  673. if link.getAttribute("rel") == "self":
  674. self.assertEqual(link.getAttribute("href")[0:5], "https")
  675. def test_item_link_error(self):
  676. """
  677. An ImproperlyConfigured is raised if no link could be found for the
  678. item(s).
  679. """
  680. msg = (
  681. "Give your Article class a get_absolute_url() method, or define "
  682. "an item_link() method in your Feed class."
  683. )
  684. with self.assertRaisesMessage(ImproperlyConfigured, msg):
  685. self.client.get("/syndication/articles/")
  686. def test_template_feed(self):
  687. """
  688. The item title and description can be overridden with templates.
  689. """
  690. response = self.client.get("/syndication/template/")
  691. doc = minidom.parseString(response.content)
  692. feed = doc.getElementsByTagName("rss")[0]
  693. chan = feed.getElementsByTagName("channel")[0]
  694. items = chan.getElementsByTagName("item")
  695. self.assertChildNodeContent(
  696. items[0],
  697. {
  698. "title": "Title in your templates: My first entry\n",
  699. "description": "Description in your templates: My first entry\n",
  700. "link": "http://example.com/blog/%s/" % self.e1.pk,
  701. },
  702. )
  703. def test_template_context_feed(self):
  704. """
  705. Custom context data can be passed to templates for title
  706. and description.
  707. """
  708. response = self.client.get("/syndication/template_context/")
  709. doc = minidom.parseString(response.content)
  710. feed = doc.getElementsByTagName("rss")[0]
  711. chan = feed.getElementsByTagName("channel")[0]
  712. items = chan.getElementsByTagName("item")
  713. self.assertChildNodeContent(
  714. items[0],
  715. {
  716. "title": "My first entry (foo is bar)\n",
  717. "description": "My first entry (foo is bar)\n",
  718. },
  719. )
  720. def test_add_domain(self):
  721. """
  722. add_domain() prefixes domains onto the correct URLs.
  723. """
  724. prefix_domain_mapping = (
  725. (("example.com", "/foo/?arg=value"), "http://example.com/foo/?arg=value"),
  726. (
  727. ("example.com", "/foo/?arg=value", True),
  728. "https://example.com/foo/?arg=value",
  729. ),
  730. (
  731. ("example.com", "http://djangoproject.com/doc/"),
  732. "http://djangoproject.com/doc/",
  733. ),
  734. (
  735. ("example.com", "https://djangoproject.com/doc/"),
  736. "https://djangoproject.com/doc/",
  737. ),
  738. (
  739. ("example.com", "mailto:uhoh@djangoproject.com"),
  740. "mailto:uhoh@djangoproject.com",
  741. ),
  742. (
  743. ("example.com", "//example.com/foo/?arg=value"),
  744. "http://example.com/foo/?arg=value",
  745. ),
  746. )
  747. for prefix in prefix_domain_mapping:
  748. with self.subTest(prefix=prefix):
  749. self.assertEqual(views.add_domain(*prefix[0]), prefix[1])
  750. def test_get_object(self):
  751. response = self.client.get("/syndication/rss2/articles/%s/" % self.e1.pk)
  752. doc = minidom.parseString(response.content)
  753. feed = doc.getElementsByTagName("rss")[0]
  754. chan = feed.getElementsByTagName("channel")[0]
  755. items = chan.getElementsByTagName("item")
  756. self.assertChildNodeContent(
  757. items[0],
  758. {
  759. "comments": "/blog/%s/article/%s/comments" % (self.e1.pk, self.a1.pk),
  760. "description": "Article description: My first article",
  761. "link": "http://example.com/blog/%s/article/%s/"
  762. % (self.e1.pk, self.a1.pk),
  763. "title": "Title: My first article",
  764. "pubDate": rfc2822_date(timezone.make_aware(self.a1.published, TZ)),
  765. },
  766. )
  767. def test_get_non_existent_object(self):
  768. response = self.client.get("/syndication/rss2/articles/0/")
  769. self.assertEqual(response.status_code, 404)