mirror of
https://github.com/JockeTF/fimfarchive.git
synced 2024-11-27 15:38:00 +01:00
348 lines
9.8 KiB
Python
348 lines
9.8 KiB
Python
"""
|
|
Selector tests.
|
|
"""
|
|
|
|
|
|
#
|
|
# Fimfarchive, preserves stories from Fimfiction.
|
|
# Copyright (C) 2015 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/>.
|
|
#
|
|
|
|
|
|
from unittest.mock import patch, Mock, PropertyMock
|
|
|
|
import pytest
|
|
|
|
from fimfarchive.exceptions import InvalidStoryError
|
|
from fimfarchive.fetchers import Fetcher
|
|
from fimfarchive.flavors import UpdateStatus
|
|
from fimfarchive.mappers import StoryDateMapper
|
|
from fimfarchive.selectors import RefetchSelector, UpdateSelector
|
|
from fimfarchive.stories import Story
|
|
|
|
|
|
class TestUpdateSelector:
|
|
"""
|
|
UpdateSelector tests.
|
|
"""
|
|
|
|
@pytest.fixture
|
|
def selector(self):
|
|
"""
|
|
Returns a new update selector.
|
|
"""
|
|
return UpdateSelector()
|
|
|
|
def populate(self, story, date=0):
|
|
"""
|
|
Returns a cloned story populated with chapter meta.
|
|
"""
|
|
meta = {
|
|
**story.meta,
|
|
'date_modified': date,
|
|
'chapters': [
|
|
{'id': 1},
|
|
{'id': 2},
|
|
{'id': 3},
|
|
],
|
|
}
|
|
|
|
return story.merge(meta=meta)
|
|
|
|
def test_filter_empty_with_chapters(self, selector, story):
|
|
"""
|
|
Tests `filter_empty` keeps stories containing chapter meta.
|
|
"""
|
|
story = self.populate(story, 1)
|
|
selected = selector.filter_empty(story)
|
|
|
|
assert selected is story
|
|
|
|
def test_filter_empty_without_chapters(self, selector, story):
|
|
"""
|
|
Tests `filter_empty` drops stories without chapter meta.
|
|
"""
|
|
selected = selector.filter_empty(story)
|
|
|
|
assert selected is None
|
|
|
|
def test_filter_empty_with_empty_chapters(self, selector, story):
|
|
"""
|
|
Tests `filter_empty` drops stories with empty chapter meta.
|
|
"""
|
|
meta = {
|
|
**story.meta,
|
|
'chapters': []
|
|
}
|
|
|
|
story = story.merge(meta=meta)
|
|
selected = selector.filter_empty(story)
|
|
|
|
assert selected is None
|
|
|
|
def test_filter_invalid_for_valid(self, selector, story):
|
|
"""
|
|
Tests `filter_invalid` keeps valid stories.
|
|
"""
|
|
selected = selector.filter_invalid(story)
|
|
|
|
assert selected is story
|
|
|
|
def test_filter_invalid_without_meta(self, selector, story):
|
|
"""
|
|
Tests `filter_invalid` drops stories raising invalid on meta.
|
|
"""
|
|
with patch.object(Story, 'meta', new_callable=PropertyMock) as m:
|
|
m.side_effect = InvalidStoryError
|
|
selected = selector.filter_invalid(story)
|
|
|
|
assert selected is None
|
|
|
|
def test_filter_invalid_without_data(self, selector, story):
|
|
"""
|
|
Tests `filter_invalid` drops stories raising invalid on data.
|
|
"""
|
|
with patch.object(Story, 'data', new_callable=PropertyMock) as m:
|
|
m.side_effect = InvalidStoryError
|
|
selected = selector.filter_invalid(story)
|
|
|
|
assert selected is None
|
|
|
|
def test_filter_unchanged_for_changed(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` keeps changed stories.
|
|
"""
|
|
old = self.populate(story, 0)
|
|
new = self.populate(story, 1)
|
|
|
|
selected = selector.filter_unchanged(old, new)
|
|
|
|
assert selected is new
|
|
|
|
def test_filter_unchanged_for_unchanged(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` drops unchanged stories.
|
|
"""
|
|
old = self.populate(story, 0)
|
|
new = self.populate(story, 0)
|
|
|
|
selected = selector.filter_unchanged(old, new)
|
|
|
|
assert selected is None
|
|
|
|
def test_filter_unchanged_default_mapper(self):
|
|
"""
|
|
Tests `date_mapper` defaults to `StoryDateMapper`.
|
|
"""
|
|
selector = UpdateSelector()
|
|
|
|
assert isinstance(selector.date_mapper, StoryDateMapper)
|
|
|
|
def test_filter_unchanged_custom_mapper(self, story):
|
|
"""
|
|
Tests `date_mapper` can be customised.
|
|
"""
|
|
old = self.populate(story, 1)
|
|
new = self.populate(story, 0)
|
|
|
|
data = {
|
|
old: 0,
|
|
new: 1,
|
|
}
|
|
|
|
date_mapper = data.get
|
|
selector = UpdateSelector(date_mapper=date_mapper)
|
|
selected = selector.filter_unchanged(old, new)
|
|
|
|
assert selector.date_mapper is date_mapper
|
|
assert selected is new
|
|
|
|
def test_filter_unchanged_missing_old_date(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` raises `ValueError` on missing old date.
|
|
"""
|
|
old = self.populate(story, None)
|
|
new = self.populate(story, 1)
|
|
|
|
with pytest.raises(ValueError):
|
|
selector.filter_unchanged(old, new)
|
|
|
|
def test_filter_unchanged_missing_new_date(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` raises `ValueError` on missing new date.
|
|
"""
|
|
old = self.populate(story, 1)
|
|
new = self.populate(story, None)
|
|
|
|
with pytest.raises(ValueError):
|
|
selector.filter_unchanged(old, new)
|
|
|
|
def test_flavored(self, selector, story):
|
|
"""
|
|
Tests `flavored` adds specified flavor and returns the story.
|
|
"""
|
|
assert UpdateStatus.CREATED not in story.flavors
|
|
|
|
flavored = selector.flavored(story, UpdateStatus.CREATED)
|
|
|
|
assert story is flavored
|
|
assert UpdateStatus.CREATED in story.flavors
|
|
|
|
def test_created_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for newly created stories.
|
|
"""
|
|
old = None
|
|
new = self.populate(story, 1)
|
|
|
|
selected = selector(old, new)
|
|
|
|
assert selected is new
|
|
assert UpdateStatus.CREATED in new.flavors
|
|
|
|
def test_revived_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for stories that have not updated.
|
|
"""
|
|
old = self.populate(story, 1)
|
|
new = self.populate(story, 1)
|
|
|
|
selected = selector(old, new)
|
|
|
|
assert selected is old
|
|
assert UpdateStatus.REVIVED in old.flavors
|
|
|
|
def test_updated_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for stories that have updated.
|
|
"""
|
|
old = self.populate(story, 0)
|
|
new = self.populate(story, 1)
|
|
|
|
selected = selector(old, new)
|
|
|
|
assert selected is new
|
|
assert UpdateStatus.UPDATED in new.flavors
|
|
|
|
def test_deleted_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for stories that have been deleted.
|
|
"""
|
|
old = self.populate(story, 1)
|
|
new = None
|
|
|
|
selected = selector(old, new)
|
|
|
|
assert selected is old
|
|
assert UpdateStatus.DELETED in old.flavors
|
|
|
|
def test_old_protected_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for stories that have become password-protected.
|
|
"""
|
|
old = self.populate(story, 0)
|
|
new = self.populate(story, 1)
|
|
|
|
fetcher = Mock(spec=Fetcher)
|
|
fetcher.fetch_data.side_effect = InvalidStoryError
|
|
new = new.merge(fetcher=fetcher, data=None)
|
|
|
|
selected = selector(old, new)
|
|
|
|
fetcher.fetch_data.assert_called_once_with(new.key)
|
|
assert selected is old
|
|
assert UpdateStatus.DELETED in old.flavors
|
|
|
|
def test_new_protected_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for new password-protected stories.
|
|
"""
|
|
new = self.populate(story, 1)
|
|
|
|
fetcher = Mock(spec=Fetcher)
|
|
fetcher.fetch_data.side_effect = InvalidStoryError
|
|
new = new.merge(fetcher=fetcher, data=None)
|
|
|
|
selected = selector(None, new)
|
|
|
|
fetcher.fetch_data.assert_called_once_with(new.key)
|
|
assert selected is None
|
|
|
|
def test_invalid_selection(self, selector):
|
|
"""
|
|
Tests behavior for invalid stories.
|
|
"""
|
|
selected = selector(None, None)
|
|
|
|
assert selected is None
|
|
|
|
|
|
class TestRefetchSelector(TestUpdateSelector):
|
|
"""
|
|
RefetchSelector tests.
|
|
"""
|
|
|
|
@pytest.fixture
|
|
def selector(self):
|
|
"""
|
|
Returns a new refetch selector.
|
|
"""
|
|
return RefetchSelector()
|
|
|
|
def test_filter_unchanged_for_unchanged(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` keeps unchanged stories.
|
|
"""
|
|
old = self.populate(story, 0)
|
|
new = self.populate(story, 0)
|
|
|
|
selected = selector.filter_unchanged(old, new)
|
|
|
|
assert selected is new
|
|
|
|
def test_filter_unchanged_missing_old_date(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` keeps stories when missing old date.
|
|
"""
|
|
old = self.populate(story, None)
|
|
new = self.populate(story, 1)
|
|
|
|
selected = selector.filter_unchanged(old, new)
|
|
|
|
assert selected is new
|
|
|
|
def test_filter_unchanged_missing_new_date(self, selector, story):
|
|
"""
|
|
Tests `filter_unchanged` keeps stories when missing new date.
|
|
"""
|
|
old = self.populate(story, 1)
|
|
new = self.populate(story, None)
|
|
|
|
selected = selector.filter_unchanged(old, new)
|
|
|
|
assert selected is new
|
|
|
|
def test_revived_selection(self, selector, story):
|
|
"""
|
|
Tests behavior for stories that have not updated.
|
|
"""
|
|
old = self.populate(story, 1)
|
|
new = self.populate(story, 1)
|
|
|
|
selected = selector(old, new)
|
|
|
|
assert selected is new
|
|
assert UpdateStatus.UPDATED in new.flavors
|