Fix: dfinke#50 API Key reset after Invoke-Pester

Agazoth · Dec 2, 2024 · 13bf1dd · 13bf1dd
1 parent 2100f09
commit 13bf1dd
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/__tests__/IntegrationTests/Integration-OpenAI.tests.ps1 b/__tests__/IntegrationTests/Integration-OpenAI.tests.ps1
@@ -3,6 +3,7 @@ BeforeDiscovery {
         $info = $null
     } else{
         $info = Get-SecretInfo -Name OpenAI
+        $env:TempOpenAIKey = $env:OpenAIKey
         $env:OpenAIKey = Get-Secret OpenAI -asPlainText # replace with clear text password if Get-Secret is not setup on your system    
     }
 }
@@ -45,7 +46,7 @@ Describe "Test chat endpoints with Environment Variable" -Skip:($null -eq $env:O
     }
 
     Context "Invoke-OAIChatCompletion with Environment Variable" {
-        It "Invoke-OAIChatCompletion generates an answer" -Skip:($null -eq $envSecret) {
+        It "Invoke-OAIChatCompletion generates an answer" -Skip:($null -eq $env:OpenAIKey) {
             $Prompt = "What is the capitol of France"
             $Message = New-OAIChatMessage -Role user -Content $Prompt
             $Answer = Invoke-OAIChatCompletion -Message $Message
@@ -55,4 +56,6 @@ Describe "Test chat endpoints with Environment Variable" -Skip:($null -eq $env:O
         }
 
     }
-}
+}
+
+AfterAll {$env:OpenAIKey = $env:TempOpenAIKey}
diff --git a/__tests__/Invoke-OAIBeta-InvokeRestMethod-OpenAI.tests.ps1 b/__tests__/Invoke-OAIBeta-InvokeRestMethod-OpenAI.tests.ps1
@@ -1,3 +1,7 @@
+BeforeAll {
+    $env:TempOpenAIKey = $env:OpenAIKey
+}
+
 Describe 'Test Invoke-OAIBeta InvokeRestMethod OpenAI Params' -Tag Invoke-OAIBetaParams-OpenAI {
     BeforeAll {
         Import-Module "$PSScriptRoot/../PSAI.psd1" -Force
@@ -209,4 +213,8 @@ Describe 'Test Invoke-OAIBeta InvokeRestMethod OpenAI Params' -Tag Invoke-OAIBet
 
         Test-UnitTestingData $UnitTestingData $ExpectedUnitTestingData
     }
+}
+
+AfterAll {
+    $env:OpenAIKey = $env:TempOpenAIKey
 }