Fix tests.

fkdosilovic · fkdosilovic · commit 38d87a809342 · 2024-11-12T22:56:03.000+01:00
diff --git a/src/judge0/api.py b/src/judge0/api.py
@@ -26,7 +26,7 @@ def resolve_client(
         submissions = [submissions]
 
     # Check which client supports all languages from the provided submissions.
-    languages = (submission.language_id for submission in submissions)
+    languages = [submission.language_id for submission in submissions]
 
     for flavor in Flavor:
         client = _get_implicit_client(flavor)
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -1,12 +1,7 @@
+import judge0
 import pytest
 
-from judge0 import (
-    Flavor,
-    JUDGE0_IMPLICIT_CE_CLIENT,
-    JUDGE0_IMPLICIT_EXTRA_CE_CLIENT,
-    Language,
-    Submission,
-)
+from judge0 import Flavor, Language, Submission
 from judge0.api import resolve_client
 
 DEFAULT_CLIENTS = (
@@ -30,19 +25,20 @@ def test_resolve_client_with_explicit_client(client, request):
     [
         [
             Flavor.CE,
-            JUDGE0_IMPLICIT_CE_CLIENT,
+            "JUDGE0_IMPLICIT_CE_CLIENT",
         ],
         [
             Flavor.EXTRA_CE,
-            JUDGE0_IMPLICIT_EXTRA_CE_CLIENT,
+            "JUDGE0_IMPLICIT_EXTRA_CE_CLIENT",
         ],
     ],
 )
 def test_resolve_client_with_flavor(
     flavor,
     expected_client,
 ):
-    assert resolve_client(client=flavor) is expected_client
+    # We have to use getattr since both implicit clients are initially None.
+    assert resolve_client(client=flavor) is getattr(judge0, expected_client)
 
 
 @pytest.mark.parametrize(
@@ -88,7 +84,7 @@ def test_resolve_client_common_ce_client():
 
     submissions = [cpp_submission, py_submission]
 
-    assert resolve_client(submissions=submissions) is JUDGE0_IMPLICIT_CE_CLIENT
+    assert resolve_client(submissions=submissions) is judge0.JUDGE0_IMPLICIT_CE_CLIENT
 
 
 def test_resolve_client_common_extra_ce_client():
@@ -104,4 +100,7 @@ def test_resolve_client_common_extra_ce_client():
 
     submissions = [cpp_submission, py_submission]
 
-    assert resolve_client(submissions=submissions) is JUDGE0_IMPLICIT_EXTRA_CE_CLIENT
+    assert (
+        resolve_client(submissions=submissions)
+        is judge0.JUDGE0_IMPLICIT_EXTRA_CE_CLIENT
+    )
