fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-06-28 03:06:14 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improved tests for lsb and lsbset module.

Browse source
This commit is contained in:
Cédric Bonhomme 2017-02-22 10:50:03 +01:00
parent 7bee922746
commit e0ed23a225
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
3 changed files with 42 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/test_lsb.py
View file

@ -20,13 +20,15 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>
__author__ = "Cedric Bonhomme"
__version__ = "$Revision: 0.1 $"
__version__ = "$Revision: 0.2 $"
__date__ = "$Date: 2016/04/12 $"
__revision__ = "$Date: 2017/02/22 $"
__license__ = "GPLv3"
import io
import os
import unittest
from unittest.mock import patch
from stegano import lsb
@ -41,7 +43,6 @@ class TestLSB(unittest.TestCase):
def test_hide_and_reveal(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = lsb.hide("./tests/sample-files/Lenna.png", message)
secret.save("./image.png")
@ -52,7 +53,6 @@ class TestLSB(unittest.TestCase):
def test_with_transparent_png(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = lsb.hide("./tests/sample-files/transparent.png", message)
secret.save("./image.png")
@ -61,6 +61,24 @@ class TestLSB(unittest.TestCase):
self.assertEqual(message, clear_message)
@patch('builtins.input', return_value='y')
def test_manual_convert_rgb(self, input):
message_to_hide = "I love 🍕 and 🍫!"
secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide)
@patch('builtins.input', return_value='n')
def test_refuse_convert_rgb(self, input):
message_to_hide = "I love 🍕 and 🍫!"
with self.assertRaises(Exception):
secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide)
def test_auto_convert_rgb(self):
message_to_hide = "I love 🍕 and 🍫!"
secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide, True)
def test_with_text_file(self):
text_file_to_hide = "./tests/sample-files/lorem_ipsum.txt"
with open(text_file_to_hide) as f: