From 031e1ca84c0a7dcbe803038d683d04c295527f71 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Bonhomme?= Date: Thu, 26 May 2016 07:32:35 +0200 Subject: [PATCH] Updated tests for lsb --- tests/test_exifHeader.py | 15 +++++++++++++++ tests/test_lsb.py | 5 +++-- 2 files changed, 18 insertions(+), 2 deletions(-) diff --git a/tests/test_exifHeader.py b/tests/test_exifHeader.py index db63963..f4853d7 100644 --- a/tests/test_exifHeader.py +++ b/tests/test_exifHeader.py @@ -63,6 +63,17 @@ class TestEXIFHeader(unittest.TestCase): self.assertEqual(b"", clear_message) + def test_with_text_file(self): + text_file_to_hide = "./tests/sample-files/lorem_ipsum.txt" + with open(text_file_to_hide, "rb") as f: + message = f.read() + secret = exifHeader.hide("./tests/sample-files/Elisha-Cuthbert.jpg", + img_enc="./image.jpg", + secret_file=text_file_to_hide) + + clear_message = exifHeader.reveal("./image.jpg") + self.assertEqual(message, clear_message) + def test_with_png_image(self): secret = exifHeader.hide("./tests/sample-files/Lenna.png", "./image.png", secret_message="Secret") @@ -76,6 +87,10 @@ class TestEXIFHeader(unittest.TestCase): os.unlink("./image.jpg") except: pass + try: + os.unlink("./image.png") + except: + pass if __name__ == '__main__': unittest.main() diff --git a/tests/test_lsb.py b/tests/test_lsb.py index bce2539..08dfd26 100644 --- a/tests/test_lsb.py +++ b/tests/test_lsb.py @@ -49,8 +49,9 @@ class TestLSB(unittest.TestCase): self.assertEqual(message, clear_message) - def test_with_long_message(self): - with open("./tests/sample-files/lorem_ipsum.txt") as f: + def test_with_text_file(self): + text_file_to_hide = "./tests/sample-files/lorem_ipsum.txt" + with open(text_file_to_hide) as f: message = f.read() secret = lsb.hide("./tests/sample-files/Lenna.png", message) secret.save("./image.png")