Fix test issues

PSModule · Mar 16, 2024 · 59fedef · 59fedef
1 parent cad0466
commit 59fedef
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/src/Store/private/Set-StoreVariable.ps1 b/src/Store/private/Set-StoreVariable.ps1
@@ -6,6 +6,7 @@
         .EXAMPLE
         Set-StoreVariable -Name 'Name' -Value 'MyName'
     #>
+    [CmdletBinding(SupportsShouldProcess)]
     param(
         # The name of the variable to set.
         [Parameter(Mandatory)]
@@ -17,10 +18,12 @@
         [object] $Value
     )
 
-    if ($null -eq $Value) {
-        $script:Store.Remove($Name)
-    } else {
-        $script:Store.$Name = $Value
+    if ($PSCmdlet.ShouldProcess("Set variable '$Name' to '$Value'")) {
+        if ($null -eq $Value) {
+            $script:Store.Remove($Name)
+        } else {
+            $script:Store.$Name = $Value
+        }
+        Save-VariableStore
     }
-    Save-VariableStore
 }
diff --git a/src/Store/public/Set-StoreConfig.ps1 b/src/Store/public/Set-StoreConfig.ps1
@@ -42,10 +42,14 @@
 
     switch ($PSCmdlet.ParameterSetName) {
         'Variable' {
-            Set-StoreVariable -Name $VariableName -Value $Value
+            if ($PSCmdlet.ShouldProcess("Set variable '$VariableName' to '$Value'")) {
+                Set-StoreVariable -Name $VariableName -Value $Value
+            }
         }
         'Secret' {
-            Set-Secret -Name $SecretName -SecretValue $Value
+            if ($PSCmdlet.ShouldProcess("Set secret '$SecretName' to '$Value'")) {
+                Set-Secret -Name $SecretName -SecretValue $Value
+            }
         }
     }
 }