Add Location parameter to Update-FabricCapacity cmdlet

This change introduces a mandatory Location parameter to the Update-FabricCapacity cmdlet, ensuring that the location is specified when creating or updating a Microsoft Fabric capacity. Additionally, the API request method has been updated from PATCH to PUT for better alignment with RESTful practices.

Author: NowinskiK

source/Private/Test-FabricApiResponse.ps1

@@ -142,11 +142,19 @@ function Test-FabricApiResponse {
         default {
             Write-Message -Message "Test-FabricApiResponse::default" -Level Debug
             Write-Message -Message "Unexpected response code: $statusCode from the API." -Level Error
-            Write-Message -Message "Error: $($response.message)" -Level Error
-            if ($response.moreDetails) {
-                Write-Message -Message "More Details: $($response.moreDetails)" -Level Error
+
+            if ($null -ne $Response.error) {
+                Write-Message -Message "Error Code: $($Response.error.code)" -Level Error
+                Write-Message -Message "Error Msg: $($Response.error.message)" -Level Error
+                $errorJson = $Response.error | ConvertTo-Json -Depth 10
+                Write-Message -Message "Error Detailed Info:`n$errorJson" -Level Debug
+            } else {
+                Write-Message -Message "Error Code: $($Response.errorCode)" -Level Error
+                Write-Message -Message "Error Msg: $($Response.message)" -Level Error
+                if ($Response.moreDetails) {
+                    Write-Message -Message "More Details: $($Response.moreDetails)" -Level Error
+                }
             }
-            Write-Message "Error Code: $($response.errorCode)" -Level Error
             throw "API request failed with status code $statusCode."
         }
     }

source/Public/Capacity/Update-FabricCapacity.ps1

@@ -69,6 +69,10 @@ function Update-FabricCapacity
         [ValidateNotNullOrEmpty()]
         [string]$SkuName,
 
+        [Parameter(Mandatory = $true)]
+        [ValidateNotNullOrEmpty()]
+        [string]$Location,
+
         [Parameter(Mandatory = $true)]
         [ValidateNotNullOrEmpty()]
         [string[]]$AdministrationMembers,
@@ -88,7 +92,6 @@ function Update-FabricCapacity
 
         # Step 2: Construct the API URL
         $apiEndpointUrl = "$($AzureSession.BaseApiUrl)/subscriptions/{0}/resourceGroups/{1}/providers/Microsoft.Fabric/capacities/{2}?api-version=2023-11-01" -f $SubscriptionId, $ResourceGroupName, $CapacityName
-        Write-Message -Message "API Endpoint: $apiEndpointUrl" -Level Debug
 
         # Step 3: Construct the request body
         $body = @{
@@ -101,6 +104,7 @@ function Update-FabricCapacity
                 name = $SkuName
                 tier = $SkuTier
             }
+            location = $Location
         }
 
         if ($Tags)
@@ -113,7 +117,7 @@ function Update-FabricCapacity
         {
             $apiParams = @{
                 Uri = $apiEndpointUrl
-                Method = 'PATCH'
+                Method = 'PUT'
                 Body = $body
                 TypeName = 'Fabric Capacity'
                 NoWait = $NoWait

source/Public/Invoke-FabricRestMethod.ps1

@@ -133,6 +133,12 @@ Function Invoke-FabricRestMethod {
         Write-Message -Message "PowerBIApi param is ignored when full Uri is provided." -Level Warning
     }
 
+    $Headers = $FabricSession.HeaderParams
+    if ($Uri.StartsWith($AzureSession.BaseApiUrl)) {
+        $Headers = $AzureSession.HeaderParams
+        Write-Message -Message "Using AzureSession headers for request." -Level Debug
+    }
+
     if ($Body -is [hashtable]) {
         $Body = $Body | ConvertTo-Json -Depth 10
         Write-Message -Message "Request Body: $Body" -Level Debug
@@ -154,7 +160,7 @@ Function Invoke-FabricRestMethod {
         Write-Message -Message "API Endpoint: $Method $apiEndpointUrl" -Level Verbose
 
         $request = @{
-            Headers = $FabricSession.HeaderParams
+            Headers = $Headers
             Uri = $Uri
             Method = $Method
             Body = $Body

tests/Unit/Update-FabricCapacity.Tests.ps1

@@ -44,8 +44,8 @@ Describe "Update-FabricCapacity" -Tag "UnitTests" {
         }
 
         It "Should have exactly the number of expected parameters $($expected.Count)" {
-            $hasparms = $command.Parameters.Values.Name
-            Compare-Object -ReferenceObject $expected -DifferenceObject $hasparms | Should -BeNullOrEmpty
+            $hasParams = $command.Parameters.Values.Name
+            Compare-Object -ReferenceObject $expected -DifferenceObject $hasParams | Should -BeNullOrEmpty
         }
     }