test(sorting): tidy re import + guard tbody match (#68)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
2026-06-21 14:17:14 +02:00
parent 4dae3a1845
commit a4b4adb392
+4 -5
View File
@@ -1,5 +1,6 @@
"""Tests for the list-view sorting system (games/sorting.py).""" """Tests for the list-view sorting system (games/sorting.py)."""
import re
from datetime import datetime from datetime import datetime
from zoneinfo import ZoneInfo from zoneinfo import ZoneInfo
@@ -10,8 +11,6 @@ from django.test import RequestFactory
from django.urls import reverse from django.urls import reverse
from games.filters import FindFilter from games.filters import FindFilter
import re
from games.models import Game, Platform, Purchase, Session from games.models import Game, Platform, Purchase, Session
from games.sorting import ( from games.sorting import (
GAME_DEFAULT_SORT, GAME_DEFAULT_SORT,
@@ -185,8 +184,6 @@ class TestListGamesSort:
class TestListSessionsSort: class TestListSessionsSort:
def test_sort_by_duration_descending(self, logged_client, two_games): def test_sort_by_duration_descending(self, logged_client, two_games):
import re
alpha, beta = two_games alpha, beta = two_games
Session.objects.create( Session.objects.create(
game=alpha, game=alpha,
@@ -251,7 +248,9 @@ class TestListPurchasesSort:
) )
assert response.status_code == 200 assert response.status_code == 200
body = response.content.decode() body = response.content.decode()
tbody = re.search(r"<tbody[^>]*>(.*?)</tbody>", body, re.DOTALL).group(1) tbody_match = re.search(r"<tbody[^>]*>(.*?)</tbody>", body, re.DOTALL)
assert tbody_match
tbody = tbody_match.group(1)
assert tbody.index("Beta") < tbody.index("Alpha") # 90 before 10 assert tbody.index("Beta") < tbody.index("Alpha") # 90 before 10
def test_name_aggregate_sort_no_duplicate_rows(self, logged_client, two_purchases): def test_name_aggregate_sort_no_duplicate_rows(self, logged_client, two_purchases):