def test_timeout(self): """ Ensure we can use the token after n days, but no greater. """ # Uses a mocked version of EmailActivationTokenGenerator # so we can change the value of 'today' class Mocked(EmailActivationTokenGenerator): def __init__(self, today): self._today_val = today def _today(self): return self._today_val user = self.create_user() token_generator = EmailActivationTokenGenerator() token = token_generator.make_token(user) p1 = Mocked(date.today() + timedelta(settings.USERS_EMAIL_CONFIRMATION_TIMEOUT_DAYS)) self.assertTrue(p1.check_token(user, token)) p2 = Mocked(date.today() + timedelta(settings.USERS_EMAIL_CONFIRMATION_TIMEOUT_DAYS + 1)) self.assertFalse(p2.check_token(user, token))
def test_make_token(self): """ Ensure that we can make a token and that it is valid """ user = self.create_user() token_generator = EmailActivationTokenGenerator() token = token_generator.make_token(user) self.assertTrue(token_generator.check_token(user, token))