# coding: utf-8 # ------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for # license information. # -------------------------------------------------------------------------- # TEST SCENARIO COVERAGE # ---------------------- # Methods Total : 16 # Methods Covered : 16 # Examples Total : 21 # Examples Tested : 21 # Coverage % : 100 # ---------------------- # current method cover: 15/16 import os import unittest import azure.mgmt.databox from devtools_testutils import AzureMgmtTestCase, ResourceGroupPreparer AZURE_LOCATION = "eastus" class MgmtDataBoxTest(AzureMgmtTestCase): def setUp(self): super(MgmtDataBoxTest, self).setUp() self.mgmt_client = self.create_mgmt_client(azure.mgmt.databox.DataBoxManagementClient) @unittest.skip("unavailable in track2") @ResourceGroupPreparer(location=AZURE_LOCATION) def test_databox(self, resource_group): SUBSCRIPTION_ID = None if self.is_live: SUBSCRIPTION_ID = os.environ.get("AZURE_SUBSCRIPTION_ID", None) if not SUBSCRIPTION_ID: SUBSCRIPTION_ID = self.settings.SUBSCRIPTION_ID RESOURCE_GROUP = resource_group.name STORAGE_ACCOUNT_NAME = "databoxaccountabc" JOB_NAME = "testjob" LOCATION_NAME = "westus" # JobsCreate[put] BODY = { "details": { "job_details_type": "DataBox", "contact_details": { "contact_name": "Public SDK Test", "phone": "1234567890", "phone_extension": "1234", "email_list": ["testing@microsoft.com"], }, "shipping_address": { "street_address1": "16 TOWNSEND ST", "street_address2": "Unit 1", "city": "San Francisco", "state_or_province": "CA", "country": "US", "postal_code": "94107", "company_name": "Microsoft", "address_type": "Commercial", }, "destination_account_details": [ { "storage_account_id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Storage/storageAccounts/" + STORAGE_ACCOUNT_NAME + "", "data_destination_type": "StorageAccount", } ], }, "location": "westus", "sku": {"name": "DataBox"}, } result = self.mgmt_client.jobs.create(resource_group.name, JOB_NAME, BODY) result = result.result() # JobsGet5[get] result = self.mgmt_client.jobs.get(resource_group.name, JOB_NAME) # JobsGet4[get] result = self.mgmt_client.jobs.get(resource_group.name, JOB_NAME) # JobsGet3[get] result = self.mgmt_client.jobs.get(resource_group.name, JOB_NAME) # JobsGet2[get] result = self.mgmt_client.jobs.get(resource_group.name, JOB_NAME) # JobsGet1[get] result = self.mgmt_client.jobs.get(resource_group.name, JOB_NAME) # JobsGet[get] result = self.mgmt_client.jobs.get(resource_group.name, JOB_NAME) # JobsListByResourceGroup[get] result = self.mgmt_client.jobs.list_by_resource_group(resource_group.name) # JobsList[get] result = self.mgmt_client.jobs.list() # OperationsGet[get] result = self.mgmt_client.operations.list() # ServiceValidateInputsByResourceGroup[post] BODY = { "validation_category": "JobCreationValidation", "individual_request_details": [ { "validation_type": "ValidateDataDestinationDetails", "location": "westus", "destination_account_details": [ { "storage_account_id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Storage/storageAccounts/" + STORAGE_ACCOUNT_NAME + "", "data_destination_type": "StorageAccount", } ], }, { "validation_type": "ValidateAddress", "shipping_address": { "street_address1": "16 TOWNSEND ST", "street_address2": "Unit 1", "city": "San Francisco", "state_or_province": "CA", "country": "US", "postal_code": "94107", "company_name": "Microsoft", "address_type": "Commercial", }, "device_type": "DataBox", }, ], } result = self.mgmt_client.service.validate_inputs_by_resource_group(resource_group.name, LOCATION_NAME, BODY) # AvailableSkusByResourceGroup[post] BODY = {"country": "US", "location": "westus", "transfer_type": "ImportToAzure"} result = self.mgmt_client.service.list_available_skus_by_resource_group( resource_group.name, LOCATION_NAME, BODY ) """ # BookShipmentPickupPost[post] now = dt.datetime.now() BODY = { # For new test, change the start time as current date # and end time as start_time + 2 days "start_time": now, "end_time": now + dt.timedelta(days=2), "shipment_location": "Front desk" } self.mgmt_client.jobs.book_shipment_pick_up(resource_group.name, JOB_NAME, BODY) """ # JobsListCredentials[post] result = self.mgmt_client.jobs.list_credentials(resource_group.name, JOB_NAME) # JobsPatch[patch] BODY = { "details": { "contact_details": { "contact_name": "Update Job", "phone": "1234567890", "phone_extension": "1234", "email_list": ["testing@microsoft.com"], }, "shipping_address": { "street_address1": "16 TOWNSEND ST", "street_address2": "Unit 1", "city": "San Francisco", "state_or_province": "CA", "country": "US", "postal_code": "94107", "company_name": "Microsoft", "address_type": "Commercial", }, } } result = self.mgmt_client.jobs.update(resource_group.name, JOB_NAME, BODY) result = result.result() # ServiceRegionConfiguration[post] # TODO: SKUs are not available in live test # BODY = { # "storage_location": "westus", # "sku_name": "DataBox" # } BODY = None result = self.mgmt_client.service.region_configuration(LOCATION_NAME, BODY) # ValidateAddressPost[post] BODY = { "validation_type": "ValidateAddress", "shipping_address": { "street_address1": "16 TOWNSEND ST", "street_address2": "Unit 1", "city": "San Francisco", "state_or_province": "CA", "country": "US", "postal_code": "94107", "company_name": "Microsoft", "address_type": "Commercial", }, "device_type": "DataBox", } result = self.mgmt_client.service.validate_address_method(LOCATION_NAME, BODY) # ServiceValidateInputs[post] BODY = { "validation_category": "JobCreationValidation", "individual_request_details": [ { "validation_type": "ValidateDataDestinationDetails", "location": "westus", "destination_account_details": [ { "storage_account_id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Storage/storageAccounts/" + STORAGE_ACCOUNT_NAME + "", "data_destination_type": "StorageAccount", } ], }, { "validation_type": "ValidateAddress", "shipping_address": { "street_address1": "16 TOWNSEND ST", "street_address2": "Unit 1", "city": "San Francisco", "state_or_province": "CA", "country": "US", "postal_code": "94107", "company_name": "Microsoft", "address_type": "Commercial", }, "device_type": "DataBox", }, ], } result = self.mgmt_client.service.validate_inputs(LOCATION_NAME, BODY) # AvailableSkusPost[post] BODY = {"country": "US", "location": "westus", "transfer_type": "ImportToAzure"} result = self.mgmt_client.service.list_available_skus(LOCATION_NAME, BODY) # JobsCancelPost[post] BODY = {"reason": "CancelTest"} result = self.mgmt_client.jobs.cancel(resource_group.name, JOB_NAME, BODY) # JobsDelete[delete] result = self.mgmt_client.jobs.delete(resource_group.name, JOB_NAME) result = result.result() # ------------------------------------------------------------------------------ if __name__ == "__main__": unittest.main()