class FileCrypterTest(unittest.TestCase):
test_file_path = os.path.realpath(__file__)
def setUp(self):
super(FileCrypterTest, self).setUp()
self.file_crypter = FileCrypter()
def test_can_encrypt_and_decrypt_file(self):
"""fab.tests.security.file_crypter_test Can encrypt and decrypt a specified file"""
password = "******"
self.file_crypter.encrypt(password, self.test_file_path)
self.file_crypter.decrypt(password, self.test_file_path + '.enc', self.test_file_path + '.dec')
self.assertEqual(self._file_contents(self.test_file_path), self._file_contents(self.test_file_path + '.dec'))
def _file_contents(self, file_path):
with open(file_path, 'r') as a_file:
return a_file.readlines()
def tearDown(self):
self._delete_file(self.test_file_path + '.enc')
self._delete_file(self.test_file_path + '.dec')
def _delete_file(self, file_path):
if os.path.exists(file_path):
os.remove(file_path)
class FileCrypterTest(unittest.TestCase):
test_file_path = os.path.realpath(__file__)
def setUp(self):
super(FileCrypterTest, self).setUp()
self.file_crypter = FileCrypter()
def test_can_encrypt_and_decrypt_file(self):
"""fab.tests.security.file_crypter_test Can encrypt and decrypt a specified file"""
password = "******"
self.file_crypter.encrypt(password, self.test_file_path)
self.file_crypter.decrypt(password, self.test_file_path + '.enc',
self.test_file_path + '.dec')
self.assertEqual(self._file_contents(self.test_file_path),
self._file_contents(self.test_file_path + '.dec'))
def _file_contents(self, file_path):
with open(file_path, 'r') as a_file:
return a_file.readlines()
def tearDown(self):
self._delete_file(self.test_file_path + '.enc')
self._delete_file(self.test_file_path + '.dec')
def _delete_file(self, file_path):
if os.path.exists(file_path):
os.remove(file_path)
def setUp(self):
super(FileCrypterTest, self).setUp()
self.file_crypter = FileCrypter()
def setUp(self):
super(FileCrypterTest, self).setUp()
self.file_crypter = FileCrypter()
