43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
|
import unittest
|
||
|
|
from datetime import datetime
|
||
|
|
import szurubooru.util
|
||
|
|
from szurubooru.util import parse_time_range
|
||
|
|
from szurubooru.errors import ValidationError
|
||
|
|
|
||
|
|
class FakeDatetime(datetime):
|
||
|
|
def now(tz=None):
|
||
|
|
return datetime(1997, 1, 2, 3, 4, 5, tzinfo=tz)
|
||
|
|
|
||
|
|
class TestParseTime(unittest.TestCase):
|
||
|
|
def test_empty(self):
|
||
|
|
self.assertRaises(ValidationError, parse_time_range, '')
|
||
|
|
|
||
|
|
def test_today(self):
|
||
|
|
szurubooru.util.datetime.datetime = FakeDatetime
|
||
|
|
date_min, date_max = parse_time_range('today')
|
||
|
|
self.assertEquals(date_min, datetime(1997, 1, 2, 0, 0, 0))
|
||
|
|
self.assertEquals(date_max, datetime(1997, 1, 2, 23, 59, 59))
|
||
|
|
|
||
|
|
def test_yesterday(self):
|
||
|
|
szurubooru.util.datetime.datetime = FakeDatetime
|
||
|
|
date_min, date_max = parse_time_range('yesterday')
|
||
|
|
self.assertEquals(date_min, datetime(1997, 1, 1, 0, 0, 0))
|
||
|
|
self.assertEquals(date_max, datetime(1997, 1, 1, 23, 59, 59))
|
||
|
|
|
||
|
|
def test_year(self):
|
||
|
|
date_min, date_max = parse_time_range('1999')
|
||
|
|
self.assertEquals(date_min, datetime(1999, 1, 1, 0, 0, 0))
|
||
|
|
self.assertEquals(date_max, datetime(1999, 12, 31, 23, 59, 59))
|
||
|
|
|
||
|
|
def test_month(self):
|
||
|
|
for text in ['1999-2', '1999-02']:
|
||
|
|
date_min, date_max = parse_time_range(text)
|
||
|
|
self.assertEquals(date_min, datetime(1999, 2, 1, 0, 0, 0))
|
||
|
|
self.assertEquals(date_max, datetime(1999, 2, 28, 23, 59, 59))
|
||
|
|
|
||
|
|
def test_day(self):
|
||
|
|
for text in ['1999-2-6', '1999-02-6', '1999-2-06', '1999-02-06']:
|
||
|
|
date_min, date_max = parse_time_range(text)
|
||
|
|
self.assertEquals(date_min, datetime(1999, 2, 6, 0, 0, 0))
|
||
|
|
self.assertEquals(date_max, datetime(1999, 2, 6, 23, 59, 59))
|