def test_getIP_cached():
bc = BlessCache(None, None, BlessCache.CACHEMODE_ENABLED)
bc.cache = {}
bc.set('lastip', '1.1.1.1')
bc.set('lastipchecktime', time.time())
user_ip = UserIP(bc, 10, IP_URLS)
assert user_ip.getIP() == '1.1.1.1'
def test_get_kmsauth_token_cached():
kmsconfig = {'awsregion': 'us-east-1', 'context': {}, 'kmskey': None}
expiration = datetime.datetime.utcnow() + datetime.timedelta(minutes=60)
kmsauth_cache = {
'token': 'KMSTOKEN',
'Expiration': expiration.strftime('%Y%m%dT%H%M%SZ')
}
bless_cache = BlessCache(None, None, BlessCache.CACHEMODE_ENABLED)
bless_cache.cache = {}
bless_cache.set('kmsauth-us-east-1', kmsauth_cache)
token = client.get_kmsauth_token(None, kmsconfig, 'foouser', bless_cache)
assert token == 'KMSTOKEN'
def test_set():
bc = BlessCache(None, None, BlessCache.CACHEMODE_DISABLED)
bc.cache = {}
bc.set('foo', 'bar')
assert bc.cache == {'foo': 'bar'}
assert bc.dirty == True