premium product purchase unit tests.

atst/domain/csp/cloud/azure_cloud_provider.py

@@ -506,12 +506,21 @@ class AzureCloudProvider(CloudProviderInterface):
                 "Could not resolve token for aad premium product purchase"
             )
 
-        create_product_purchase_body = payload.dict(by_alias=True)
+        payload_as_dict = payload.dict(by_alias=True)
+
+        create_product_purchase_body =  {
+            "type": "AADPremium",
+            "sku": "AADP1",
+            "productProperties": {
+                "beneficiaryTenantId": payload_as_dict["productProperties"]["beneficiaryTenantId"],
+            },
+            "quantity": payload_as_dict.get("quantity"),
+        }
         create_product_purchase_headers = {
             "Authorization": f"Bearer {sp_token}",
         }
 
-        product_purchase_url = f"https://management.azure.com//providers/Microsoft.Billing/billingAccounts/{payload.billing_account_name}/billingProfiles/{payload.billing_profile_name}/products?api-version=2019-10-01-preview"
+        product_purchase_url = f"https://management.azure.com/providers/Microsoft.Billing/billingAccounts/{payload.billing_account_name}/billingProfiles/{payload.billing_profile_name}/products?api-version=2019-10-01-preview"
 
         result = self.sdk.requests.post(
             product_purchase_url,
@@ -524,7 +533,7 @@ class AzureCloudProvider(CloudProviderInterface):
             return self._ok(ProductPurchaseCSPResult(**result.headers))
         elif result.status_code == 200:
             # NB: Swagger docs imply call can sometimes resolve immediately
-            return self._ok(ProductPurchaseCSPResult(**result.json()))
+            return self._ok(ProductPurchaseVerificationCSPResult(**result.json()))
         else:
             return self._error(result.json())
 
@@ -535,7 +544,7 @@ class AzureCloudProvider(CloudProviderInterface):
         sp_token = self._get_sp_token(payload.creds)
         if sp_token is None:
             raise AuthenticationException(
-                "Could not resolve token for task order billing validation"
+                "Could not resolve token for aad premium product purchase validation"
             )
 
         auth_header = {
@@ -545,12 +554,13 @@ 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:
-            return self._ok(ProductPurchaseVerificationCSPResult(**result.json()))
+            return self._ok(ProductPurchaseVerificationCSPResult(premium_purchase_date=premium_purchase_date))
         else:
             return self._error(result.json())
 

atst/domain/csp/cloud/mock_cloud_provider.py

@@ -282,15 +282,22 @@ class MockCloudProvider(CloudProviderInterface):
         )
 
 
-    def create_product_purchase(self, payload: ProductPurchaseCSPPayload):
+    def create_product_purchase(
+        self, payload: ProductPurchaseCSPPayload
+    ):
         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 ProductPurchaseCSPResult(
-            **{"Location": "https://somelocation", "Retry-After": "10"}
+            **dict(
+                product_purchase_verify_url="https://zombo.com",
+                product_purchase_retry_after=10,
+            )
         )
 
+
+
     def create_product_purchase_verification(
         self, payload: ProductPurchaseVerificationCSPPayload
     ):
@@ -299,11 +306,9 @@ class MockCloudProvider(CloudProviderInterface):
         self._maybe_raise(self.UNAUTHORIZED_RATE, self.AUTHORIZATION_EXCEPTION)
 
         return ProductPurchaseVerificationCSPResult(
-            **{
-            }
+            **dict(premium_purchase_date="2020-01-30T18:57:05.981Z")
         )
 
-
     def create_or_update_user(self, auth_credentials, user_info, csp_role_id):
         self._authorize(auth_credentials)
 

atst/domain/csp/cloud/models.py

@@ -342,25 +342,15 @@ class KeyVaultCredentials(BaseModel):
 
         return values
 
-class AadPremiumProductParameter(AliasModel):
+
+class ProductPurchaseCSPPayload(BaseCSPPayload):
     type: str
     sku: str
     quantity: int
     productProperties: Dict
-
-    #{
-    #  "type": "string",
-    #  "sku": "string",
-    #  "quantity": 0,
-    #  "productProperties": {
-    #    "beneficiaryTenantId": "string"
-    #  }
-    #}
-
-class ProductPurchaseCSPPayload(BaseCSPPayload):
     billing_account_name: str
     billing_profile_name: str
-    parameters: List[AadPremiumProductParameter]
+
 
 class ProductPurchaseCSPResult(AliasModel):
     product_purchase_verify_url: str
@@ -378,13 +368,4 @@ class ProductPurchaseVerificationCSPPayload(BaseCSPPayload):
 
 
 class ProductPurchaseVerificationCSPResult(AliasModel):
-    billing_profile_id: str
-    billing_profile_name: str
-    billing_profile_enabled_plan_details: BillingProfileEnabledPlanDetails
-
-    class Config:
-        fields = {
-            "billing_profile_id": "id",
-            "billing_profile_name": "name",
-            "billing_profile_enabled_plan_details": "properties",
-        }
+    premium_purchase_date: str