def test_schedule_file_option(self):
args = ArgHandler.get_args(['-f', filename, '-S', filename])
self.assertTrue(args.schedule_file)
args = ArgHandler.get_args([
'--file', filename,
'--schedule-file', filename,
])
self.assertTrue(args.schedule_file)
def test_file_option_and_filename_both_required(self):
"""should cause argparse error if file opt specified w/o file"""
expected = 'error: argument -f/--file: expected one argument'
try:
ArgHandler.get_args(['--file'])
except SystemExit:
pass
self.redirecterr.seek(0)
actual = self.redirecterr.read()
self.assertTrue(expected in actual)
def test_schedule_filename_required_with_schedule_option(self):
"""argparse error if sched file opt specified w/o file"""
expected = (
'error: argument -S/--schedule-file: expected one argument'
)
try:
ArgHandler.get_args(['--file', filename, '--schedule-file'])
except SystemExit:
pass
self.redirecterr.seek(0)
actual = self.redirecterr.read()
self.assertTrue(expected in actual)
def main(argv=None):
if argv is None:
argv = sys.argv[1:] # pragma: no cover
args = ArgHandler.get_args(argv)
ledgerbil = Ledgerbil(args)
return ledgerbil.process_file()
def test_no_sorting_option(self):
"""should not set parse args 'sort' var"""
args = ArgHandler.get_args(['--file', filename])
self.assertFalse(args.sort)
def test_sort_option(self):
args = ArgHandler.get_args(['-f', filename, '-s'])
self.assertTrue(args.sort)
args = ArgHandler.get_args(['--file', filename, '--sort'])
self.assertTrue(args.sort)
def test_file_option(self):
args = ArgHandler.get_args(['-f', filename])
self.assertTrue(args.file)
args = ArgHandler.get_args(['--file', filename])
self.assertTrue(args.file)