Make use of .raise_for_status()

Co-authored-by: Hassan Abouelela <[email protected]>

1 files changed, 6 insertions, 8 deletions

diff --git a/backend/routes/forms/unittesting.py b/backend/routes/forms/unittesting.py
index 57bf5db..cc9f814 100644
--- a/backend/routes/forms/unittesting.py
+++ b/backend/routes/forms/unittesting.py
@@ -2,9 +2,9 @@ import ast
 from collections import namedtuple
 from itertools import count
 from textwrap import indent
-from typing import Optional
 
 import httpx
+from httpx import HTTPStatusError
 
 from backend.constants import SNEKBOX_URL
 from backend.models import FormResponse, Form
@@ -46,15 +46,13 @@ def _make_user_code(code: str) -> str:
     return f'USER_CODE = r"""{code}"""'
 
 
-async def _post_eval(code: str) -> Optional[dict[str, str]]:
+async def _post_eval(code: str) -> dict[str, str]:
     """Post the eval to snekbox and return the response."""
     async with httpx.AsyncClient() as client:
         data = {"input": code}
         response = await client.post(SNEKBOX_URL, json=data, timeout=10)
 
-        if not response.status_code == 200:
-            return
-
+        response.raise_for_status()
         return response.json()
 
 
@@ -89,9 +87,9 @@ async def execute_unittest(form_response: FormResponse, form: Form) -> list[Unit
                 result = "Invalid generated unit code."
             # The runner is correctly formatted, we can run it.
             else:
-                response = await _post_eval(code)
-
-                if not response:
+                try:
+                    response = await _post_eval(code)
+                except HTTPStatusError:
                     return_code = 99
                     result = "Unable to contact code runner."
                 else: