1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
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
|
