Add tests

Author: dihan0604

src/ServiceManagement/ExpressRoute/Commands.ExpressRoute/DedicatedCircuitProperties/SetAzureDedicatedCircuitProperties.cs

@@ -44,7 +44,7 @@ public override void ExecuteCmdlet()
         {
             ConfirmAction(
                 Force.IsPresent,
-                string.Format(Resources.SetAzureDedicatedCircuitBandwidthWarning, ServiceKey, Bandwidth, BillingType),
+                string.Format(Resources.SetAzureDedicatedCircuitBandwidthWarning, ServiceKey, Bandwidth, Sku, BillingType),
                 Resources.SetAzureDedicatedCircuitBandwidthMessage,
                 ServiceKey.ToString(),
                 () =>

src/ServiceManagement/Services/Commands.Test/Commands.Test.csproj

@@ -133,7 +133,7 @@
       <HintPath>..\..\..\packages\Microsoft.WindowsAzure.Management.Compute.12.6.0\lib\net40\Microsoft.WindowsAzure.Management.Compute.dll</HintPath>
     </Reference>
     <Reference Include="Microsoft.WindowsAzure.Management.ExpressRoute, Version=0.9.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
-      <HintPath>..\..\..\packages\Microsoft.WindowsAzure.Management.ExpressRoute.0.18.14-preview\lib\net40\Microsoft.WindowsAzure.Management.ExpressRoute.dll</HintPath>
+      <HintPath>..\..\..\packages\Microsoft.WindowsAzure.Management.ExpressRoute.0.19.0-preview\lib\net40\Microsoft.WindowsAzure.Management.ExpressRoute.dll</HintPath>
       <Private>True</Private>
     </Reference>
     <Reference Include="Microsoft.WindowsAzure.Management.MediaServices">

src/ServiceManagement/Services/Commands.Test/ExpressRoute/AzureDedicatedCircuitTests.cs

@@ -56,7 +56,8 @@ public void NewAzureDedicatedCircuitSuccessful()
             string serviceProviderName = "TestProvider";
             string location = "us-west";
             string serviceKey = "aa28cd19-b10a-41ff-981b-53c6bbf15ead";
-            BillingType billingType = BillingType.UnlimitedData;
+            BillingType billingType = BillingType.MeteredData;
+            CircuitSku sku = CircuitSku.Premium;
 
             MockCommandRuntime mockCommandRuntime = new MockCommandRuntime();
             Mock<ExpressRouteManagementClient> client = InitExpressRouteManagementClient();
@@ -75,6 +76,7 @@ public void NewAzureDedicatedCircuitSuccessful()
                         ServiceKey = serviceKey,
                         ServiceProviderProvisioningState = ProviderProvisioningState.NotProvisioned,
                         Status = DedicatedCircuitState.Enabled,
+                        Sku = sku
                     },
                     RequestId = "",
                     StatusCode = new HttpStatusCode()
@@ -91,8 +93,27 @@ public void NewAzureDedicatedCircuitSuccessful()
             var tNew = new Task<ExpressRouteOperationStatusResponse>(() => expectedStatus);
             tNew.Start();
 
-            dcMock.Setup(f => f.NewAsync(It.Is<DedicatedCircuitNewParameters>(x => x.Bandwidth == bandwidth && x.CircuitName == circuitName && x.Location == location && x.ServiceProviderName == serviceProviderName), It.IsAny<CancellationToken>())).Returns((DedicatedCircuitNewParameters param, CancellationToken cancellation) => tNew);
+            ExpressRouteOperationStatusResponse expectedUpdateStatus = new ExpressRouteOperationStatusResponse()
+            {
+                HttpStatusCode = HttpStatusCode.OK,
+                Data = serviceKey
+            };
+
+            var tUpdate = new Task<ExpressRouteOperationStatusResponse>(() => expectedUpdateStatus);
+            tUpdate.Start();
+
+            //Task<ExpressRouteOperationStatusResponse> NewAsync(DedicatedCircuitNewParameters parameters, CancellationToken cancellationToken);
+            dcMock.Setup(f => f.NewAsync(It.Is<DedicatedCircuitNewParameters>(x => x.Bandwidth == bandwidth && x.CircuitName == circuitName && x.Location == location && x.ServiceProviderName == serviceProviderName), 
+                It.IsAny<CancellationToken>())).Returns((DedicatedCircuitNewParameters param, CancellationToken cancellation) => tNew);
             dcMock.Setup(f => f.GetAsync(It.Is<string>(sKey => sKey == serviceKey), It.IsAny<CancellationToken>())).Returns((string sKey, CancellationToken cancellation) => tGet);
+            
+            //Task<ExpressRouteOperationStatusResponse> UpdateAsync(string serviceKey, DedicatedCircuitUpdateParameters parameters, CancellationToken cancellationToken);
+            // UpdateAsync(this IDedicatedCircuitOperations operations, string serviceKey, DedicatedCircuitUpdateParameters parameters)
+            
+            dcMock.Setup(f => f.UpdateAsync(It.Is<string>(sKey => sKey == serviceKey), 
+                It.Is<DedicatedCircuitUpdateParameters>(y => y.Bandwidth == bandwidth.ToString() && y.BillingType == billingType && y.Sku == sku.ToString()),
+                It.IsAny<CancellationToken>())).Returns((string sKey, DedicatedCircuitUpdateParameters updateParam, 
+                    CancellationToken cancellation) => tUpdate);
             client.SetupGet(f => f.DedicatedCircuits).Returns(dcMock.Object);
 
             NewAzureDedicatedCircuitCommand cmdlet = new NewAzureDedicatedCircuitCommand()
@@ -102,6 +123,7 @@ public void NewAzureDedicatedCircuitSuccessful()
                 BillingType = billingType,
                 Location = location,
                 ServiceProviderName = serviceProviderName,
+                Sku = sku,
                 CommandRuntime = mockCommandRuntime,
                 ExpressRouteClient = new ExpressRouteClient(client.Object)
             };
