1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
from azure.mgmt.automation import AutomationClient
from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy
import pytest
AZURE_LOCATION = "eastus"
@pytest.mark.live_test_only
class TestMgmtRunbook(AzureMgmtRecordedTestCase):
def setup_method(self, method):
self.mgmt_client = self.create_mgmt_client(AutomationClient)
def create_automation_account(self, group_name, location, account_name):
automation_account_creation = self.mgmt_client.automation_account.create_or_update(
group_name,
account_name,
{"location": location, "name": account_name, "properties": {"sku": {"name": "Free"}}},
)
return automation_account_creation.name
def create_runbook(self, group_name, location, account_name, runbook_name):
runbook_creation = self.mgmt_client.runbook.create_or_update(
group_name,
account_name,
runbook_name,
{
"location": location,
"name": runbook_name,
"properties": {
"publishContentLink": {
"uri": "https://github.com/Azure/azure-quickstart-templates/blob/master/quickstarts/microsoft.automation/101-automation/scripts/AzureAutomationTutorial.ps1"
},
"runbookType": "PowerShell",
},
},
)
return runbook_creation.name
@RandomNameResourceGroupPreparer(location=AZURE_LOCATION)
@recorded_by_proxy
def test_get_runbook_content(self, resource_group):
ACCOUNT_NAME = self.get_resource_name("automationtest07")
RUNBOOK_NAME = self.get_resource_name("runbooktest07")
account_name = self.create_automation_account(resource_group.name, AZURE_LOCATION, ACCOUNT_NAME)
runbook_name = self.create_runbook(resource_group.name, AZURE_LOCATION, account_name, RUNBOOK_NAME)
result = self.mgmt_client.runbook.get_content(resource_group.name, account_name, runbook_name)
|
