Optimize checker

Author: inkustr

checker/src/checker.py

@@ -17,7 +17,7 @@
     FlagSearcher,
     BaseCheckerTaskMessage,
     PutflagCheckerTaskMessage,
-    GetflagCheckerTaskMessage,
+    GetflagCheckerTaskMessage, 
     PutnoiseCheckerTaskMessage,
     GetnoiseCheckerTaskMessage,
     HavocCheckerTaskMessage,
@@ -48,9 +48,12 @@ def __init__(self, logger: LoggerAdapter, client: AsyncClient):
         self.client = client
         self.logger = logger
 
+        self.JSON_HEADERS = {"Accept": "application/json"}
+
     async def register_user(self, username: str, password: str):
         response = await self.client.post(
             "/register",
+            headers=self.JSON_HEADERS,
             data={"username": username, "password": password}
         )
         data = response.json()
@@ -61,6 +64,7 @@ async def register_user(self, username: str, password: str):
     async def login_user(self, username: str, password: str):
         response = await self.client.post(
             "/login",
+            headers=self.JSON_HEADERS,
             data={"username": username, "password": password}
         )
         data = response.json()
@@ -147,10 +151,9 @@ async def putflag_solutions(
     problem_title = generate_title()
     problem_description, test_cases_json, expected_outputs_json = generate_problem_from_scenario()
 
-    headers = {"Accept": "application/json"}
     response = await conn.client.post(
             "/problems/create",
-            headers=headers,
+            headers=conn.JSON_HEADERS,
             data={
                 "title": problem_title,
                 "description": problem_description,
@@ -159,7 +162,7 @@ async def putflag_solutions(
                 "expectedOutputs": expected_outputs_json,
                 "isPublished": "true",
                 "isPrivate": "false",
-                "maxRuntime": "1.0"
+                "maxRuntime": "0.2"
             }
         )
 
@@ -503,17 +506,14 @@ async def havoc_feedback_image(task: HavocCheckerTaskMessage, client: AsyncClien
     if response.status_code not in [200, 201, 302]:
         raise MumbleException(f"Failed to submit feedback: {response.status_code}")
 
-    await asyncio.sleep(2)
+    await asyncio.sleep(0.2)
 
     response = await conn.client.get("/feedback")
 
     if response.status_code not in [200, 201, 302]:
         raise MumbleException(f"Failed to retrieve feedback page: {response.status_code}")
 
-    content = response.text
-    logger.debug(f"Feedback page content: {content}")
-    
-    if "Test feedback with simple SVG image" in content:
+    if "Test feedback with simple SVG image" in response.text:
         logger.info("Simple SVG feedback successfully appears on feedback page")
     else:
         logger.warning("Simple SVG feedback does not appear on feedback page")