def test_simple_text(self):
text = " Hello World "
key = "44 52 d7 16 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb ba 62 24 af fb 76 76 e1 33 79 26 cd d6 02"
encrypted_text = encrypt_data(text, key)
decrypted_text = decrypt_data(encrypted_text, key)
self.assertEqual(text, decrypted_text)
def testLongText(self):
text = "สวัสดีชาวโลก $ ¢ € Й ﯚ Å || The quick brown fox jumps over the lazy dog. 1234567890! \n This is second line! \t this is tab!"
key = "4f 07 76 f2 72 0b c4 ce 31 e2 6e bb 19 26 f7 9f cc 38 fa f0 e3 68 b6 81 cb 09 d6 df f3 e5 eb 49"
encrypted_text = encrypt_data(text, key)
decrypted_text = decrypt_data(encrypted_text, key)
self.assertEqual(text, decrypted_text)
def testUTF8Text(self):
text = "สวัสดีชาวโลก $ ¢ € Й ﯚ Å"
key = "44 52 d7 16 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb ba 62 24 af fb 76 76 e1 33 79 26 cd d6 02"
encrypted_text = encrypt_data(text, key)
decrypted_text = decrypt_data(encrypted_text, key)
self.assertEqual(text, decrypted_text)
def testDecryptOnly(self):
encrypted_text = "df78cedebfb70624bb663a2eb8ba828adf78cedebfb70624bb663a2eb8ba828a9B354CBA0AB6F7763A88F04D83A237462BE1B68F8D2CFCD4FCDC575BE51333B431A45E86D9B9B5D21DBD91C044DA118C783D98F89E28911463F957E2DA3188F1F4615F637B7F089F6DAEFA352EABAF1AC6731C6E67D0644E36E8D5AFACEF3DC7795E0909DFF84CF92E773A0268F9C2F01232F0AC7E1152F4E3802B09CAC5B0E421DDEC41F55A56001E1D37CBD9E2865567E1ABF5A2B2F160EAD80B147F0A1A9B"
key = "3c 36 38 36 cf 4e 16 66 66 69 a2 5d a2 80 a1 86 1b 81 bd ff f9 75 71 38 c3 a7 06 05 a4 b2 f2 ce"
decrypted_text = decrypt_data(encrypted_text, key)
self.assertEqual(
decrypted_text,
"สวัสดีชาวโลก || $ ¢ € Й ﯚ Å || 1234567890 || The quick brown fox jumps over the lazy dog."
)
def testWrongKey(self):
text = " Hello World "
key = "44 52 d7 16 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb ba 62 24 af fb 76 76 e1 33 79 26 cd d6 02"
encrypted_text = encrypt_data(text, key)
key = "4e 52 d7 16 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb ba 62 24 af fb 76 76 e1 33 79 26 cd d6 02"
decrypted_text = decrypt_data(encrypted_text, key)
self.assertEqual(decrypted_text, None)
def test_same_key(self):
text = " Hello World "
key = "44 52 d7 16 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb ba 62 24 af fb 76 76 e1 33 79 26 cd d6 02"
encrypted_text = encrypt_data(text, key)
key = key.upper()
key = key.replace(" ", "")
decrypted_text = decrypt_data(encrypted_text, key)
self.assertEqual(text, decrypted_text)
def view_password():
global data
print '\n\n\tPlease Select an Application'
print '--------------------------------------------------\n'
for i in range(len(data)):
print str(i+1) + " - " + data[i][0]
selected_app = int(input('\nApplication : '))
clipboard.clipboard(encryption.decrypt_data(data[selected_app-1][1], 'paradise'))
print 'Selected Application Password Copied to Clipboard'
def testZeroByteKey(self):
text = " Hello World "
key = "44 52 d7 00 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb b0 02 24 af fb 76 76 e1 33 79 26 cd d6 02"
try:
encrypted_text = encrypt_data(text, key)
self.assertTrue(
False,
"Exception thrown expected because because key is invalid")
except Exception as exc:
self.assertEqual(str(exc), "key cannot contain zero byte block")
try:
decrypted_text = decrypt_data(
"249ec2837616f746f0055f3cc43adbddB403B528EBDF9AF29126BD70D85C904BF3F25748FB22898E1E837FE9D67133ED6DF2F70882CF2EF5060A1E87250AF92762479E4858159AADFC512052DA00630339663ED7DC359A9B02EC40B67E0E12F3E3CA6723D2E18CA1F239CDF0486F0B7D28C2A730C3DA4A0342A2D1C02D833839BDF6819610A2FBC34CFA0AB4C64A9352",
key)
self.assertTrue(
False,
"Exception thrown expected because because key is invalid")
except Exception as exc:
self.assertEqual(str(exc), "key cannot contain zero byte block")
def testInvalidKey(self):
text = " Hello World "
key = "40 02 d7 87 b6 bc 2c 93 89 c3 34 9f dc 17 fb 3d fb ba 62 24 af fb 76 76 e1 33 79 26 cd d6 02"
try:
encrypted_text = encrypt_data(text, key)
self.assertTrue(
False,
"Exception thrown expected because because key is invalid")
except Exception as exc:
self.assertEqual(str(exc),
"key length is not 256 bit (64 hex characters)")
try:
decrypted_text = decrypt_data(
"df78cedebfb70624bb663a2eb8ba828adf78cedebfb70624bb663a2eb8ba828a9B354CBA0AB6F7763A88F04D83A237462BE1B68F8D2CFCD4FCDC575BE51333B431A45E86D9B9B5D21DBD91C044DA118C783D98F89E28911463F957E2DA3188F1F4615F637B7F089F6DAEFA352EABAF1AC6731C6E67D0644E36E8D5AFACEF3DC7795E0909DFF84CF92E773A0268F9C2F01232F0AC7E1152F4E3802B09CAC5B0E421DDEC41F55A56001E1D37CBD9E2865567E1ABF5A2B2F160EAD80B147F0A1A9B",
key)
self.assertTrue(
False,
"Exception thrown expected because because key is invalid")
except Exception as exc:
self.assertEqual(str(exc),
"key length is not 256 bit (64 hex characters)")
def recv_data_encrypted(usocket, session_key):
enc_data = recv_data(usocket)
data = decrypt_data(enc_data, session_key)
return data