single workspace role for task order officers

2019-01-08 14:36:32 -05:00
parent d41f233f5a
commit 95e7adfcf3
6 changed files with 41 additions and 36 deletions
--- a/tests/domain/test_task_orders.py
+++ b/tests/domain/test_task_orders.py
@@ -48,3 +48,15 @@ def test_add_officer_with_nonexistent_role():
    owner = task_order.workspace.owner
    with pytest.raises(TaskOrderError):
        TaskOrders.add_officer(owner, task_order, "pilot", ko.to_dictionary())
+
+
+def test_add_officer_who_is_already_workspace_member():
+    task_order = TaskOrderFactory.create()
+    owner = task_order.workspace.owner
+    TaskOrders.add_officer(
+        owner, task_order, "contracting_officer", owner.to_dictionary()
+    )
+
+    assert task_order.contracting_officer == owner
+    member = task_order.workspace.members[0]
+    assert member.user == owner and member.role_name == "owner"