def test_week(self):
now = datetime.now().date()
first_day = now - timedelta(days=now.weekday())
last_day = first_day + timedelta(days=6)
days = [first_day, last_day]
args = parse_args(['search', 'task', '--week'])
self.assertListEqual(dates_to_search(args), days)
def test_month(self):
now = datetime.now()
num_days = calendar.monthrange(now.year, now.month)[1]
days = [
date(now.year, now.month, 1),
date(now.year, now.month, num_days)
]
args = parse_args(['search', 'task', '--month'])
self.assertListEqual(dates_to_search(args), days)
def __init__(self, args: Namespace) -> None:
""" SearchTask Constructor
Parameters
----------
args : NameSpace
Arguments of argparse
"""
self.console = Console()
self.db = SQLiteConnection()
self.search_date = dates_to_search(args)
self.search_text = format_argument_text(args.text)
if not date_args_empty(args):
self.__search_task()
else:
PrintFormatted.print_help(help_text.SEARCH_USAGE_TEXT)
def test_yesterday(self):
search_date = datetime.now().date() - timedelta(days=1)
args = parse_args(['search', 'task', '--yesterday'])
self.assertEqual(dates_to_search(args), search_date)
def test_today(self):
search_date = datetime.now().date()
args = parse_args(['search', 'task', '--today'])
self.assertEqual(dates_to_search(args), search_date)