handle exception

Author: sergey-shandar

src/ResourceManager/Common/Commands.Common.Strategies/SyncTaskScheduler.cs

@@ -47,6 +47,10 @@ public void Wait(Task task)
                 Thread.Yield();
             }
             HandleActions();
+            if (task.IsFaulted)
+            {
+                throw task.Exception.InnerException;
+            }
         }
 
         void HandleActions()

src/ResourceManager/Compute/Commands.Compute/Models/PSVirtualMachine.cs

@@ -98,5 +98,7 @@ public string ResourceGroupName
 
         // Gets or sets the virtual machine zones.
         public IList<string> Zones { get; set; }
+
+        public string Fqdn { get; set; }
     }
 }

src/ResourceManager/Compute/Commands.Compute/Strategies/AsyncCmdletExtensions.cs

@@ -1,4 +1,18 @@
-using Microsoft.Azure.Commands.Common.Strategies;
+// ----------------------------------------------------------------------------------
+//
+// 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.Common.Strategies;
 using System;
 using System.Management.Automation;
 using System.Threading.Tasks;
@@ -34,6 +48,9 @@ public void WriteVerbose(string message)
 
             public Task<bool> ShouldProcessAsync(string target, string action)
                 => Scheduler.Invoke(() => _Cmdlet.ShouldProcess(target, action));
+
+            public void WriteObject(object value)
+                => Scheduler.BeginInvoke(() => _Cmdlet.WriteObject(value));
         }
     }
 }

src/ResourceManager/Compute/Commands.Compute/Strategies/Client.cs

@@ -1,4 +1,18 @@
-using Microsoft.Azure.Commands.Common.Authentication;
+// ----------------------------------------------------------------------------------
+//
+// 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.Common.Authentication;
 using Microsoft.Azure.Commands.Common.Authentication.Abstractions;
 using Microsoft.Azure.Commands.Common.Strategies;
 using Microsoft.Rest;

src/ResourceManager/Compute/Commands.Compute/Strategies/IAsyncCmdlet.cs

@@ -1,4 +1,18 @@
-using System.Threading.Tasks;
+// ----------------------------------------------------------------------------------
+//
+// 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 System.Threading.Tasks;
 
 namespace Microsoft.Azure.Commands.Compute.Strategies
 {
@@ -7,5 +21,7 @@ interface IAsyncCmdlet
         void WriteVerbose(string message);
 
         Task<bool> ShouldProcessAsync(string target, string action);
+
+        void WriteObject(object value);
     }
 }

src/ResourceManager/Compute/Commands.Compute/VirtualMachine/Operation/NewAzureVMCommand.cs

@@ -49,7 +49,7 @@ namespace Microsoft.Azure.Commands.Compute
         ProfileNouns.VirtualMachine,
         SupportsShouldProcess = true,
         DefaultParameterSetName = "DefaultParameterSet")]
-    [OutputType(typeof(PSAzureOperationResponse))]
+    [OutputType(typeof(PSAzureOperationResponse), typeof(PSVirtualMachine))]
     public class NewAzureVMCommand : VirtualMachineBaseCmdlet
     {
         public const string DefaultParameterSet = "DefaultParameterSet";
@@ -210,8 +210,9 @@ async Task StrategyExecuteCmdletAsync(IAsyncCmdlet asyncCmdlet)
                 adminPassword: new NetworkCredential(string.Empty, Credential.Password).Password,
                 image: image.Image);
 
-            // get state
             var client = new Client(DefaultProfile.DefaultContext);
+
+            // get current Azure state
             var current = await virtualMachine.GetStateAsync(client, new CancellationToken());
 
             if (Location == null)
@@ -227,14 +228,17 @@ async Task StrategyExecuteCmdletAsync(IAsyncCmdlet asyncCmdlet)
             var target = virtualMachine.GetTargetState(current, client.SubscriptionId, Location);
 
             // apply target state
-            var result = await virtualMachine
+            var newState = await virtualMachine
                 .UpdateStateAsync(
                     client,
                     target,
                     new CancellationToken(),
                     new ShouldProcess(asyncCmdlet),
                     new ProgressReport(asyncCmdlet));
-            WriteObject(result);
+
+            var result = newState.Get(virtualMachine);
+            var psResult = ComputeAutoMapperProfile.Mapper.Map<PSVirtualMachine>(result);
+            asyncCmdlet.WriteObject(psResult);
         }
 
         public void DefaultExecuteCmdlet()