diff options
Diffstat (limited to '')
| -rw-r--r-- | tests/test_main.py | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/tests/test_main.py b/tests/test_main.py index 1017123..196c196 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -2,6 +2,7 @@ import contextlib import io import unittest from argparse import Namespace +from subprocess import CompletedProcess from unittest.mock import patch import snekbox.__main__ as snekbox_main @@ -49,3 +50,51 @@ class ArgParseTests(unittest.TestCase): self.assertEqual(cm.exception.code, 2) self.assertIn("the following arguments are required: code", stderr.getvalue()) + + +class EntrypointTests(unittest.TestCase): + @patch("sys.argv", ["", "code"]) + @patch("snekbox.__main__.NsJail", autospec=True) + def test_main_prints_stdout(self, mock_nsjail): + mock_nsjail.return_value.python3.return_value = CompletedProcess( + args=[], + returncode=0, + stdout="output", + stderr=None + ) + + with contextlib.redirect_stdout(io.StringIO()) as stdout: + snekbox_main.main() + + self.assertEqual(stdout.getvalue(), "output\n") + + @patch("sys.argv", ["", "code"]) + @patch("snekbox.__main__.NsJail", autospec=True) + def test_main_exits_with_returncode(self, mock_nsjail): + mock_nsjail.return_value.python3.return_value = CompletedProcess( + args=[], + returncode=137, + stdout="output", + stderr=None + ) + + with self.assertRaises(SystemExit) as cm: + snekbox_main.main() + + self.assertEqual(cm.exception.code, 137) + + @patch("sys.argv", ["", "code", "--time_limit", "0", "---", "-m", "timeit"]) + @patch("snekbox.__main__.NsJail", autospec=True) + def test_main_forwards_args(self, mock_nsjail): + mock_nsjail.return_value.python3.return_value = CompletedProcess( + args=[], + returncode=0, + stdout="output", + stderr=None + ) + + snekbox_main.main() + + mock_nsjail.return_value.python3.assert_called_once_with( + "code", nsjail_args=["--time_limit", "0"], py_args=["-m", "timeit"] + ) |