rename state machine data class

This commit is contained in:
Philip Kalinsky 2020-02-06 10:44:32 -05:00
parent 7559875d64
commit c3c2f2e1c5
3 changed files with 10 additions and 10 deletions

View File

@ -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,

View File

@ -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}"

View File

@ -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