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

provisioning error handling applied to all requests calls

Browse Source
This commit is contained in:
Philip Kalinsky
2020-02-06 14:44:49 -05:00
parent 1da7d42eb7
commit 50ab045353
3 changed files with 484 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
tests/domain/cloud/test_azure_csp.py
View File

@@ -154,30 +154,29 @@ def test_create_tenant(mock_azure: AzureCloudProvider):
)
)
mock_azure = mock_get_secret(mock_azure)
result = mock_azure.create_tenant(payload)
body: TenantCSPResult = result.get("body")
assert body.tenant_id == "60ff9d34-82bf-4f21-b565-308ef0533435"
result: TenantCSPResult = mock_azure.create_tenant(payload)
assert result.tenant_id == "60ff9d34-82bf-4f21-b565-308ef0533435"
def test_create_tenant_fails(mock_azure: AzureCloudProvider):
mock_result = Mock()
mock_result.json.return_value = {"error": "body"}
mock_result.status_code = 403
mock_azure.sdk.requests.post.return_value = mock_result
payload = TenantCSPPayload(
**dict(
user_id="admin",
password="JediJan13$coot", # pragma: allowlist secret
domain_name="jediccpospawnedtenant2",
first_name="Tedry",
last_name="Tenet",
country_code="US",
password_recovery_email_address="thomas@promptworks.com",
)
)
mock_azure = mock_get_secret(mock_azure)
result = mock_azure.create_tenant(payload)
assert result.get("status") == "error"
# def test_create_tenant_fails(mock_azure: AzureCloudProvider):
# mock_result = Mock()
# mock_result.status_code = 200
# mock_azure.sdk.requests.post.return_value = mock_result
# payload = TenantCSPPayload(
# **dict(
# user_id="admin",
# password="JediJan13$coot", # pragma: allowlist secret
# domain_name="jediccpospawnedtenant2",
# first_name="Tedry",
# last_name="Tenet",
# country_code="US",
# password_recovery_email_address="thomas@promptworks.com",
# )
# )
# mock_azure = mock_get_secret(mock_azure)
# with pytest.raises(pydantic.ValidationError):
# assert mock_azure.create_tenant(payload)
def test_create_billing_profile_creation(mock_azure: AzureCloudProvider):
@@ -207,8 +206,9 @@ def test_create_billing_profile_creation(mock_azure: AzureCloudProvider):
billing_account_name=BILLING_ACCOUNT_NAME,
)
)
result = mock_azure.create_billing_profile_creation(payload)
body: BillingProfileCreationCSPResult = result.get("body")
body: BillingProfileCreationCSPResult = mock_azure.create_billing_profile_creation(
payload
)
assert body.billing_profile_retry_after == 10
@@ -257,8 +257,9 @@ def test_validate_billing_profile_creation(mock_azure: AzureCloudProvider):
)
)
result = mock_azure.create_billing_profile_verification(payload)
body: BillingProfileVerificationCSPResult = result.get("body")
body: BillingProfileVerificationCSPResult = mock_azure.create_billing_profile_verification(
payload
)
assert body.billing_profile_name == "KQWI-W2SU-BG7-TGB"
assert (
body.billing_profile_properties.billing_profile_display_name
@@ -298,8 +299,9 @@ def test_create_billing_profile_tenant_access(mock_azure: AzureCloudProvider):
)
)
result = mock_azure.create_billing_profile_tenant_access(payload)
body: BillingProfileTenantAccessCSPResult = result.get("body")
body: BillingProfileTenantAccessCSPResult = mock_azure.create_billing_profile_tenant_access(
payload
)
assert (
body.billing_role_assignment_name
== "40000000-aaaa-bbbb-cccc-100000000000_0a5f4926-e3ee-4f47-a6e3-8b0a30a40e3d"
@@ -328,8 +330,9 @@ def test_create_task_order_billing_creation(mock_azure: AzureCloudProvider):
)
)
result = mock_azure.create_task_order_billing_creation(payload)
body: TaskOrderBillingCreationCSPResult = result.get("body")
body: TaskOrderBillingCreationCSPResult = mock_azure.create_task_order_billing_creation(
payload
)
assert (
body.task_order_billing_verify_url
@@ -388,8 +391,9 @@ def test_create_task_order_billing_verification(mock_azure):
)
)
result = mock_azure.create_task_order_billing_verification(payload)
body: TaskOrderBillingVerificationCSPResult = result.get("body")
body: TaskOrderBillingVerificationCSPResult = mock_azure.create_task_order_billing_verification(
payload
)
assert body.billing_profile_name == "KQWI-W2SU-BG7-TGB"
assert (
body.billing_profile_enabled_plan_details.enabled_azure_plans[0].get("skuId")
@@ -428,8 +432,7 @@ def test_create_billing_instruction(mock_azure: AzureCloudProvider):
billing_profile_name="KQWI-W2SU-BG7-TGB",
)
)
result = mock_azure.create_billing_instruction(payload)
body: BillingInstructionCSPResult = result.get("body")
body: BillingInstructionCSPResult = mock_azure.create_billing_instruction(payload)
assert body.reported_clin_name == "TO1:CLIN001"
@@ -455,8 +458,7 @@ def test_create_product_purchase(mock_azure: AzureCloudProvider):
)
)
result = mock_azure.create_product_purchase(payload)
body: ProductPurchaseCSPResult = result.get("body")
body: ProductPurchaseCSPResult = mock_azure.create_product_purchase(payload)
assert (
body.product_purchase_verify_url
== "https://management.azure.com/providers/Microsoft.Billing/billingAccounts/7c89b735-b22b-55c0-ab5a-c624843e8bf6:de4416ce-acc6-44b1-8122-c87c4e903c91_2019-05-31/operationResults/patchBillingProfile_KQWI-W2SU-BG7-TGB:02715576-4118-466c-bca7-b1cd3169ff46?api-version=2019-10-01-preview"
@@ -500,8 +502,9 @@ def test_create_product_purchase_verification(mock_azure):
)
)
result = mock_azure.create_product_purchase_verification(payload)
body: ProductPurchaseVerificationCSPResult = result.get("body")
body: ProductPurchaseVerificationCSPResult = mock_azure.create_product_purchase_verification(
payload
)
assert body.premium_purchase_date == "01/31/2020"

1
tests/mock_azure.py
View File

@@ -63,6 +63,7 @@ def mock_policy():
return Mock(spec=policy)
def mock_azure_exceptions():
from azure.core import exceptions