blob: 84dffcc714491a3f3b0170b80545b4dd2aa2dba9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from mitmproxy.contentviews import image
from .. import full_eval
def test_view_image(tdata):
v = full_eval(image.ViewImage())
for img in [
"mitmproxy/data/image.png",
"mitmproxy/data/image.gif",
"mitmproxy/data/all.jpeg",
"mitmproxy/data/image.ico",
]:
with open(tdata.path(img), "rb") as f:
viewname, lines = v(f.read())
assert img.split(".")[-1].upper() in viewname
assert v(b"flibble") == ('Unknown Image', [[('header', 'Image Format: '), ('text', 'unknown')]])
|
