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)