1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
import contextlib
import io
import unittest
from argparse import Namespace
from unittest.mock import patch
import snekbox.__main__ as snekbox_main
class ArgParseTests(unittest.TestCase):
def test_parse_args(self):
subtests = (
(
["", "code"],
Namespace(code="code", nsjail_args=[], py_args=["-c"])
),
(
["", "code", "--time_limit", "0"],
Namespace(code="code", nsjail_args=["--time_limit", "0"], py_args=["-c"])
),
(
["", "code", "---", "-m", "timeit"],
Namespace(code="code", nsjail_args=[], py_args=["-m", "timeit"])
),
(
["", "code", "--time_limit", "0", "---", "-m", "timeit"],
Namespace(code="code", nsjail_args=["--time_limit", "0"], py_args=["-m", "timeit"])
),
(
["", "code", "--time_limit", "0", "---"],
Namespace(code="code", nsjail_args=["--time_limit", "0"], py_args=[])
),
(
["", "code", "---"],
Namespace(code="code", nsjail_args=[], py_args=[])
)
)
for argv, expected in subtests:
with self.subTest(argv=argv, expected=expected), patch("sys.argv", argv):
args = snekbox_main.parse_args()
self.assertEqual(args, expected)
@patch("sys.argv", [""])
def test_parse_args_code_missing_exits(self):
with self.assertRaises(SystemExit) as cm:
with contextlib.redirect_stderr(io.StringIO()) as stderr:
snekbox_main.parse_args()
self.assertEqual(cm.exception.code, 2)
self.assertIn("the following arguments are required: code", stderr.getvalue())
|
