mirror of
https://github.com/topydo/topydo.git
synced 2024-05-20 13:58:33 +00:00
187 lines
6.1 KiB
Python
187 lines
6.1 KiB
Python
# Topydo - A todo.txt client written in Python.
|
|
# Copyright (C) 2014 - 2015 Bram Schoenmakers <bram@topydo.org>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import unittest
|
|
from datetime import date
|
|
|
|
from freezegun import freeze_time
|
|
|
|
from topydo.lib.RelativeDate import relative_date_to_date
|
|
|
|
from .topydo_testcase import TopydoTest
|
|
|
|
|
|
@freeze_time('2015, 11, 06')
|
|
class RelativeDateTester(TopydoTest):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.yesterday = date(2015, 11, 5)
|
|
self.today = date(2015, 11, 6)
|
|
self.tomorrow = date(2015, 11, 7)
|
|
self.monday = date(2015, 11, 9)
|
|
self.friday = date(2015, 11, 13)
|
|
|
|
def test_zero_days(self):
|
|
result = relative_date_to_date('0d')
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_one_day(self):
|
|
result = relative_date_to_date('1d')
|
|
self.assertEqual(result, self.tomorrow)
|
|
|
|
def test_zero_bdays(self):
|
|
result = relative_date_to_date('0b')
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_one_bday(self):
|
|
result = relative_date_to_date('1b')
|
|
self.assertEqual(result, self.monday)
|
|
|
|
def test_one_bweek(self):
|
|
result = relative_date_to_date('5b')
|
|
self.assertEqual(result, self.friday)
|
|
|
|
def test_one_week(self):
|
|
result = relative_date_to_date('1w')
|
|
self.assertEqual(result, date(2015, 11, 13))
|
|
|
|
def test_one_month(self):
|
|
test_date = date(2015, 1, 10)
|
|
result = relative_date_to_date('1m', test_date)
|
|
self.assertEqual(result, date(2015, 2, 10))
|
|
|
|
def test_one_month_ext(self):
|
|
test_date1 = date(2015, 1, 29)
|
|
test_date2 = date(2016, 1, 31)
|
|
test_date3 = date(2015, 12, 31)
|
|
test_date4 = date(2015, 7, 31)
|
|
test_date5 = date(2015, 10, 31)
|
|
|
|
result1 = relative_date_to_date('1m', test_date1)
|
|
result2 = relative_date_to_date('1m', test_date2)
|
|
result3 = relative_date_to_date('1m', test_date3)
|
|
result4 = relative_date_to_date('1m', test_date4)
|
|
result5 = relative_date_to_date('1m', test_date5)
|
|
|
|
self.assertEqual(result1, date(2015, 2, 28))
|
|
self.assertEqual(result2, date(2016, 2, 29))
|
|
self.assertEqual(result3, date(2016, 1, 31))
|
|
self.assertEqual(result4, date(2015, 8, 31))
|
|
self.assertEqual(result5, date(2015, 11, 30))
|
|
|
|
def test_one_year(self):
|
|
test_date = date(2015, 1, 10)
|
|
result = relative_date_to_date('1y', test_date)
|
|
self.assertEqual(result, date(2016, 1, 10))
|
|
|
|
def test_leap_year(self):
|
|
test_date = date(2016, 2, 29)
|
|
result1 = relative_date_to_date('1y', test_date)
|
|
result2 = relative_date_to_date('4y', test_date)
|
|
self.assertEqual(result1, date(2017, 2, 28))
|
|
self.assertEqual(result2, date(2020, 2, 29))
|
|
|
|
def test_zero_months(self):
|
|
result = relative_date_to_date('0m')
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_zero_years(self):
|
|
result = relative_date_to_date('0y')
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_garbage1(self):
|
|
result = relative_date_to_date('0dd')
|
|
self.assertFalse(result)
|
|
|
|
def test_one_day_capital(self):
|
|
result = relative_date_to_date('1D')
|
|
self.assertEqual(result, self.tomorrow)
|
|
|
|
def test_today1(self):
|
|
result = relative_date_to_date('today')
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_today2(self):
|
|
result = relative_date_to_date('tod')
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_today3(self):
|
|
result = relative_date_to_date('today', self.tomorrow)
|
|
self.assertEqual(result, self.today)
|
|
|
|
def test_tomorrow1(self):
|
|
result = relative_date_to_date('Tomorrow')
|
|
self.assertEqual(result, self.tomorrow)
|
|
|
|
def test_tomorrow2(self):
|
|
result = relative_date_to_date('tom')
|
|
self.assertEqual(result, self.tomorrow)
|
|
|
|
def test_yesterday1(self):
|
|
result = relative_date_to_date('yesterday')
|
|
self.assertEqual(result, self.yesterday)
|
|
|
|
def test_yesterday2(self):
|
|
result = relative_date_to_date('yes')
|
|
self.assertEqual(result, self.yesterday)
|
|
|
|
def test_monday1(self):
|
|
result = relative_date_to_date('monday')
|
|
self.assertEqual(result, self.monday)
|
|
|
|
def test_monday2(self):
|
|
result = relative_date_to_date('mo')
|
|
self.assertEqual(result, self.monday)
|
|
|
|
def test_monday3(self):
|
|
result = relative_date_to_date('mon')
|
|
self.assertEqual(result, self.monday)
|
|
|
|
def test_monday4(self):
|
|
result = relative_date_to_date('mondayy')
|
|
self.assertFalse(result)
|
|
|
|
def test_offset1(self):
|
|
result = relative_date_to_date('1d', self.tomorrow)
|
|
self.assertEqual(result, date(2015, 11, 8))
|
|
|
|
def test_negative_period1(self):
|
|
result = relative_date_to_date('-1d')
|
|
self.assertEqual(result, date(2015, 11, 5))
|
|
|
|
def test_negative_period2(self):
|
|
result = relative_date_to_date('-0d')
|
|
self.assertTrue(result, self.today)
|
|
|
|
def test_negative_period3(self):
|
|
result = relative_date_to_date('-1b')
|
|
self.assertEqual(result, date(2015, 11, 5))
|
|
|
|
def test_negative_period4(self):
|
|
result = relative_date_to_date('-5b')
|
|
self.assertEqual(result, date(2015, 10, 30))
|
|
|
|
def test_weekday_next_week(self):
|
|
"""
|
|
When entering "Friday" on a Friday, return next week Friday instead of
|
|
today.
|
|
"""
|
|
result = relative_date_to_date("fri")
|
|
self.assertTrue(result, self.friday)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|