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

chg: [flake] Addressed some flake and mypy issues.

Browse source
This commit is contained in:
Cédric Bonhomme 2022-11-13 23:28:32 +01:00
parent fc53f2a6b9
commit e290d1260e
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
6 changed files with 17 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

21
tests/test_exifHeader.py
View file

@ -35,10 +35,9 @@ from stegano import exifHeader
class TestEXIFHeader(unittest.TestCase):
def test_hide_empty_message(self):
"""Test hiding the empty string."""
secret = exifHeader.hide(
exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg", "./image.jpg", secret_message=""
)
# secret.save(""./image.png"")
clear_message = exifHeader.reveal("./image.jpg")
@ -48,7 +47,7 @@ class TestEXIFHeader(unittest.TestCase):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = exifHeader.hide(
exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg",
"./image.jpg",
secret_message=message,
@ -56,13 +55,12 @@ class TestEXIFHeader(unittest.TestCase):
clear_message = exifHeader.reveal("./image.jpg")
self.assertEqual(message, message)
self.assertEqual(message, clear_message.decode())
def test_with_image_without_exif_data(self):
secret = exifHeader.hide(
exifHeader.hide(
"./tests/sample-files/Lenna.jpg", "./image.jpg", secret_message=""
)
# secret.save(""./image.png"")
clear_message = exifHeader.reveal("./image.jpg")
@ -72,7 +70,7 @@ 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(
exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg",
img_enc="./image.jpg",
secret_file=text_file_to_hide,
@ -82,13 +80,12 @@ class TestEXIFHeader(unittest.TestCase):
self.assertEqual(message, clear_message)
def test_with_png_image(self):
secret = exifHeader.hide(
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")
exifHeader.reveal("./image.png")
def test_with_bytes(self):
outputBytes = io.BytesIO()
@ -102,11 +99,11 @@ class TestEXIFHeader(unittest.TestCase):
def tearDown(self):
try:
os.unlink("./image.jpg")
except:
except Exception:
pass
try:
os.unlink("./image.png")
except:
except Exception:
pass