From 3279b0c52ac9e71f3e1b30261f6b271303644405 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Bonhomme?= Date: Thu, 26 May 2016 07:35:23 +0200 Subject: [PATCH] Updated tests for lsbset. --- tests/test_lsbset.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_lsbset.py b/tests/test_lsbset.py index d6e0309..8d16f2f 100644 --- a/tests/test_lsbset.py +++ b/tests/test_lsbset.py @@ -53,6 +53,13 @@ class TestLSBSet(unittest.TestCase): self.assertEqual(message, clear_message) + def test_with_too_long_message(self): + with open("./tests/sample-files/lorem_ipsum.txt") as f: + message = f.read() + with self.assertRaises(Exception): + lsbset.hide("./tests/sample-files/Lenna.png", message, + generators.eratosthenes()) + def test_hide_and_reveal_with_bad_generator(self): message_to_hide = "Hello World!" secret = lsbset.hide("./tests/sample-files/Lenna.png", message_to_hide,