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)
def test_colorImageXml(self):
"""
Test the XML representation of an encrypted color image to match the reference.
"""
sourcePath = 'files/color_mona' + self.ext
expectedPath = 'files/color_mona_enc.xml'
message = Message(filePath=sourcePath, messageType='ColorImage')
encryptedMessage = AesMessage(message, self.password)
with open(self.targetTextFilePath, 'w') as xmlFile:
xmlString = encryptedMessage.getXmlString()
xmlFile.write(xmlString)
actualTextFile, expectedTextFile = loadTwoTextFiles(self.targetTextFilePath, expectedPath)
self.assertEqual(actualTextFile, expectedTextFile)
def test_colorImageXml(self):
"""
Test the XML representation of an encrypted color image to match the reference.
"""
sourcePath = 'files/color_mona' + self.ext
expectedPath = 'files/color_mona_enc.xml'
message = Message(filePath=sourcePath, messageType='ColorImage')
encryptedMessage = AesMessage(message, self.password)
with open(self.targetTextFilePath, 'w') as xmlFile:
xmlString = encryptedMessage.getXmlString()
xmlFile.write(xmlString)
actualTextFile, expectedTextFile = loadTwoTextFiles(
self.targetTextFilePath, expectedPath)
self.assertEqual(actualTextFile, expectedTextFile)
def test_embedEncryptedColorImage(self):
"""
Test the embedding of an encrypted color image in a medium.
"""
sourcePath = 'files/color_mona' + self.ext
expectedPath = 'files/bridge_color_mona_enc' + self.ext
mediumPath = 'files/bridge' + self.ext
message = Message(filePath=sourcePath, messageType='ColorImage')
encryptedMessage = AesMessage(message, self.password)
medium = Steganography(mediumPath)
medium.embedMessageInMedium(encryptedMessage, self.targetImageFilePath)
actualImage, expectedImage = loadTwoImageFiles(
self.targetImageFilePath, expectedPath)
self.assertEqual(actualImage, expectedImage)
def test_embedEncryptedText(self):
"""
Test the embedding of an encrypted text file in a medium.
"""
sourcePath = 'files/full.txt'
expectedPath = 'files/lena_full_enc' + self.ext
mediumPath = 'files/lena' + self.ext
message = Message(filePath=sourcePath, messageType='Text')
encryptedMessage = AesMessage(message, self.password)
medium = Steganography(mediumPath)
medium.embedMessageInMedium(encryptedMessage, self.targetImageFilePath)
actualImage, expectedImage = loadTwoImageFiles(
self.targetImageFilePath, expectedPath)
self.assertEqual(actualImage, expectedImage)
