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_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() print actual_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() print actual_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)