Add png parser

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"