def test_rotate_key(module_client, function_empty_db):
# Arrange
key = create_fake_key(function_empty_db.session)
function_empty_db.session.add(key)
function_empty_db.session.commit()
# Act
rotate_key(key, function_empty_db.session)
# Assert
assert key.apikey != FAKE_APIKEY
def rotate_apikey():
new_key = rotate_key(g.auth_key, db.session)
if not new_key:
return utils.standardize_response(status_code=500)
return utils.standardize_response(
payload=dict(data=new_key.serialize),
datatype="credentials")
def rotate(apikey_or_email):
key = find_key_by_apikey_or_email(apikey_or_email)
if not key:
print('Could not find apikey or email.')
return 1
key = rotate_key(key, db.session)
if not key:
print('Error trying to rotate key.')
return -1
print(f'Rotated {key}')
