def test_multiple_items(self):
entry_list = [{'title': 'First'}, {'title': 'Second'}]
atom_xml = atom_formatter.format_atom(entries=entry_list)
root_elem = ElementTree.fromstring(atom_xml)
entries = root_elem.findall('{http://www.w3.org/2005/Atom}entry')
self.assertEqual(2, len(entries))
def test_include_single_item(self):
entry_list = [{'title': 'Title', 'link': '/link/', 'updated': '2013-03-14T00:00:00+01:00'}]
atom_xml = atom_formatter.format_atom(title='Feed Title', entries=entry_list)
root_elem = ElementTree.fromstring(atom_xml)
entry_elem = root_elem.find('{http://www.w3.org/2005/Atom}entry')
self.assertEqual('Title', entry_elem.find('{http://www.w3.org/2005/Atom}title').text)
link_elem = entry_elem.find('{http://www.w3.org/2005/Atom}link')
self.assertEqual('/link/', link_elem.get('href'))
def get(self):
word_lists = yield gen.Task(self.list_db.get_lists)
entry_list = []
for word_list in word_lists:
entry = {'title': word_list.get('title'),
'link': '/list/%s' % make_stub(word_list.get('id'),
word_list.get('stub'))}
entry_list.append(entry)
self.write(atom_formatter.format_atom(title='CiHui',
entries=entry_list))
self.finish()
def get(self):
word_lists = yield gen.Task(self.list_db.get_lists)
entry_list = []
for word_list in word_lists:
entry = {
'title':
word_list.get('title'),
'link':
'/list/%s' %
make_stub(word_list.get('id'), word_list.get('stub'))
}
entry_list.append(entry)
self.write(
atom_formatter.format_atom(title='CiHui', entries=entry_list))
self.finish()
def test_set_title(self):
atom_xml = atom_formatter.format_atom(title=u'大 feed title')
root_elem = ElementTree.fromstring(atom_xml)
title_elem = root_elem.find('{http://www.w3.org/2005/Atom}title')
self.assertEqual(u'大 feed title', title_elem.text)
def test_empty_list(self):
self.assertIn(b'http://www.w3.org/2005/Atom', atom_formatter.format_atom())
