From 5d2629fa941de4b58e5f3557db5c32fedb853eb7 Mon Sep 17 00:00:00 2001
From: Sergey Shandar <sergey@live.com.au>
Date: Thu, 14 Sep 2017 16:59:07 -0700
Subject: [PATCH] ShouldProcess

---
 .../AzureRM.Compute.Experiments.psd1               |  2 +-
 .../AzureRM.Compute.Experiments.psm1               | 14 ++++++++------
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psd1 b/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psd1
index f6987ea25571..895311f327f5 100644
--- a/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psd1
+++ b/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psd1
@@ -12,7 +12,7 @@
 RootModule = ".\AzureRM.Compute.Experiments.psm1"
 
 # Version number of this module.
-ModuleVersion = '1.0.4'
+ModuleVersion = '1.0.5'
 
 # Supported PSEditions
 # CompatiblePSEditions = @()
diff --git a/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psm1 b/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psm1
index f255ebd300b4..82d8cafd3868 100644
--- a/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psm1
+++ b/experiments/Compute.Experiments/AzureRM.Compute.Experiments.psm1
@@ -2,7 +2,7 @@
 .ExternalHelp AzureRM.Compute.Experiments-help.xml
 #>
 function New-AzVm {
-    [CmdletBinding()]
+    [CmdletBinding(SupportsShouldProcess = $true)]
     param (
         [Parameter(Mandatory=$true, Position=0)][string] $Name = "VM",
         [Parameter()][PSCredential] $Credential,
@@ -42,12 +42,14 @@ function New-AzVm {
             $locationi.Value = $Location;
         }
 
-        $resourceGroup = $rgi.GetOrCreate($Name + "ResourceGroup", $locationi.Value, $null);
-        $vmResponse = $vmi.Create($Name, $locationi.Value, $resourceGroup.ResourceGroupName);
+        if ($PSCmdlet.ShouldProcess($Name, "Creating a virtual machine")) {
+            $resourceGroup = $rgi.GetOrCreate($Name + "ResourceGroup", $locationi.Value, $null);
+            $vmResponse = $vmi.Create($Name, $locationi.Value, $resourceGroup.ResourceGroupName);
 
-        New-PsObject @{
-            ResourceId = $resourceGroup.ResourceId;
-            Response = $vmResponse;
+            New-PsObject @{
+                ResourceId = $resourceGroup.ResourceId;
+                Response = $vmResponse;
+            }
         }
     }
 }