Add cmdlets for Availability Group Listener resource of SqlVirtualMachine

src/SqlVirtualMachine/SqlVirtualMachine.Test/ScenarioTests/AvailabilityGroupListenerTests.cs

@@ -0,0 +1,57 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using Microsoft.Azure.Commands.SqlVirtualMachine.Test.ScenarioTests;
+using Microsoft.WindowsAzure.Commands.ScenarioTest;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace Microsoft.Azure.Commands.AvailabilityGroupListener.Test.ScenarioTests
+{
+    [Collection("SqlVirtualMachineTests")]
+    public class AvailabilityGroupListenerTests : SqlVirtualMachineTestsBase
+    {
+        public AvailabilityGroupListenerTests(ITestOutputHelper output) : base(output)
+        {
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAvailabilityGroupListenerCreate()
+        {
+            RunPowerShellTest("Test-CreateAvailabilityGroupListener");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAvailabilityGroupListenerGet()
+        {
+            RunPowerShellTest("Test-GetAvailabilityGroupListener");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAvailabilityGroupListenerUpdate()
+        {
+            RunPowerShellTest("Test-UpdateAvailabilityGroupListener");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAvailabilityGroupListenerRemove()
+        {
+            RunPowerShellTest("Test-RemoveAvailabilityGroupListener");
+        }
+    }
+}

src/SqlVirtualMachine/SqlVirtualMachine.Test/ScenarioTests/AvailabilityGroupListenerTests.ps1

@@ -0,0 +1,165 @@
+# ----------------------------------------------------------------------------------
+#
+# Copyright Microsoft Corporation
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ----------------------------------------------------------------------------------
+
+<#
+	.SYNOPSIS
+	Create tests for Availability Group Listener
+#>
+function Test-CreateAvailabilityGroupListener
+{
+	# Setup
+	$previousErrorActionPreferenceValue = $ErrorActionPreference
+	$ErrorActionPreference = "SilentlyContinue"
+	$loadBalancer = Get-LoadBalancerResourceId
+	$subnetId = Get-SubnetId
+	$probePort = Get-ProbePort
+	$sqlVMIds = Get-SqlVirtualMachineId
+	$name = Get-AGListenerName
+	$rgName = Get-AGListenerResourceGroup
+	$groupName = Get-AgListenerGroupName
+	$ipAddress = Get-IpAddress
+	$port = Get-DefaultPort
+	$agName = Get-AgName
+
+	try
+	{
+		# Create Sql VM with parameters
+		$listener = New-AzAvailabilityGroupListener -AvailabilityGroupName $agName -LoadBalancerResourceId $loadBalancer -SubnetId $subnetId `
+					-ProbePort $probePort -SqlVirtualMachineId $sqlVMIds -Name $name -ResourceGroupName $rgName -SqlVMGroupName $groupName -IpAddress $ipAddress
+		Assert-NotNull $listener
+		Assert-AreEqual $listener.Name $name
+		Assert-AreEqual $listener.Port $port
+	}
+	finally
+	{
+		Remove-AzAvailabilityGroupListener -ResourceGroupName $rgName -SqlVMGroupName $groupName -Name $name
+		$ErrorActionPreference = $previousErrorActionPreferenceValue
+	}
+}
+
+<#
+	.SYNOPSIS
+	Get\list tests for Availability Group Listener
+#>
+function Test-GetAvailabilityGroupListener
+{
+	#Setup
+	$previousErrorActionPreferenceValue = $ErrorActionPreference
+	$ErrorActionPreference = "SilentlyContinue"
+	$loadBalancer = Get-LoadBalancerResourceId
+	$subnetId = Get-SubnetId
+	$probePort = Get-ProbePort
+	$sqlVMIds = Get-SqlVirtualMachineId
+	$name = Get-AGListenerName
+	$rgName = Get-AGListenerResourceGroup
+	$groupName = Get-AgListenerGroupName
+	$ipAddress = Get-IpAddress
+	$port = Get-DefaultPort
+	$agName = Get-AgName
+
+	try 
+	{
+		$listener1 = New-AzAvailabilityGroupListener -AvailabilityGroupName $agName -LoadBalancerResourceId $loadBalancer -SubnetId $subnetId `
+					-ProbePort $probePort -SqlVirtualMachineId $sqlVMIds -Name $name -ResourceGroupName $rgName -SqlVMGroupName $groupName -IpAddress $ipAddress
+		$listener2 = Get-AzAvailabilityGroupListener -Name $name -ResourceGroupName $rgName -SqlVMGroupName $groupName
+		Assert-NotNull $listener1
+		Assert-NotNull $listener1
+		Assert-AreEqual $listener1.Name $listener2.Name
+		Assert-AreEqual $listener1.AvailabilityGroupName $listener2.AvailabilityGroupName
+		Assert-AreEqual $listener1.Port $listener2.Port
+		Assert-AreEqual $listener1.GroupName $listener2.GroupName
+	}
+	finally
+	{
+		Remove-AzAvailabilityGroupListener -ResourceGroupName $rgName -SqlVMGroupName $groupName -Name $name
+		$ErrorActionPreference = $previousErrorActionPreferenceValue
+	}
+}
+
+<#
+	.SYNOPSIS
+	Update tests for Availability Group Listener
+#>
+function Test-UpdateAvailabilityGroupListener
+{
+	#Setup
+	$previousErrorActionPreferenceValue = $ErrorActionPreference
+	$ErrorActionPreference = "SilentlyContinue"
+	$loadBalancer = Get-LoadBalancerResourceId
+	$subnetId = Get-SubnetId
+	$probePort = Get-ProbePort
+	$sqlVMIds = Get-SqlVirtualMachineId
+	$name = Get-AGListenerName
+	$rgName = Get-AGListenerResourceGroup
+	$groupName = Get-AgListenerGroupName
+	$ipAddress = Get-IpAddress
+	$port = Get-DefaultPort
+	$agName = Get-AgName
+
+	try 
+	{
+		$listener1 = New-AzAvailabilityGroupListener -AvailabilityGroupName $agName -LoadBalancerResourceId $loadBalancer -SubnetId $subnetId `
+					-ProbePort $probePort -SqlVirtualMachineId $sqlVMIds -Name $name -ResourceGroupName $rgName -SqlVMGroupName $groupName -IpAddress $ipAddress
+		$listener1 = Get-AzAvailabilityGroupListener -ResourceId $listener1.ResourceId
+		$listener2 = Update-AzAvailabilityGroupListener -Name $name -ResourceGroupName $rgName -SqlVMGroupName $groupName -SqlVirtualMachineId $sqlVMIds
+		Assert-NotNull $listener1
+		Assert-NotNull $listener1
+		Assert-AreEqual $listener1.Name $listener2.Name
+		Assert-AreEqual $listener1.AvailabilityGroupName $listener2.AvailabilityGroupName
+		Assert-AreEqual $listener1.Port $listener2.Port
+		Assert-AreEqual $listener1.GroupName $listener2.GroupName
+		Assert-AreEqual $listener1.LoadBalancerConfigurations[0].SqlVirtualMachineInstances.count $listener2.LoadBalancerConfigurations[0].SqlVirtualMachineInstances.count
+	}
+	finally
+	{
+		Remove-AzAvailabilityGroupListener -ResourceGroupName $rgName -SqlVMGroupName $groupName -Name $name
+		$ErrorActionPreference = $previousErrorActionPreferenceValue
+	}
+}
+
+<#
+	.SYNOPSIS
+	Remove tests for Availability Group Listener
+#>
+function Test-RemoveAvailabilityGroupListener
+{
+	#Setup
+	$previousErrorActionPreferenceValue = $ErrorActionPreference
+	$ErrorActionPreference = "SilentlyContinue"
+	$loadBalancer = Get-LoadBalancerResourceId
+	$subnetId = Get-SubnetId
+	$probePort = Get-ProbePort
+	$sqlVMIds = Get-SqlVirtualMachineId
+	$name = Get-AGListenerName
+	$rgName = Get-AGListenerResourceGroup
+	$groupName = Get-AgListenerGroupName
+	$ipAddress = Get-IpAddress
+	$port = Get-DefaultPort
+	$agName = Get-AgName
+
+	try 
+	{
+		$listener1 = New-AzAvailabilityGroupListener -AvailabilityGroupName $agName -LoadBalancerResourceId $loadBalancer -SubnetId $subnetId `
+					-ProbePort $probePort -SqlVirtualMachineId $sqlVMIds -Name $name -ResourceGroupName $rgName -SqlVMGroupName $groupName -IpAddress $ipAddress
+		Assert-NotNull $listener1
+
+		$listener1 = Remove-AzAvailabilityGroupListener -ResourceGroupName $rgName -SqlVMGroupName $groupName -Name $name
+		$listenerList = Get-AzAvailabilityGroupListener -ResourceGroupName $rgName -SqlVMGroupName $groupName
+		Assert-False {$listener1.Name -in $listenerList.Name}
+	}
+	finally
+	{
+		$ErrorActionPreference = $previousErrorActionPreferenceValue
+	}
+}

src/SqlVirtualMachine/SqlVirtualMachine.Test/ScenarioTests/Common.ps1

@@ -150,6 +150,62 @@ function Get-StorageaccountNameForTest()
 	return $nr
 }
 
+# Functions to get resources for Availability Group Listener
+# Currently there is not function to create Setup
+
+function Get-LoadBalancerResourceId()
+{
+	$loadBalancerResourceId = "/subscriptions/0009fc4d-e310-4e40-8e63-c48a23e9cdc1/resourceGroups/karthik-wcus2/providers/Microsoft.Network/loadBalancers/rlnaonlb4"
+	return $loadBalancerResourceId
+}
+
+function Get-SubnetId()
+{
+	$subnetId = "/subscriptions/0009fc4d-e310-4e40-8e63-c48a23e9cdc1/resourceGroups/karthik-wcus2/providers/Microsoft.Network/virtualNetworks/DecTest-vnet/subnets/DecTest-subnet"
+	return $subnetId
+}
+
+function Get-ProbePort()
+{
+	return 59999
+}
+
+function Get-SqlVirtualMachineId()
+{
+	$sqlVirtualMachineIds = "/subscriptions/0009fc4d-e310-4e40-8e63-c48a23e9cdc1/resourceGroups/rlnalwayson/providers/Microsoft.SqlVirtualMachine/SqlVirtualMachines/rlnvmag2","/subscriptions/0009fc4d-e310-4e40-8e63-c48a23e9cdc1/resourceGroups/rlnalwayson/providers/Microsoft.SqlVirtualMachine/SqlVirtualMachines/rlnagvm1"
+	return $sqlVirtualMachineIds
+}
+
+function Get-AGListenerName
+{
+	return "rlnaonagl11"
+}
+
+function Get-AGListenerResourceGroup
+{
+	return "rlnAlwaysOn"
+}
+
+function Get-AgListenerGroupName
+{
+	return "rlnaonsqlgr2"
+}
+
+function Get-IpAddress
+{
+	return "10.0.0.31"
+}
+
+function Get-DefaultPort()
+{
+	return 1433
+}
+
+function Get-AgName()
+{
+	return "rlnaonag"
+}
+
 <#
 	.SYNOPSIS
 	Checks that the sql virtual machines provided are equal