Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions backend/tests/fakes.py
Original file line number Diff line number Diff line change
Expand Up @@ -323,7 +323,7 @@ def get_latest_for_assembly(self, assembly_id: uuid.UUID) -> SelectionRunRecord
if not assembly_records:
return None
# Sort by created_at, return the most recent
return max(assembly_records, key=lambda r: r.created_at or "")
return max(assembly_records, key=lambda r: r.created_at or datetime.min)

def get_running_tasks(self) -> Iterable[SelectionRunRecord]:
"""Get all currently running selection tasks."""
Expand All @@ -343,7 +343,7 @@ def get_by_assembly_id_paginated(
# Get all records for the assembly
all_records = sorted(
[item for item in self._items if item.assembly_id == assembly_id],
key=lambda r: r.created_at or "",
key=lambda r: r.created_at or datetime.min,
reverse=True, # Newest first
)

Expand Down Expand Up @@ -453,11 +453,11 @@ def get_by_token(self, token: str) -> EmailConfirmationToken | None:

def count_recent_requests(self, user_id: uuid.UUID, since: datetime) -> int:
"""Count email confirmation requests for a user since a given datetime."""
count = 0
for item in self._items:
if item.user_id == user_id and item.created_at >= since:
count += 1
return count
return sum(
1
for item in self._items
if item.user_id == user_id and item.created_at >= since
)

def delete_old_tokens(self, before: datetime) -> int:
"""Delete tokens created before a given datetime. Returns count deleted."""
Expand Down
Loading