mirror of https://github.com/django/django.git
119 lines
5.0 KiB
Python
119 lines
5.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import datetime
|
|
from xml.dom import minidom
|
|
from django.test import TestCase
|
|
from django.test.client import Client
|
|
from django.utils import tzinfo
|
|
from models import Entry
|
|
try:
|
|
set
|
|
except NameError:
|
|
from sets import Set as set
|
|
|
|
class SyndicationFeedTest(TestCase):
|
|
fixtures = ['feeddata.json']
|
|
|
|
def assertChildNodes(self, elem, expected):
|
|
actual = set([n.nodeName for n in elem.childNodes])
|
|
expected = set(expected)
|
|
self.assertEqual(actual, expected)
|
|
|
|
def test_rss_feed(self):
|
|
response = self.client.get('/syndication/feeds/rss/')
|
|
doc = minidom.parseString(response.content)
|
|
|
|
# Making sure there's only 1 `rss` element and that the correct
|
|
# RSS version was specified.
|
|
feed_elem = doc.getElementsByTagName('rss')
|
|
self.assertEqual(len(feed_elem), 1)
|
|
feed = feed_elem[0]
|
|
self.assertEqual(feed.getAttribute('version'), '2.0')
|
|
|
|
# Making sure there's only one `channel` element w/in the
|
|
# `rss` element.
|
|
chan_elem = feed.getElementsByTagName('channel')
|
|
self.assertEqual(len(chan_elem), 1)
|
|
chan = chan_elem[0]
|
|
self.assertChildNodes(chan, ['title', 'link', 'description', 'language', 'lastBuildDate', 'item'])
|
|
|
|
items = chan.getElementsByTagName('item')
|
|
self.assertEqual(len(items), Entry.objects.count())
|
|
for item in items:
|
|
self.assertChildNodes(item, ['title', 'link', 'description', 'guid'])
|
|
|
|
def test_atom_feed(self):
|
|
response = self.client.get('/syndication/feeds/atom/')
|
|
doc = minidom.parseString(response.content)
|
|
|
|
feed = doc.firstChild
|
|
self.assertEqual(feed.nodeName, 'feed')
|
|
self.assertEqual(feed.getAttribute('xmlns'), 'http://www.w3.org/2005/Atom')
|
|
self.assertChildNodes(feed, ['title', 'link', 'id', 'updated', 'entry'])
|
|
|
|
entries = feed.getElementsByTagName('entry')
|
|
self.assertEqual(len(entries), Entry.objects.count())
|
|
for entry in entries:
|
|
self.assertChildNodes(entry, ['title', 'link', 'id', 'summary'])
|
|
summary = entry.getElementsByTagName('summary')[0]
|
|
self.assertEqual(summary.getAttribute('type'), 'html')
|
|
|
|
def test_custom_feed_generator(self):
|
|
response = self.client.get('/syndication/feeds/custom/')
|
|
doc = minidom.parseString(response.content)
|
|
|
|
feed = doc.firstChild
|
|
self.assertEqual(feed.nodeName, 'feed')
|
|
self.assertEqual(feed.getAttribute('django'), 'rocks')
|
|
self.assertChildNodes(feed, ['title', 'link', 'id', 'updated', 'entry', 'spam'])
|
|
|
|
entries = feed.getElementsByTagName('entry')
|
|
self.assertEqual(len(entries), Entry.objects.count())
|
|
for entry in entries:
|
|
self.assertEqual(entry.getAttribute('bacon'), 'yum')
|
|
self.assertChildNodes(entry, ['title', 'link', 'id', 'summary', 'ministry'])
|
|
summary = entry.getElementsByTagName('summary')[0]
|
|
self.assertEqual(summary.getAttribute('type'), 'html')
|
|
|
|
def test_complex_base_url(self):
|
|
"""
|
|
Tests that that the base url for a complex feed doesn't raise a 500
|
|
exception.
|
|
"""
|
|
response = self.client.get('/syndication/feeds/complex/')
|
|
self.assertEquals(response.status_code, 404)
|
|
|
|
def test_title_escaping(self):
|
|
"""
|
|
Tests that titles are escaped correctly in RSS feeds.
|
|
"""
|
|
response = self.client.get('/syndication/feeds/rss/')
|
|
doc = minidom.parseString(response.content)
|
|
for item in doc.getElementsByTagName('item'):
|
|
link = item.getElementsByTagName('link')[0]
|
|
if link.firstChild.wholeText == 'http://example.com/blog/4/':
|
|
title = item.getElementsByTagName('title')[0]
|
|
self.assertEquals(title.firstChild.wholeText, u'A & B < C > D')
|
|
|
|
def test_naive_datetime_conversion(self):
|
|
"""
|
|
Test that datetimes are correctly converted to the local time zone.
|
|
"""
|
|
# Naive date times passed in get converted to the local time zone, so
|
|
# check the recived zone offset against the local offset.
|
|
response = self.client.get('/syndication/feeds/naive-dates/')
|
|
doc = minidom.parseString(response.content)
|
|
updated = doc.getElementsByTagName('updated')[0].firstChild.wholeText
|
|
tz = tzinfo.LocalTimezone(datetime.datetime.now())
|
|
now = datetime.datetime.now(tz)
|
|
self.assertEqual(updated[-6:], str(now)[-6:])
|
|
|
|
def test_aware_datetime_conversion(self):
|
|
"""
|
|
Test that datetimes with timezones don't get trodden on.
|
|
"""
|
|
response = self.client.get('/syndication/feeds/aware-dates/')
|
|
doc = minidom.parseString(response.content)
|
|
updated = doc.getElementsByTagName('updated')[0].firstChild.wholeText
|
|
self.assertEqual(updated[-6:], '+00:42')
|
|
|