Update MockKeyManager to use given algorithm
MockKeyManager will now use the given algorithm when creating a key.
This is beneficial for testing.
Change-Id: I878efa01363f00a5b50fd55f7b944b0b1263d10a
Kaitlin Farr
8 years ago
64 | 64 |
|
65 | 65 |
def _generate_key(self, **kwargs):
|
66 | 66 |
name = kwargs.get('name', None)
|
|
67 |
algorithm = kwargs.get('algorithm', 'AES')
|
67 | 68 |
key_length = kwargs.get('key_length', 256)
|
68 | 69 |
_hex = self._generate_hex_key(key_length)
|
69 | 70 |
return sym_key.SymmetricKey(
|
70 | |
'AES',
|
|
71 |
algorithm,
|
71 | 72 |
key_length,
|
72 | 73 |
bytes(binascii.unhexlify(_hex)),
|
73 | 74 |
name)
|
70 | 70 |
key_id = self.key_mgr.create_key(self.context, name=name)
|
71 | 71 |
key = self.key_mgr.get(self.context, key_id)
|
72 | 72 |
self.assertEqual(name, key.name)
|
|
73 |
|
|
74 |
def test_create_key_with_algorithm(self):
|
|
75 |
algorithm = 'DES'
|
|
76 |
key_id = self.key_mgr.create_key(self.context, algorithm=algorithm)
|
|
77 |
key = self.key_mgr.get(self.context, key_id)
|
|
78 |
self.assertEqual(algorithm, key.algorithm)
|
73 | 79 |
|
74 | 80 |
def test_create_key_null_context(self):
|
75 | 81 |
self.assertRaises(exception.Forbidden,
|