PonyMirrors/fimfarchive
1
0
Fork 0
mirror of https://github.com/JockeTF/fimfarchive.git synced 2024-11-22 05:17:59 +01:00
Code Issues Projects Releases Packages Wiki Activity

Fix blacklist not working for older archives

Browse source
This commit is contained in:
Joakim Soderlund 2020-03-28 20:42:08 +01:00
parent feb0cbacaf
commit 00afaedb60
2 changed files with 33 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
fimfarchive/utils.py
View file

@ -246,8 +246,8 @@ def is_blacklisted(story: Story) -> bool:
Returns:
True if a story has been blacklisted.
"""
story_id = story.key
author_id = story.meta['author']['id']
story_id = int(story.meta['id'])
author_id = int(story.meta['author']['id'])
if story_id in STORY_WHITELIST:
return False

52
tests/test_utils.py
View file

@ -37,13 +37,6 @@ from fimfarchive.utils import (
)
BLACKLISTED_AUTHOR = 1
BLACKLISTED_STORY = 2
WHITELISTED_STORY = 3
UNLISTED_AUTHOR = 4
UNLISTED_STORY = 5
class TestEmpty:
"""
Empty tests.
@ -327,27 +320,34 @@ class TestIsBlacklisted:
"""
is_blacklisted tests.
"""
BLACKLISTED_AUTHOR = 1
BLACKLISTED_STORY = 2
WHITELISTED_STORY = 3
UNLISTED_AUTHOR = 4
UNLISTED_STORY = 5
@pytest.fixture
def utils(self):
"""
Patches the blacklists and whitelists.
"""
ab = patch.object(utils, 'AUTHOR_BLACKLIST', {BLACKLISTED_AUTHOR})
sb = patch.object(utils, 'STORY_BLACKLIST', {BLACKLISTED_STORY})
sw = patch.object(utils, 'STORY_WHITELIST', {WHITELISTED_STORY})
with ab, sb, sw:
yield utils
@pytest.mark.parametrize('key,author,result', [
COMBINATIONS = [
(BLACKLISTED_STORY, BLACKLISTED_AUTHOR, True),
(BLACKLISTED_STORY, UNLISTED_AUTHOR, True),
(UNLISTED_STORY, BLACKLISTED_AUTHOR, True),
(UNLISTED_STORY, UNLISTED_AUTHOR, False),
(WHITELISTED_STORY, BLACKLISTED_AUTHOR, False),
(WHITELISTED_STORY, UNLISTED_AUTHOR, False),
])
]
@pytest.fixture
def utils(self):
"""
Patches the blacklists and whitelists.
"""
ab = patch.object(utils, 'AUTHOR_BLACKLIST', {self.BLACKLISTED_AUTHOR})
sb = patch.object(utils, 'STORY_BLACKLIST', {self.BLACKLISTED_STORY})
sw = patch.object(utils, 'STORY_WHITELIST', {self.WHITELISTED_STORY})
with ab, sb, sw:
yield utils
@pytest.mark.parametrize('key,author,result', COMBINATIONS)
def test_blacklisted(self, utils, story, key, author, result):
"""
Tests the various blacklist combinations.
@ -356,3 +356,13 @@ class TestIsBlacklisted:
story = story.merge(key=key, meta=meta)
assert result is is_blacklisted(story)
@pytest.mark.parametrize('key,author,result', COMBINATIONS)
def test_blacklisted_string(self, utils, story, key, author, result):
"""
Tests the various blacklist combinations when IDs are strings.
"""
meta = {'id': str(key), 'author': {'id': str(author)}}
story = story.merge(key=key, meta=meta)
assert result is is_blacklisted(story)