diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/mitmproxy/contentviews/test_image_parser.py | 68 | ||||
| -rw-r--r-- | test/mitmproxy/data/png_parser/ctzn0g04.png | bin | 0 -> 753 bytes |
2 files changed, 50 insertions, 18 deletions
diff --git a/test/mitmproxy/contentviews/test_image_parser.py b/test/mitmproxy/contentviews/test_image_parser.py index 2544e3f1..3d987b9d 100644 --- a/test/mitmproxy/contentviews/test_image_parser.py +++ b/test/mitmproxy/contentviews/test_image_parser.py @@ -4,7 +4,7 @@ from mitmproxy.contentviews.image import image_parser from mitmproxy.test import tutils class TestPngParser: - def test_png_parser(self): + def test_png_size(self): img = "mitmproxy/data/image.png" with open(tutils.test_data.path(img), "rb") as f: parts = image_parser.parse_png(io.BytesIO(f.read())) @@ -12,22 +12,6 @@ class TestPngParser: assert tuple(['Size', '174 x 174 px']) in parts assert tuple(["Format", "Portable network graphics"]) in parts - def test_textual_data(self): - img = "mitmproxy/data/png_parser/ct1n0g04.png" - with open(tutils.test_data.path(img), "rb") as f: - parts = image_parser.parse_png(io.BytesIO(f.read())) - assert parts - expected = [ - ('Title', 'PngSuite'), - ('Author', 'Willem A.J. van Schaik\n(willem@schaik.com)'), - ('Copyright', 'Copyright Willem van Schaik, Singapore 1995-96'), - ('Description', 'A compilation of a set of images created to test the\nvarious color-types of the PNG format. Included are\nblack&white, color, paletted, with alpha channel, with\ntransparency formats. All bit-depths allowed according\nto the spec are present.'), - ('Software', 'Created on a NeXTstation color using "pnmtopng".'), - ('Disclaimer', 'Freeware.') - ] - for data in expected: - assert data in parts - def test_no_textual_data(self): img = "mitmproxy/data/png_parser/ct0n0g04.png" with open(tutils.test_data.path(img), "rb") as f: @@ -41,6 +25,54 @@ class TestPngParser: parts = [data for data in parts if data not in metadata] assert not parts + def test_textual_data(self): + img = "mitmproxy/data/png_parser/ct1n0g04.png" + expected = [ + ('Title', 'PngSuite'), + ('Author', 'Willem A.J. van Schaik\n(willem@schaik.com)'), + ('Copyright', 'Copyright Willem van Schaik, Singapore 1995-96'), + ('Description', 'A compilation of a set of images created to test the\nvarious color-types of the PNG format. Included are\nblack&white, color, paletted, with alpha channel, with\ntransparency formats. All bit-depths allowed according\nto the spec are present.'), + ('Software', 'Created on a NeXTstation color using "pnmtopng".'), + ('Disclaimer', 'Freeware.') + ] + with open(tutils.test_data.path(img), "rb") as f: + parts = image_parser.parse_png(io.BytesIO(f.read())) + assert parts + for data in expected: + assert data in parts + + def test_compressed_textual_data(self): + img = "mitmproxy/data/png_parser/ctzn0g04.png" + expected = [ + ('Title', 'PngSuite'), + ('Author', 'Willem A.J. van Schaik\n(willem@schaik.com)'), + ('Copyright', 'Copyright Willem van Schaik, Singapore 1995-96'), + ('Description', 'A compilation of a set of images created to test the\nvarious color-types of the PNG format. Included are\nblack&white, color, paletted, with alpha channel, with\ntransparency formats. All bit-depths allowed according\nto the spec are present.'), + ('Software', 'Created on a NeXTstation color using "pnmtopng".'), + ('Disclaimer', 'Freeware.') + ] + with open(tutils.test_data.path(img), "rb") as f: + parts = image_parser.parse_png(io.BytesIO(f.read())) + assert parts + for data in expected: + assert data in parts + + def test_international_textual_data(self): + img = "mitmproxy/data/png_parser/cten0g04.png" + expected = [ + ('Author', 'Willem van Schaik (willem@schaik.com)'), + ('Copyright', 'Copyright Willem van Schaik, Canada 2011'), + ('Description', 'A compilation of a set of images created to test the various color-types of the PNG format. Included are black&white, color, paletted, with alpha channel, with transparency formats. All bit-depths allowed according to the spec are present.'), + ('Software', 'Created on a NeXTstation color using "pnmtopng".'), + ('Disclaimer', 'Freeware.') + ] + with open(tutils.test_data.path(img), "rb") as f: + parts = image_parser.parse_png(io.BytesIO(f.read())) + print(parts) + assert parts + for data in expected: + assert data in parts + def test_gamma(self): img = "mitmproxy/data/png_parser/g07n0g16.png" with open(tutils.test_data.path(img), "rb") as f: @@ -48,7 +80,7 @@ class TestPngParser: assert parts assert ('gamma', '0.7') in parts - def test_gamma(self): + def test_aspect(self): img = "mitmproxy/data/png_parser/aspect.png" with open(tutils.test_data.path(img), "rb") as f: parts = image_parser.parse_png(io.BytesIO(f.read())) diff --git a/test/mitmproxy/data/png_parser/ctzn0g04.png b/test/mitmproxy/data/png_parser/ctzn0g04.png Binary files differnew file mode 100644 index 00000000..b4401c9c --- /dev/null +++ b/test/mitmproxy/data/png_parser/ctzn0g04.png |
