mock csp disable user test correct variable names

This commit is contained in:
Philip Kalinsky 2020-02-26 08:56:52 -05:00
parent e9d120ecae
commit d627031902
2 changed files with 5 additions and 3 deletions

View File

@ -417,14 +417,14 @@ class MockCloudProvider(CloudProviderInterface):
self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION) self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
return self._id() return self._id()
def disable_user(self, tenant_id, role_assignment_cloud_id): def disable_user(self, auth_credentials, tenant_id, role_assignment_cloud_id):
self._authorize(auth_credentials) self._authorize(auth_credentials)
self._maybe_raise(self.NETWORK_FAILURE_PCT, self.NETWORK_EXCEPTION) self._maybe_raise(self.NETWORK_FAILURE_PCT, self.NETWORK_EXCEPTION)
self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION) self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION)
self._maybe_raise( self._maybe_raise(
self.ATAT_ADMIN_CREATE_FAILURE_PCT, self.ATAT_ADMIN_CREATE_FAILURE_PCT,
UserRemovalException(csp_user_id, "Could not disable user."), UserRemovalException(tenant_id, "Could not disable user."),
) )
return self._maybe(12) return self._maybe(12)

View File

@ -79,4 +79,6 @@ def test_create_or_update_user(mock_csp: MockCloudProvider):
def test_disable_user(mock_csp: MockCloudProvider): def test_disable_user(mock_csp: MockCloudProvider):
assert mock_csp.disable_user(CREDENTIALS, "csp_user_id") assert mock_csp.disable_user(
CREDENTIALS, "tenant_id", "role_assignment_cloud_id"
)