wip

tests/domain/test_invitations.py

@@ -7,6 +7,7 @@ from atst.domain.invitations import (
     InvitationError,
     WrongUserError,
     ExpiredError,
+    NotFoundError,
 )
 from atst.models.invitation import Status
 
@@ -144,3 +145,15 @@ def test_audit_event_for_accepted_invite():
     accepted_event = AuditLog.get_by_resource(invite.id)[0]
     assert "email" in accepted_event.event_details
     assert "dod_id" in accepted_event.event_details
+
+
+def test_lookup_by_user_and_portfolio():
+    portfolio = PortfolioFactory.create()
+    user = UserFactory.create()
+    ws_role = PortfolioRoleFactory.create(user=user, portfolio=portfolio)
+    invite = Invitations.create(portfolio.owner, ws_role, user.email)
+
+    assert Invitations.lookup_by_portfolio_and_user(portfolio, user) == invite
+
+    with pytest.raises(NotFoundError):
+        Invitations.lookup_by_portfolio_and_user(portfolio, UserFactory.create())

tests/domain/test_task_orders.py

@@ -1,6 +1,11 @@
 import pytest
 
-from atst.domain.task_orders import TaskOrders, TaskOrderError, DD254s
+from atst.domain.task_orders import (
+    TaskOrders,
+    TaskOrderError,
+    InvalidOfficerError,
+    DD254s,
+)
 from atst.domain.exceptions import UnauthorizedError
 from atst.models.attachment import Attachment
 
@@ -127,6 +132,26 @@ def test_task_order_access():
     )
 
 
+def test_remove_valid_officer():
+    task_order = TaskOrderFactory.create()
+    owner = task_order.portfolio.owner
+    TaskOrders.add_officer(
+        owner, task_order, "contracting_officer", owner.to_dictionary()
+    )
+
+    assert task_order.contracting_officer == owner
+
+    TaskOrders.remove_officer(task_order, "contracting_officer")
+    assert task_order.contracting_officer is None
+
+
+def test_remove_invalid_officer():
+    task_order = TaskOrderFactory.create()
+
+    with pytest.raises(InvalidOfficerError):
+        TaskOrders.remove_officer(task_order, "invalid_officer_type")
+
+
 def test_dd254_complete():
     finished = DD254Factory.create()
     unfinished = DD254Factory.create(certifying_official=None)