def test_success_url_as_url(self, get_context_data):
url = '/account/two_factor/'
view = PhoneSetupView()
view.success_url = url
def return_kwargs(form, **kwargs):
return kwargs
get_context_data.side_effect = return_kwargs
context = view.get_context_data(None)
self.assertIn('cancel_url', context)
self.assertEqual(url, context['cancel_url'])
def test_success_url_as_named_url(self, get_context_data):
url_name = 'two_factor:profile'
url = reverse(url_name)
view = PhoneSetupView()
view.success_url = url_name
def return_kwargs(form, **kwargs):
return kwargs
get_context_data.side_effect = return_kwargs
context = view.get_context_data(None)
self.assertIn('cancel_url', context)
self.assertEqual(url, context['cancel_url'])