a3x/nyaa
1
0
Fork 0
mirror of https://gitlab.com/SIGBUS/nyaa.git synced 2024-12-22 20:50:00 +00:00
Code Issues Projects Releases Wiki Activity

Test modify_query, create_magnet_from_es_info

Browse source
This commit is contained in:
Kfir Hadas 2017-08-03 00:40:25 +03:00
parent fc226ed5f6
commit 85688629b7
1 changed files with 34 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
tests/test_template_utils.py
View file

@ -2,19 +2,47 @@ import datetime
import os.path import os.path
import posixpath import posixpath
import unittest import unittest
from base64 import b32encode
from email.utils import formatdate from email.utils import formatdate
from urllib.parse import urlencode
from tests import NyaaTestCase from tests import NyaaTestCase
from nyaa.template_utils import (_jinja2_filter_rfc822, _jinja2_filter_rfc822_es, get_utc_timestamp, from nyaa.template_utils import (_jinja2_filter_rfc822, _jinja2_filter_rfc822_es, get_utc_timestamp,
get_display_time, timesince, filter_truthy, category_name, get_display_time, timesince, filter_truthy, category_name,
static_cachebuster, _static_cache) static_cachebuster, _static_cache, modify_query,
create_magnet_from_es_info)
from nyaa.torrents import get_default_trackers
class TestTemplateUtils(NyaaTestCase): class TestTemplateUtils(NyaaTestCase):
@unittest.skip('Not yet implemented')
def test_create_magnet_from_es_info(self): def test_create_magnet_from_es_info(self):
pass with self.request_context():
outer_func = create_magnet_from_es_info()
self.assertIn('create_magnet_from_es_info', outer_func)
magnet_creator = outer_func['create_magnet_from_es_info']
self.assertTrue(callable(magnet_creator))
download_name = 'ubuntu-16.04.2-desktop-amd64.iso'
es_info_hash = 'DA775E4AAF5635EF72583A391977E5ED6F14617E'
max_trackers = 5
trackers = None
magnet_hash = b32encode(bytes.fromhex(es_info_hash)).decode('utf-8')
magnet_parts = [
('dn', download_name)
]
for tracker in get_default_trackers()[:5]:
magnet_parts.append(('tr', tracker))
expected_magnet = 'magnet:?xt=urn:btih:' + magnet_hash + '&' + urlencode(magnet_parts)
magnet_uri = magnet_creator(
display_name=download_name,
info_hash=es_info_hash,
max_trackers=max_trackers,
trackers=trackers
)
self.assertEquals(magnet_uri, expected_magnet)
def test_static_cachebuster(self): def test_static_cachebuster(self):
with self.request_context(): with self.request_context():
@ -43,9 +71,10 @@ class TestTemplateUtils(NyaaTestCase):
self.app.debug = orig_debug self.app.debug = orig_debug
@unittest.skip('Not yet implemented')
def test_modify_query(self): def test_modify_query(self):
pass with self.request_context(path='/?q=hello&c=1_0'):
self.assertEqual(modify_query(q='world', u='HorribleSubs'),
'/?q=world&c=1_0&u=HorribleSubs')
def test_filter_truthy(self): def test_filter_truthy(self):
my_list = [ my_list = [