Merge #108 - allow custom arguments to be passed to eval

2 files changed, 31 insertions, 0 deletions

diff --git a/tests/api/test_eval.py b/tests/api/test_eval.py
index 3350763..bdeee3e 100644
--- a/tests/api/test_eval.py
+++ b/tests/api/test_eval.py
@@ -25,6 +25,24 @@ class TestEvalResource(SnekAPITestCase):
 
         self.assertEqual(expected, result.json)
 
+    def test_post_invalid_data_400(self):
+        bodies = (
+            {"input": 400}, {"input": "", "args": [400]}
+        )
+
+        for body in bodies:
+            with self.subTest():
+                result = self.simulate_post(self.PATH, json=body)
+
+                self.assertEqual(result.status_code, 400)
+
+                expected = {
+                    "title": "Request data failed validation",
+                    "description": "400 is not of type 'string'"
+                }
+
+                self.assertEqual(expected, result.json)
+
     def test_post_invalid_content_type_415(self):
         body = "{'input': 'foo'}"
         headers = {"Content-Type": "application/xml"}
diff --git a/tests/test_nsjail.py b/tests/test_nsjail.py
index 46193b2..8955b4a 100644
--- a/tests/test_nsjail.py
+++ b/tests/test_nsjail.py
@@ -217,3 +217,16 @@ class NsJailTests(unittest.TestCase):
 
         output = self.nsjail._consume_stdout(nsjail_subprocess)
         self.assertEqual(output, chunk * expected_chunks)
+
+    def test_nsjail_args(self):
+        args = ("foo", "bar")
+        result = self.nsjail.python3("", nsjail_args=args)
+
+        self.assertEqual(result.args[9:11], args)
+
+    def test_py_args(self):
+        args = ("-m", "timeit")
+        result = self.nsjail.python3("", py_args=args)
+
+        self.assertEqual(result.returncode, 0)
+        self.assertEqual(result.args[-3:-1], args)