Adds Code Question Model

Adds an explicit model for questions of type `code`, to allow more
complex parsing. Updates schema docs.

Signed-off-by: Hassan Abouelela <[email protected]>

2 files changed, 25 insertions, 1 deletions

diff --git a/backend/models/__init__.py b/backend/models/__init__.py
index 29ccb24..8ad7f7f 100644
--- a/backend/models/__init__.py
+++ b/backend/models/__init__.py
@@ -2,13 +2,14 @@ from .antispam import AntiSpam
 from .discord_user import DiscordUser
 from .form import Form, FormList
 from .form_response import FormResponse, ResponseList
-from .question import Question
+from .question import CodeQuestion, Question
 
 __all__ = [
     "AntiSpam",
     "DiscordUser",
     "Form",
     "FormResponse",
+    "CodeQuestion",
     "Question",
     "FormList",
     "ResponseList"
diff --git a/backend/models/question.py b/backend/models/question.py
index 7daeb5a..9829843 100644
--- a/backend/models/question.py
+++ b/backend/models/question.py
@@ -4,6 +4,25 @@ from pydantic import BaseModel, Field, root_validator, validator
 
 from backend.constants import QUESTION_TYPES, REQUIRED_QUESTION_TYPE_DATA
 
+_TESTS_TYPE = t.Union[t.Dict[str, str], int]
+
+
+class Unittests(BaseModel):
+    allow_failure: bool = False
+    tests: _TESTS_TYPE
+
+    @validator("tests")
+    def validate_tests(cls, value: _TESTS_TYPE) -> _TESTS_TYPE:
+        if isinstance(value, dict) and not len(value.keys()):
+            raise ValueError("Must have at least one test in a test suite.")
+
+        return value
+
+
+class CodeQuestion(BaseModel):
+    language: str
+    unittests: t.Optional[Unittests]
+
 
 class Question(BaseModel):
     """Schema model for form question."""
@@ -49,4 +68,8 @@ class Question(BaseModel):
                     f"got {type(value['data'][key]).__name__} instead."
                 )
 
+            # Validate unittest options
+            if value.get("type").lower() == "code":
+                value["data"] = CodeQuestion(**value.get("data")).dict()
+
         return value