fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-12 17:18:30 +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
Show stats Download patch file Download diff file Expand all files Collapse all files

BIN
tests/sample-files/Lenna-grayscale.png Normal file
View file

Binary file not shown.
Side by side

After

Width:  |  Height:  |  Size: 151 KiB

24
tests/test_lsb.py
View file

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

24
tests/test_lsbset.py
View file

@ -20,13 +20,14 @@
# along with this program. If not, see <http://www.gnu.org/licenses/> # along with this program. If not, see <http://www.gnu.org/licenses/>
__author__ = "Cedric Bonhomme" __author__ = "Cedric Bonhomme"
__version__ = "$Revision: 0.2 $" __version__ = "$Revision: 0.3 $"
__date__ = "$Date: 2016/04/13 $" __date__ = "$Date: 2016/04/13 $"
__revision__ = "$Date: 2016/05/22 $" __revision__ = "$Date: 2017/02/22 $"
__license__ = "GPLv3" __license__ = "GPLv3"
import os import os
import unittest import unittest
from unittest.mock import patch
from stegano import lsbset from stegano import lsbset
from stegano.lsbset import generators from stegano.lsbset import generators
@ -55,7 +56,6 @@ class TestLSBSet(unittest.TestCase):
def test_with_transparent_png(self): def test_with_transparent_png(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"] messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide: for message in messages_to_hide:
secret = lsbset.hide("./tests/sample-files/transparent.png", secret = lsbset.hide("./tests/sample-files/transparent.png",
message, generators.eratosthenes()) message, generators.eratosthenes())
@ -66,6 +66,24 @@ class TestLSBSet(unittest.TestCase):
self.assertEqual(message, clear_message) 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 = lsbset.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide, generators.eratosthenes())
@patch('builtins.input', return_value='n')
def test_refuse_convert_rgb(self, input):
message_to_hide = "I love 🍕 and 🍫!"
with self.assertRaises(Exception):
secret = lsbset.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide, generators.eratosthenes())
def test_auto_convert_rgb(self):
message_to_hide = "I love 🍕 and 🍫!"
secret = lsbset.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide, generators.eratosthenes(), True)
def test_with_too_long_message(self): def test_with_too_long_message(self):
with open("./tests/sample-files/lorem_ipsum.txt") as f: with open("./tests/sample-files/lorem_ipsum.txt") as f:
message = f.read() message = f.read()