diff --git a/atst/domain/csp/cloud/mock_cloud_provider.py b/atst/domain/csp/cloud/mock_cloud_provider.py index f73044f5..b8bae79d 100644 --- a/atst/domain/csp/cloud/mock_cloud_provider.py +++ b/atst/domain/csp/cloud/mock_cloud_provider.py @@ -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) diff --git a/tests/domain/cloud/test_mock_csp.py b/tests/domain/cloud/test_mock_csp.py index 098e91b1..fb66b153 100644 --- a/tests/domain/cloud/test_mock_csp.py +++ b/tests/domain/cloud/test_mock_csp.py @@ -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" + )