122 lines
5.0 KiB
Python
122 lines
5.0 KiB
Python
import os
|
|
import pytest
|
|
import mock
|
|
import feedparser
|
|
import tempfile
|
|
|
|
from django.core.files.storage import DefaultStorage
|
|
from django.utils import timezone
|
|
from django.urls import reverse
|
|
from django.conf import settings
|
|
from django.test import override_settings
|
|
|
|
from corbo.models import Category, Announce
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
CATEGORIES = ('Alerts',)
|
|
|
|
ATOM_FEED = """<?xml version="1.0" encoding="UTF-8"?>
|
|
<feed xml:lang="en-US" xmlns="http://www.w3.org/2005/Atom" xmlns:wfw="http://wellformedweb.org/CommentAPI/">
|
|
<id>tag:linuxfr.org,2005:/news</id>
|
|
<title>Sample RSS Feeds</title>
|
|
<updated>2016-09-16T10:29:46+02:00</updated>
|
|
<entry>
|
|
<id>tag:linuxfr.org,2005:News/37537</id>
|
|
<published>2016-09-16T10:29:46+02:00</published>
|
|
<updated>2016-09-16T11:27:00+02:00</updated>
|
|
<title>Feed entry sample</title>
|
|
<content type="html">
|
|
<img src="http://example.com/logo.png">
|
|
Feed entry content
|
|
<img src="http://site.com/error.png">
|
|
Another logo
|
|
<img src="http://example.com/folder/logo.png">
|
|
</content>
|
|
<author>
|
|
<name>Foo Bar</name>
|
|
</author>
|
|
</entry>
|
|
</feed>
|
|
"""
|
|
|
|
|
|
def mocked_request_get(*args, **kwargs):
|
|
|
|
class MockResponse:
|
|
|
|
def __init__(self, content, ok=True):
|
|
self.ok = ok
|
|
self.content = content
|
|
|
|
if args[0] == 'http://example.com/atom':
|
|
return MockResponse(ATOM_FEED)
|
|
elif args[0] == 'http://example.com/logo.png':
|
|
logo_path = os.path.join(os.path.dirname(__file__), 'media', 'logo.png')
|
|
return MockResponse(open(logo_path, 'rb').read())
|
|
elif args[0] == 'http://example.com/folder/logo.png':
|
|
logo_path = os.path.join(os.path.dirname(__file__), 'media', 'another_logo.png')
|
|
return MockResponse(open(logo_path, 'rb').read())
|
|
else:
|
|
return MockResponse(None, ok=False)
|
|
|
|
|
|
@mock.patch('corbo.models.requests.get', side_effect=mocked_request_get)
|
|
def test_announces_from_feed(mocked_get):
|
|
tmpdirname = tempfile.mkdtemp(suffix='_corbo')
|
|
with override_settings(MEDIA_ROOT=tmpdirname):
|
|
storage = DefaultStorage()
|
|
feed_content = feedparser.parse(ATOM_FEED)
|
|
for category in CATEGORIES:
|
|
c = Category.objects.create(name=category, rss_feed_url='http://example.com/atom')
|
|
assert c.announce_set.count() == len(feed_content['entries'])
|
|
for announce in c.announce_set.all():
|
|
assert announce.title in [feed['title'] for feed in feed_content['entries']]
|
|
assert storage.url(os.path.join('images', str(announce.id), '01_logo.png')) in announce.text
|
|
assert os.path.exists(storage.path(os.path.join('images', str(announce.id), '01_logo.png')))
|
|
assert storage.url(os.path.join('images', str(announce.id), '02_logo.png')) in announce.text
|
|
assert os.path.exists(storage.path(os.path.join('images', str(announce.id), '02_logo.png')))
|
|
assert storage.url('error.png') not in announce.text
|
|
assert 'http://site.com/error.png' in announce.text
|
|
# cleanup uploaded images
|
|
os.unlink(storage.path(os.path.join('images', str(announce.id), '01_logo.png')))
|
|
os.unlink(storage.path(os.path.join('images', str(announce.id), '02_logo.png')))
|
|
os.rmdir(storage.path(os.path.join('images', str(announce.id))))
|
|
os.rmdir(storage.path('images'))
|
|
os.rmdir(tmpdirname)
|
|
|
|
|
|
def test_announces_publishing(app):
|
|
c = Category.objects.create(name='Test announces')
|
|
a = Announce.objects.create(category=c, title='Test 1',
|
|
text='text')
|
|
feed_content = feedparser.parse(app.get(reverse('atom')).text)
|
|
assert len(feed_content['entries']) == 0
|
|
a.publication_time = timezone.now()
|
|
a.save()
|
|
feed_content = feedparser.parse(app.get(reverse('atom')).text)
|
|
assert len(feed_content['entries']) == 1
|
|
a.publication_time = timezone.now() + timezone.timedelta(days=1)
|
|
a.save()
|
|
feed_content = feedparser.parse(app.get(reverse('atom')).text)
|
|
assert len(feed_content['entries']) == 0
|
|
a.publication_time = timezone.now()
|
|
a.expiration_time = timezone.now()
|
|
a.save()
|
|
feed_content = feedparser.parse(app.get(reverse('atom')).text)
|
|
assert len(feed_content['entries']) == 0
|
|
|
|
def test_rss_feed_items(app):
|
|
c = Category.objects.create(name='Test announces')
|
|
for i in range(10):
|
|
a = Announce.objects.create(category=c, title='Test %s' % i,
|
|
publication_time=timezone.now(),
|
|
text='text of announce %s' % i)
|
|
feed_content = feedparser.parse(app.get(reverse('atom')).text)
|
|
assert len(feed_content['entries']) <= settings.RSS_ITEMS_LIMIT
|
|
for i in range(i, 10):
|
|
a = Announce.objects.create(category=c, title='Test %s' % i,
|
|
publication_time=timezone.now(),
|
|
text='text of announce %s' % i)
|
|
assert len(feed_content['entries']) <= settings.RSS_ITEMS_LIMIT
|