def test_weekly_tl_with_logfile(self):
'''testing weekly_tl with --logfile argument'''
self.args.logfile = [self.LogFile]
with patch('argparse.ArgumentParser.parse_args',
return_value=self.args):
with patch('weekly_tl.get_time',
return_value=(self.week_first, self.week_last)):
weekly_tl.main()
output = sys.stdout.getvalue().strip()
self.assertRegex(output, self.regex)
def test_weekly_tl_with_env_variable(self):
'''testing only 'new' argument with GTIMELOG_FILE env variable'''
with patch('argparse.ArgumentParser.parse_args',
return_value=self.args):
with patch('weekly_tl.get_time',
return_value=(self.week_first, self.week_last)):
with patch('os.environ.get', return_value=self.LogFile):
weekly_tl.main()
output = sys.stdout.getvalue().strip()
self.assertRegex(output, self.regex)
def test_weekly_tl_with_user_arg(self):
'''testing weekly_tl with GTIMELOG_USER env variable'''
self.args.user = ['my-user']
with patch('argparse.ArgumentParser.parse_args',
return_value=self.args):
with patch('weekly_tl.get_time',
return_value=(self.week_first, self.week_last)):
weekly_tl.main()
user_regex = re.compile(r'my-user')
output = sys.stdout.getvalue().strip()
self.assertRegex(output, user_regex)
def test_weekly_tl_with_minutes(self):
'''testing with --minutes argument'''
self.args.minutes = True
with patch('argparse.ArgumentParser.parse_args',
return_value=self.args):
with patch('weekly_tl.get_time',
return_value=(self.week_first, self.week_last)):
with patch('weekly_tl.set_logfile', return_value=self.LogFile):
weekly_tl.main()
output = sys.stdout.getvalue().strip()
self.assertRegex(output, self.regex_with_minutes)
self.assertRegex(output, self.regex_defuser)