python-feedgen/tests/test_entry.py

# -*- coding: utf-8 -*-

"""
Tests for a basic entry

These are test cases for a basic entry.
"""

import unittest

from feedgen.feed import FeedGenerator


class TestSequenceFunctions(unittest.TestCase):

    def setUp(self):
        fg = FeedGenerator()
        self.feedId = 'http://example.com'
        self.title = 'Some Testfeed'

        fg.id(self.feedId)
        fg.title(self.title)
        fg.link(href='http://lkiesow.de', rel='alternate')[0]
        fg.description('...')

        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654321/1')
        fe.title('The First Episode')
        fe.content(u'…')

        # Use also the different name add_item
        fe = fg.add_item()
        fe.id('http://lernfunk.de/media/654321/1')
        fe.title('The Second Episode')
        fe.content(u'…')

        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654321/1')
        fe.title('The Third Episode')
        fe.content(u'…')

        self.fg = fg

    def test_setEntries(self):
        fg2 = FeedGenerator()
        fg2.entry(self.fg.entry())
        assert len(fg2.entry()) == 3
        assert self.fg.entry() == fg2.entry()

    def test_loadExtension(self):
        fe = self.fg.add_item()
        fe.id('1')
        fe.title(u'…')
        fe.content(u'…')
        fe.load_extension('base')
        assert fe.base
        assert self.fg.atom_str()

    def test_checkEntryNumbers(self):
        fg = self.fg
        assert len(fg.entry()) == 3

    def test_TestEntryItems(self):
        fe = self.fg.add_item()
        fe.title('qwe')
        assert fe.title() == 'qwe'
        author = fe.author(email='ldoe@example.com')[0]
        assert not author.get('name')
        assert author.get('email') == 'ldoe@example.com'
        author = fe.author(name='John Doe', email='jdoe@example.com',
                           replace=True)[0]
        assert author.get('name') == 'John Doe'
        assert author.get('email') == 'jdoe@example.com'
        contributor = fe.contributor(name='John Doe', email='jdoe@ex.com')[0]
        assert contributor == fe.contributor()[0]
        assert contributor.get('name') == 'John Doe'
        assert contributor.get('email') == 'jdoe@ex.com'
        link = fe.link(href='http://lkiesow.de', rel='alternate')[0]
        assert link == fe.link()[0]
        assert link.get('href') == 'http://lkiesow.de'
        assert link.get('rel') == 'alternate'
        fe.guid('123')
        assert fe.guid().get('guid') == '123'
        fe.updated('2017-02-05 13:26:58+01:00')
        assert fe.updated().year == 2017
        fe.summary('asdf')
        assert fe.summary() == {'summary': 'asdf'}
        fe.description('asdfx')
        assert fe.description() == 'asdfx'
        fe.pubDate('2017-02-05 13:26:58+01:00')
        assert fe.pubDate().year == 2017
        fe.rights('asdfx')
        assert fe.rights() == 'asdfx'
        source = fe.source(url='https://example.com', title='Test')
        assert source.get('title') == 'Test'
        assert source.get('url') == 'https://example.com'
        fe.comments('asdfx')
        assert fe.comments() == 'asdfx'
        fe.enclosure(url='http://lkiesow.de', type='text/plain', length='1')
        assert fe.enclosure().get('url') == 'http://lkiesow.de'
        fe.ttl(8)
        assert fe.ttl() == 8

        self.fg.rss_str()
        self.fg.atom_str()

    def test_checkItemNumbers(self):
        fg = self.fg
        assert len(fg.item()) == 3

    def test_checkEntryContent(self):
        fg = self.fg
        assert fg.entry()

    def test_removeEntryByIndex(self):
        fg = FeedGenerator()
        self.feedId = 'http://example.com'
        self.title = 'Some Testfeed'

        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654321/1')
        fe.title('The Third Episode')
        assert len(fg.entry()) == 1
        fg.remove_entry(0)
        assert len(fg.entry()) == 0

    def test_removeEntryByEntry(self):
        fg = FeedGenerator()
        self.feedId = 'http://example.com'
        self.title = 'Some Testfeed'

        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654321/1')
        fe.title('The Third Episode')

        assert len(fg.entry()) == 1
        fg.remove_entry(fe)
        assert len(fg.entry()) == 0

    def test_categoryHasDomain(self):
        fg = FeedGenerator()
        fg.title('some title')
        fg.link(href='http://www.dontcare.com', rel='alternate')
        fg.description('description')
        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654321/1')
        fe.title('some title')
        fe.category([
             {'term': 'category',
              'scheme': 'http://www.somedomain.com/category',
              'label': 'Category',
              }])

        result = fg.rss_str()
        assert b'domain="http://www.somedomain.com/category"' in result

    def test_content_cdata_type(self):
        fg = FeedGenerator()
        fg.title('some title')
        fg.id('http://lernfunk.de/media/654322/1')
        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654322/1')
        fe.title('some title')
        fe.content('content', type='CDATA')
        result = fg.atom_str()
        assert b'<content type="CDATA"><![CDATA[content]]></content>' in result

    def test_summary_html_type(self):
        fg = FeedGenerator()
        fg.title('some title')
        fg.id('http://lernfunk.de/media/654322/1')
        fe = fg.add_entry()
        fe.id('http://lernfunk.de/media/654322/1')
        fe.title('some title')
        fe.link(href='http://lernfunk.de/media/654322/1')
        fe.summary('<p>summary</p>', type='html')
        result = fg.atom_str()
        expected = b'<summary type="html">&lt;p&gt;summary&lt;/p&gt;</summary>'
        assert expected in result