def test_get_by_email(client):
""" Get user by email ."""
user = User('*****@*****.**', 'foo')
user.save()
retrieved = User.get_by_email(user.email)
assert retrieved == user
def register_url():
logging.info('Processing shorten request')
url = request.json.get('url')
current_user = User.get_by_email(get_jwt_identity())
if not current_user:
abort(HTTPStatus.BAD_REQUEST, INVALID_ACCOUNT)
# create short url
new_short_url = Url(user=current_user,
long_url=url,
short_id=Url.create_id())
new_short_url.save()
response = create_response({'short_url': new_short_url.get_short_url()},
SHORT_URL_CREATED, HTTPStatus.CREATED)
return response
