57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
import testtools
|
|
|
|
from keystoneclient.middleware import memcache_crypt
|
|
|
|
|
|
class MemcacheCryptPositiveTests(testtools.TestCase):
|
|
def test_generate_aes_key(self):
|
|
self.assertEqual(
|
|
len(memcache_crypt.generate_aes_key('Gimme Da Key', 'hush')), 32)
|
|
|
|
def test_compute_mac(self):
|
|
self.assertEqual(
|
|
memcache_crypt.compute_mac('mykey', 'This is a test!'),
|
|
'tREu41yR5tEgeBWIuv9ag4AeKA8=')
|
|
|
|
def test_sign_data(self):
|
|
expected = '{MAC:SHA1}eyJtYWMiOiAiM0FrQmdPZHRybGo1RFFESHA1eUxqcDVq' +\
|
|
'Si9BPSIsICJzZXJpYWxpemVkX2RhdGEiOiAiXCJUaGlzIGlzIGEgdG' +\
|
|
'VzdCFcIiJ9'
|
|
self.assertEqual(
|
|
memcache_crypt.sign_data('mykey', 'This is a test!'),
|
|
expected)
|
|
|
|
def test_verify_signed_data(self):
|
|
signed = memcache_crypt.sign_data('mykey', 'Testz')
|
|
self.assertEqual(
|
|
memcache_crypt.verify_signed_data('mykey', signed),
|
|
'Testz')
|
|
self.assertEqual(
|
|
memcache_crypt.verify_signed_data('aasSFWE13WER', 'not MACed'),
|
|
'not MACed')
|
|
|
|
def test_encrypt_data(self):
|
|
expected = '{ENCRYPT:AES256}'
|
|
self.assertEqual(
|
|
memcache_crypt.encrypt_data('mykey', 'mysecret',
|
|
'This is a test!')[:16],
|
|
expected)
|
|
|
|
def test_decrypt_data(self):
|
|
encrypted = memcache_crypt.encrypt_data('mykey', 'mysecret', 'Testz')
|
|
self.assertEqual(
|
|
memcache_crypt.decrypt_data('mykey', 'mysecret', encrypted),
|
|
'Testz')
|
|
self.assertEqual(
|
|
memcache_crypt.decrypt_data('mykey', 'mysecret',
|
|
'Not Encrypted!'),
|
|
'Not Encrypted!')
|
|
|
|
def test_no_pycrypt(self):
|
|
aes = memcache_crypt.AES
|
|
memcache_crypt.AES = None
|
|
self.assertRaises(memcache_crypt.CryptoUnavailableError,
|
|
memcache_crypt.encrypt_data, 'token', 'secret',
|
|
'data')
|
|
memcache_crypt.AES = aes
|