Merge pull request #4331 from dhiltgen/fix_unit

Fix envconfig unit test

Merge pull request #4331 from dhiltgen/fix_unit
Fix envconfig unit test
c60a0866 · Daniel Hiltgen · GitHub · 92ca2cca · 824ee544 · c60a0866
Unverified Commit c60a0866 authored May 11, 2024 by Daniel Hiltgen Committed by GitHub May 11, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

server/envconfig/config_test.go server/envconfig/config_test.go +4 -4

No files found.
--- a/server/envconfig/config_test.go
+++ b/server/envconfig/config_test.go
 package envconfig

 import (
-	"os"
 	"testing"

 	"github.com/stretchr/testify/require"
 )

 func TestConfig(t *testing.T) {
-	os.Setenv("OLLAMA_DEBUG", "")
+	Debug = false // Reset whatever was loaded in init()
+	t.Setenv("OLLAMA_DEBUG", "")
 	LoadConfig()
 	require.False(t, Debug)
-	os.Setenv("OLLAMA_DEBUG", "false")
+	t.Setenv("OLLAMA_DEBUG", "false")
 	LoadConfig()
 	require.False(t, Debug)
-	os.Setenv("OLLAMA_DEBUG", "1")
+	t.Setenv("OLLAMA_DEBUG", "1")
 	LoadConfig()
 	require.True(t, Debug)
 }