def test_isnt_safe(self):
"""If next isn't safe, return None."""
request = self.factory.post('/', {'next': '/asdf'})
request.get_host = lambda: 'myhost'
with patch.object(views, 'is_safe_url', return_value=False) as is_safe_url:
eq_(views._get_next(request), None)
is_safe_url.assert_called_with('/asdf', host='myhost')
def test_is_safe(self):
"""Return the value of next if it is considered safe."""
request = self.factory.post('/', {'next': '/asdf'})
request.get_host = lambda: 'myhost'
with patch.object(views, 'is_safe_url', return_value=True) as is_safe_url:
eq_(views._get_next(request), '/asdf')
is_safe_url.assert_called_with('/asdf', host='myhost')
def test_isnt_safe(self):
"""If next isn't safe, return None."""
request = self.factory.post('/', {'next': '/asdf'})
request.get_host = lambda: 'myhost'
with patch.object(views, 'is_safe_url',
return_value=False) as is_safe_url:
self.assertEqual(views._get_next(request), None)
is_safe_url.assert_called_with('/asdf', host='myhost')
def test_is_safe(self):
"""Return the value of next if it is considered safe."""
request = self.factory.post('/', {'next': '/asdf'})
request.get_host = lambda: 'myhost'
with patch.object(views, 'is_safe_url',
return_value=True) as is_safe_url:
self.assertEqual(views._get_next(request), '/asdf')
is_safe_url.assert_called_with('/asdf', host='myhost')
def test_no_param(self):
"""If next isn't in the POST params, return None."""
request = self.factory.post('/')
eq_(views._get_next(request), None)
def test_no_param(self):
"""If next isn't in the POST params, return None."""
request = self.factory.post('/')
self.assertEqual(views._get_next(request), None)
