Added test that the token can be signed and retrieved

This commit is contained in:
Rebecca Law
2016-01-13 14:34:45 +00:00
parent 725b976d31
commit a6dda26ba1
2 changed files with 12 additions and 2 deletions

View File

@@ -113,8 +113,9 @@ def delete_token(service_id):
return jsonify(result="error", message="Token not found"), 404
def _generate_token():
token = uuid.uuid4()
def _generate_token(token=None):
if not token:
token = uuid.uuid4()
serializer = URLSafeSerializer(current_app.config.get('SECRET_KEY'))
return serializer.dumps(str(token), current_app.config.get('DANGEROUS_SALT'))