diff --git a/stegano/generators.py b/stegano/generators.py index 2adbc43..d158614 100644 --- a/stegano/generators.py +++ b/stegano/generators.py @@ -152,4 +152,5 @@ if __name__ == "__main__": # Point of entry in execution mode. f = fibonacci() for x in range(13): - print(next(f), end=' ') # 0 1 1 2 3 5 8 13 21 34 55 89 144 + #print(next(f), end=' ') # 0 1 1 2 3 5 8 13 21 34 55 89 144 + pass diff --git a/tests/test_slsbset.py b/tests/test_slsbset.py new file mode 100644 index 0000000..d5e04b5 --- /dev/null +++ b/tests/test_slsbset.py @@ -0,0 +1,76 @@ +#!/usr/bin/env python +#-*- coding: utf-8 -*- + +# Stéganô - Stéganô is a basic Python Steganography module. +# Copyright (C) 2010-2016 Cédric Bonhomme - https://www.cedricbonhomme.org +# +# For more information : https://github.com/cedricbonhomme/Stegano +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see + +__author__ = "Cedric Bonhomme" +__version__ = "$Revision: 0.1 $" +__date__ = "$Date: 2016/04/13 $" +__license__ = "GPLv3" + +import os +import unittest + +from stegano import slsbset + +class TestSLSBSET(unittest.TestCase): + + def test_hide_empty_message(self): + """ + Test hiding the empty string. + """ + secret = slsbset.hide("./examples/pictures/Lenna.png", "", + "eratosthenes") + secret.save("./image.png") + + clear_message = slsbset.reveal("./image.png", "eratosthenes") + + self.assertEqual("", clear_message) + + def test_hide_and_reveal(self): + messages_to_hide = ["a", "foo", "Hello World!", ":Python:"] + + for message in messages_to_hide: + secret = slsbset.hide("./examples/pictures/Lenna.png", message, + "eratosthenes") + secret.save("./image.png") + + clear_message = slsbset.reveal("./image.png", "eratosthenes") + + self.assertEqual(message, clear_message) + + def test_hide_and_reveal_with_bad_generator(self): + message_to_hide = "Hello World!" + + secret = slsbset.hide("./examples/pictures/Lenna.png", message_to_hide, + "eratosthenes") + secret.save("./image.png") + + with self.assertRaises(IndexError): + clear_message = slsbset.reveal("./image.png", "identity") + + def tearDown(self): + try: + os.unlink("./image.png") + except: + pass + + +if __name__ == '__main__': + unittest.main()