trentpalmerdotorg/audio/tests/test_rss.py
2021-04-25 20:07:13 -07:00

190 lines
8.8 KiB
Python

from django.test import TestCase
from django.contrib.auth.models import User
from audio.models import Feed, Episode
from accounts.models import Account
from datetime import date, timedelta
from dateutil import parser
from django.urls import reverse
import xml.etree.ElementTree as ET
from tp.settings import IMAGES_URL, MP3_URL
from .string_test_string import get_ep_description
class TestRssTestCase(TestCase):
def test_rss(self):
feed_a = Feed.objects.get(title='Caesar-Pompey Civil War')
kw_args = {'slug': feed_a.slug}
response = self.client.get(reverse('audio:rss', kwargs=kw_args))
self.assertEquals(response.status_code, 200)
self.assertEquals(len(response.templates), 0)
self.assertEquals(
response.request['PATH_INFO'],
f'/rss/{feed_a.slug}.xml'
)
rss = ET.fromstring(response.content)
self.assertEquals(rss.attrib['version'], '2.0')
rss_children = [t.tag for t in rss.findall('*')]
self.assertEquals(rss_children, ['channel'])
channel = rss.find('channel')
channel_children = [t.tag for t in channel.findall('*')]
self.assertEquals(channel_children, [
'title', 'link', 'description',
'{http://www.w3.org/2005/Atom}link',
'language', 'lastBuildDate', 'image',
'item', 'item'
])
channel_title = channel.find('title')
self.assertEquals(channel_title.text, 'Caesar-Pompey Civil War')
cdt = channel.find('description').text
self.assertEquals(cdt, 'Civil War between Pompey and Caesar.')
channel_link = channel.find('link')
self.assertEquals(channel_link.text[-75:], reverse(
'audio:feed', kwargs={'pk': feed_a.pk, 'slug': feed_a.slug})
)
cdlang = channel.find('language').text
self.assertEquals(cdlang, 'en-us')
latest_post = str(date.today() - timedelta(days=7))
cdlbd = parser.parse(channel.find('lastBuildDate').text)
cdlbd_string = cdlbd.strftime('%Y-%m-%d')
self.assertEquals(latest_post, cdlbd_string)
f_image = channel.find('image')
f_image_children = [t.tag for t in f_image.findall('*')]
self.assertEquals(f_image_children, [
'url', 'title', 'link', 'description'
])
f_image_url = f_image.find('url').text
self.assertEquals(f_image_url, f'{IMAGES_URL}{feed_a.image}')
f_image_title = f_image.find('title').text
self.assertEquals(f_image_title, feed_a.title)
f_image_link = f_image.find('link').text
self.assertEquals(
f_image_link,
f'testserver/feed/{feed_a.pk}/{feed_a.slug}'
)
f_image_description = f_image.find('description').text
self.assertEquals(
f_image_description, 'Image for: Caesar-Pompey Civil War'
)
episodes = [t for t in channel.findall('item')]
for i, ep in enumerate(episodes):
ep_children = [t.tag for t in ep.findall('*')]
self.assertEquals(ep_children, [
'title', 'link', 'description',
'pubDate', 'guid', 'enclosure', 'image'
])
episode = Episode.objects.get(feed=feed_a, episode_number=i + 1)
ep_title = ep.find('title').text
self.assertEquals(ep_title, f'{episode.episode_number}: {episode.title}')
ep_link = ep.find('link').text
self.assertEquals(
ep_link,
f'http://testserver/episode/{episode.pk}/{episode.slug}'
)
ep_desc = ep.find('description').text
self.assertEquals(
ep_desc,
get_ep_description(feed_a, episode)
)
ep_pub_date = parser.parse(ep.find('pubDate').text).strftime('%Y-%m-%d')
self.assertEquals(ep_pub_date, str(episode.pub_date))
ep_guid = ep.find('guid').text
self.assertEquals(
ep_guid,
f'http://testserver/episode/{episode.pk}/{episode.slug}'
)
ep_enclosure = ep.find('enclosure')
self.assertEquals(ep_enclosure.keys(), ['length', 'type', 'url'])
self.assertEquals(
ep_enclosure.attrib['length'], str(episode.image.size)
)
self.assertEquals(ep_enclosure.attrib['type'], 'audio/mpeg')
self.assertEquals(
ep_enclosure.attrib['url'], f'{MP3_URL}{episode.mp3}'
)
ep_image = ep.find('image')
epim_children = [t.tag for t in ep_image.findall('*')]
self.assertEquals(
epim_children, ['url', 'title', 'link', 'description']
)
epim_url = ep_image.find('url').text
self.assertEquals(epim_url, f'{IMAGES_URL}{episode.image.name}')
epim_title = ep_image.find('title').text
self.assertEquals(epim_title, episode.title)
epim_link = ep_image.find('link').text
self.assertEquals(
epim_link,
f'testserver/episode/{episode.pk}/{episode.slug}'
)
epim_desc = ep_image.find('description').text
self.assertEquals(epim_desc, f'Image for: {episode.title}')
def setUp(self):
user_a = User.objects.create(username='user_a')
user_a.set_password('password_user_a')
user_a.save()
Account.objects.create(user=user_a)
self.client.login(username='user_a', password='password_user_a')
with open('bicycle.jpg', 'rb') as image_f:
self.client.post(reverse('audio:new_feed'), {
'title': "Caesar-Pompey Civil War",
'author': "Gaius Julius Caesar",
'ebook_title': "Caesar's De Bello Gallico & Other Commentaries",
'ebook_url': "https://gutenberg.org/ebooks/10657",
'author_url': "https://gutenberg.org/ebooks/author/3621",
'translator': 'McDevitte, W. A. (William Alexander)',
'translator_url': 'https://gutenberg.org/ebooks/author/37952',
'intro_author': 'De Quincey, Thomas',
'intro_author_url': 'https://gutenberg.org/ebooks/author/797',
'license': 1,
'license_jurisdiction': 'in the USA',
'description': "Civil War between Pompey and Caesar.",
'image_title': 'A Penny Farthing',
'image_attribution': 'Agnieszka Kwiecień',
'image_attribution_url': 'https://commons.wikimedia.org/wiki/User:Nova',
'original_image_url': 'https://commons.wikimedia.org/wiki/File:Ordinary_bicycle01.jpg',
'image_license': 1,
'image_license_jurisdiction': 'in the USA',
'image': image_f,
})
feed_a = Feed.objects.get(title='Caesar-Pompey Civil War')
kw_args = {'feed_pk': feed_a.pk, 'feed_title_slug': feed_a.slug}
with (
open('bicycle.jpg', 'rb') as image_f,
open('Mark Twain Taming The Bicycle.mp3', 'rb') as mp3_f
):
self.client.post(reverse('audio:new_episode', kwargs=kw_args), {
'title': "Caesar-Pompey Civil War Book I",
'author': "Gaius Julius Caesar",
'pub_date': str(date.today() - timedelta(days=8)),
'episode_number': 1,
'description': 'Caesar confronts Afranius and Petreius',
'mp3': mp3_f,
'image_title': 'A Penny Farthing',
'image_attribution': 'Agnieszka Kwiecień',
'image_attribution_url': 'https://commons.wikimedia.org/wiki/User:Nova',
'original_image_url': 'https://commons.wikimedia.org/wiki/File:Ordinary_bicycle01.jpg',
'image_license': 1,
'image_license_jurisdiction': 'in the USA',
'image': image_f,
})
with (
open('bicycle.jpg', 'rb') as image_f,
open('Mark Twain Taming The Bicycle.mp3', 'rb') as mp3_f
):
self.client.post(reverse('audio:new_episode', kwargs=kw_args), {
'title': "Caesar-Pompey Civil War Book II",
'author': "Gaius Julius Caesar",
'pub_date': str(date.today() - timedelta(days=7)),
'episode_number': 2,
'description': 'Trebonius confronts Domitius',
'mp3': mp3_f,
'image_title': 'A Penny Farthing',
'image_attribution': 'Agnieszka Kwiecień',
'image_attribution_url': 'https://commons.wikimedia.org/wiki/User:Nova',
'original_image_url': 'https://commons.wikimedia.org/wiki/File:Ordinary_bicycle01.jpg',
'image_license': 1,
'image_license_jurisdiction': 'in the USA',
'image': image_f,
})