AzSK.Test

3.0.0

TestCases/SVT/Automation/AutomationResource.ps1

Set-StrictMode -Version Latest 
class AutomationResource:TestResource{
    [bool] $RetainResource = $false    
    [string] $RunbookName = "TestSuiteRunbook"
    AutomationResource([TestCase] $testcase, [TestSettings] $testsettings):Base($testcase, $testsettings){
      }

    #Setting the properties as required by this resource type.
    [void]SetDerivedResourceProps()
    {
        #Set resource parameters
        $this.ResourceName    = "automationaccount" + $(get-date -f MMddyyHHmm) 
        $this.ResourceType = "Microsoft.Automation/automationAccounts" 
    }
    [void] InitializeResource()
    {
        $this.DeployAutomationRunbook()    
    }
    #Deploy Automation
    [void] DeployAutomationRunbook()
    {
        try
        {
            $automationAccount = New-AzureRmAutomationAccount `
                 -ResourceGroupName $this.ResourceGroupName `
                 -Name $this.ResourceName `
                 -Location $this.Location `
                 -Plan Free -ErrorAction Stop
            $runbook = New-AzureRmAutomationRunbook -AutomationAccountName $this.ResourceName -Name $this.RunbookName -ResourceGroupName $this.ResourceGroupName -Type PowerShell -ErrorAction Stop
            Publish-AzureRmAutomationRunbook -AutomationAccountName $this.ResourceName -Name $this.RunbookName -ResourceGroupName $this.ResourceGroupName -ErrorAction Stop
            $this.ProvisioningState = "Succeeded"
            [CommonHelper]::Log("Automation account deployed successfully " + $this.ResourceName, [MessageType]::Information)
        }
        catch
        {
            $this.ProvisioningState = "Failed"
            [CommonHelper]::Log("Error while deploying Automation account: " + $this.ResourceName, [MessageType]::Error)
        }
    }
    #Set valid webhook
    [void] SetValidWebhook()
    {
        try{
            if($this.ProvisioningState -eq "Succeeded")
            {
                $AutomationResource = Get-AzureRmResource -ResourceName $this.ResourceName -ResourceGroupName $this.ResourceGroupName -ErrorAction Stop
                
                 New-AzureRmAutomationWebhook -AutomationAccountName $this.ResourceName -ExpiryTime $(get-date).AddDays(10) `
                -IsEnabled $true -Name "TestSuiteWebhook" -ResourceGroupName $this.ResourceGroupName `
                -RunbookName $this.RunbookName -Force -ErrorAction Stop
            }
        }
        catch
        {
            $this.ProvisioningState = "Failed"
            [CommonHelper]::Log("Error while setting the webhook for: " + $this.ResourceName, [MessageType]::Error)
        }
    }
    #Set encrypted variables
    [void] SetEncryptedVariables(){
        try{
            if($this.ProvisioningState -eq "Succeeded")
            {    
                New-AzureRmAutomationVariable -Name "variable1" -Encrypted $true -Value "variable1" -ResourceGroupName $this.ResourceGroupName -AutomationAccountName $this.ResourceName -ErrorAction Stop
                New-AzureRmAutomationVariable -Name "variable2" -Encrypted $true -Value "variable2" -ResourceGroupName $this.ResourceGroupName -AutomationAccountName $this.ResourceName 
            }
        }
        catch{
        $this.ProvisioningState = "Failed"
            [CommonHelper]::Log("Error while creating encrypted variables for: " + $this.ResourceName, [MessageType]::Error)
        }
    }
    #Set invalid webhook
    [void] SetInvalidWebhook()
    {
        try{
            if($this.ProvisioningState -eq "Succeeded")
            {
                $AutomationResource = Get-AzureRmResource -ResourceName $this.ResourceName -ResourceGroupName $this.ResourceGroupName -ErrorAction Stop
                
                 New-AzureRmAutomationWebhook -AutomationAccountName $this.ResourceName -ExpiryTime $(get-date).AddDays(90) `
                -IsEnabled $true -Name "TestSuiteWebhook" -ResourceGroupName $this.ResourceGroupName `
                -RunbookName $this.RunbookName -Force -ErrorAction Stop
            }
        }
        catch
        {
            $this.ProvisioningState = "Failed"
            [CommonHelper]::Log("Error while setting the webhook for: " + $this.ResourceName, [MessageType]::Error)
        }
    }
    #Set unencrypted variables
    [void] SetUnencryptedVariables()
    {
        try{
            if($this.ProvisioningState -eq "Succeeded")
            {    
                New-AzureRmAutomationVariable -Name "variable1" -Encrypted $false -Value "variable1" -ResourceGroupName $this.ResourceGroupName -AutomationAccountName $this.ResourceName -ErrorAction Stop
                New-AzureRmAutomationVariable -Name "variable2" -Encrypted $false -Value "variable2" -ResourceGroupName $this.ResourceGroupName -AutomationAccountName $this.ResourceName 
            }
        }
        catch{
        $this.ProvisioningState = "Failed"
            [CommonHelper]::Log("Error while creating encrypted variables for: " + $this.ResourceName, [MessageType]::Error)
        }
    }
}