diff --git a/atst/domain/csp/cloud/azure_cloud_provider.py b/atst/domain/csp/cloud/azure_cloud_provider.py index 17c55a30..3b49ddb1 100644 --- a/atst/domain/csp/cloud/azure_cloud_provider.py +++ b/atst/domain/csp/cloud/azure_cloud_provider.py @@ -26,9 +26,9 @@ from .models import ( BillingProfileVerificationCSPResult, CostManagementQueryCSPResult, InitialMgmtGroupCSPPayload, - InitialMgmtGroupCSPResponse, + InitialMgmtGroupCSPResult, InitialMgmtGroupVerificationCSPPayload, - InitialMgmtGroupVerificationCSPResponse, + InitialMgmtGroupVerificationCSPResult, EnvironmentCSPPayload, EnvironmentCSPResult, KeyVaultCredentials, @@ -207,7 +207,7 @@ class AzureCloudProvider(CloudProviderInterface): credentials, payload.management_group_name, payload.display_name, ) - return InitialMgmtGroupCSPResponse(**response) + return InitialMgmtGroupCSPResult(**response) def create_initial_mgmt_group_verification( self, payload: InitialMgmtGroupVerificationCSPPayload @@ -223,7 +223,7 @@ class AzureCloudProvider(CloudProviderInterface): ) response = self._get_management_group(credentials, payload.tenant_id,) - return InitialMgmtGroupVerificationCSPResponse(**response.result()) + return InitialMgmtGroupVerificationCSPResult(**response.result()) def _create_management_group( self, credentials, management_group_id, display_name, parent_id=None, diff --git a/atst/domain/csp/cloud/mock_cloud_provider.py b/atst/domain/csp/cloud/mock_cloud_provider.py index 70e54304..c03dbbf5 100644 --- a/atst/domain/csp/cloud/mock_cloud_provider.py +++ b/atst/domain/csp/cloud/mock_cloud_provider.py @@ -24,9 +24,9 @@ from .models import ( BillingProfileVerificationCSPPayload, BillingProfileVerificationCSPResult, InitialMgmtGroupCSPPayload, - InitialMgmtGroupCSPResponse, + InitialMgmtGroupCSPResult, InitialMgmtGroupVerificationCSPPayload, - InitialMgmtGroupVerificationCSPResponse, + InitialMgmtGroupVerificationCSPResult, CostManagementQueryCSPResult, CostManagementQueryProperties, ProductPurchaseCSPPayload, @@ -289,7 +289,7 @@ class MockCloudProvider(CloudProviderInterface): self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION) self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION) - return InitialMgmtGroupCSPResponse( + return InitialMgmtGroupCSPResult( id=f"{AZURE_MGMNT_PATH}{payload.management_group_name}", ) @@ -300,7 +300,7 @@ class MockCloudProvider(CloudProviderInterface): self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION) self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION) - return InitialMgmtGroupVerificationCSPResponse( + return InitialMgmtGroupVerificationCSPResult( **dict( id="Test Id" # id=f"{AZURE_MGMNT_PATH}{payload.management_group_name}" diff --git a/atst/domain/csp/cloud/models.py b/atst/domain/csp/cloud/models.py index e8ee45ec..4c31fce5 100644 --- a/atst/domain/csp/cloud/models.py +++ b/atst/domain/csp/cloud/models.py @@ -371,7 +371,7 @@ class InitialMgmtGroupCSPPayload(ManagementGroupCSPPayload): pass -class InitialMgmtGroupCSPResponse(ManagementGroupCSPResponse): +class InitialMgmtGroupCSPResult(ManagementGroupCSPResponse): pass @@ -379,7 +379,7 @@ class InitialMgmtGroupVerificationCSPPayload(ManagementGroupGetCSPPayload): pass -class InitialMgmtGroupVerificationCSPResponse(ManagementGroupGetCSPResponse): +class InitialMgmtGroupVerificationCSPResult(ManagementGroupGetCSPResponse): pass