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)