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

Restore original save-manager, web-history at end of test.

Browse Source
This commit is contained in:
Imran Sobir 2017-02-08 16:18:33 +05:00
parent ec0e95969e
commit 0ab7fd4581
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/unit/browser/test_qutehistory.py
View File

@ -31,6 +31,16 @@ class TestHistoryHandler:
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def fake_objects(self, fake_save_manager): def fake_objects(self, fake_save_manager):
""" Create fake web-history with history for three different days """ """ Create fake web-history with history for three different days """
try:
original_save_manager = objreg.get('save-manager')
except KeyError:
original_save_manager = None
try:
original_web_history = objreg.get('web-history')
except KeyError:
original_web_history = None
temp_dir = tempfile.TemporaryDirectory() temp_dir = tempfile.TemporaryDirectory()
fake_web_history = history.WebHistory(temp_dir.name, 'fake-history') fake_web_history = history.WebHistory(temp_dir.name, 'fake-history')
objreg.register('web-history', fake_web_history, update=True) objreg.register('web-history', fake_web_history, update=True)
@ -55,6 +65,18 @@ class TestHistoryHandler:
web_history._add_entry(yesterday) web_history._add_entry(yesterday)
web_history.save() web_history.save()
yield
if original_save_manager:
objreg.register('save-manager', original_save_manager, update=True)
else:
objreg.delete('save-manager')
if original_web_history:
objreg.register('web-history', original_web_history, update=True)
else:
objreg.delete('web-history')
def test_history_without_query(self): def test_history_without_query(self):
""" Test qute://history shows today's history when it has no query """ """ Test qute://history shows today's history when it has no query """
_mimetype, data = qutescheme.qute_history(QUrl("qute://history")) _mimetype, data = qutescheme.qute_history(QUrl("qute://history"))