Updated tests for lsbset.

This commit is contained in:
Cédric Bonhomme 2016-05-26 07:35:23 +02:00
parent 031e1ca84c
commit 3279b0c52a
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D

View file

@ -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,