Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-16 05:34:47 +01:00
Code Issues Projects Releases Wiki Activity

fetchart: flake8 fixes

Browse source
This commit is contained in:
wordofglass 2016-06-25 21:55:44 +02:00
parent 5d00ca8bda
commit 2db1530ef8
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
test/test_art.py
View file

@ -46,22 +46,24 @@ class UseThePlugin(_common.TestCase):
class FetchImageHelper(_common.TestCase):
"""Helper mixin for mocking requests when fetching images
"""Helper mixin for mocking requests when fetching images
with remote art sources.
"""
@responses.activate
def run(self, *args, **kwargs):
super(FetchImageHelper, self).run(*args, **kwargs)
IMAGEHEADER = {'image/jpeg': b'\x00' * 6 + b'JFIF',
'image/png': b'\211PNG\r\n\032\n', }
def mock_response(self, url, content_type='image/jpeg', file_type=None):
IMAGEHEADER = {'image/jpeg': b'\x00' * 6 + b'JFIF',
'image/png': b'\211PNG\r\n\032\n', }
if file_type is None:
file_type = content_type
responses.add(responses.GET, url,
content_type=content_type,
# imghdr reads 32 bytes
body=IMAGEHEADER.get(file_type, b'').ljust(32, b'\x00'))
body=self.IMAGEHEADER.get(
file_type, b'').ljust(32, b'\x00'))
class FetchImageTest(FetchImageHelper, UseThePlugin):