feat: new API route to apply status changes after RSVP deadline

apps/api/src/routers/admin.py

@@ -1,7 +1,7 @@
 import asyncio
 from datetime import datetime
 from logging import getLogger
-from typing import Any, Optional
+from typing import Any, Optional, Sequence
 
 from fastapi import APIRouter, Body, Depends, HTTPException, status
 from pydantic import BaseModel, EmailStr, Field, TypeAdapter, ValidationError
@@ -127,7 +127,6 @@ async def release_decisions() -> None:
         group = [record for record in records if record["decision"] == decision]
         if not group:
             continue
-
         await asyncio.gather(
             *(_process_batch(batch, decision) for batch in batched(group, 100))
         )
@@ -166,6 +165,49 @@ async def rsvp_reminder() -> None:
     )
 
 
+@router.post(
+    "/confirm-attendance", dependencies=[Depends(require_role([Role.DIRECTOR]))]
+)
+async def confirm_attendance() -> None:
+    """Update applicant status to void or attending based on their current status."""
+    records = await mongodb_handler.retrieve(
+        Collection.USERS,
+        {"role": Role.APPLICANT},
+        ["_id", "status"],
+    )
+
+    statuses = {
+        Status.CONFIRMED: Status.ATTENDING,
+        Decision.ACCEPTED: Status.VOID,
+        Status.WAIVER_SIGNED: Status.VOID,
+    }
+
+    for status_from, status_to in statuses.items():
+        current_record = [
+            record for record in records if record["status"] == status_from
+        ]
+
+        for record in current_record:
+            record["status"] = status_to
+
+        await asyncio.gather(
+            *(
+                _process_status(batch, status_to)
+                for batch in batched(
+                    [str(record["_id"]) for record in current_record], 100
+                )
+            )
+        )
+
+
+async def _process_status(uids: Sequence[str], status: Status) -> None:
+    ok = await mongodb_handler.update(
+        Collection.USERS, {"_id": {"$in": uids}}, {"status": status}
+    )
+    if not ok:
+        raise RuntimeError("gg wp")
+
+
 async def _process_batch(batch: tuple[dict[str, Any], ...], decision: Decision) -> None:
     uids: list[str] = [record["_id"] for record in batch]
     log.info(f"Setting {','.join(uids)} as {decision}")

apps/api/tests/test_admin.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from unittest.mock import ANY, AsyncMock, patch
+from unittest.mock import ANY, AsyncMock, call, patch
 
 from fastapi import FastAPI
 
@@ -8,6 +8,7 @@
 from models.ApplicationData import Decision
 from routers import admin
 from services.mongodb_handler import Collection
+from utils.user_record import Status
 
 user_identity.JWT_SECRET = "not a good idea"
 
@@ -30,11 +31,22 @@
     "role": "reviewer",
 }
 
+USER_DIRECTOR = NativeUser(
+    ucinetid="dir",
+    display_name="Dir",
+    email="[email protected]",
+    affiliations=["student"],
+)
+
+DIRECTOR_IDENTITY = {"_id": "edu.uci.dir", "role": "director", "status": "CONFIRMED"}
+
 app = FastAPI()
 app.include_router(admin.router)
 
 reviewer_client = UserTestClient(USER_REVIEWER, app)
 
+director_client = UserTestClient(USER_DIRECTOR, app)
+
 
 @patch("services.mongodb_handler.retrieve_one", autospec=True)
 def test_restricted_admin_route_is_forbidden(
@@ -155,3 +167,66 @@ def test_can_submit_review(
         },
     )
     assert res.status_code == 200
+
+
+@patch("services.mongodb_handler.update", autospec=True)
+@patch("services.mongodb_handler.retrieve_one", autospec=True)
+@patch("services.mongodb_handler.retrieve", autospec=True)
+def test_confirm_attendance_route(
+    mock_mongodb_handler_retrieve: AsyncMock,
+    mock_mongodb_handler_retrieve_one: AsyncMock,
+    mock_mognodb_handler_update: AsyncMock,
+) -> None:
+    """Test that confirmed status changes to void with accepted."""
+
+    mock_mongodb_handler_retrieve_one.return_value = DIRECTOR_IDENTITY
+    mock_mongodb_handler_retrieve.return_value = [
+        {
+            "_id": "edu.uc.tester",
+            "role": "applicant",
+            "status": Decision.ACCEPTED,
+        },
+        {
+            "_id": "edu.uc.tester2",
+            "role": "applicant",
+            "status": Status.WAIVER_SIGNED,
+        },
+        {
+            "_id": "edu.uc.tester3",
+            "role": "applicant",
+            "status": Status.CONFIRMED,
+        },
+        {
+            "_id": "edu.uc.tester4",
+            "role": "applicant",
+            "status": Decision.WAITLISTED,
+        },
+    ]
+
+    res = director_client.post("/confirm-attendance")
+
+    mock_mongodb_handler_retrieve.assert_awaited()
+    mock_mognodb_handler_update.assert_has_calls(
+        [
+            call(
+                Collection.USERS,
+                {"_id": {"$in": ("edu.uc.tester3",)}},
+                {"status": Status.ATTENDING},
+            ),
+            call().__bool__(),
+            call(
+                Collection.USERS,
+                {"_id": {"$in": ("edu.uc.tester",)}},
+                {"status": Status.VOID},
+            ),
+            call().__bool__(),
+            call(
+                Collection.USERS,
+                {"_id": {"$in": ("edu.uc.tester2",)}},
+                {"status": Status.VOID},
+            ),
+            call().__bool__(),
+        ]
+    )
+
+    assert res.status_code == 200