def scenario_parse_args_exits(self, argv):
with self.assertRaises(SystemExit) as cm:
stdout = StringIO()
stderr = StringIO()
with Redirect(stdout=stdout, stderr=stderr):
parsed_args = args.parse(argv)
return (stdout.getvalue(), stderr.getvalue(), cm.exception.code)
def test_if_file_is_given_file_is_set_to_file(self):
file = "tmp/text"
parsed_args = args.parse([PROG_NAME, file])
self.assertEqual(parsed_args.file, file)
def test_if_no_arguments_are_given_file_is_set_to_none(self):
parsed_args = args.parse([PROG_NAME])
self.assertEqual(parsed_args.file, None)
