diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/mitmproxy/contentviews/test_image.py | 4 | ||||
-rw-r--r-- | test/mitmproxy/contentviews/test_image_parser.py | 14 |
2 files changed, 16 insertions, 2 deletions
diff --git a/test/mitmproxy/contentviews/test_image.py b/test/mitmproxy/contentviews/test_image.py index 9e7e28f5..000b9da5 100644 --- a/test/mitmproxy/contentviews/test_image.py +++ b/test/mitmproxy/contentviews/test_image.py @@ -1,10 +1,10 @@ -from mitmproxy.contentviews import image +from mitmproxy.contentviews.image import pillow from mitmproxy.test import tutils from . import full_eval def test_view_image(): - v = full_eval(image.ViewImage()) + v = full_eval(pillow.ViewImage()) for img in [ "mitmproxy/data/image.png", "mitmproxy/data/image.gif", diff --git a/test/mitmproxy/contentviews/test_image_parser.py b/test/mitmproxy/contentviews/test_image_parser.py new file mode 100644 index 00000000..d4116392 --- /dev/null +++ b/test/mitmproxy/contentviews/test_image_parser.py @@ -0,0 +1,14 @@ +import io + +from mitmproxy.contentviews.image import image_parser +from mitmproxy.test import tutils + +def test_png_parser(): + img = "mitmproxy/data/image.png" + with open(tutils.test_data.path(img), "rb") as f: + fmt, parts = image_parser.get_png(io.BytesIO(f.read())) + assert fmt == "PNG" + assert parts + assert parts["width"] == 174 + assert parts["height"] == 174 + assert parts["format"] == "Portable network graphics" |