def test_encrypt_RSA(self):
from launchkey import encrypt_RSA
private_key, public_key = generate_RSA()
encrypted = encrypt_RSA(public_key, "#" * 214)
assert len(encrypted) > 300
assert "#" * 214 not in encrypted
self.assertRaises(ValueError, encrypt_RSA, public_key, "#" * 215)
def _post_ping(result, my_data, signature):
to_encrypt = {"secret": self.app_secret, "stamped": str(result['launchkey_time'])}
encrypted_secret = launchkey.encrypt_RSA(result['key'], str(to_encrypt))
my_data['secret_key'] = encrypted_secret
if signature:
signature = launchkey.sign_data(self.private_key, encrypted_secret)
my_data['signature'] = signature
my_data['app_key'] = self.app_key
return my_data
def test_sign_data(self):
from launchkey import encrypt_RSA, decrypt_RSA, sign_data
from Crypto.PublicKey import RSA
private_key, public_key = generate_RSA()
encrypted = encrypt_RSA(public_key, "test message")
private_key2, public_key2 = generate_RSA()
signature = sign_data(private_key2, encrypted)
self.assertTrue(verify_sign(public_key2, signature, encrypted))
self.assertFalse(verify_sign(public_key, signature, encrypted))
o_signature = sign_data(private_key, encrypted)
self.assertFalse(verify_sign(public_key2, o_signature, encrypted))
self.assertTrue(verify_sign(public_key, o_signature, encrypted))
def test_decrypt_RSA(self):
from launchkey import encrypt_RSA, decrypt_RSA
private_key, public_key = generate_RSA()
encrypted = encrypt_RSA(public_key, "test message " * 5)
decrypted = decrypt_RSA(private_key, encrypted)
self.assertEquals(decrypted, "test message " * 5)
