def test_generate_month_list_same_year(self):
date0 = Mongotools.TimeObject(
datetime.datetime(year=2011, month=2, day=1))
date1 = Mongotools.TimeObject(
datetime.datetime(year=2011, month=5, day=1))
backup = Backup.BackupHelper(date0, date1)
month_list = [(2011, k) for k in xrange(2, 6)]
self.assertEqual(list(backup.iterate_months()), month_list)
def test_iterate_timestamp_ranges(self):
date0 = Mongotools.TimeObject(
datetime.datetime(year=2011, month=2, day=1))
date1 = Mongotools.TimeObject(
datetime.datetime(year=2011, month=5, day=1))
backup = Backup.BackupHelper(date0, date1)
expected_dateranges = [(1296518400000, 1298937599999),
(1298937600000, 1301615999999),
(1301616000000, 1304207999999),
(1304208000000, 1306886399999)]
calculated_dateranges = list(backup.iterate_timestamp_ranges())
self.assertEqual(expected_dateranges, calculated_dateranges)
def test_iterate_monthly_date_ranges(self):
date0 = Mongotools.TimeObject(
datetime.datetime(year=2011, month=2, day=1))
date1 = Mongotools.TimeObject(
datetime.datetime(year=2011, month=5, day=1))
backup = Backup.BackupHelper(date0, date1)
expected_dateranges = [
(datetime.datetime(2011, 2, 1),
datetime.datetime(2011, 2, 28, 23, 59, 59, 999999)),
(datetime.datetime(2011, 3, 1),
datetime.datetime(2011, 3, 31, 23, 59, 59, 999999)),
(datetime.datetime(2011, 4, 1),
datetime.datetime(2011, 4, 30, 23, 59, 59, 999999)),
(datetime.datetime(2011, 5, 1),
datetime.datetime(2011, 5, 31, 23, 59, 59, 999999))
]
calculated_dateranges = list(backup.iterate_monthly_date_ranges())
self.assertEqual(calculated_dateranges, expected_dateranges)