@@ -3011,15 +3011,15 @@ def test_add_func_with_id(self):
30113011 entity = self .entity
30123012 entity .id = 1
30133013 func_param_dict = {entity .add_ansible_role : 'ansible_role_id' }
3014- for func in func_param_dict :
3014+ for func , param in func_param_dict . items () :
30153015 self .assertEqual (inspect .getfullargspec (func ), EXPECTED_ARGSPEC )
3016- kwargs = {'kwarg' : gen_integer (), 'data' : {func_param_dict [ func ] : gen_integer ()}}
3016+ kwargs = {'kwarg' : gen_integer (), 'data' : {param : gen_integer ()}}
30173017 with mock .patch .object (entities , '_handle_response' ) as handlr :
30183018 with mock .patch .object (client , 'put' ) as client_request :
30193019 response = func (** kwargs )
30203020 self .assertEqual (client_request .call_count , 1 )
30213021 self .assertEqual (len (client_request .call_args [0 ]), 1 )
3022- self .assertNotIn (func_param_dict [ func ] , client_request .call_args [1 ]['data' ])
3022+ self .assertNotIn (param , client_request .call_args [1 ]['data' ])
30233023 self .assertEqual (client_request .call_args [1 ], kwargs )
30243024 self .assertEqual (handlr .call_count , 1 )
30253025 self .assertEqual (handlr .return_value , response )
@@ -3042,15 +3042,15 @@ def test_delete_func_with_id(self):
30423042 entity .delete_puppetclass : 'puppetclass_id' ,
30433043 entity .remove_ansible_role : 'ansible_role_id' ,
30443044 }
3045- for func in func_param_dict :
3045+ for func , param in func_param_dict . items () :
30463046 self .assertEqual (inspect .getfullargspec (func ), EXPECTED_ARGSPEC )
3047- kwargs = {'kwarg' : gen_integer (), 'data' : {func_param_dict [ func ] : gen_integer ()}}
3047+ kwargs = {'kwarg' : gen_integer (), 'data' : {param : gen_integer ()}}
30483048 with mock .patch .object (entities , '_handle_response' ) as handlr :
30493049 with mock .patch .object (client , 'delete' ) as client_request :
30503050 response = func (** kwargs )
30513051 self .assertEqual (client_request .call_count , 1 )
30523052 self .assertEqual (len (client_request .call_args [0 ]), 1 )
3053- self .assertNotIn (func_param_dict [ func ] , client_request .call_args [1 ]['data' ])
3053+ self .assertNotIn (param , client_request .call_args [1 ]['data' ])
30543054 self .assertEqual (client_request .call_args [1 ], kwargs )
30553055 self .assertEqual (handlr .call_count , 1 )
30563056 self .assertEqual (handlr .return_value , response )
@@ -3188,15 +3188,15 @@ def test_add_func_with_id(self):
31883188 """
31893189 entity = entities .Host (self .cfg , id = 1 )
31903190 func_param_dict = {entity .add_ansible_role : 'ansible_role_id' }
3191- for func in func_param_dict :
3191+ for func , param in func_param_dict . items () :
31923192 self .assertEqual (inspect .getfullargspec (func ), EXPECTED_ARGSPEC )
3193- kwargs = {'kwarg' : gen_integer (), 'data' : {func_param_dict [ func ] : gen_integer ()}}
3193+ kwargs = {'kwarg' : gen_integer (), 'data' : {param : gen_integer ()}}
31943194 with mock .patch .object (entities , '_handle_response' ) as handlr :
31953195 with mock .patch .object (client , 'put' ) as client_request :
31963196 response = func (** kwargs )
31973197 self .assertEqual (client_request .call_count , 1 )
31983198 self .assertEqual (len (client_request .call_args [0 ]), 1 )
3199- self .assertNotIn (func_param_dict [ func ] , client_request .call_args [1 ]['data' ])
3199+ self .assertNotIn (param , client_request .call_args [1 ]['data' ])
32003200 self .assertEqual (client_request .call_args [1 ], kwargs )
32013201 self .assertEqual (handlr .call_count , 1 )
32023202 self .assertEqual (handlr .return_value , response )
@@ -3218,15 +3218,15 @@ def test_delete_func_with_id(self):
32183218 entity .delete_puppetclass : 'puppetclass_id' ,
32193219 entity .remove_ansible_role : 'ansible_role_id' ,
32203220 }
3221- for func in func_param_dict :
3221+ for func , param in func_param_dict . items () :
32223222 self .assertEqual (inspect .getfullargspec (func ), EXPECTED_ARGSPEC )
3223- kwargs = {'kwarg' : gen_integer (), 'data' : {func_param_dict [ func ] : gen_integer ()}}
3223+ kwargs = {'kwarg' : gen_integer (), 'data' : {param : gen_integer ()}}
32243224 with mock .patch .object (entities , '_handle_response' ) as handlr :
32253225 with mock .patch .object (client , 'delete' ) as client_request :
32263226 response = func (** kwargs )
32273227 self .assertEqual (client_request .call_count , 1 )
32283228 self .assertEqual (len (client_request .call_args [0 ]), 1 )
3229- self .assertNotIn (func_param_dict [ func ] , client_request .call_args [1 ]['data' ])
3229+ self .assertNotIn (param , client_request .call_args [1 ]['data' ])
32303230 self .assertEqual (client_request .call_args [1 ], kwargs )
32313231 self .assertEqual (handlr .call_count , 1 )
32323232 self .assertEqual (handlr .return_value , response )
0 commit comments