def get_logged_in_user():
public_id = get_logged_in_user.public_id
user = User.find_by_public_id(public_id)
expires_at = get_logged_in_user.expires_at
user.token_expires_in = format_timespan_digits(
remaining_fromtimestamp(expires_at))
return user
def test_login(client, db):
register_user(client)
response = login_user(client)
assert response.status_code == HTTPStatus.OK
assert "status" in response.json and response.json["status"] == "success"
assert "message" in response.json and response.json["message"] == SUCCESS
assert "access_token" in response.json
access_token = response.json["access_token"]
result = User.decode_access_token(access_token)
assert result.success
user_dict = result.value
assert not user_dict["admin"]
user = User.find_by_public_id(user_dict["public_id"])
assert user and user.email == EMAIL
def create_widget(widget_dict):
name = widget_dict["name"]
if Widget.find_by_name(name):
error = f"Widget name: {name} already exists, must be unique."
abort(HTTPStatus.CONFLICT, error, status="fail")
widget = Widget(**widget_dict)
owner = User.find_by_public_id(create_widget.public_id)
widget.owner_id = owner.id
db.session.add(widget)
db.session.commit()
response = jsonify(status="success", message=f"New widget added: {name}.")
response.status_code = HTTPStatus.CREATED
response.headers["Location"] = url_for("api.widget", name=name)
return response
def test_auth_register(client, db):
response = register_user(client)
assert response.status_code == HTTPStatus.CREATED
assert "status" in response.json and response.json["status"] == "success"
assert "message" in response.json and response.json["message"] == SUCCESS
assert "token_type" in response.json and response.json["token_type"] == "bearer"
assert "expires_in" in response.json and response.json["expires_in"] == 5
assert "access_token" in response.json
access_token = response.json["access_token"]
result = User.decode_access_token(access_token)
assert result.success
user_dict = result.value
assert not user_dict["admin"]
user = User.find_by_public_id(user_dict["public_id"])
assert user and user.email == EMAIL
