diff --git a/ckan/new_tests/logic/auth/test_create.py b/ckan/new_tests/logic/auth/test_create.py index 04a05c7fd6a..039ba3d96c1 100644 --- a/ckan/new_tests/logic/auth/test_create.py +++ b/ckan/new_tests/logic/auth/test_create.py @@ -8,6 +8,8 @@ import ckan.new_tests.helpers as helpers import ckan.new_tests.factories as factories +logic = helpers.logic + class TestCreate(object): @@ -26,11 +28,12 @@ def test_user_invite_delegates_correctly_to_group_member_create(self, gmc): group_member_create_data_dict = data_dict.copy() group_member_create_data_dict['id'] = data_dict['group_id'] - with nose.tools.assert_raises(helpers.logic.NotAuthorized): - gmc.return_value = {'success': False} - helpers.call_auth('user_invite', context=context, **data_dict) + gmc.return_value = {'success': False} + nose.tools.assert_raises(logic.NotAuthorized, helpers.call_auth, + 'user_invite', context=context, **data_dict) gmc.return_value = {'success': True} - assert helpers.call_auth('user_invite', context=context, **data_dict) + result = helpers.call_auth('user_invite', context=context, **data_dict) + assert result is True gmc.assert_called_twice_with(context, group_member_create_data_dict)