class TestDjangoTokenManager(GoDjangoTestCase):
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.user_pk = self.user_helper.get_django_user().pk
self.client = Client()
self.tm = DjangoTokenManager(TokenManager(
self.vumi_helper.get_redis_manager().sub_manager('token_manager')))
def test_generate_callback(self):
token = self.tm.generate_callback_token('/foo/', 'It worked',
callback_for_test, callback_args=['arg'],
callback_kwargs={'kwarg': 'kwarg'}, message_level=messages.SUCCESS,
user_id=self.user_pk)
token_data = self.tm.get(token)
self.assertEqual(token_data['redirect_to'], reverse('token_task'))
self.assertEqual(token_data['user_id'], str(self.user_pk))
self.assertEqual(token_data['extra_params'], {
'callback_args': ['arg'],
'callback_kwargs': {'kwarg': 'kwarg'},
'callback_name': '%s.%s' % (callback_for_test.__module__,
callback_for_test.__name__),
'return_to': '/foo/',
'message_level': messages.SUCCESS,
'message': 'It worked',
})
def test_url_for_token(self):
url = self.tm.url_for_token('foo')
self.assertTrue(url.endswith(reverse('token',
kwargs={'token': 'foo'})))
def test_token_require_login(self):
token = self.tm.generate('/path/', user_id=self.user_pk)
token_url = reverse('token', kwargs={'token': token})
response = self.client.get(token_url)
self.assertRedirects(response,
'%s?%s' % (reverse('auth_login'), urllib.urlencode({
'next': reverse('token', kwargs={'token': token}),
})))
def test_token_with_login(self):
token = self.tm.generate('/path/', user_id=self.user_pk)
token_url = reverse('token', kwargs={'token': token})
token_data = self.tm.get(token)
self.client.login(email='*****@*****.**', password='******')
response = self.client.get(token_url)
self.assertTrue(
response['Location'].endswith('/path/?token=%s-%s%s' % (
len(token), token, token_data['system_token'])))
def test_token_with_invalid_login(self):
token = self.tm.generate('/path/', user_id=-1)
token_url = reverse('token', kwargs={'token': token})
self.client.login(username='******', password='******')
response = self.client.get(token_url)
self.assertRedirects(response,
'%s?%s' % (reverse('auth_login'), urllib.urlencode({
'next': reverse('token', kwargs={'token': token}),
})))
def test_invalid_token(self):
token = self.tm.generate('/foo/', user_id=self.user_pk)
token_url = reverse('token', kwargs={'token': token})
response = self.client.get(token_url)
self.assertTrue(response.status_code, 404)