@@ -118,6 +140,28 @@ public void NewAzureDedicatedCircuitSuccessful()
             Assert.Equal(expected.DedicatedCircuit.ServiceProviderProvisioningState, actual.ServiceProviderProvisioningState);
             Assert.Equal(expected.DedicatedCircuit.Status, actual.Status);
             Assert.Equal<string>(expected.DedicatedCircuit.ServiceKey, actual.ServiceKey);
+            Assert.Equal<CircuitSku>(expected.DedicatedCircuit.Sku, actual.Sku);
+
+            SetAzureDedicatedCircuitPropertiesCommand setCmdlet = new SetAzureDedicatedCircuitPropertiesCommand()
+                                                                      {
+                                                                          ServiceKey = Guid.Parse(actual.ServiceKey),
+                                                                          Bandwidth =  bandwidth,
+                                                                          BillingType = billingType,
+                                                                          Sku = sku,
+                                                                          CommandRuntime = mockCommandRuntime,
+                                                                          ExpressRouteClient = new ExpressRouteClient(client.Object)
+                                                                      };
+            setCmdlet.ExecuteCmdlet();
+            actual = mockCommandRuntime.OutputPipeline[0] as AzureDedicatedCircuit;
+            Assert.Equal<string>(expected.DedicatedCircuit.CircuitName, actual.CircuitName);
+            Assert.Equal<string>(expected.DedicatedCircuit.BillingType, actual.BillingType);
+            Assert.Equal<uint>(expected.DedicatedCircuit.Bandwidth, actual.Bandwidth);
+            Assert.Equal<string>(expected.DedicatedCircuit.Location, actual.Location);
+            Assert.Equal<string>(expected.DedicatedCircuit.ServiceProviderName, actual.ServiceProviderName);
+            Assert.Equal(expected.DedicatedCircuit.ServiceProviderProvisioningState, actual.ServiceProviderProvisioningState);
+            Assert.Equal(expected.DedicatedCircuit.Status, actual.Status);
+            Assert.Equal<string>(expected.DedicatedCircuit.ServiceKey, actual.ServiceKey);
+            Assert.Equal<CircuitSku>(expected.DedicatedCircuit.Sku, actual.Sku);
         }
 
         [Fact]

src/ServiceManagement/Services/Commands.Test/packages.config

@@ -20,7 +20,7 @@
   <package id="Microsoft.WindowsAzure.ConfigurationManager" version="3.2.0" targetFramework="net45" />
   <package id="Microsoft.WindowsAzure.Management" version="4.1.1" targetFramework="net45" />
   <package id="Microsoft.WindowsAzure.Management.Compute" version="12.6.0" targetFramework="net45" />
-  <package id="Microsoft.WindowsAzure.Management.ExpressRoute" version="0.18.14-preview" targetFramework="net45" />
+  <package id="Microsoft.WindowsAzure.Management.ExpressRoute" version="0.19.0-preview" targetFramework="net45" />
   <package id="Microsoft.WindowsAzure.Management.MediaServices" version="4.0.0" targetFramework="net45" />
   <package id="Microsoft.WindowsAzure.Management.ServiceBus" version="0.18.0-preview" targetFramework="net45" />
   <package id="Microsoft.WindowsAzure.Management.Storage" version="5.1.1" targetFramework="net45" />