def test_redirect_decorator_doesnt_redirect(self):
"""
Test that the redirect is triggered only when needed.
"""
decorated_view = redirect_to_default_cms_site(
lambda request: "untouched response")
response = decorated_view(self.factory.get('/example/'))
self.assertEqual(response, "untouched response")
def test_redirect_decorator(self):
"""
Test that the decorator redirects.
"""
decorated_view = redirect_to_default_cms_site(lambda request: None)
response = decorated_view(
self.factory.get('/example/', HTTP_HOST='secondary-site.com'))
self.assertEqual(response.status_code, 302)
