108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (C) 2004-2008 Tristan Seligmann and Jonathan Jacobs
|
|
# Copyright (C) 2012-2014 Bastian Kleineidam
|
|
# Copyright (C) 2015-2019 Tobias Gruetzmacher
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
import json
|
|
import re
|
|
|
|
import pytest
|
|
import responses
|
|
|
|
import dosagelib.cmd
|
|
import httpmocks
|
|
|
|
|
|
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
|
|
|
|
|
|
@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)
|
|
|
|
def test_display_version(self):
|
|
cmd_ok("--version")
|
|
|
|
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",
|
|
"-o", "json", "xkcd")
|
|
|
|
@responses.activate
|
|
def test_fetch_html_and_rss_2(self, tmpdir):
|
|
httpmocks.page('http://www.bloomingfaeries.com/', 'bf-home')
|
|
httpmocks.page(re.compile('http://www.*faeries-405/'), 'bf-405')
|
|
httpmocks.png(re.compile(r'http://www\.bloomingfaeries\.com/.*\.jpg'))
|
|
|
|
cmd_ok("--numstrips", "2", "--baseurl", "bla", "--basepath",
|
|
str(tmpdir), "--output", "rss", "--output", "html", "--adult",
|
|
"BloomingFaeries")
|
|
|
|
@responses.activate
|
|
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")
|
|
|
|
directory = tmpdir.join('ZenPencils')
|
|
f = directory.join('dosage.json').open(encoding='utf-8')
|
|
data = json.load(f)
|
|
f.close()
|
|
|
|
pages = data['pages']
|
|
assert len(pages) == 1
|
|
|
|
page = list(pages.keys())[0]
|
|
assert page == 'https://zenpencils.com/comic/missing/'
|
|
|
|
images = data['pages'][page]['images']
|
|
assert len(images) == 2
|
|
|
|
for imgurl, imgfile in images.items():
|
|
assert directory.join(imgfile).check(file=1)
|