Fix input append to args

2 files changed, 3 insertions, 2 deletions

diff --git a/snekbox/api/resources/eval.py b/snekbox/api/resources/eval.py
index c3e553a..28b739d 100644
--- a/snekbox/api/resources/eval.py
+++ b/snekbox/api/resources/eval.py
@@ -122,6 +122,7 @@ class EvalResource:
         # If `input` is supplied, default `args` to `-c`
         if "input" in body:
             body.setdefault("args", ["-c"])
+            body["args"].append(body["input"])
         try:
             result = self.nsjail.python3(
                 py_args=body["args"],
diff --git a/tests/test_integration.py b/tests/test_integration.py
index 5aa1b43..01fa657 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -56,8 +56,8 @@ class IntegrationTests(unittest.TestCase):
         """Test normal eval requests without files."""
         with run_gunicorn():
             cases = [
-                ({"input": "print('Hello')"}, "Hello"),
-                ({"args": ["-c", "print('Hello')"]}, "Hello"),
+                ({"input": "print('Hello')"}, "Hello\n"),
+                ({"args": ["-c", "print('abc12')"]}, "abc12\n"),
             ]
             for body, expected in cases:
                 with self.subTest(body=body):