Strengthen tests

jdabtieu · Mar 5, 2023 · 982fad7 · 982fad7
1 parent 3de85ac
commit 982fad7
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/src/tests/test_problems.py b/src/tests/test_problems.py
@@ -50,14 +50,38 @@ def test_problem(client, database):
     assert b'edited' in result.data
 
     result = client.post('/problem/helloworldtesting/edit', data={
-        'name': 'hello world 2',
         'description': 'a short fun problem 2',
         'hints': 'try looking at the title 2',
         'point_value': 2,
         'rejudge': True,
         'category': 'web',
         'flag': 'ctf{hello}'
     })
+    assert result.status_code == 400
+    assert b'required' in result.data
+
+    result = client.post('/problem/helloworldtesting/edit', data={
+        'name': 'hello world 2',
+        'description': 'a short fun problem 2',
+        'hints': 'try looking at the title 2',
+        'point_value': 2,
+        'rejudge': True,
+        'category': 'web',
+        'flag': '\x2f\x10'
+    })
+    assert result.status_code == 400
+    assert b'Invalid' in result.data
+
+    result = client.post('/problem/helloworldtesting/edit', data={
+        'name': 'hello world 2',
+        'description': 'a short fun problem 2',
+        'hints': 'try looking at the title 2',
+        'point_value': 2,
+        'rejudge': True,
+        'category': 'web',
+        'flag': 'ctf{hello}'
+    }, follow_redirects=True)
+    assert result.status_code == 200
 
     result = client.get('/users/admin/profile')
     assert result.status_code == 200