def test_clear_keys() -> None:
"""Test clearance of specified keys from credential store."""
credential_store = CredentialStore()
test_key = "test"
test_permanent_key = "locale"
# Try to get value from empty store
assert test_key not in credential_store
assert test_permanent_key not in credential_store
# Set value
test_value = Credential("test_value")
test_pemanent_value = Credential("test_locale")
credential_store[test_key] = test_value
credential_store[test_permanent_key] = test_pemanent_value
# Try to get values from filled store
assert test_key in credential_store
assert credential_store[test_key] == test_value
assert test_permanent_key in credential_store
assert credential_store[test_permanent_key] == test_pemanent_value
# Clear the store
credential_store.clear_keys([test_permanent_key])
# Try to get values from filled store
assert test_key in credential_store
assert test_permanent_key not in credential_store
def get_logged_in_credential_store() -> CredentialStore:
"""Get credential store initialised with Gigya credentials."""
credential_store = CredentialStore()
credential_store[GIGYA_LOGIN_TOKEN] = Credential(TEST_LOGIN_TOKEN)
credential_store[GIGYA_PERSON_ID] = Credential(TEST_PERSON_ID)
credential_store[GIGYA_JWT] = JWTCredential(get_jwt())
return credential_store
def test_jwt_credential() -> None:
"""Test get/set with jwt credential."""
credential_store = CredentialStore()
test_key = "test"
# Try to get value from empty store
assert test_key not in credential_store
assert not credential_store.get(test_key)
with pytest.raises(KeyError):
credential_store[test_key]
# Set value
test_value = JWTCredential(get_jwt())
credential_store[test_key] = test_value
# Try to get values from filled store
assert test_key in credential_store
assert credential_store.get(test_key) == test_value
assert credential_store.get_value(test_key) == test_value.value
assert credential_store[test_key] == test_value
# Try again with expired
expired_time = time.time() + 3600
with mock.patch("time.time", mock.MagicMock(return_value=expired_time)):
assert test_key not in credential_store
assert credential_store.get_value(test_key) is None
assert not credential_store.get(test_key)
with pytest.raises(KeyError):
credential_store[test_key]
def test_invalid_credential() -> None:
"""Test set with invalid types."""
credential_store = CredentialStore()
test_key = "test"
with pytest.raises(TypeError):
credential_store[test_key] = test_key # type:ignore
test_value = Credential("test_value")
with pytest.raises(TypeError):
credential_store[test_value] = test_value # type:ignore
def test_simple_credential() -> None:
"""Test get/set with simple credential."""
credential_store = CredentialStore()
test_key = "test"
# Try to get value from empty store
assert test_key not in credential_store
assert not credential_store.get(test_key)
with pytest.raises(KeyError):
credential_store[test_key]
# Set value
test_value = Credential("test_value")
credential_store[test_key] = test_value
# Try to get values from filled store
assert test_key in credential_store
assert credential_store.get(test_key) == test_value
assert credential_store[test_key] == test_value
# Delete value
del credential_store[test_key]
# Try to get values from filled store
assert test_key not in credential_store
assert credential_store.get(test_key) is None
assert credential_store.get_value(test_key) is None