def test_encrypt_dict_keep_type():
encrypted_dict = security.encrypt({'token': 123123123},
secure_keys=['token'])
assert encrypted_dict['token']
decrypted_dict = security.decrypt(encrypted_dict, secure_keys=['token'])
assert decrypted_dict['token'] == 123123123
def test_encrypt_dict_inner_list():
encrypted_dict = security.encrypt({'tokens': [1, 2, 3]},
secure_keys=['tokens'])
assert security.is_encrypted(encrypted_dict['tokens'][0])
assert security.is_encrypted(encrypted_dict['tokens'][1])
assert security.is_encrypted(encrypted_dict['tokens'][2])
decrypted_dict = security.decrypt(encrypted_dict, secure_keys=['tokens'])
assert decrypted_dict['tokens'] == [1, 2, 3]
def test_encrypt_dict_nested():
encrypted_dict = security.encrypt(
{'db': {
'user': '******',
'pass': '******',
}}, secure_keys=['db.pass'])
assert encrypted_dict['db']['user'] == 'admin'
assert security.is_encrypted(encrypted_dict['db']['pass'])
decrypted_dict = security.decrypt(encrypted_dict, secure_keys=['db.pass'])
assert decrypted_dict['db']['user'] == 'admin'
assert decrypted_dict['db']['pass'] == 'secret'
def update_settings(environment: Environment,
component: Component,
data: dict,
validate: bool = True):
"""
Update settings.
"""
if validate:
validate_settings(
environment=environment,
component=component,
data=data,
)
if settings.ENCRYPT_ENABLED:
data = security.encrypt(data, secure_keys=settings.SECURE_KEYS)
configstore.update_data(environment=environment.alias,
component=component.alias,
data=data)