diff --git a/atst/routes/task_orders/invite.py b/atst/routes/task_orders/invite.py index c43f14b5..78e2daf5 100644 --- a/atst/routes/task_orders/invite.py +++ b/atst/routes/task_orders/invite.py @@ -1,14 +1,13 @@ -from flask import redirect, url_for +from flask import redirect, url_for, g from . import task_orders_bp from atst.domain.task_orders import TaskOrders from atst.utils.flash import formatted_flash as flash -# TODO: add a real implementation for this @task_orders_bp.route("/task_orders/invite/", methods=["POST"]) def invite(task_order_id): - task_order = TaskOrders.get(task_order_id) + task_order = TaskOrders.get(g.current_user, task_order_id) flash("task_order_submitted", task_order=task_order) return redirect( url_for("workspaces.workspace_members", workspace_id=task_order.workspace.id) diff --git a/tests/routes/task_orders/test_invite.py b/tests/routes/task_orders/test_invite.py new file mode 100644 index 00000000..39cfeb76 --- /dev/null +++ b/tests/routes/task_orders/test_invite.py @@ -0,0 +1,11 @@ +import pytest +from flask import url_for + +from tests.factories import TaskOrderFactory + + +def test_invite(client): + to = TaskOrderFactory.create() + response = client.post( + url_for("task_orders.invite", task_order_id=to.id), follow_redirects=False + )