def test_check_authorization_not_id(self, mocked_get_user_id):
expected_response = redirect('/')
real_func = mock.MagicMock(__name__='real_func')
func = auth.check_authorization(real_func)
mocked_get_user_id.return_value = None
actual_response = func(None)
self.assertEqual(actual_response.get('Location'), expected_response.get('Location'))
self.assertEqual(real_func.call_count, 0)
self.assertEqual(actual_response.status_code, 302)
def test_check_authorization(self, mocked_get_user_id):
expected_response = mock.MagicMock()
request = mock.MagicMock()
real_func = mock.MagicMock(return_value=expected_response, __name__='real_func')
func = auth.check_authorization(real_func)
user_id = 123
mocked_get_user_id.return_value = user_id
actual_response = func(request)
real_func.assert_called_with(request, user_id=user_id)
self.assertIs(actual_response, expected_response)
