def test_set_no_expiration(self):
# create a random value to put into registry
test_value = "test-value-" + str(random.randint(1, 1000))
test_key = "test-key-" + str(random.randint(1, 1000))
set_cache_value_ms(test_key, test_value)
# pull value back out
returned_value = get_cache_value(test_key)
# test that value comes back
self.assertTrue(test_value == returned_value)
def load_cached_vault_path(self):
vault_path = get_cache_value('keepass-vault-path')
if not vault_path:
vault_path = string_validation(
"KeePass Vault Path",
"Path to the KeePass vault file for secrets lookup", [],
path_validation, True)
set_cache_value_ms('keepass-vault-path', vault_path)
return vault_path
def load_cached_vault_pwd(self):
vault_pwd = get_cache_value('keepass-vault-pwd')
if not vault_pwd:
vault_pwd = string_validation(
"KeePass Vault Password",
"The master key for the KeePass vault", [], string_validation,
True, 100, True)
set_cache_value_ms('keepass-vault-pwd', vault_pwd)
return vault_pwd
def test_set_unexpired_dt(self):
# create a random value to put into registry
test_value = "test-value-" + str(random.randint(1, 1000))
test_key = "test-key-" + str(random.randint(1, 1000))
# set expiration to current time plus a few seconds
expiration_dt = datetime.datetime.now() + datetime.timedelta(seconds=5)
set_cache_value_dt(test_key, test_value, expiration_dt)
# pull value back out
returned_value = get_cache_value(test_key)
# test that value comes back
self.assertTrue(test_value == returned_value)
def test_set_unexpired_ms(self):
# create a random value to put into registry
test_value = "test-value-" + str(random.randint(1, 1000))
test_key = "test-key-" + str(random.randint(1, 1000))
# set expiration to current time plus a few seconds
expiration_ms = int(time.time() * 1000) + 5000
set_cache_value_ms(test_key, test_value, expiration_ms)
# pull value back out
returned_value = get_cache_value(test_key)
# test that value comes back
self.assertTrue(test_value == returned_value)
def test_set_expired_dt(self):
# create a random value to put into registry
test_value = "test-value-" + str(random.randint(1, 1000))
test_key = "test-key-" + str(random.randint(1, 1000))
# set expiration to current time minus a day
expiration_dt = datetime.datetime.now() - datetime.timedelta(days=1)
set_cache_value_dt(test_key, test_value, expiration_dt)
# pull value back out
returned_value = get_cache_value(test_key)
# test that a null value comes back
self.assertTrue(not returned_value)