def test_incorrect_key_decrypt():
token = b'some-token'
key = b'incorrect-key'
with pytest.raises(ValueError) as exc:
decrypt_string(token, key=key)
assert 'Wrong secret key' in str(exc)
def test_decrypt_string_transforms_to_bytes():
value = b'some-text'
key = settings.GTIO_SECRET_KEY
f = Fernet(key)
token = f.encrypt(value)
our_value = decrypt_string(token.decode('utf-8'))
assert our_value == value.decode('utf-8')
def test_decrypt_string_uses_key_in_settings_by_default():
value = b'some-text'
key = settings.GTIO_SECRET_KEY
f = Fernet(key)
token = f.encrypt(value)
our_value = decrypt_string(token)
assert our_value == value.decode('utf-8')
def _get_encrypted(self, key):
token = self.conf.get(key)
if token:
try:
token = decrypt_string(token)
except cryptography.fernet.InvalidToken:
# This happens if someone sets the key directly
# on the conf property (ie it should never happen)
token = None
return token
def get_token(self):
token = self.conf.get('github_oauth_token')
if token:
try:
token = decrypt_string(token)
except cryptography.fernet.InvalidToken:
# This happens if someone sets the github_oauth_token directly
# on the conf property (ie it should never happen)
token = None
return token
def test_decrypt_string():
value = b'some-text'
key = Fernet.generate_key()
f = Fernet(key)
token = f.encrypt(value)
our_value = decrypt_string(token, key=key)
assert isinstance(our_value, str)
assert our_value == value.decode('utf-8')
