Add new testing clients. Change Submission.status to be Status enum.

fkdosilovic · fkdosilovic · commit e3e94e60a823 · 2024-11-14T18:57:11.000+01:00
diff --git a/src/judge0/common.py b/src/judge0/common.py
@@ -2,6 +2,8 @@
 from enum import IntEnum
 
 
+# TODO: Consider moving enumerations and other possible future types to separate
+# module (_types.py).
 class Language(IntEnum):
     PYTHON = 0
     CPP = 1
diff --git a/src/judge0/submission.py b/src/judge0/submission.py
@@ -125,6 +125,8 @@ def set_attributes(self, attributes):
 
             if attr in ENCODED_FIELDS:
                 setattr(self, attr, decode(value) if value else None)
+            elif attr == "status":
+                self.status = Status(value["id"])
             else:
                 setattr(self, attr, value)
 
@@ -151,5 +153,4 @@ def is_done(self) -> bool:
         if self.status is None:
             return False
         else:
-            # TODO: When status is changed to `Status`, refactor this as well.
-            return self.status["id"] not in (Status.IN_QUEUE, Status.PROCESSING)
+            return self.status not in (Status.IN_QUEUE, Status.PROCESSING)
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -8,6 +8,30 @@
 load_dotenv()
 
 
+@pytest.fixture(scope="session")
+def judge0_ce_client():
+    api_key = os.getenv("JUDGE0_TEST_API_KEY")
+    api_key_header = os.getenv("JUDGE0_TEST_API_KEY_HEADER")
+    endpoint = os.getenv("JUDGE0_TEST_CE_ENDPOINT")
+    client = clients.Client(
+        endpoint=endpoint,
+        auth_headers={api_key_header: api_key},
+    )
+    return client
+
+
+@pytest.fixture(scope="session")
+def judge0_extra_ce_client():
+    api_key = os.getenv("JUDGE0_TEST_API_KEY")
+    api_key_header = os.getenv("JUDGE0_TEST_API_KEY_HEADER")
+    endpoint = os.getenv("JUDGE0_TEST_EXTRA_CE_ENDPOINT")
+    client = clients.Client(
+        endpoint=endpoint,
+        auth_headers={api_key_header: api_key},
+    )
+    return client
+
+
 @pytest.fixture(scope="session")
 def atd_ce_client():
     api_key = os.getenv("JUDGE0_ATD_API_KEY")
diff --git a/tests/test_submission.py b/tests/test_submission.py
@@ -0,0 +1,28 @@
+import pytest
+
+from judge0 import Status, Submission, wait
+
+
+def test_status_before_and_after_submission(request):
+    client = request.getfixturevalue("judge0_ce_client")
+    submission = Submission('print("Hello World!")')
+
+    assert submission.status is None
+
+    client.create_submission(submission)
+    client.get_submission(submission)
+
+    assert submission.status.__class__ == Status
+    assert submission.status >= Status.IN_QUEUE
+
+
+def test_is_done(request):
+    client = request.getfixturevalue("judge0_ce_client")
+    submission = Submission('print("Hello World!")')
+
+    assert submission.status is None
+
+    client.create_submission(submission)
+    wait(client, submission)
+
+    assert submission.is_done()
