def test_a_string_encrypted_with_a_custom_key_can_be_decrypted_again(
random_sovereign_key, random_string):
encrypted_secret = cipher_suite.encrypt(random_string,
key=random_sovereign_key)
decrypted_string = cipher_suite.decrypt(encrypted_secret,
key=random_sovereign_key)
assert decrypted_string == random_string
def validate_authentication_string(s: str) -> bool:
try:
password = cipher_suite.decrypt(s)
except Exception:
stats.increment("discovery.auth.failed")
raise
if password in config.passwords:
stats.increment("discovery.auth.success")
return True
stats.increment("discovery.auth.failed")
return False
def test_an_encrypted_string_can_be_decrypted_again(random_string):
encrypted_secret = cipher_suite.encrypt(random_string)
decrypted_string = cipher_suite.decrypt(encrypted_secret)
assert decrypted_string == random_string
def test_decrypting_with_the_wrong_key_raises_an_exception(
auth_string, random_sovereign_key):
with pytest.raises(HTTPException) as e:
cipher_suite.decrypt(auth_string, random_sovereign_key)
assert e.status_code == 400
assert e.detail == "Decryption failed"
async def _decryptable(request: DecryptableRequest = Body(
None)) -> JSONResponse:
cipher_suite.decrypt(request.data)
return json_response_class({})
async def _decrypt(request: DecryptionRequest = Body(None)) -> Dict[str, str]:
return {"result": cipher_suite.decrypt(request.data, request.key)}