From 621c835f28cd3fcd1e5c3cccae9b5057647f9fa1 Mon Sep 17 00:00:00 2001 From: ionite34 Date: Tue, 22 Nov 2022 13:07:15 -0500 Subject: Fixed leading empty filter for py_args --- tests/test_nsjail.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) (limited to 'tests') diff --git a/tests/test_nsjail.py b/tests/test_nsjail.py index dca0a8f..da2afea 100644 --- a/tests/test_nsjail.py +++ b/tests/test_nsjail.py @@ -361,11 +361,18 @@ class NsJailTests(unittest.TestCase): self.assertEqual(result.args[end - len(args) : end], args) def test_py_args(self): - args = ["-m", "timeit"] - result = self.nsjail.python3(args) - - self.assertEqual(result.returncode, 0) - self.assertEqual(result.args[-2:], args) + expected = ["-m", "timeit"] + args = [ + ["", "-m", "timeit"], + ["", "", "-m", "timeit"], + ["", "", "", "-m", "timeit"], + ] + # Leading empty strings should be removed + for case in args: + with self.subTest(args=args): + result = self.nsjail.python3(case) + self.assertEqual(result.returncode, 0) + self.assertEqual(result.args[-2:], expected) class NsJailArgsTests(unittest.TestCase): -- cgit v1.2.3