New resource: Automation Account/Credential/Runbook/Schedule

azurerm/config.go

@@ -8,6 +8,7 @@ import (
 	"net/http/httputil"
 
 	"github.com/Azure/azure-sdk-for-go/arm/appinsights"
+	"github.com/Azure/azure-sdk-for-go/arm/automation"
 	"github.com/Azure/azure-sdk-for-go/arm/cdn"
 	"github.com/Azure/azure-sdk-for-go/arm/compute"
 	"github.com/Azure/azure-sdk-for-go/arm/containerregistry"
@@ -58,8 +59,12 @@ type ArmClient struct {
 	vmClient               compute.VirtualMachinesClient
 	imageClient            compute.ImagesClient
 
-	diskClient     disk.DisksClient
-	cosmosDBClient cosmosdb.DatabaseAccountsClient
+	diskClient                 disk.DisksClient
+	cosmosDBClient             cosmosdb.DatabaseAccountsClient
+	automationAccountClient    automation.AccountClient
+	automationRunbookClient    automation.RunbookClient
+	automationCredentialClient automation.CredentialClient
+	automationScheduleClient   automation.ScheduleClient
 
 	appGatewayClient             network.ApplicationGatewaysClient
 	ifaceClient                  network.InterfacesClient
@@ -617,6 +622,30 @@ func (c *Config) getArmClient() (*ArmClient, error) {
 	spc.Sender = autorest.CreateSender(withRequestLogging())
 	client.servicePrincipalsClient = spc
 
+	aadb := automation.NewAccountClientWithBaseURI(endpoint, c.SubscriptionID)
+	setUserAgent(&aadb.Client)
+	aadb.Authorizer = auth
+	aadb.Sender = autorest.CreateSender(withRequestLogging())
+	client.automationAccountClient = aadb
+
+	arc := automation.NewRunbookClientWithBaseURI(endpoint, c.SubscriptionID)
+	setUserAgent(&arc.Client)
+	arc.Authorizer = auth
+	arc.Sender = autorest.CreateSender(withRequestLogging())
+	client.automationRunbookClient = arc
+
+	acc := automation.NewCredentialClientWithBaseURI(endpoint, c.SubscriptionID)
+	setUserAgent(&acc.Client)
+	acc.Authorizer = auth
+	acc.Sender = autorest.CreateSender(withRequestLogging())
+	client.automationCredentialClient = acc
+
+	aschc := automation.NewScheduleClientWithBaseURI(endpoint, c.SubscriptionID)
+	setUserAgent(&aschc.Client)
+	aschc.Authorizer = auth
+	aschc.Sender = autorest.CreateSender(withRequestLogging())
+	client.automationScheduleClient = aschc
+
 	ac := web.NewAppsClientWithBaseURI(endpoint, c.SubscriptionID)
 	setUserAgent(&ac.Client)
 	ac.Authorizer = auth

azurerm/import_arm_automation_account_test.go

@@ -0,0 +1,56 @@
+package azurerm
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccAzureRMAutomationAccount_importAccountWithFreeSku(t *testing.T) {
+	resourceName := "azurerm_automation_account.test"
+
+	ri := acctest.RandInt()
+	config := testAccAzureRMAutomationAccount_skuFree(ri, testLocation())
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testCheckAzureRMAutomationAccountDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+			},
+
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}
+
+func TestAccAzureRMAutomationAccount_importAccountWithBasicSku(t *testing.T) {
+	resourceName := "azurerm_automation_account.test"
+
+	ri := acctest.RandInt()
+	config := testAccAzureRMAutomationAccount_skuBasic(ri, testLocation())
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testCheckAzureRMAutomationAccountDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+			},
+
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}

azurerm/import_arm_automation_credential_test.go

@@ -0,0 +1,34 @@
+package azurerm
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccAzureRMAutomationCredential_importCredential(t *testing.T) {
+	resourceName := "azurerm_automation_credential.test"
+
+	ri := acctest.RandInt()
+	config := testAccAzureRMAutomationCredential_testCredential(ri, testLocation())
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testCheckAzureRMAutomationCredentialDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				// publish content link is not returned after the runbook is created
+				ExpectNonEmptyPlan: true,
+			},
+
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}

azurerm/import_arm_automation_runbook_test.go

@@ -0,0 +1,34 @@
+package azurerm
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccAzureRMAutomationRunbook_importRunbookPSWorkflow(t *testing.T) {
+	resourceName := "azurerm_automation_runbook.test"
+
+	ri := acctest.RandInt()
+	config := testAccAzureRMAutomationRunbook_PSWorkflow(ri, testLocation())
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testCheckAzureRMAutomationRunbookDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				// publish content link is not returned after the runbook is created
+				ExpectNonEmptyPlan: true,
+			},
+
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}

azurerm/import_arm_automation_schedule_test.go

@@ -0,0 +1,34 @@
+package azurerm
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccAzureRMAutomationSchedule_importScheduleOneTime(t *testing.T) {
+	resourceName := "azurerm_automation_schedule.test"
+
+	ri := acctest.RandInt()
+	config := testAccAzureRMAutomationSchedule_oneTime(ri, testLocation())
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testCheckAzureRMAutomationScheduleDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				// publish content link is not returned after the runbook is created
+				ExpectNonEmptyPlan: true,
+			},
+
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}

azurerm/provider.go

@@ -69,6 +69,10 @@ func Provider() terraform.ResourceProvider {
 		ResourcesMap: map[string]*schema.Resource{
 			"azurerm_application_insights":        resourceArmApplicationInsights(),
 			"azurerm_app_service_plan":            resourceArmAppServicePlan(),
+			"azurerm_automation_account":          resourceArmAutomationAccount(),
+			"azurerm_automation_runbook":          resourceArmAutomationRunbook(),
+			"azurerm_automation_credential":       resourceArmAutomationCredential(),
+			"azurerm_automation_schedule":         resourceArmAutomationSchedule(),
 			"azurerm_availability_set":            resourceArmAvailabilitySet(),
 			"azurerm_cdn_endpoint":                resourceArmCdnEndpoint(),
 			"azurerm_cdn_profile":                 resourceArmCdnProfile(),
@@ -243,6 +247,7 @@ var providerRegistrationOnce sync.Once
 
 func determineAzureResourceProvidersToRegister(providerList []resources.Provider) map[string]struct{} {
 	providers := map[string]struct{}{
+		"Microsoft.Automation":        {},
 		"Microsoft.Cache":             {},
 		"Microsoft.Cdn":               {},
 		"Microsoft.Compute":           {},