# 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 : 147 # Methods Covered : 146 # Examples Total : 167 # Examples Tested : 166 # Coverage % : 98.72499898162857 # ---------------------- # current coverage: 85 import unittest import azure.mgmt.automation from devtools_testutils import AzureMgmtTestCase, ResourceGroupPreparer AZURE_LOCATION = 'eastus' class MgmtAutomationClientTest(AzureMgmtTestCase): def setUp(self): super(MgmtAutomationClientTest, self).setUp() self.mgmt_client = self.create_mgmt_client( azure.mgmt.automation.AutomationClient ) @ResourceGroupPreparer(location=AZURE_LOCATION) def test_automation(self, resource_group): SERVICE_NAME = "myapimrndxyz" AUTOMATION_ACCOUNT_NAME = 'myAutomationAccount9' JOB_NAME = "job1" MODULE_NAME = "module1" WEBHOOK_NAME = "webhook1" WATCHER_NAME = "MyTestWatcher" SCHEDULE_NAME = "mySchedule" # Create or update automation account[put] BODY = { "sku": { "name": "Free" }, "name": "myAutomationAccount9", "location": "East US 2" } result = self.mgmt_client.automation_account.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, BODY) RUNBOOK_DRAFT_NAME = "runbook_draft" # Create runbook as draft[put] BODY = { "log_verbose": False, "log_progress": False, "runbook_type": "PowerShellWorkflow", "draft": {}, "description": "Description of the Runbook", "name": RUNBOOK_DRAFT_NAME, "location": "East US 2", "tags": { "tag01": "value01", "tag02": "value02", } } result = self.mgmt_client.runbook.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_DRAFT_NAME, BODY) # Create or update a module[put] BODY = { "content_link": { "uri": "https://teststorage.blob.core.windows.net/dsccomposite/OmsCompositeResources.zip", "content_hash": { "algorithm": "sha265", "value": "07E108A962B81DD9C9BAA89BB47C0F6EE52B29E83758B07795E408D258B2B87A" }, "version": "1.0.0.0" } } result = self.mgmt_client.module.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME, BODY) RUNBOOK_NAME = "Get-AzureVMTutorial" # Create or update runbook and publish it[put] BODY = { "log_verbose": False, "log_progress": True, "runbook_type": "PowerShellWorkflow", "publish_content_link": { "uri": "https://raw.githubusercontent.com/Azure/azure-quickstart-templates/0.0.0.3/101-automation-runbook-getvms/Runbooks/Get-AzureVMTutorial.ps1", "content_hash": { "algorithm": "SHA256", "value": "4fab357cab33adbe9af72ae4b1203e601e30e44de271616e376c08218fd10d1c" }, }, "description": "Description of the Runbook", "log_activity_trace": "1", "name": RUNBOOK_NAME, "location": "East US 2", "tags": { "tag01": "value01", "tag02": "value02" } } result = self.mgmt_client.runbook.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, BODY) # Create job[put] BODY = { "runbook": { "name": RUNBOOK_NAME }, "parameters": { "key01": "value01", "key02": "value02" }, "run_on": "" } result = self.mgmt_client.job.create(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME, BODY) # Create or update webhook[put] BODY = { "name": "TestWebhook", "is_enabled": True, "uri": "https://s1events.azure-automation.net/webhooks?token=7u3KfQvM1vUPWaDMFRv2%2fAA4Jqx8QwS8aBuyO6Xsdcw%3d", "expiry_time": "2021-03-29T22:18:13.7002872Z", "runbook": { "name": RUNBOOK_NAME } } result = self.mgmt_client.webhook.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, WEBHOOK_NAME, BODY) """ TODO: response 500 error # Create or update watcher[put] BODY = { "name": "MyTestWatcher", "type": None, "location": None, "tags": {}, "etag": None, "execution_frequency_in_seconds": "60", "script_name": RUNBOOK_NAME, "description": "This is a test watcher.", "script_run_on": "MyTestHybridWorkerGroup", "scriptParameters": None, "creation_time": "2020-03-01T11:22:47.7333333+00:00", "lastModifiedBy": None, "last_modified_time": "2020-03-01T11:22:47.7333333+00:00" } result = self.mgmt_client.watcher.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, WATCHER_NAME, BODY) """ # Create or update a schedule[put] BODY = { "name": "mySchedule", "description": "my description of schedule goes here", "start_time": "2020-12-27T17:28:57.2494819Z", "expiry_time": "2021-01-01T17:28:57.2494819Z", "interval": "1", "frequency": "Hour" } result = self.mgmt_client.schedule.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, SCHEDULE_NAME, BODY) VARIABLE_NAME = "sampleVariable" # Create or update a variable[put] BODY = { "name": "sampleVariable", "value": "\"ComputerName.domain.com\"", "description": "my description", "is_encrypted": False } result = self.mgmt_client.variable.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, VARIABLE_NAME, BODY) CREDENTIAL_NAME = "myCredential" # Create a credential[put] BODY = { "name": "myCredential", "user_name": "mylingaiah", "password": "myPassw0rd", "description": "my description goes here" } result = self.mgmt_client.credential.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CREDENTIAL_NAME, BODY) """ CERTIFICATE_NAME = "testCert" # Create or update a certificate[put] BODY = { "name": "testCert", "base64value": "base 64 value of cert", "description": "Sample Cert", "thumbprint": "thumbprint of cert", "is_exportable": False } result = self.mgmt_client.certificate.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CERTIFICATE_NAME, BODY) CONNECTION_NAME = "mysConnection" # Create or update connection[put] BODY = { "name": "mysConnection", "description": "my description goes here", "connection_type": { "name": "Azure" }, "field_definition_values": { "automation_certificate_name": CERTIFICATE_NAME, "subscription_id": "subid" } } result = self.mgmt_client.connection.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_NAME, BODY) JOB_SCHEDULE_NAME = "ScheduleNameGoesHere332204b5-debe-4348-a5c7-6357457189f2" # Create a job schedule[put] BODY = { "schedule": { "name": "ScheduleNameGoesHere332204b5-debe-4348-a5c7-6357457189f2" }, "runbook": { "name": RUNBOOK_NAME }, "parameters": { "jobscheduletag01": "jobschedulevalue01", "jobscheduletag02": "jobschedulevalue02" } } result = self.mgmt_client.job_schedule.create(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_SCHEDULE_NAME, BODY) CONFIGURATION_NAME = "SetupServer" # Create or Update Configuration[put] BODY = { "source": { "hash": { "algorithm": "sha256", "value": "A9E5DB56BA21513F61E0B3868816FDC6D4DF5131F5617D7FF0D769674BD5072F" }, "type": "embeddedContent", "value": "Configuration SetupServer {\r\n Node localhost {\r\n WindowsFeature IIS {\r\n Name = \"Web-Server\";\r\n Ensure = \"Present\"\r\n }\r\n }\r\n}" }, "description": "sample configuration" "name": "SetupServer", "location": "East US 2" } result = self.mgmt_client.dsc_configuration.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONFIGURATION_NAME, BODY) SOURCE_CONTROL_NAME = "sourceControl" # Create or update a source control[put] BODY = { "repo_url": "https://sampleUser.visualstudio.com/myProject/_git/myRepository", "branch": "master", "folder_path": "/folderOne/folderTwo", "auto_sync": True, "publish_runbook": True, "source_type": "VsoGit", "security_token": { "access_token": "3a326f7a0dcd343ea58fee21f2fd5fb4c1234567", "token_type": "PersonalAccessToken" }, "description": "my description" } result = self.mgmt_client.source_control.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME, BODY) PYTHON2PACKAGE_NAME = "pythonPackage" # Create or update a python 2 package[put] BODY = { "content_link": { "uri": "https://teststorage.blob.core.windows.net/dsccomposite/OmsCompositeResources.zip", "content_hash": { "algorithm": "sha265", "value": "07E108A962B81DD9C9BAA89BB47C0F6EE52B29E83758B07795E408D258B2B87A" }, "version": "1.0.0.0" } } result = self.mgmt_client.python2package.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, PYTHON2PACKAGE_NAME, BODY) COMPILATIONJOB_NAME = "compilationJob" # Create or update a DSC Compilation job[put] BODY = { "configuration": { "name": "SetupServer" } } result = self.mgmt_client.dsc_compilation_job.create(resource_group.name, AUTOMATION_ACCOUNT_NAME, COMPILATIONJOB_NAME, BODY) result = result.result() CONNECTION_TYPE_NAME = "myCT" # Create or update connection type[put] BODY = { "name": "myCT", "is_global": False, "field_definitions": { "my_string_field": { "is_encrypted": False, "is_optional": False, "type": "string" }, "my_bool_field": { "is_encrypted": False, "is_optional": False, "type": "bool" }, "my_string_field_encrypted": { "is_encrypted": True, "is_optional": False, "type": "string" } } } result = self.mgmt_client.connection_type.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_TYPE_NAME, BODY) NODE_CONFIGURATION_NAME = "configName.nodeConfigName" # Create node configuration[put] BODY = { "name": "configName.nodeConfigName", "source": { "hash": { "algorithm": "sha256", "value": "6DE256A57F01BFA29B88696D5E77A383D6E61484C7686E8DB955FA10ACE9FFE5" }, "type": "embeddedContent", "value": "\r\ninstance of MSFT_RoleResource as $MSFT_RoleResource1ref\r\n{\r\nResourceID = \"[WindowsFeature]IIS\";\r\n Ensure = \"Present\";\r\n SourceInfo = \"::3::32::WindowsFeature\";\r\n Name = \"Web-Server\";\r\n ModuleName = \"PsDesiredStateConfiguration\";\r\n\r\nModuleVersion = \"1.0\";\r\r\n ConfigurationName = \"configName\";\r\r\n};\r\ninstance of OMI_ConfigurationDocument\r\n\r\r\n {\r\n Version=\"2.0.0\";\r\n \r\r\n MinimumCompatibleVersion = \"1.0.0\";\r\n \r\r\n CompatibleVersionAdditionalProperties= {\"Omi_BaseResource:ConfigurationName\"};\r\n \r\r\n Author=\"weijiel\";\r\n \r\r\n GenerationDate=\"03/30/2017 13:40:25\";\r\n \r\r\n GenerationHost=\"TEST-BACKEND\";\r\n \r\r\n Name=\"configName\";\r\n\r\r\n };\r\n", "version": "1.0" }, "increment_node_configuration_build": True, "configuration": { "name": "configName" } } result = self.mgmt_client.dsc_node_configuration.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_CONFIGURATION_NAME, BODY) result = result.result() DRAFT_NAME = "Get-AzureVMTutorial" # Create or update runbook draft[put] result = self.mgmt_client.runbook_draft.replace_content(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) result = result.result() # Create test job[put] BODY = { "key01": "value01", "key02": "value02" "run_on": "" } result = self.mgmt_client.test_job.create(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME, BODY) SOFTWARE_UPDATE_CONFIGURATION_NAME = "softconfig" # Create software update configuration[put] BODY = { "update_configuration": { "operating_system": "Windows", "duration": "PT2H0M", "windows": { "excluded_kb_numbers": [ "168934", "168973" ], "included_update_classifications": "Critical", "reboot_setting": "IfRequired" }, "azure_virtual_machines": [ "/subscriptions/5ae68d89-69a4-454f-b5ce-e443cc4e0067/resourceGroups/myresources/providers/Microsoft.Compute/virtualMachines/vm-01", "/subscriptions/5ae68d89-69a4-454f-b5ce-e443cc4e0067/resourceGroups/myresources/providers/Microsoft.Compute/virtualMachines/vm-02", "/subscriptions/5ae68d89-69a4-454f-b5ce-e443cc4e0067/resourceGroups/myresources/providers/Microsoft.Compute/virtualMachines/vm-03" ], "non_azure_computer_names": [ "box1.contoso.com", "box2.contoso.com" ], "targets": { "azure_queries": [ { "scope": [ "/subscriptions/5ae68d89-69a4-454f-b5ce-e443cc4e0067/resourceGroups/myresources", "/subscriptions/5ae68d89-69a4-454f-b5ce-e443cc4e0067" ], "tag_settings": { "tags": [ { "tag1": [ "tag1Value1", "tag1Value2", "tag1Value3" ] }, { "tag2": [ "tag2Value1", "tag2Value2", "tag2Value3" ] } ], "filter_operator": "All" }, "locations": [ "Japan East", "UK South" ] } ], "non_azure_queries": [ { "function_alias": "SavedSearch1", "workspace_id": "WorkspaceId1" }, { "function_alias": "SavedSearch2", "workspace_id": "WorkspaceId2" } ] } }, "schedule_info": { "frequency": "Hour", "start_time": "2017-10-19T12:22:57+00:00", "time_zone": "America/Los_Angeles", "interval": "1", "expiry_time": "2018-11-09T11:22:57+00:00", "advanced_schedule": { "week_days": [ "Monday", "Thursday" ] } }, "tasks": { "pre_task": { "source": "HelloWorld", "parameters": { "computername": "Computer1" } }, "post_task": { "source": "GetCache" } } } result = self.mgmt_client.software_update_configurations.create(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOFTWARE_UPDATE_CONFIGURATION_NAME, BODY) SOURCE_CONTROL_SYNC_JOB_NAME = "sourceControlJobName" # Create or update a source control sync job[put] BODY = { "commit_id": "9de0980bfb45026a3d97a1b0522d98a9f604226e" } result = self.mgmt_client.source_control_sync_job.create(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME, SOURCE_CONTROL_SYNC_JOB_NAME, BODY) # # Get a sync job stream identified by sync job stream id.[get] # result = self.mgmt_client.source_control_sync_job_streams.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME, SOURCE_CONTROL_SYNC_JOB_NAME, STREAM_NAME) # Get a list of sync job streams identified by sync job id[get] result = self.mgmt_client.source_control_sync_job_streams.list_by_sync_job(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME, SOURCE_CONTROL_SYNC_JOB_NAME) # Get a source control sync job by job id[get] result = self.mgmt_client.source_control_sync_job.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME, SOURCE_CONTROL_SYNC_JOB_NAME) # # Get software update configuration machine run[get] # result = self.mgmt_client.software_update_configuration_machine_runs.get_by_id(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOFTWARE_UPDATE_CONFIGURATION_MACHINE_RUN_NAME) # # Get test job stream[get] # result = self.mgmt_client.test_job_streams.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME, STREAM_NAME) # # Get software update configuration runs by Id[get] # result = self.mgmt_client.software_update_configuration_runs.get_by_id(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOFTWARE_UPDATE_CONFIGURATION_RUN_NAME) # # Get a list of fields of a given type[get] # result = self.mgmt_client.object_data_types.list_fields_by_module_and_type(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME, OBJECT_DATA_TYPE_NAME) # Get software update configuration by name[get] result = self.mgmt_client.software_update_configurations.get_by_name(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOFTWARE_UPDATE_CONFIGURATION_NAME) # # Get a DSC Compilation job stream by job stream id[get] # result = self.mgmt_client.dsc_compilation_job.get_stream(resource_group.name, AUTOMATION_ACCOUNT_NAME, COMPILATIONJOB_NAME, STREAM_NAME) # # Get a hybrid worker group[get] # result = self.mgmt_client.hybrid_runbook_worker_group.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, HYBRID_RUNBOOK_WORKER_GROUP_NAME) # Get a list of source control sync jobs[get] result = self.mgmt_client.source_control_sync_job.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME) """ # List job streams by job name[get] result = self.mgmt_client.job_stream.list_by_job(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) """ # # Get Activity in a module[get] # result = self.mgmt_client.activity.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME, ACTIVITY_NAME) # # Get content of node[get] # result = self.mgmt_client.node_reports.get_content(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_NAME, REPORT_NAME) # # Get a list of fields of a given type[get] # result = self.mgmt_client.object_data_types.list_fields_by_module_and_type(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME, OBJECT_DATA_TYPE_NAME) # # Get a list of fields of a given type across all accessible modules[get] # result = self.mgmt_client.object_data_types.list_fields_by_type(resource_group.name, AUTOMATION_ACCOUNT_NAME, OBJECT_DATA_TYPE_NAME) # Get runbook draft content[get] result = self.mgmt_client.runbook_draft.get_content(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) # # Get test job[get] # result = self.mgmt_client.test_job.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) # List DSC Compilation job streams[get] result = self.mgmt_client.dsc_compilation_job_stream.list_by_job(resource_group.name, AUTOMATION_ACCOUNT_NAME, COMPILATIONJOB_NAME) # Get a DSC node configuration[get] result = self.mgmt_client.dsc_node_configuration.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_CONFIGURATION_NAME) # Get DSC Configuration content[get] result = self.mgmt_client.dsc_configuration.get_content(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONFIGURATION_NAME) # # Get job stream[get] # result = self.mgmt_client.job_stream.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME, STREAM_NAME) # Get connection type[get] result = self.mgmt_client.connection_type.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_TYPE_NAME) # Get a DSC Compilation job[get] result = self.mgmt_client.dsc_compilation_job.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, COMPILATIONJOB_NAME) # Get a python 2 package[get] result = self.mgmt_client.python2package.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, PYTHON2PACKAGE_NAME) # Get a source control[get] result = self.mgmt_client.source_control.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME) # Get a DSC Configuration[get] result = self.mgmt_client.dsc_configuration.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONFIGURATION_NAME) """ # List software update configuration machine runs for a specific software update configuration run[get] result = self.mgmt_client.software_update_configuration_machine_runs.list(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List software update configuration machine runs[get] result = self.mgmt_client.software_update_configuration_runs.list(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List activities by a module[get] result = self.mgmt_client.activity.list_by_module(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME) """ # Get a job schedule[get] result = self.mgmt_client.job_schedule.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_SCHEDULE_NAME) """ # Get runbook content[get] # result = self.mgmt_client.runbook.get_content(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME) """ # Get a certificate[get] result = self.mgmt_client.certificate.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, CERTIFICATE_NAME) """ # Get Job Runbook Content[get] # result = self.mgmt_client.job.get_runbook_content(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) # Get runbook draft[get] result = self.mgmt_client.runbook_draft.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_DRAFT_NAME) """ # Get a connection[get] result = self.mgmt_client.connection.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_NAME) """ # Get a credential[get] result = self.mgmt_client.credential.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, CREDENTIAL_NAME) """ # # Get node's status counts[get] # result = self.mgmt_client.node_count_information.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODECOUNT_NAME) # # Get node's node configuration counts[get] # result = self.mgmt_client.node_count_information.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODECOUNT_NAME) """ # List software update configuration machine run with status equal to 'Failed'[get] result = self.mgmt_client.software_update_configuration_runs.list(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List software update configuration machine runs[get] result = self.mgmt_client.software_update_configuration_runs.list(resource_group.name, AUTOMATION_ACCOUNT_NAME) # # List DSC reports by node id.[get] # result = self.mgmt_client.node_reports.list_by_node(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_NAME) # Get a variable[get] result = self.mgmt_client.variable.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, VARIABLE_NAME) # Get a schedule[get] result = self.mgmt_client.schedule.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, SCHEDULE_NAME) """ # Get the agent registration information[get] result = self.mgmt_client.agent_registration_information.get(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List software update configurations Targeting a specific azure virtual machine[get] result = self.mgmt_client.software_update_configurations.list(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List software update configurations[get] result = self.mgmt_client.software_update_configurations.list(resource_group.name, AUTOMATION_ACCOUNT_NAME) """ # Get runbook[get] result = self.mgmt_client.runbook.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME) # List job streams by job name[get] result = self.mgmt_client.job_stream.list_by_job(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) """ # Get watcher[get] result = self.mgmt_client.watcher.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, WATCHER_NAME) """ # Get webhook[get] result = self.mgmt_client.webhook.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, WEBHOOK_NAME) # Get Job Output[get] result = self.mgmt_client.job.get_output(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) """ # List hybrid worker groups by Automation Account[get] result = self.mgmt_client.hybrid_runbook_worker_group.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) """ # Get a module[get] result = self.mgmt_client.module.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME) """ # # Get a node[get] # result = self.mgmt_client.dsc_node.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_NAME) """ # Get job[get] result = self.mgmt_client.job.get(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) # List Paged DSC node configurations by Automation Account with name filter[get] result = self.mgmt_client.dsc_node_configuration.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List DSC node configurations by Automation Account[get] result = self.mgmt_client.dsc_node_configuration.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC node configurations by Automation Account with no filter[get] result = self.mgmt_client.dsc_node_configuration.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # TODO: AttributeError: 'AutomationClient' object has no attribute 'python2package' # # List python 2 packages by automation account[get] # result = self.mgmt_client.python2package.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get connection types, first 100[get] result = self.mgmt_client.connection_type.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List DSC Compilation job in Automation Account[get] result = self.mgmt_client.dsc_compilation_job.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get connection types, next 100[get] result = self.mgmt_client.connection_type.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get the linked workspace of an automation account[get] result = self.mgmt_client.linked_workspace.get(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get DSC Configuration[get] result = self.mgmt_client.dsc_configuration.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC Configurations with name filter[get] result = self.mgmt_client.dsc_configuration.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List sourceControls[get] result = self.mgmt_client.source_control.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC Configurations with no filter[get] result = self.mgmt_client.dsc_configuration.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List certificates[get] result = self.mgmt_client.certificate.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List all job schedules by automation account[get] result = self.mgmt_client.job_schedule.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List credentials by automation account, next 100[get] result = self.mgmt_client.credential.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List credentials by automation account, first 100[get] result = self.mgmt_client.credential.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List connections by automation account, next 100[get] result = self.mgmt_client.connection.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List connections by automation account, first 100[get] result = self.mgmt_client.connection.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get statistics of an automation account[get] result = self.mgmt_client.statistics.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List variables, First 100[get] result = self.mgmt_client.variable.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List variables, Next 100[get] result = self.mgmt_client.variable.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List schedules by automation account, first 100[get] result = self.mgmt_client.schedule.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List schedules by automation account, next 100[get] result = self.mgmt_client.schedule.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List watchers by Automation Account[get] result = self.mgmt_client.watcher.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List runbooks by automation account[get] result = self.mgmt_client.runbook.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List webhooks by Automation Account[get] result = self.mgmt_client.webhook.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List modules by automation account[get] result = self.mgmt_client.module.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get usages of an automation account[get] result = self.mgmt_client.usages.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes by Automation Account with version filter[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes by Automation Account with name filter[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes by Automation Account with no filters[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List DSC nodes by Automation Account[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes by Automation Account with node status filter[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes by Automation Account with Node Configuration Custom filter[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes by Automation Account where Node Configurations are not assigned filter[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List Paged DSC nodes with filters separated by 'and'[get] result = self.mgmt_client.dsc_node.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List jobs by automation account[get] result = self.mgmt_client.job.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Get automation account[get] result = self.mgmt_client.automation_account.get(resource_group.name, AUTOMATION_ACCOUNT_NAME) # List automation accounts by resource group[get] result = self.mgmt_client.automation_account.list_by_resource_group(resource_group.name) # List automation accounts by resource group[get] result = self.mgmt_client.automation_account.list_by_resource_group(resource_group.name) """ # # Regenerate registration key[post] # BODY = { # "key_name": "primary" # } # result = self.mgmt_client.agent_registration_information.regenerate_key(resource_group.name, AUTOMATION_ACCOUNT_NAME, AGENT_REGISTRATION_INFORMATION_NAME, BODY) # # Update hybrid worker group[patch] # BODY = { # "credential": { # "name": "myRunAsCredentialName" # } # } # result = self.mgmt_client.hybrid_runbook_worker_group.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, HYBRID_RUNBOOK_WORKER_GROUP_NAME, BODY) # Suspend test job[post] result = self.mgmt_client.test_job.suspend(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) # Resume test job[post] result = self.mgmt_client.test_job.resume(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) # Stop test job[post] result = self.mgmt_client.test_job.stop(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) # Undo draft edit to last known published state[post] result = self.mgmt_client.runbook_draft.undo_edit(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, DRAFT_NAME) """ # Update a module[patch] BODY = { "content_link": { "uri": "https://teststorage.blob.core.windows.net/mycontainer/MyModule.zip", "content_hash": { "algorithm": "sha265", "value": "07E108A962B81DD9C9BAA89BB47C0F6EE52B29E83758B07795E408D258B2B87A" }, "version": "1.0.0.0" } } result = self.mgmt_client.module.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME, BODY) """ # Update a source control[patch] BODY = { "branch": "master", "folder_path": "/folderOne/folderTwo", "auto_sync": True, "publish_runbook": True, "security_token": { "access_token": "3a326f7a0dcd343ea58fee21f2fd5fb4c1234567", "token_type": "PersonalAccessToken" }, "description": "my description" } result = self.mgmt_client.source_control.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME, BODY) # Create or Update Configuration[put] BODY = { "source": { "hash": { "algorithm": "sha256", "value": "A9E5DB56BA21513F61E0B3868816FDC6D4DF5131F5617D7FF0D769674BD5072F" }, "type": "embeddedContent", "value": "Configuration SetupServer {\r\n Node localhost {\r\n WindowsFeature IIS {\r\n Name = \"Web-Server\";\r\n Ensure = \"Present\"\r\n }\r\n }\r\n}" }, "description": "sample configuration", "name": "SetupServer", "location": "East US 2" } result = self.mgmt_client.dsc_configuration.create_or_update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONFIGURATION_NAME, BODY) # Update a certificate[patch] BODY = { "name": "testCert", "properties": { "description": "sample certificate. Description updated" } } result = self.mgmt_client.certificate.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CERTIFICATE_NAME, BODY) """ # Publish runbook draft[post] result = self.mgmt_client.runbook.begin_publish(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_DRAFT_NAME) result = result.result() # Update a credential[patch] BODY = { "name": "myCredential", "user_name": "mylingaiah", "password": "myPassw0rd3", "description": "my description goes here" } result = self.mgmt_client.credential.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CREDENTIAL_NAME, BODY) """ # Update a connection[patch] BODY = { "name": "myConnection", "description": "my description goes here", "field_definition_values": { "automation_certificate_name": "myCertificateName", "subscription_id": "b5e4748c-f69a-467c-8749-e2f9c8cd3009" } } result = self.mgmt_client.connection.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_NAME, BODY) # Start Watcher[post] result = self.mgmt_client.watcher.start(resource_group.name, AUTOMATION_ACCOUNT_NAME, WATCHER_NAME) # Start Watcher[post] result = self.mgmt_client.watcher.start(resource_group.name, AUTOMATION_ACCOUNT_NAME, WATCHER_NAME) """ # Update a variable[patch] BODY = { "name": "sampleVariable", "properties": { "value": "\"ComputerName3.domain.com\"" } } result = self.mgmt_client.variable.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, VARIABLE_NAME, BODY) # Update a schedule[patch] BODY = { "name": "mySchedule", "description": "my updated description of schedule goes here", "is_enabled": False } result = self.mgmt_client.schedule.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, SCHEDULE_NAME, BODY) # Update runbook[patch] BODY = { "description": "Updated Description of the Runbook", "log_verbose": False, "log_progress": True, "log_activity_trace": "1" } result = self.mgmt_client.runbook.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME, BODY) """ # Update watcher[patch] BODY = { "name": "MyTestWatcher", "execution_frequency_in_seconds": "600" } result = self.mgmt_client.watcher.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, WATCHER_NAME, BODY) """ # Suspend job[post] result = self.mgmt_client.job.suspend(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) # Update webhook[patch] BODY = { "name": WEBHOOK_NAME, "is_enabled": False, "description": "updated webhook" } result = self.mgmt_client.webhook.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, WEBHOOK_NAME, BODY) """ # Generate webhook uri[post] result = self.mgmt_client.webhook.generate_uri(resource_group.name, AUTOMATION_ACCOUNT_NAME, WEBHOOK_NAME) # Resume job[post] result = self.mgmt_client.job.resume(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) """ # Update a module[patch] BODY = { "content_link": { "uri": "https://teststorage.blob.core.windows.net/mycontainer/MyModule.zip", "content_hash": { "algorithm": "sha265", "value": "07E108A962B81DD9C9BAA89BB47C0F6EE52B29E83758B07795E408D258B2B87A" }, "version": "1.0.0.0" } } result = self.mgmt_client.module.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME, BODY) # Stop job[post] result = self.mgmt_client.job.stop(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_NAME) """ # Update a node[patch] BODY = { "node_id": "nodeId", "node_configuration": { "name": "SetupServer.localhost" } } result = self.mgmt_client.dsc_node.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_NAME, BODY) """ # Get lists of an automation account[post] result = self.mgmt_client.keys.list_by_automation_account(resource_group.name, AUTOMATION_ACCOUNT_NAME) # Update an automation account[patch] BODY = { "sku": { "name": "Free" }, "name": "myAutomationAccount9", "location": "East US 2" } result = self.mgmt_client.automation_account.update(resource_group.name, AUTOMATION_ACCOUNT_NAME, BODY) """ # # Delete software update configuration[delete] # result = self.mgmt_client.software_update_configurations.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOFTWARE_UPDATE_CONFIGURATION_NAME) # # Delete a hybrid worker group[delete] # result = self.mgmt_client.hybrid_runbook_worker_group.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, HYBRID_RUNBOOK_WORKER_GROUP_NAME) # # Delete a DSC node configuration[delete] # result = self.mgmt_client.dsc_node_configuration.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_CONFIGURATION_NAME) # Delete an existing connection type[delete] result = self.mgmt_client.connection_type.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_TYPE_NAME) # Delete a python 2 package[delete] result = self.mgmt_client.python2package.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, PYTHON2PACKAGE_NAME) # Delete a source control[delete] result = self.mgmt_client.source_control.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, SOURCE_CONTROL_NAME) # Delete DSC Configuration[delete] result = self.mgmt_client.dsc_configuration.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONFIGURATION_NAME) # Delete a job schedule[delete] result = self.mgmt_client.job_schedule.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, JOB_SCHEDULE_NAME) # Delete an existing connection[delete] result = self.mgmt_client.connection.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, CONNECTION_NAME) # Delete a certificate[delete] result = self.mgmt_client.certificate.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, CERTIFICATE_NAME) """ # Delete a credential[delete] result = self.mgmt_client.credential.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, CREDENTIAL_NAME) # Delete a variable[delete] result = self.mgmt_client.variable.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, VARIABLE_NAME) # Delete schedule[delete] result = self.mgmt_client.schedule.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, SCHEDULE_NAME) # Delete webhook[delete] result = self.mgmt_client.webhook.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, WEBHOOK_NAME) # Delete a runbook[delete] result = self.mgmt_client.runbook.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, RUNBOOK_NAME) # # Delete watcher[delete] # result = self.mgmt_client.watcher.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, WATCHER_NAME) # Delete a module[delete] result = self.mgmt_client.module.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, MODULE_NAME) # # Delete a DSC Node[delete] # result = self.mgmt_client.dsc_node.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME, NODE_NAME) # Delete automation account[delete] result = self.mgmt_client.automation_account.delete(resource_group.name, AUTOMATION_ACCOUNT_NAME) #------------------------------------------------------------------------------ if __name__ == '__main__': unittest.main()