def test_get_session_expiry(self): three_am = old_datetime(2011, 8, 20, 3) now = three_am FrozenTime.freeze_time(now) tomorrow_4am = old_datetime(2011, 8, 21, 4) t = Token() expected_expiry = tomorrow_4am self.assertEqual(expected_expiry.year, t.get_session_expiry().year) self.assertEqual(expected_expiry.month, t.get_session_expiry().month) self.assertEqual(expected_expiry.day, t.get_session_expiry().day) self.assertEqual(expected_expiry.hour, t.get_session_expiry().hour)
def test_get_session_expiry_near_expiry(self): now = old_datetime(2011, 8, 20, 3) FrozenTime.freeze_time(now) t = Token() t.expiry_date = now.date() expected_expiry = old_datetime(2011, 8, 20, 23, 59, 59) actual_expiry = t.get_session_expiry() self.assertEqual(expected_expiry.year, actual_expiry.year) self.assertEqual(expected_expiry.month, actual_expiry.month) self.assertEqual(expected_expiry.day, actual_expiry.day) self.assertEqual(expected_expiry.hour, actual_expiry.hour)
def test_get_session_expiry_near_expiry(self): now = old_datetime(2011,8,20,3) FrozenTime.freeze_time(now) t = Token() t.expiry_date = now.date() expected_expiry = old_datetime(2011, 8, 20, 23, 59, 59) actual_expiry = t.get_session_expiry() self.assertEqual(expected_expiry.year, actual_expiry.year) self.assertEqual(expected_expiry.month, actual_expiry.month) self.assertEqual(expected_expiry.day, actual_expiry.day) self.assertEqual(expected_expiry.hour, actual_expiry.hour)
def test_default_expiry(self): the_time = old_datetime(2011, 9, 20, 1, 52, 31, 570376) FrozenTime.freeze_time(the_time) expected_expiry = the_time.date() + \ datetime.timedelta(settings.TOKEN_EXPIRY_DAYS) t = Token() self.assertEqual(expected_expiry, t.expiry_date)
def test_get_session_expiry_expired_token(self): now = old_datetime(2011, 8, 20, 13) FrozenTime.freeze_time(now) yesterday = (now - datetime.timedelta(1)).date() t = Token() t.expiry_date = yesterday self.assertEqual(now.year, t.get_session_expiry().year) self.assertEqual(now.month, t.get_session_expiry().month) self.assertEqual(now.day, t.get_session_expiry().day) self.assertEqual(now.hour, t.get_session_expiry().hour)
def test_is_expired(self): now = old_datetime(2011, 8, 20, 0, 0, 0, 1) FrozenTime.freeze_time(now) today = now.date() yesterday = today - datetime.timedelta(1) tomorrow = today + datetime.timedelta(1) t = Token() t.expiry_date = yesterday self.assertTrue(t.is_expired()) t.expiry_date = today self.assertFalse(t.is_expired()) t.expiry_date = tomorrow self.assertFalse(t.is_expired())
def __lt__(self, b): return old_datetime(self.year, self.month, self.day, self.hour) < \ old_datetime(b.year, b.month, b.day, b.hour)
def __lt__(a, b): return old_datetime(a.year, a.month, a.day, a.hour) < old_datetime(b.year, b.month, b.day, b.hour)