def test_extractEncryptedColorImage(self):
"""
Test the extraction of an encrypted color image `in a medium.
"""
sourcePath = 'files/bridge_color_mona_enc' + self.ext
expectedPath = 'files/color_mona' + self.ext
medium = Steganography(sourcePath)
extractedMessage = medium.extractMessageFromMedium()
encryptedMessage = AesMessage(extractedMessage, self.password)
encryptedMessage.saveToTarget(self.targetImageFilePath)
actualImage, expectedImage = loadTwoImageFiles(self.targetImageFilePath, expectedPath)
self.assertEqual(actualImage, expectedImage)
def test_extractEncryptedText(self):
"""
Test the extraction of an encrypted text file in a medium.
"""
sourcePath = 'files/lena_full_enc' + self.ext
expectedPath = 'files/full.txt'
medium = Steganography(sourcePath)
extractedMessage = medium.extractMessageFromMedium()
encryptedMessage = AesMessage(extractedMessage, self.password)
encryptedMessage.saveToTarget(self.targetTextFilePath)
actualTextFile, expectedTextFile = loadTwoTextFiles(self.targetTextFilePath, expectedPath)
self.assertEqual(actualTextFile, expectedTextFile)
def test_extractEncryptedColorImage(self):
"""
Test the extraction of an encrypted color image `in a medium.
"""
sourcePath = 'files/bridge_color_mona_enc' + self.ext
expectedPath = 'files/color_mona' + self.ext
medium = Steganography(sourcePath)
extractedMessage = medium.extractMessageFromMedium()
encryptedMessage = AesMessage(extractedMessage, self.password)
encryptedMessage.saveToTarget(self.targetImageFilePath)
actualImage, expectedImage = loadTwoImageFiles(
self.targetImageFilePath, expectedPath)
self.assertEqual(actualImage, expectedImage)
def test_extractEncryptedText(self):
"""
Test the extraction of an encrypted text file in a medium.
"""
sourcePath = 'files/lena_full_enc' + self.ext
expectedPath = 'files/full.txt'
medium = Steganography(sourcePath)
extractedMessage = medium.extractMessageFromMedium()
encryptedMessage = AesMessage(extractedMessage, self.password)
encryptedMessage.saveToTarget(self.targetTextFilePath)
actualTextFile, expectedTextFile = loadTwoTextFiles(
self.targetTextFilePath, expectedPath)
self.assertEqual(actualTextFile, expectedTextFile)
