Add LocalNetworkGateway tests and session records

Author: lamchester

src/ServiceManagement/Network/Commands.Network.Test/ScenarioTests/Gateway/GatewayScenarioTests.cs

@@ -0,0 +1,82 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.
+// ----------------------------------------------------------------------------------
+
+
+namespace Microsoft.WindowsAzure.Commands.ServiceManagement.Network.Test.ScenarioTests
+{
+    using Microsoft.Azure.Commands.Common.Authentication;
+    using Microsoft.Azure.Test;
+    using Microsoft.WindowsAzure.Commands.ScenarioTest;
+    using Microsoft.WindowsAzure.Management;
+    using Microsoft.WindowsAzure.Management.Compute;
+    using Microsoft.WindowsAzure.Management.Network;
+    using Microsoft.WindowsAzure.Management.Storage;
+    using Microsoft.WindowsAzure.Commands.Utilities.Common;
+    using System;
+    using System.Collections.Generic;
+    using System.IO;
+    using System.Linq;
+    using Xunit;
+
+    public class GatewayScenarioTests
+    {   
+        private readonly EnvironmentSetupHelper helper = new EnvironmentSetupHelper();
+
+        [Fact]
+        [Trait(Category.Service, Category.Network)]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void LocalNetworkGateway()
+        {
+            RunPowerShellTest("Test-LocalNetworkGateway");
+        }
+
+        [Fact]
+        [Trait(Category.Service, Category.Network)]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void LocalNetworkGatewayBgp()
+        {
+            RunPowerShellTest("Test-LocalNetworkGatewayBgp");
+        }
+
+        #region Test setup
+
+        protected void SetupManagementClients()
+        {
+            var client = TestBase.GetServiceClient<NetworkManagementClient>(new RDFETestEnvironmentFactory());
+            var client2 = TestBase.GetServiceClient<ManagementClient>(new RDFETestEnvironmentFactory());
+            var client3 = TestBase.GetServiceClient<StorageManagementClient>(new RDFETestEnvironmentFactory());
+            var client4 = TestBase.GetServiceClient<ComputeManagementClient>(new RDFETestEnvironmentFactory());
+            helper.SetupManagementClients(client, client2, client3, client4);
+        }
+
+        protected void RunPowerShellTest(params string[] scripts)
+        {
+            using (UndoContext context = UndoContext.Current)
+            {
+                context.Start(TestUtilities.GetCallingClass(2), TestUtilities.GetCurrentMethodName(2));
+
+                List<string> modules = Directory.GetFiles("ScenarioTests\\Gateway".AsAbsoluteLocation(), "*.ps1").ToList();
+                modules.AddRange(Directory.GetFiles("ScenarioTests".AsAbsoluteLocation(), "*.ps1"));
+
+                SetupManagementClients();
+
+                helper.SetupEnvironment(AzureModule.AzureServiceManagement);
+                helper.SetupModules(AzureModule.AzureServiceManagement, modules.ToArray());
+
+                helper.RunPowerShellTest(scripts);
+            }
+        }
+        #endregion
+    }
+}

src/ServiceManagement/Network/Commands.Network.Test/ScenarioTests/Gateway/GatewayTests.ps1

@@ -0,0 +1,84 @@
+# ----------------------------------------------------------------------------------
+#
+# 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.
+# ----------------------------------------------------------------------------------
+
+########################################################################### Gateway tests ###################################################################
+
+<#
+.SYNOPSIS
+    Tests for local network gateway right now
+#>
+
+function Test-LocalNetworkGateway
+{
+    # Setup
+	$lngname = getAssetName
+    $location="West US"
+
+    $subscription = Get-AzureSubscription -Current
+    Set-AzureSubscription -SubscriptionId $subscription.SubscriptionId
+        
+	# Create a local network gateway
+	$ipAddr = "1.2.3.4"
+	$addrSpace = "10.0.0.0/8"
+    New-AzureLocalNetworkGateway -GatewayName $lngname -IpAddress $ipAddr -AddressSpace $addrSpace
+
+	# Retrieve a list of local network gateways, and get the one created above by name
+	$lnglist = Get-AzureLocalNetworkGateway
+	$lng = $lnglist | Where-Object { $_.GatewayName -eq $lngname }
+	Assert-AreEqual $lng.IpAddress $ipAddr
+	Assert-AreEqual $lng.AddressSpace $addrSpace
+
+	# Retrieve the local network gateway by ID
+	$lng = Get-AzureLocalNetworkGateway -GatewayId $lng.GatewayId
+	Assert-AreEqual $lng.IpAddress $ipAddr
+	Assert-AreEqual $lng.AddressSpace $addrSpace
+	
+	# Delete the local network gateway
+    Remove-AzureLocalNetworkGateway -GatewayId $lng.GatewayId
+}
+
+function Test-LocalNetworkGatewayBgp
+{
+# Setup
+	$lngname = getAssetName
+    $location="West US"
+
+    $subscription = Get-AzureSubscription -Current
+    Set-AzureSubscription -SubscriptionId $subscription.SubscriptionId
+        
+	# Create a local network gateway
+	$ipAddr = "1.2.3.4"
+	$addrSpace = "10.0.0.0/8"
+	$asn = "1234"
+	$bgpaddr = "10.0.0.1"
+    New-AzureLocalNetworkGateway -GatewayName $lngname -IpAddress $ipAddr -AddressSpace $addrSpace -Asn $asn -BgpPeeringAddress $bgpaddr
+
+	# Retrieve a list of local network gateways, and get the one created above by name
+	$lnglist = Get-AzureLocalNetworkGateway
+	$lng = $lnglist | Where-Object { $_.GatewayName -eq $lngname }
+	Assert-AreEqual $lng.IpAddress $ipAddr
+	Assert-AreEqual $lng.AddressSpace $addrSpace
+	Assert-AreEqual $lng.Asn $asn
+	Assert-AreEqual $lng.BgpPeeringAddress $bgpaddr
+
+	# Retrieve the local network gateway by ID
+	$lng = Get-AzureLocalNetworkGateway -GatewayId $lng.GatewayId
+	Assert-AreEqual $lng.IpAddress $ipAddr
+	Assert-AreEqual $lng.AddressSpace $addrSpace
+	Assert-AreEqual $lng.Asn $asn
+	Assert-AreEqual $lng.BgpPeeringAddress $bgpaddr
+	
+	# Delete the local network gateway
+    Remove-AzureLocalNetworkGateway -GatewayId $lng.GatewayId
+}