mock csp disable user test correct variable names

This commit is contained in:
Philip Kalinsky 2020-02-26 08:56:52 -05:00
parent 90bf69afd3
commit a977beb7f6
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)
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._maybe_raise(self.NETWORK_FAILURE_PCT, self.NETWORK_EXCEPTION)
self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION)
self._maybe_raise(
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)

View File

@ -79,4 +79,6 @@ def test_create_or_update_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"
)