@@ -25,26 +25,32 @@ class TestRole(identity_fakes.TestIdentityv3):
2525 def setUp (self ):
2626 super (TestRole , self ).setUp ()
2727
28+ identity_client = self .app .client_manager .identity
29+
2830 # Get a shortcut to the UserManager Mock
29- self .users_mock = self . app . client_manager . identity .users
31+ self .users_mock = identity_client .users
3032 self .users_mock .reset_mock ()
3133
3234 # Get a shortcut to the UserManager Mock
33- self .groups_mock = self . app . client_manager . identity .groups
35+ self .groups_mock = identity_client .groups
3436 self .groups_mock .reset_mock ()
3537
3638 # Get a shortcut to the DomainManager Mock
37- self .domains_mock = self . app . client_manager . identity .domains
39+ self .domains_mock = identity_client .domains
3840 self .domains_mock .reset_mock ()
3941
4042 # Get a shortcut to the ProjectManager Mock
41- self .projects_mock = self . app . client_manager . identity .projects
43+ self .projects_mock = identity_client .projects
4244 self .projects_mock .reset_mock ()
4345
4446 # Get a shortcut to the RoleManager Mock
45- self .roles_mock = self . app . client_manager . identity .roles
47+ self .roles_mock = identity_client .roles
4648 self .roles_mock .reset_mock ()
4749
50+ # Get a shortcut to the InferenceRuleManager Mock
51+ self .inference_rules_mock = identity_client .inference_rules
52+ self .inference_rules_mock .reset_mock ()
53+
4854 def _is_inheritance_testcase (self ):
4955 return False
5056
@@ -67,12 +73,13 @@ def setUp(self):
6773 ),
6874 ]
6975
70- self . roles_mock . create_implied . return_value = fakes .FakeResource (
76+ fake_resource = fakes .FakeResource (
7177 None ,
7278 {'prior_role' : copy .deepcopy (identity_fakes .ROLES [0 ]),
7379 'implied' : copy .deepcopy (identity_fakes .ROLES [1 ]), },
7480 loaded = True ,
7581 )
82+ self .inference_rules_mock .create .return_value = fake_resource
7683
7784 self .cmd = implied_role .CreateImpliedRole (self .app , None )
7885
@@ -93,8 +100,8 @@ def test_implied_role_create(self):
93100 # data to be shown.
94101 columns , data = self .cmd .take_action (parsed_args )
95102
96- # RoleManager.create_implied (prior, implied)
97- self .roles_mock . create_implied .assert_called_with (
103+ # InferenceRuleManager.create (prior, implied)
104+ self .inference_rules_mock . create .assert_called_with (
98105 identity_fakes .ROLES [0 ]['id' ],
99106 identity_fakes .ROLES [1 ]['id' ]
100107 )
@@ -126,12 +133,13 @@ def setUp(self):
126133 ),
127134 ]
128135
129- self . roles_mock . delete_implied . return_value = fakes .FakeResource (
136+ fake_resource = fakes .FakeResource (
130137 None ,
131138 {'prior-role' : copy .deepcopy (identity_fakes .ROLES [0 ]),
132139 'implied' : copy .deepcopy (identity_fakes .ROLES [1 ]), },
133140 loaded = True ,
134141 )
142+ self .inference_rules_mock .delete .return_value = fake_resource
135143
136144 self .cmd = implied_role .DeleteImpliedRole (self .app , None )
137145
@@ -147,7 +155,7 @@ def test_implied_role_delete(self):
147155 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
148156 self .cmd .take_action (parsed_args )
149157
150- self .roles_mock . delete_implied .assert_called_with (
158+ self .inference_rules_mock . delete .assert_called_with (
151159 identity_fakes .ROLES [0 ]['id' ],
152160 identity_fakes .ROLES [1 ]['id' ]
153161 )
@@ -158,7 +166,7 @@ class TestImpliedRoleList(TestRole):
158166 def setUp (self ):
159167 super (TestImpliedRoleList , self ).setUp ()
160168
161- self .roles_mock .list_inference_roles .return_value = (
169+ self .inference_rules_mock .list_inference_roles .return_value = (
162170 identity_fakes .FakeImpliedRoleResponse .create_list ())
163171
164172 self .cmd = implied_role .ListImpliedRole (self .app , None )
@@ -168,7 +176,7 @@ def test_implied_role_list(self):
168176 verifylist = []
169177 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
170178 columns , data = self .cmd .take_action (parsed_args )
171- self .roles_mock .list_inference_roles .assert_called_with ()
179+ self .inference_rules_mock .list_inference_roles .assert_called_with ()
172180
173181 collist = ['Prior Role ID' , 'Prior Role Name' ,
174182 'Implied Role ID' , 'Implied Role Name' ]
0 commit comments