Fill in subscription creation mocks

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

View File

@ -28,7 +28,9 @@ from .models import (
PrincipalAdminRoleCSPPayload,
PrincipalAdminRoleCSPResult,
SubscriptionCreationCSPPayload,
SubscriptionCreationCSPResult,
SubscriptionVerificationCSPPayload,
SuscriptionVerificationCSPResult,
TaskOrderBillingCreationCSPPayload,
TaskOrderBillingCreationCSPResult,
TaskOrderBillingVerificationCSPPayload,
@ -112,15 +114,27 @@ class MockCloudProvider(CloudProviderInterface):
return csp_environment_id
def create_subscription(self, payload: SubscriptionCreationCSPPayload):
pass
return self.create_subscription_creation(payload)
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(
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):
self._authorize(auth_credentials)

View File

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