96 lines
3.9 KiB
Python
96 lines
3.9 KiB
Python
# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:
|
|
|
|
# Copyright 2017 Imran Sobir
|
|
#
|
|
# This file is part of qutebrowser.
|
|
#
|
|
# qutebrowser is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# qutebrowser is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with qutebrowser. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from PyQt5.QtCore import QUrl
|
|
from qutebrowser.browser import history, qutescheme
|
|
from qutebrowser.utils import objreg
|
|
import datetime
|
|
import pytest
|
|
import tempfile
|
|
|
|
|
|
class TestHistoryHandler:
|
|
"""Test the qute://history endpoint."""
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def fake_objects(self, fake_save_manager):
|
|
""" Create fake web-history with history for three different days """
|
|
temp_dir = tempfile.TemporaryDirectory()
|
|
fake_web_history = history.WebHistory(temp_dir.name, 'fake-history')
|
|
objreg.register('web-history', fake_web_history, update=True)
|
|
objreg.register('save-manager', fake_save_manager, update=True)
|
|
|
|
# Add fake history items for different days
|
|
one_day = datetime.timedelta(days=1)
|
|
self.curr_date = datetime.datetime.today()
|
|
self.next_date = self.curr_date + one_day
|
|
self.prev_date = self.curr_date - one_day
|
|
|
|
today = history.Entry(atime=str(self.curr_date.timestamp()),
|
|
url=QUrl('www.today.com'), title='today')
|
|
tomorrow = history.Entry(atime=str(self.next_date.timestamp()),
|
|
url=QUrl('www.tomorrow.com'), title='tomorrow')
|
|
yesterday = history.Entry(atime=str(self.prev_date.timestamp()),
|
|
url=QUrl('www.yesterday.com'), title='yesterday')
|
|
|
|
web_history = objreg.get('web-history')
|
|
web_history._add_entry(today)
|
|
web_history._add_entry(tomorrow)
|
|
web_history._add_entry(yesterday)
|
|
web_history.save()
|
|
|
|
def test_history_without_query(self):
|
|
""" Test qute://history shows today's history when it has no query """
|
|
_mimetype, data = qutescheme.qute_history(QUrl("qute://history"))
|
|
key = "<span class=\"date\">{}</span>".format(
|
|
datetime.date.today().strftime("%a, %d %B %Y"))
|
|
assert key in data
|
|
|
|
def test_history_with_bad_query(self):
|
|
""" Test qute://history shows today's history when given bad query """
|
|
url = QUrl("qute://history?date=204-blaah")
|
|
_mimetype, data = qutescheme.qute_history(url)
|
|
key = "<span class=\"date\">{}</span>".format(
|
|
datetime.date.today().strftime("%a, %d %B %Y"))
|
|
assert key in data
|
|
|
|
def test_history_today(self):
|
|
""" Test qute://history shows history for today """
|
|
url = QUrl("qute://history")
|
|
_mimetype, data = qutescheme.qute_history(url)
|
|
assert "today" in data
|
|
assert "tomorrow" not in data
|
|
assert "yesterday" not in data
|
|
|
|
def test_history_yesterday(self):
|
|
""" Test qute://history shows history for yesterday """
|
|
url = QUrl("qute://history?date=" + self.prev_date.strftime("%Y-%m-%d"))
|
|
_mimetype, data = qutescheme.qute_history(url)
|
|
assert "today" not in data
|
|
assert "tomorrow" not in data
|
|
assert "yesterday" in data
|
|
|
|
def test_history_tomorrow(self):
|
|
""" Test qute://history shows history for tomorrow """
|
|
url = QUrl("qute://history?date=" + self.next_date.strftime("%Y-%m-%d"))
|
|
_mimetype, data = qutescheme.qute_history(url)
|
|
assert "today" not in data
|
|
assert "tomorrow" in data
|
|
assert "yesterday" not in data
|