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)