Update completers

sethworks · Feb 27, 2024 · baf9b1b · baf9b1b
1 parent bf3be79
commit baf9b1b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/functions/DiskSmartInfo.completers.ps1 b/functions/DiskSmartInfo.completers.ps1
@@ -114,10 +114,18 @@ class DiskCompleter : IArgumentCompleter
             {
                 $sessionParameters.Add('Credential', $fakeBoundParameters.Credential)
             }
-            if ($cimSession = New-CimSession @sessionParameters)
+
+            $option = [Microsoft.Management.Infrastructure.Options.WSManSessionOptions]::new()
+            $option.Timeout = New-TimeSpan -Seconds 1
+
+            if ($cimSession = New-CimSession -SessionOption $option @sessionParameters)
             {
                 $instanceParameters.Add('CimSession', $cimSession)
             }
+            else
+            {
+                return $result
+            }
         }
         elseif  ($fakeBoundParameters.ContainsKey('CimSession'))
         {