Fix testing with csrf token

jdabtieu · Dec 18, 2023 · 3fb4d8b · 3fb4d8b
1 parent f5af4da
commit 3fb4d8b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/tests/test_contest.py b/src/tests/test_contest.py
@@ -473,7 +473,7 @@ def test_contest_rejudge(client, database):
     result = client.get('/contest/testingcontest/scoreboard')
     assert result.status_code == 200
     assert b'None' not in result.data
-    assert result.data.replace(b'501', b'1') == result2.data  # Check points and last AC
+    assert result.data[result.data.index(b'table'):].replace(b'501', b'1') == result2.data[result2.data.index(b'table'):]  # Check points and last AC
 
 
     result = client.post('/contest/testingcontest/problem/static/edit', data={
@@ -491,7 +491,7 @@ def test_contest_rejudge(client, database):
     result = client.get('/contest/testingcontest/scoreboard')
     assert result.status_code == 200
     assert b'None' not in result.data
-    assert result.data == result1.data  # Check points and last AC
+    assert result.data[result.data.index(b'table'):] == result1.data[result1.data.index(b'table'):]  # Check points and last AC
 
     client.post('/contest/testingcontest/delete', follow_redirects=True)
-    assert result.status_code == 200
+    assert result.status_code == 200