rnhmjoj/qutebrowser
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix test_qutescheme.py

Browse Source
This commit is contained in:
Florian Bruhin 2016-09-14 12:05:15 +02:00
parent 5501d90268
commit cc1e134f25
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/unit/browser/webkit/network/test_qutescheme.py
View File

@ -23,13 +23,13 @@ import logging
from PyQt5.QtCore import QUrl from PyQt5.QtCore import QUrl
from PyQt5.QtNetwork import QNetworkRequest, QNetworkReply from PyQt5.QtNetwork import QNetworkRequest, QNetworkReply
from qutebrowser.browser import pdfjs from qutebrowser.browser import pdfjs, qutescheme
from qutebrowser.browser.webkit.network import qutescheme from qutebrowser.browser.webkit.network import webkitqutescheme
@pytest.fixture @pytest.fixture
def handler(): def handler():
return qutescheme.QuteSchemeHandler(win_id=0) return webkitqutescheme.QuteSchemeHandler()
class TestPDFJSHandler: class TestPDFJSHandler:
@ -45,18 +45,21 @@ class TestPDFJSHandler:
monkeypatch.setattr('qutebrowser.browser.pdfjs.get_pdfjs_res', monkeypatch.setattr('qutebrowser.browser.pdfjs.get_pdfjs_res',
get_pdfjs_res) get_pdfjs_res)
def test_existing_resource(self, handler): @pytest.fixture(autouse=True)
"""Test with a resource that exists.""" def patch_args(self, fake_args):
req = QNetworkRequest(QUrl('qute://pdfjs/existing/file')) fake_args.backend = 'webkit'
reply = handler.createRequest(None, req, None)
assert reply.readAll() == b'foobar'
def test_nonexisting_resource(self, handler, caplog): def test_existing_resource(self):
"""Test with a resource that exists."""
_mimetype, data = qutescheme.data_for_url(
QUrl('qute://pdfjs/existing/file'))
assert data == b'foobar'
def test_nonexisting_resource(self, caplog):
"""Test with a resource that does not exist.""" """Test with a resource that does not exist."""
req = QNetworkRequest(QUrl('qute://pdfjs/no/file'))
with caplog.at_level(logging.WARNING, 'misc'): with caplog.at_level(logging.WARNING, 'misc'):
reply = handler.createRequest(None, req, None) with pytest.raises(qutescheme.QuteSchemeError):
assert reply.error() == QNetworkReply.ContentNotFoundError qutescheme.data_for_url(QUrl('qute://pdfjs/no/file'))
assert len(caplog.records) == 1 assert len(caplog.records) == 1
assert (caplog.records[0].message == assert (caplog.records[0].message ==
'pdfjs resource requested but not found: /no/file') 'pdfjs resource requested but not found: /no/file')