From 73ddd323652d75b33bc06b016beb7a735321da64 Mon Sep 17 00:00:00 2001 From: tomdds Date: Fri, 31 Jan 2020 11:51:13 -0500 Subject: [PATCH] Only unpack AADP purchase date after verifying response code. --- atst/domain/csp/cloud/azure_cloud_provider.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/atst/domain/csp/cloud/azure_cloud_provider.py b/atst/domain/csp/cloud/azure_cloud_provider.py index 048b1d5b..6b058b11 100644 --- a/atst/domain/csp/cloud/azure_cloud_provider.py +++ b/atst/domain/csp/cloud/azure_cloud_provider.py @@ -572,12 +572,14 @@ class AzureCloudProvider(CloudProviderInterface): result = self.sdk.requests.get( payload.product_purchase_verify_url, headers=auth_header ) - premium_purchase_date = result.json()["product"]["properties"]["purchaseDate"] if result.status_code == 202: # 202 has location/retry after headers return self._ok(ProductPurchaseCSPResult(**result.headers)) elif result.status_code == 200: + premium_purchase_date = result.json()["product"]["properties"][ + "purchaseDate" + ] return self._ok( ProductPurchaseVerificationCSPResult( premium_purchase_date=premium_purchase_date