0a18d8bbdf
This is the suggested location for tests and is adopted by most projects. As part of this change relative imports to package imports. Fix all the test running and coverage code to point to the new location. Change-Id: I01264aed14f396ab9a7242e3e72b71e1bc332675
89 lines
3.7 KiB
Python
89 lines
3.7 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import testtools
|
|
|
|
from keystoneclient.middleware import memcache_crypt
|
|
|
|
|
|
class MemcacheCryptPositiveTests(testtools.TestCase):
|
|
def _setup_keys(self, strategy):
|
|
return memcache_crypt.derive_keys('token', 'secret', strategy)
|
|
|
|
def test_constant_time_compare(self):
|
|
# make sure it works as a compare, the "constant time" aspect
|
|
# isn't appropriate to test in unittests
|
|
ctc = memcache_crypt.constant_time_compare
|
|
self.assertTrue(ctc('abcd', 'abcd'))
|
|
self.assertTrue(ctc('', ''))
|
|
self.assertFalse(ctc('abcd', 'efgh'))
|
|
self.assertFalse(ctc('abc', 'abcd'))
|
|
self.assertFalse(ctc('abc', 'abc\x00'))
|
|
self.assertFalse(ctc('', 'abc'))
|
|
|
|
def test_derive_keys(self):
|
|
keys = memcache_crypt.derive_keys('token', 'secret', 'strategy')
|
|
self.assertEqual(len(keys['ENCRYPTION']),
|
|
len(keys['CACHE_KEY']))
|
|
self.assertEqual(len(keys['CACHE_KEY']),
|
|
len(keys['MAC']))
|
|
self.assertNotEqual(keys['ENCRYPTION'],
|
|
keys['MAC'])
|
|
self.assertIn('strategy', keys.keys())
|
|
|
|
def test_key_strategy_diff(self):
|
|
k1 = self._setup_keys('MAC')
|
|
k2 = self._setup_keys('ENCRYPT')
|
|
self.assertNotEqual(k1, k2)
|
|
|
|
def test_sign_data(self):
|
|
keys = self._setup_keys('MAC')
|
|
sig = memcache_crypt.sign_data(keys['MAC'], 'data')
|
|
self.assertEqual(len(sig), memcache_crypt.DIGEST_LENGTH_B64)
|
|
|
|
def test_encryption(self):
|
|
keys = self._setup_keys('ENCRYPT')
|
|
# what you put in is what you get out
|
|
for data in ['data', '1234567890123456', '\x00\xFF' * 13
|
|
] + [chr(x % 256) * x for x in range(768)]:
|
|
crypt = memcache_crypt.encrypt_data(keys['ENCRYPTION'], data)
|
|
decrypt = memcache_crypt.decrypt_data(keys['ENCRYPTION'], crypt)
|
|
self.assertEqual(data, decrypt)
|
|
self.assertRaises(memcache_crypt.DecryptError,
|
|
memcache_crypt.decrypt_data,
|
|
keys['ENCRYPTION'], crypt[:-1])
|
|
|
|
def test_protect_wrappers(self):
|
|
data = 'My Pretty Little Data'
|
|
for strategy in ['MAC', 'ENCRYPT']:
|
|
keys = self._setup_keys(strategy)
|
|
protected = memcache_crypt.protect_data(keys, data)
|
|
self.assertNotEqual(protected, data)
|
|
if strategy == 'ENCRYPT':
|
|
self.assertNotIn(data, protected)
|
|
unprotected = memcache_crypt.unprotect_data(keys, protected)
|
|
self.assertEqual(data, unprotected)
|
|
self.assertRaises(memcache_crypt.InvalidMacError,
|
|
memcache_crypt.unprotect_data,
|
|
keys, protected[:-1])
|
|
self.assertIsNone(memcache_crypt.unprotect_data(keys, None))
|
|
|
|
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
|