Skip to content

Commit

Permalink
fix scheduling tests
Browse files Browse the repository at this point in the history
Signed-off-by: Chris “Kwpolska” Warrick <kwpolska@gmail.com>
  • Loading branch information
Kwpolska committed Apr 22, 2014
1 parent edd8e4b commit 33775bf
Showing 1 changed file with 2 additions and 18 deletions.
20 changes: 2 additions & 18 deletions tests/test_scheduling.py
Expand Up @@ -62,83 +62,68 @@ def test_get_date(self):
# No last date
expected = TODAY.replace(day=23).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_FR, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_FR, tz=UTC, force_today=True))
self.assertEqual(expected, get_date(True, RULE_FR, tz=UTC))

# Last date in the past; doesn't match rule
date = TODAY.replace(hour=7)
expected = TODAY.replace(day=23, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_FR, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_FR, date, True, tz=UTC))

# Last date in the future; doesn't match rule
date = TODAY.replace(day=24, hour=7)
expected = TODAY.replace(day=30, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_FR, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_FR, date, True, tz=UTC))

# Last date in the past; matches rule
date = TODAY.replace(day=16, hour=8)
expected = TODAY.replace(day=23, hour=8).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_FR, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_FR, date, True, tz=UTC))

# Last date in the future; matches rule
date = TODAY.replace(day=23, hour=18)
expected = TODAY.replace(day=30, hour=18).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_FR, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_FR, date, True, tz=UTC))

# NOW matches rule ################################################
# Not scheduling should return NOW
self.assertEqual(NOW, get_date(False, RULE_TH, tz=UTC))
# No last date
self.assertEqual(NOW, get_date(True, RULE_TH, tz=UTC))
self.assertEqual(NOW, get_date(True, RULE_TH, force_today=True, tz=UTC))
self.assertEqual(NOW, get_date(True, RULE_TH, tz=UTC))

# Last date in the past; doesn't match rule
# Corresponding time has already passed, today
date = TODAY.replace(day=21, hour=7)
expected = TODAY.replace(day=29, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
expected = TODAY.replace(day=22, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))
# Corresponding time has not passed today
date = TODAY.replace(day=21, hour=18)
expected = TODAY.replace(day=22, hour=18).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))

# Last date in the future; doesn't match rule
# Corresponding time has already passed, today
date = TODAY.replace(day=24, hour=7)
expected = TODAY.replace(day=29, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))
# Corresponding time has not passed today
date = TODAY.replace(day=24, hour=18)
expected = TODAY.replace(day=29, hour=18).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))

# Last date in the past; matches rule
# Corresponding time has already passed, today
date = TODAY.replace(day=15, hour=7)
expected = TODAY.replace(day=29, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
expected = TODAY.replace(day=22, hour=7).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))
# Corresponding time has already passed, today; rule specifies HOUR
date = TODAY.replace(day=15, hour=7)
expected = TODAY.replace(day=29, hour=9).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH + ';BYHOUR=9', date, tz=UTC))
expected = TODAY.replace(day=22, hour=9).strftime(FMT)
self.assertEqual(expected,
get_date(True, RULE_TH + ';BYHOUR=9', date, True, tz=UTC))
# Corresponding time has not passed today
date = TODAY.replace(day=15, hour=18)
expected = TODAY.replace(day=22, hour=18).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))

# Last date in the future; matches rule
# Corresponding time has already passed, today
Expand All @@ -149,7 +134,6 @@ def test_get_date(self):
date = TODAY.replace(day=22, hour=18)
expected = TODAY.replace(day=29, hour=18).strftime(FMT)
self.assertEqual(expected, get_date(True, RULE_TH, date, tz=UTC))
self.assertEqual(expected, get_date(True, RULE_TH, date, True, tz=UTC))

if __name__ == '__main__':
import unittest
Expand Down

0 comments on commit 33775bf

Please sign in to comment.