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, BillingProfileVerificationCSPResult,
CostManagementQueryCSPResult, CostManagementQueryCSPResult,
InitialMgmtGroupCSPPayload, InitialMgmtGroupCSPPayload,
InitialMgmtGroupCSPResponse, InitialMgmtGroupCSPResult,
InitialMgmtGroupVerificationCSPPayload, InitialMgmtGroupVerificationCSPPayload,
InitialMgmtGroupVerificationCSPResponse, InitialMgmtGroupVerificationCSPResult,
EnvironmentCSPPayload, EnvironmentCSPPayload,
EnvironmentCSPResult, EnvironmentCSPResult,
KeyVaultCredentials, KeyVaultCredentials,
@ -207,7 +207,7 @@ class AzureCloudProvider(CloudProviderInterface):
credentials, payload.management_group_name, payload.display_name, credentials, payload.management_group_name, payload.display_name,
) )
return InitialMgmtGroupCSPResponse(**response) return InitialMgmtGroupCSPResult(**response)
def create_initial_mgmt_group_verification( def create_initial_mgmt_group_verification(
self, payload: InitialMgmtGroupVerificationCSPPayload self, payload: InitialMgmtGroupVerificationCSPPayload
@ -223,7 +223,7 @@ class AzureCloudProvider(CloudProviderInterface):
) )
response = self._get_management_group(credentials, payload.tenant_id,) response = self._get_management_group(credentials, payload.tenant_id,)
return InitialMgmtGroupVerificationCSPResponse(**response.result()) return InitialMgmtGroupVerificationCSPResult(**response.result())
def _create_management_group( def _create_management_group(
self, credentials, management_group_id, display_name, parent_id=None, self, credentials, management_group_id, display_name, parent_id=None,

View File

@ -24,9 +24,9 @@ from .models import (
BillingProfileVerificationCSPPayload, BillingProfileVerificationCSPPayload,
BillingProfileVerificationCSPResult, BillingProfileVerificationCSPResult,
InitialMgmtGroupCSPPayload, InitialMgmtGroupCSPPayload,
InitialMgmtGroupCSPResponse, InitialMgmtGroupCSPResult,
InitialMgmtGroupVerificationCSPPayload, InitialMgmtGroupVerificationCSPPayload,
InitialMgmtGroupVerificationCSPResponse, InitialMgmtGroupVerificationCSPResult,
CostManagementQueryCSPResult, CostManagementQueryCSPResult,
CostManagementQueryProperties, CostManagementQueryProperties,
ProductPurchaseCSPPayload, ProductPurchaseCSPPayload,
@ -289,7 +289,7 @@ class MockCloudProvider(CloudProviderInterface):
self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION) self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION)
self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION) self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
return InitialMgmtGroupCSPResponse( return InitialMgmtGroupCSPResult(
id=f"{AZURE_MGMNT_PATH}{payload.management_group_name}", 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.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION)
self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION) self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
return InitialMgmtGroupVerificationCSPResponse( return InitialMgmtGroupVerificationCSPResult(
**dict( **dict(
id="Test Id" id="Test Id"
# id=f"{AZURE_MGMNT_PATH}{payload.management_group_name}" # id=f"{AZURE_MGMNT_PATH}{payload.management_group_name}"

View File

@ -371,7 +371,7 @@ class InitialMgmtGroupCSPPayload(ManagementGroupCSPPayload):
pass pass
class InitialMgmtGroupCSPResponse(ManagementGroupCSPResponse): class InitialMgmtGroupCSPResult(ManagementGroupCSPResponse):
pass pass
@ -379,7 +379,7 @@ class InitialMgmtGroupVerificationCSPPayload(ManagementGroupGetCSPPayload):
pass pass
class InitialMgmtGroupVerificationCSPResponse(ManagementGroupGetCSPResponse): class InitialMgmtGroupVerificationCSPResult(ManagementGroupGetCSPResponse):
pass pass