162 lines
5.2 KiB
Python
162 lines
5.2 KiB
Python
# SPDX-FileCopyrightText: Florian Bruhin (The Compiler) <mail@qutebrowser.org>
|
|
#
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import os
|
|
import sys
|
|
import shlex
|
|
|
|
import pytest
|
|
import pytest_bdd as bdd
|
|
from qutebrowser.qt.network import QSslSocket
|
|
bdd.scenarios('downloads.feature')
|
|
|
|
|
|
PROMPT_MSG = ("Asking question <qutebrowser.utils.usertypes.Question "
|
|
"default={!r} mode=<PromptMode.download: 5> option=None "
|
|
"text=* title='Save file to:'>, *")
|
|
|
|
|
|
@pytest.fixture
|
|
def download_dir(tmpdir):
|
|
downloads = tmpdir / 'downloads'
|
|
downloads.ensure(dir=True)
|
|
(downloads / 'subdir').ensure(dir=True)
|
|
try:
|
|
os.mkfifo(downloads / 'fifo')
|
|
except AttributeError:
|
|
pass
|
|
unwritable = downloads / 'unwritable'
|
|
unwritable.ensure(dir=True)
|
|
unwritable.chmod(0)
|
|
|
|
yield downloads
|
|
|
|
unwritable.chmod(0o755)
|
|
|
|
|
|
@bdd.given("I set up a temporary download dir")
|
|
def temporary_download_dir(quteproc, download_dir):
|
|
quteproc.set_setting('downloads.location.prompt', 'false')
|
|
quteproc.set_setting('downloads.location.remember', 'false')
|
|
quteproc.set_setting('downloads.location.directory', str(download_dir))
|
|
|
|
|
|
@bdd.given("I clean old downloads")
|
|
def clean_old_downloads(quteproc):
|
|
quteproc.send_cmd(':download-cancel --all')
|
|
quteproc.send_cmd(':download-clear')
|
|
|
|
|
|
@bdd.when("SSL is supported")
|
|
def check_ssl():
|
|
if not QSslSocket.supportsSsl():
|
|
pytest.skip("QtNetwork SSL not supported")
|
|
|
|
|
|
@bdd.when("I download an SSL redirect page")
|
|
def download_ssl_redirect(server, ssl_server, quteproc):
|
|
path = "data/downloads/download.bin"
|
|
url = f"https://localhost:{ssl_server.port}/redirect-http/{path}?port={server.port}"
|
|
quteproc.send_cmd(f":download {url}")
|
|
|
|
|
|
@bdd.when("the unwritable dir is unwritable")
|
|
def check_unwritable(tmpdir):
|
|
unwritable = tmpdir / 'downloads' / 'unwritable'
|
|
if os.access(unwritable, os.W_OK):
|
|
# Docker container or similar
|
|
pytest.skip("Unwritable dir was writable")
|
|
|
|
|
|
@bdd.when("I wait until the download is finished")
|
|
def wait_for_download_finished(quteproc):
|
|
quteproc.wait_for(category='downloads', message='Download * finished')
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse("I wait until the download {name} is finished"))
|
|
def wait_for_download_finished_name(quteproc, name):
|
|
quteproc.wait_for(category='downloads',
|
|
message='Download {} finished'.format(name))
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse('I wait for the download prompt for "{path}"'))
|
|
def wait_for_download_prompt(tmpdir, quteproc, path):
|
|
full_path = path.replace('(tmpdir)', str(tmpdir)).replace('/', os.sep)
|
|
quteproc.wait_for(message=PROMPT_MSG.format(full_path))
|
|
quteproc.wait_for(message="Entering mode KeyMode.prompt "
|
|
"(reason: question asked)")
|
|
|
|
|
|
@bdd.then(bdd.parsers.parse("The downloaded file {filename} should not exist"))
|
|
def download_should_not_exist(filename, tmpdir):
|
|
path = tmpdir / 'downloads' / filename
|
|
assert not path.check()
|
|
|
|
|
|
@bdd.then(bdd.parsers.parse("The downloaded file {filename} should exist"))
|
|
def download_should_exist(filename, tmpdir):
|
|
path = tmpdir / 'downloads' / filename
|
|
assert path.check()
|
|
|
|
|
|
@bdd.then(bdd.parsers.parse("The downloaded file {filename} should be "
|
|
"{size} bytes big"))
|
|
def download_size(filename, size, tmpdir):
|
|
path = tmpdir / 'downloads' / filename
|
|
assert path.size() == int(size)
|
|
|
|
|
|
@bdd.then(bdd.parsers.parse("The downloaded file {filename} should contain "
|
|
"{text}"))
|
|
def download_contents(filename, text, tmpdir):
|
|
path = tmpdir / 'downloads' / filename
|
|
assert text in path.read()
|
|
|
|
|
|
@bdd.then(bdd.parsers.parse('The download prompt should be shown with '
|
|
'"{path}"'))
|
|
def download_prompt(tmpdir, quteproc, path):
|
|
full_path = path.replace('(tmpdir)', str(tmpdir)).replace('/', os.sep)
|
|
quteproc.wait_for(message=PROMPT_MSG.format(full_path))
|
|
quteproc.send_cmd(':mode-leave')
|
|
|
|
|
|
@bdd.when("I set a test python open_dispatcher")
|
|
def default_open_dispatcher_python(quteproc, tmpdir):
|
|
cmd = '{} -c "import sys; print(sys.argv[1])"'.format(
|
|
shlex.quote(sys.executable))
|
|
quteproc.set_setting('downloads.open_dispatcher', cmd)
|
|
|
|
|
|
@bdd.when("I open the download")
|
|
def download_open(quteproc):
|
|
cmd = '{} -c "import sys; print(sys.argv[1])"'.format(
|
|
shlex.quote(sys.executable))
|
|
quteproc.send_cmd(':download-open {}'.format(cmd))
|
|
|
|
|
|
@bdd.when("I open the download with a placeholder")
|
|
def download_open_placeholder(quteproc):
|
|
cmd = '{} -c "import sys; print(sys.argv[1])"'.format(
|
|
shlex.quote(sys.executable))
|
|
quteproc.send_cmd(':download-open {} {{}}'.format(cmd))
|
|
|
|
|
|
@bdd.when("I directly open the download")
|
|
def download_open_with_prompt(quteproc):
|
|
cmd = '{} -c pass'.format(shlex.quote(sys.executable))
|
|
quteproc.send_cmd(':prompt-open-download {}'.format(cmd))
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse("I delete the downloaded file {filename}"))
|
|
def delete_file(tmpdir, filename):
|
|
(tmpdir / 'downloads' / filename).remove()
|
|
|
|
|
|
@bdd.then("the FIFO should still be a FIFO")
|
|
def fifo_should_be_fifo(tmpdir):
|
|
download_dir = tmpdir / 'downloads'
|
|
assert download_dir.exists()
|
|
assert not os.path.isfile(download_dir / 'fifo')
|