pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fill in subscription creation mocks

Browse Source
This commit is contained in:
tomdds 2020-01-31 14:28:08 -05:00
parent 9acbeeb824
commit d5e739ea68
2 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
atst/domain/csp/cloud/mock_cloud_provider.py
View File

@ -28,7 +28,9 @@ from .models import (
PrincipalAdminRoleCSPPayload, PrincipalAdminRoleCSPPayload,
PrincipalAdminRoleCSPResult, PrincipalAdminRoleCSPResult,
SubscriptionCreationCSPPayload, SubscriptionCreationCSPPayload,
SubscriptionCreationCSPResult,
SubscriptionVerificationCSPPayload, SubscriptionVerificationCSPPayload,
SuscriptionVerificationCSPResult,
TaskOrderBillingCreationCSPPayload, TaskOrderBillingCreationCSPPayload,
TaskOrderBillingCreationCSPResult, TaskOrderBillingCreationCSPResult,
TaskOrderBillingVerificationCSPPayload, TaskOrderBillingVerificationCSPPayload,
@ -112,15 +114,27 @@ class MockCloudProvider(CloudProviderInterface):
return csp_environment_id return csp_environment_id
def create_subscription(self, payload: SubscriptionCreationCSPPayload): def create_subscription(self, payload: SubscriptionCreationCSPPayload):
pass return self.create_subscription_creation(payload)
def create_subscription_creation(self, payload: SubscriptionCreationCSPPayload): def create_subscription_creation(self, payload: SubscriptionCreationCSPPayload):
pass self._maybe_raise(self.NETWORK_FAILURE_PCT, self.NETWORK_EXCEPTION)
self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION)
self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
return SubscriptionCreationCSPResult(
subscription_verify_url="https://zombo.com", subscription_retry_after=10
)
def create_subscription_verification( def create_subscription_verification(
self, payload: SubscriptionVerificationCSPPayload self, payload: SubscriptionVerificationCSPPayload
): ):
pass self._maybe_raise(self.NETWORK_FAILURE_PCT, self.NETWORK_EXCEPTION)
self._maybe_raise(self.SERVER_FAILURE_PCT, self.SERVER_EXCEPTION)
self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
return SuscriptionVerificationCSPResult(
subscription_id="subscriptions/60fbbb72-0516-4253-ab18-c92432ba3230"
)
def create_atat_admin_user(self, auth_credentials, csp_environment_id): def create_atat_admin_user(self, auth_credentials, csp_environment_id):
self._authorize(auth_credentials) self._authorize(auth_credentials)

2
atst/domain/csp/cloud/models.py
View File

@ -418,7 +418,7 @@ class SubscriptionCreationCSPPayload(BaseCSPPayload):
class SubscriptionCreationCSPResult(AliasModel): class SubscriptionCreationCSPResult(AliasModel):
subscription_verify_url: str subscription_verify_url: str
subscription_retry_after: str subscription_retry_after: int
class Config: class Config:
fields = { fields = {