correct data class name

This commit is contained in:
Philip Kalinsky 2020-02-06 11:26:08 -05:00
parent c3c2f2e1c5
commit eb4201d6e3

View File

@ -23,9 +23,9 @@ from atst.domain.csp.cloud.models import (
BillingProfileVerificationCSPPayload, BillingProfileVerificationCSPPayload,
BillingProfileVerificationCSPResult, BillingProfileVerificationCSPResult,
InitialMgmtGroupCSPPayload, InitialMgmtGroupCSPPayload,
InitialMgmtGroupCSPResponse, InitialMgmtGroupCSPResult,
InitialMgmtGroupVerificationCSPPayload, InitialMgmtGroupVerificationCSPPayload,
InitialMgmtGroupVerificationCSPResponse, InitialMgmtGroupVerificationCSPResult,
CostManagementQueryCSPResult, CostManagementQueryCSPResult,
EnvironmentCSPPayload, EnvironmentCSPPayload,
EnvironmentCSPResult, EnvironmentCSPResult,
@ -128,7 +128,7 @@ def test_create_initial_mgmt_group_succeeds(mock_azure: AzureCloudProvider):
display_name=application.name, display_name=application.name,
management_group_name=str(uuid4()), management_group_name=str(uuid4()),
) )
result: InitialMgmtGroupCSPResponse = mock_azure.create_initial_mgmt_group(payload) result: InitialMgmtGroupCSPResult = mock_azure.create_initial_mgmt_group(payload)
assert result.id == "Test Id" assert result.id == "Test Id"
@ -145,7 +145,7 @@ def test_create_initial_mgmt_group_verification_succeeds(
payload = InitialMgmtGroupVerificationCSPPayload( payload = InitialMgmtGroupVerificationCSPPayload(
tenant_id="1234", management_group_name=management_group_name tenant_id="1234", management_group_name=management_group_name
) )
result: InitialMgmtGroupVerificationCSPResponse = mock_azure.create_initial_mgmt_group_verification( result: InitialMgmtGroupVerificationCSPResult = mock_azure.create_initial_mgmt_group_verification(
payload payload
) )