mirror of
https://github.com/JockeTF/fimfarchive.git
synced 2024-11-24 22:27:59 +01:00
168 lines
5.3 KiB
Python
168 lines
5.3 KiB
Python
"""
|
|
Fimfiction fetcher tests.
|
|
"""
|
|
|
|
|
|
#
|
|
# Fimfarchive, preserves stories from Fimfiction.
|
|
# Copyright (C) 2018 Joakim Soderlund
|
|
#
|
|
# This program 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.
|
|
#
|
|
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
|
|
import pytest
|
|
|
|
from fimfarchive.exceptions import InvalidStoryError, StorySourceError
|
|
from fimfarchive.fetchers import FimfictionFetcher
|
|
from fimfarchive.utils import JayWalker
|
|
|
|
from tests.fixtures.responses import Recorder
|
|
|
|
|
|
VALID_STORY_KEY = 9
|
|
INVALID_STORY_KEY = 7
|
|
EMPTY_STORY_KEY = 8
|
|
PROTECTED_STORY_KEY = 208799
|
|
|
|
|
|
class Redactor(JayWalker):
|
|
"""
|
|
Redacts recorded responses.
|
|
"""
|
|
|
|
def wrap(self, text: str) -> str:
|
|
"""
|
|
Wraps text in a valid HTML body.
|
|
"""
|
|
return f"<html><body>{text}</body></html>"
|
|
|
|
def handle(self, data, key, value) -> None:
|
|
if key in ('description', 'short_description'):
|
|
data[key] = "REDACTED"
|
|
elif key == 'text' and "<h1><a name='1'></a>" in value:
|
|
data[key] = self.wrap("<h1><a name='1'></a>REDACTED</h1>")
|
|
elif key == 'text':
|
|
data[key] = self.wrap("REDACTED")
|
|
else:
|
|
self.walk(value)
|
|
|
|
|
|
class TestFimfictionFetcher:
|
|
"""
|
|
FimfictionFetcher tests.
|
|
"""
|
|
|
|
@pytest.fixture
|
|
def fetcher(self, responses):
|
|
"""
|
|
Returns the fetcher instance to test.
|
|
"""
|
|
if isinstance(responses, Recorder):
|
|
responses.walker = Redactor()
|
|
|
|
yield FimfictionFetcher()
|
|
|
|
def test_with_statment(self):
|
|
"""
|
|
Tests fetcher can be used in with statements.
|
|
"""
|
|
with FimfictionFetcher() as fetcher:
|
|
fetcher.get('http://example.com/')
|
|
|
|
@pytest.mark.xfail(reason="Recorder issue")
|
|
def test_get_for_invalid_host(self, fetcher):
|
|
"""
|
|
Tests `StorySourceError` is raised if server is unreachable.
|
|
"""
|
|
with pytest.raises(StorySourceError):
|
|
fetcher.get('http://-')
|
|
|
|
def test_get_for_invalid_page(self, fetcher):
|
|
"""
|
|
Tests `StorySourceError` is raised if page does not exist.
|
|
|
|
Since Fimfiction always returns HTTP 200 OK, even for invalid stories,
|
|
other HTTP status codes indicate that something else has gone wrong.
|
|
"""
|
|
with pytest.raises(StorySourceError):
|
|
fetcher.get('http://example.com/-')
|
|
|
|
def test_fetch_meta_for_valid_story(self, fetcher):
|
|
"""
|
|
Tests meta is returned if story is valid
|
|
"""
|
|
meta = fetcher.fetch_meta(VALID_STORY_KEY)
|
|
assert meta['id'] == VALID_STORY_KEY
|
|
assert meta['words'] != 0
|
|
|
|
def test_fetch_meta_for_invalid_story(self, fetcher):
|
|
"""
|
|
Tests `InvalidStoryError` is raised if story is invalid.
|
|
"""
|
|
with pytest.raises(InvalidStoryError):
|
|
fetcher.fetch_meta(INVALID_STORY_KEY)
|
|
|
|
def test_fetch_meta_for_empty_story(self, fetcher):
|
|
"""
|
|
Tests meta is returned if story is empty.
|
|
|
|
It would be feasible for the fetcher to raise `InvalidStoryError` here
|
|
since the story data is useless. But because the meta of such stories
|
|
may still be of interest, it was decided that meta should be returned.
|
|
"""
|
|
meta = fetcher.fetch_meta(EMPTY_STORY_KEY)
|
|
assert meta['id'] == EMPTY_STORY_KEY
|
|
assert meta['words'] == 0
|
|
|
|
def test_fetch_meta_for_protected_story(self, fetcher):
|
|
"""
|
|
Tests meta is returned if story is protected.
|
|
|
|
It would be feasible for the fetcher to raise `InvalidStoryError` here
|
|
since the story data is inaccessible. However, there is no way to
|
|
determine that a story is password-protected from its meta data.
|
|
"""
|
|
meta = fetcher.fetch_meta(PROTECTED_STORY_KEY)
|
|
assert meta['id'] == PROTECTED_STORY_KEY
|
|
assert meta['words'] != 0
|
|
|
|
def test_fetch_data_for_valid_story(self, fetcher):
|
|
"""
|
|
Tests data is returned if story is valid.
|
|
"""
|
|
data = fetcher.fetch_data(VALID_STORY_KEY)
|
|
assert len(data) != 0
|
|
|
|
@pytest.mark.xfail(reason='knighty/fimfiction-issues#139')
|
|
def test_fetch_data_for_invalid_story(self, fetcher):
|
|
"""
|
|
Tests `InvalidStoryError` is raised if story is invalid.
|
|
"""
|
|
with pytest.raises(InvalidStoryError):
|
|
fetcher.fetch_data(INVALID_STORY_KEY)
|
|
|
|
def test_fetch_data_for_empty_story(self, fetcher):
|
|
"""
|
|
Tests `InvalidStoryError` is raised if story is empty.
|
|
"""
|
|
with pytest.raises(InvalidStoryError):
|
|
fetcher.fetch_data(EMPTY_STORY_KEY)
|
|
|
|
def test_fetch_data_for_protected_story(self, fetcher):
|
|
"""
|
|
Tests `InvalidStoryError` is raised if story is protected.
|
|
"""
|
|
with pytest.raises(InvalidStoryError):
|
|
fetcher.fetch_data(PROTECTED_STORY_KEY)
|