diff --git a/tests/test_climat2.py b/tests/test_climat2.py index 64345eb..0536646 100644 --- a/tests/test_climat2.py +++ b/tests/test_climat2.py @@ -1,5 +1,7 @@ -import unittest +import os +import shutil import subprocess +import unittest class TestHelp(unittest.TestCase): @@ -14,6 +16,27 @@ class TestHelp(unittest.TestCase): self.assertIn(b'usage: main.py [-h] [-c] [-l] [-s] [-L] [files [files ...]]', stdout) +class TestCleanMeta(unittest.TestCase): + def test_jpg(self): + shutil.copy('./tests/data/dirty.jpg', './tests/data/clean.jpg') + + proc = subprocess.Popen(['./main.py', '--show', './tests/data/clean.jpg'], + stdout=subprocess.PIPE) + stdout, _ = proc.communicate() + self.assertIn(b'Comment: Created with GIMP', stdout) + + proc = subprocess.Popen(['./main.py', './tests/data/clean.jpg'], + stdout=subprocess.PIPE) + stdout, _ = proc.communicate() + + proc = subprocess.Popen(['./main.py', '--show', './tests/data/clean.jpg.cleaned'], + stdout=subprocess.PIPE) + stdout, _ = proc.communicate() + self.assertNotIn(b'Comment: Created with GIMP', stdout) + + os.remove('./tests/data/clean.jpg') + + class TestGetMeta(unittest.TestCase): def test_pdf(self): proc = subprocess.Popen(['./main.py', '--show', './tests/data/dirty.pdf'],