@@ -46,22 +46,24 @@ def setUp(self):
46
46
47
47
48
48
class FetchImageHelper (_common .TestCase ):
49
- """Helper mixin for mocking requests when fetching images
49
+ """Helper mixin for mocking requests when fetching images
50
50
with remote art sources.
51
51
"""
52
52
@responses .activate
53
53
def run (self , * args , ** kwargs ):
54
54
super (FetchImageHelper , self ).run (* args , ** kwargs )
55
55
56
+ IMAGEHEADER = {'image/jpeg' : b'\x00 ' * 6 + b'JFIF' ,
57
+ 'image/png' : b'\211 PNG\r \n \032 \n ' , }
58
+
56
59
def mock_response (self , url , content_type = 'image/jpeg' , file_type = None ):
57
- IMAGEHEADER = {'image/jpeg' : b'\x00 ' * 6 + b'JFIF' ,
58
- 'image/png' : b'\211 PNG\r \n \032 \n ' , }
59
60
if file_type is None :
60
61
file_type = content_type
61
62
responses .add (responses .GET , url ,
62
63
content_type = content_type ,
63
64
# imghdr reads 32 bytes
64
- body = IMAGEHEADER .get (file_type , b'' ).ljust (32 , b'\x00 ' ))
65
+ body = self .IMAGEHEADER .get (
66
+ file_type , b'' ).ljust (32 , b'\x00 ' ))
65
67
66
68
67
69
class FetchImageTest (FetchImageHelper , UseThePlugin ):
0 commit comments