fix test

loanDepot · Aug 18, 2020 · 4ef5bcf · 4ef5bcf
1 parent 57c0130
commit 4ef5bcf
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/SnowSQL/public/Open-SnowSqlConnection.ps1 b/SnowSQL/public/Open-SnowSqlConnection.ps1
@@ -32,7 +32,7 @@ function Open-SnowSqlConnection
         [int]
         $Timeout = 10
     )
-    begging
+    begin
     {
         if ($Endpoint -match '(http[s]?)(:\/\/)([^\s,]+)')
         {
@@ -55,13 +55,21 @@ function Open-SnowSqlConnection
             Connection  = $SnowSqlConnection
             timeout     = $Timeout
         }
-        $Result = Invoke-SnowSql @invokeSnowSqlSplat
+        if ($PSCmdlet.ShouldProcess("Execute SnowSql query [$($invokeSnowSqlSplat.query)] on [$Endpoint] as [$($Credential.UserName)]. Use -Debug to see full command"))
+        {
+            $Result = Invoke-SnowSql @invokeSnowSqlSplat
+        }
+        else
+        {
+            $Result = $true
+        }
         if (-not $Result)
         {
             Write-Error ("Unable to connect to SnowSql endpoint {0}" -f $Endpoint) -ErrorAction Stop
         }
         else
         {
+            $Script:SnowSqlConnection = $SnowSqlConnection
             return $Script:SnowSqlConnection
         }
     }

diff --git a/Tests/Get-SnowSqlConnection.Tests.ps1 b/Tests/Get-SnowSqlConnection.Tests.ps1
@@ -2,7 +2,7 @@ InModuleScope SnowSQL {
     Describe 'Function Get-SnowSqlConnection' -Tag Build {
         It 'Get SnowSql Connection' {
             Mock Invoke-SnowSql -Verifiable {}
-            Open-SnowSqlConnection -Endpoint "test" -Credential ([pscredential]::Empty)
+            Open-SnowSqlConnection -Endpoint "test" -Credential ([pscredential]::Empty) -WhatIf
             Get-SnowSqlConnection | Should -Not -BeNullOrEmpty
         }
     }

diff --git a/Tests/Open-SnowSqlConnection.Tests.ps1 b/Tests/Open-SnowSqlConnection.Tests.ps1
@@ -2,7 +2,7 @@ InModuleScope SnowSql {
     Describe 'Function Open-SnowSqlConnection' -Tag Build {
         It 'Open SnowSql Connection ' {
             Mock Invoke-SnowSql -Verifiable {}
-            Open-SnowSqlConnection -Endpoint "test" -Credential ([pscredential]::Empty)
+            Open-SnowSqlConnection -Endpoint "test" -Credential ([pscredential]::Empty) -Whatif
         }
     }
 }