dosage/tests/test_dosage.py

165 lines
5.4 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
2016-10-28 22:21:41 +00:00
# Copyright (C) 2004-2008 Tristan Seligmann and Jonathan Jacobs
# Copyright (C) 2012-2014 Bastian Kleineidam
2020-01-04 14:51:01 +00:00
# Copyright (C) 2015-2020 Tobias Gruetzmacher
from __future__ import absolute_import, division, print_function
import json
import os
import re
import pytest
import responses
import dosagelib.cmd
import httpmocks
2019-11-03 23:16:25 +00:00
def cmd(*options):
"""'Fake' run dosage with given options."""
return dosagelib.cmd.main(('--allow-multiple',) + options)
def cmd_ok(*options):
assert cmd(*options) == 0
def cmd_err(*options):
assert cmd(*options) == 1
2019-12-03 22:52:14 +00:00
@pytest.mark.usefixtures("nosleep")
class TestDosage(object):
"""Test the dosage commandline client."""
# This shouldn't hit the network at all, so add responses without mocks to
# make sure it doesn't do that
@responses.activate
def test_list_comics(self):
for option in ("-l", "--list", "--singlelist"):
cmd_ok(option)
2019-12-05 21:23:48 +00:00
@responses.activate
def test_display_version(self):
cmd_ok("--version")
2019-12-05 21:23:48 +00:00
@responses.activate
def test_update_available(self, capsys):
responses.add(responses.GET, re.compile(r'https://api\.github\.com/'),
json={'tag_name': '9999.0', 'assets': [
{'browser_download_url': 'TEST.whl'},
{'browser_download_url': 'TEST.exe'},
]})
2019-12-05 21:23:48 +00:00
cmd_ok('--version', '-v')
captured = capsys.readouterr()
best = 'TEST.exe' if os.name == 'nt' else 'TEST.whl'
assert best in captured.out
2019-12-05 21:23:48 +00:00
assert 'A new version' in captured.out
@responses.activate
def test_no_update_available(self, capsys):
responses.add(responses.GET, re.compile(r'https://api\.github\.com/'),
json={'tag_name': '1.0'})
2019-12-05 21:23:48 +00:00
cmd_ok('--version', '-v')
captured = capsys.readouterr()
assert 'Detected local or development' in captured.out
@responses.activate
def test_current(self, capsys):
responses.add(responses.GET, re.compile(r'https://api\.github\.com/'),
json={'tag_name': dosagelib.__version__})
cmd_ok('--version', '-v')
captured = capsys.readouterr()
assert captured.out.endswith('issues\n')
2019-12-05 21:23:48 +00:00
@responses.activate
def test_update_broken(self, capsys):
responses.add(responses.GET, re.compile(r'https://api\.github\.com/'),
json={})
cmd_ok('--version', '-v')
captured = capsys.readouterr()
assert 'invalid update file' in captured.out
def test_display_help(self):
for option in ("-h", "--help"):
with pytest.raises(SystemExit):
cmd(option)
def test_module_help(self):
cmd_ok("-m", "xkcd")
def test_no_comics_specified(self):
cmd_err()
def test_unknown_option(self):
with pytest.raises(SystemExit):
cmd('--imadoofus')
def test_multiple_comics_match(self):
cmd_err('Garfield')
@responses.activate
def test_fetch_html_and_rss_json(self, tmpdir):
httpmocks.xkcd()
cmd_ok("-n", "2", "-v", "-b", str(tmpdir), "-o", "html", "-o", "rss",
2020-01-04 14:51:01 +00:00
"-o", "json", "--no-downscale", "xkcd")
@responses.activate
2020-01-04 14:51:01 +00:00
def test_fetch_html_and_rss_2(self, tmp_path):
httpmocks.page('http://www.bloomingfaeries.com/', 'bf-home')
httpmocks.page(re.compile('http://www.*faeries-405/'), 'bf-405')
2020-01-04 14:51:01 +00:00
httpmocks.png(re.compile(r'http://www\.blooming.*405.*jpg'))
httpmocks.png(re.compile(r'http://www\.blooming.*406.*jpg'), 'tall')
cmd_ok("--numstrips", "2", "--baseurl", "bla", "--basepath",
2020-01-04 14:51:01 +00:00
str(tmp_path), "--output", "rss", "--output", "html", "--adult",
"BloomingFaeries")
2020-01-04 14:51:01 +00:00
html = next((tmp_path / 'html').glob('*.html')).read_text()
assert "width=" in html
@responses.activate
def test_fetch_html_broken_img(self, tmp_path):
httpmocks.page('http://www.bloomingfaeries.com/', 'bf-home')
httpmocks.page(re.compile('http://www.*faeries-405/'), 'bf-405')
responses.add(responses.GET, re.compile(r'.*\.jpg'), body=b'\377\330',
content_type='image/jpeg')
cmd_ok("--numstrips", "2", "--baseurl", "bla", "--basepath",
str(tmp_path), "--output", "html", "--adult", "BloomingFaeries")
html = next((tmp_path / 'html').glob('*.html')).read_text()
assert "width=" not in html
@responses.activate
2016-03-28 14:29:57 +00:00
def test_fetch_indexed(self, tmpdir):
httpmocks.xkcd()
cmd_ok("-n", "2", "-v", "-b", str(tmpdir), "xkcd:303")
@responses.activate
def test_json_page_key_bounce_and_multi_image(self, tmpdir):
httpmocks.page('https://zenpencils.com/', 'zp-home')
httpmocks.page('https://zenpencils.com/comic/missing/', 'zp-223')
httpmocks.page('https://zenpencils.com/comic/lifejacket/', 'zp-222')
httpmocks.jpeg(re.compile(r'https://cdn-.*\.jpg'))
cmd_ok("-v", "-b", str(tmpdir), "-o", "json", "ZenPencils")
2019-11-03 23:16:25 +00:00
directory = tmpdir.join('ZenPencils')
f = directory.join('dosage.json').open(encoding='utf-8')
data = json.load(f)
f.close()
2019-11-03 23:16:25 +00:00
pages = data['pages']
assert len(pages) == 1
2019-11-03 23:16:25 +00:00
page = list(pages.keys())[0]
assert page == 'https://zenpencils.com/comic/missing/'
2019-11-03 23:16:25 +00:00
images = data['pages'][page]['images']
assert len(images) == 2
for imgurl, imgfile in images.items():
assert directory.join(imgfile).check(file=1)