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

Only unpack AADP purchase date after verifying response code.

Browse Source
This commit is contained in:
tomdds 2020-01-31 11:51:13 -05:00
parent 6b8ea41925
commit 73ddd32365
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
atst/domain/csp/cloud/azure_cloud_provider.py
View File

@ -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