def test_create_rsa_signature(self):
global private_rsa
global public_rsa
data = 'The quick brown fox jumps over the lazy dog'.encode('utf-8')
signature, method = crypto_keys.create_rsa_signature(private_rsa, data)
# Verify format of returned values.
self.assertNotEqual(None, signature)
self.assertEqual(None, tuf.formats.NAME_SCHEMA.check_match(method),
FORMAT_ERROR_MSG)
self.assertEqual('RSASSA-PSS', method)
# Check for improperly formatted arguments.
self.assertRaises(tuf.FormatError,
crypto_keys.create_rsa_signature, 123, data)
self.assertRaises(ValueError,
crypto_keys.create_rsa_signature, '', data)
# Check for invalid 'data'.
self.assertRaises(tuf.FormatError,
crypto_keys.create_rsa_signature, private_rsa, '')
self.assertRaises(tuf.FormatError,
crypto_keys.create_rsa_signature, private_rsa, 123)
# Check for missing private key.
self.assertRaises(tuf.CryptoError,
crypto_keys.create_rsa_signature, public_rsa, data)
def test_create_rsa_signature(self):
global private_rsa
global public_rsa
data = 'The quick brown fox jumps over the lazy dog'.encode('utf-8')
signature, method = crypto_keys.create_rsa_signature(private_rsa, data)
# Verify format of returned values.
self.assertNotEqual(None, signature)
self.assertEqual(None, tuf.formats.NAME_SCHEMA.check_match(method),
FORMAT_ERROR_MSG)
self.assertEqual('RSASSA-PSS', method)
# Check for improperly formatted arguments.
self.assertRaises(tuf.FormatError,
crypto_keys.create_rsa_signature, 123, data)
self.assertRaises(ValueError,
crypto_keys.create_rsa_signature, '', data)
# Check for invalid 'data'.
self.assertRaises(tuf.FormatError,
crypto_keys.create_rsa_signature, private_rsa, '')
self.assertRaises(tuf.FormatError,
crypto_keys.create_rsa_signature, private_rsa, 123)
# Check for missing private key.
self.assertRaises(tuf.CryptoError,
crypto_keys.create_rsa_signature, public_rsa, data)
def test_verify_rsa_signature(self):
global public_rsa
global private_rsa
data = 'The quick brown fox jumps over the lazy dog'.encode('utf-8')
signature, method = crypto_keys.create_rsa_signature(private_rsa, data)
valid_signature = crypto_keys.verify_rsa_signature(signature, method, public_rsa,
data)
self.assertEqual(True, valid_signature)
# Check for improperly formatted arguments.
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature, signature,
123, public_rsa, data)
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature, signature,
method, 123, data)
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature, 123, method,
public_rsa, data)
self.assertRaises(tuf.UnknownMethodError, crypto_keys.verify_rsa_signature,
signature,
'invalid_method',
public_rsa, data)
# Check for invalid 'signature', 'public_key', and 'data' arguments.
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature,
signature, method, public_rsa, 123)
self.assertRaises(tuf.CryptoError, crypto_keys.verify_rsa_signature,
signature, method, 'bad_key', data)
self.assertEqual(False, crypto_keys.verify_rsa_signature(signature, method,
public_rsa, b'mismatched data'))
mismatched_signature, method = crypto_keys.create_rsa_signature(private_rsa,
b'mismatched data')
self.assertEqual(False, crypto_keys.verify_rsa_signature(mismatched_signature,
method, public_rsa, data))
def test_verify_rsa_signature(self):
global public_rsa
global private_rsa
data = 'The quick brown fox jumps over the lazy dog'.encode('utf-8')
signature, method = crypto_keys.create_rsa_signature(private_rsa, data)
valid_signature = crypto_keys.verify_rsa_signature(signature, method, public_rsa,
data)
self.assertEqual(True, valid_signature)
# Check for improperly formatted arguments.
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature, signature,
123, public_rsa, data)
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature, signature,
method, 123, data)
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature, 123, method,
public_rsa, data)
self.assertRaises(tuf.UnknownMethodError, crypto_keys.verify_rsa_signature,
signature,
'invalid_method',
public_rsa, data)
# Check for invalid 'signature' and 'data' arguments.
self.assertRaises(tuf.FormatError, crypto_keys.verify_rsa_signature,
signature, method, public_rsa, 123)
self.assertEqual(False, crypto_keys.verify_rsa_signature(signature, method,
public_rsa, b'mismatched data'))
mismatched_signature, method = crypto_keys.create_rsa_signature(private_rsa,
b'mismatched data')
self.assertEqual(False, crypto_keys.verify_rsa_signature(mismatched_signature,
method, public_rsa, data))
