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 code style.

Browse source
This commit is contained in:
Cédric Bonhomme 2019-12-17 09:18:37 +01:00
parent 9b216d9d59
commit 71f6c08c28
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
19 changed files with 335 additions and 253 deletions
Download patch file Download diff file Expand all files Collapse all files

43
tests/test_exifHeader.py
View file

@ -31,14 +31,15 @@ import io
from stegano import exifHeader
class TestEXIFHeader(unittest.TestCase):
class TestEXIFHeader(unittest.TestCase):
def test_hide_empty_message(self):
"""Test hiding the empty string.
"""
secret = exifHeader.hide("./tests/sample-files/20160505T130442.jpg",
"./image.jpg", secret_message="")
#secret.save(""./image.png"")
secret = exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg", "./image.jpg", secret_message=""
)
# secret.save(""./image.png"")
clear_message = exifHeader.reveal("./image.jpg")
@ -48,17 +49,21 @@ class TestEXIFHeader(unittest.TestCase):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = exifHeader.hide("./tests/sample-files/20160505T130442.jpg",
"./image.jpg", secret_message=message)
secret = exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg",
"./image.jpg",
secret_message=message,
)
clear_message = exifHeader.reveal("./image.jpg")
self.assertEqual(message, message)
def test_with_image_without_exif_data(self):
secret = exifHeader.hide("./tests/sample-files/Lenna.jpg",
"./image.jpg", secret_message="")
#secret.save(""./image.png"")
secret = exifHeader.hide(
"./tests/sample-files/Lenna.jpg", "./image.jpg", secret_message=""
)
# secret.save(""./image.png"")
clear_message = exifHeader.reveal("./image.jpg")
@ -68,17 +73,20 @@ class TestEXIFHeader(unittest.TestCase):
text_file_to_hide = "./tests/sample-files/lorem_ipsum.txt"
with open(text_file_to_hide, "rb") as f:
message = f.read()
secret = exifHeader.hide("./tests/sample-files/20160505T130442.jpg",
img_enc="./image.jpg",
secret_file=text_file_to_hide)
secret = exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg",
img_enc="./image.jpg",
secret_file=text_file_to_hide,
)
clear_message = exifHeader.reveal("./image.jpg")
self.assertEqual(message, clear_message)
def test_with_png_image(self):
secret = exifHeader.hide("./tests/sample-files/Lenna.png",
"./image.png", secret_message="Secret")
#secret.save(""./image.png"")
secret = exifHeader.hide(
"./tests/sample-files/Lenna.png", "./image.png", secret_message="Secret"
)
# secret.save(""./image.png"")
with self.assertRaises(ValueError):
clear_message = exifHeader.reveal("./image.png")
@ -86,7 +94,7 @@ class TestEXIFHeader(unittest.TestCase):
def test_with_bytes(self):
outputBytes = io.BytesIO()
message = b"Secret"
with open("./tests/sample-files/20160505T130442.jpg", 'rb') as f:
with open("./tests/sample-files/20160505T130442.jpg", "rb") as f:
exifHeader.hide(f, outputBytes, secret_message=message)
clear_message = exifHeader.reveal(outputBytes)
@ -102,5 +110,6 @@ class TestEXIFHeader(unittest.TestCase):
except:
pass
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main()