Refactoring Invoke-RestMethod to facilitate mocking. This closes #15

Azure · Dec 9, 2016 · 9008cdd · 9008cdd
1 parent 26c7f09
commit 9008cdd
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tools/AzureRM.BootStrapper/AzureRM.Bootstrapper.psm1 b/tools/AzureRM.BootStrapper/AzureRM.Bootstrapper.psm1
@@ -13,13 +13,18 @@ else
   $script:BootStrapRepo = $existingRepos[0].Name
 }
 
-
 function Get-ProfileCachePath
 {
   $ProfileCache = Join-Path -path $env:LOCALAPPDATA -childpath "Microsoft\AzurePowerShell\ProfileCache"
   return $ProfileCache
 }
 
+# Make Rest-Call
+function Get-RestResponse
+{
+  $response = Invoke-RestMethod -ea SilentlyContinue -Uri $PSProfileMapEndpoint -ErrorVariable RestError -OutFile "$ProfileCache\ProfileMap.Json"
+  return $RestError
+}
 
 # Get-ProfileMap from Azure Endpoint
 function Get-AzureProfileMap
@@ -30,7 +35,7 @@ function Get-AzureProfileMap
     New-Item -ItemType Directory -Force -Path $ProfileCache | Out-Null
   }
 
-  $response = Invoke-RestMethod -ea SilentlyContinue -Uri $PSProfileMapEndpoint -ErrorVariable RestError -OutFile "$ProfileCache\ProfileMap.Json"
+  $RestError = Get-RestResponse  
 
   if ($RestError)
   {