From 824ee5446f498092653e647fdef2167d28e33b4b Mon Sep 17 00:00:00 2001 From: Daniel Hiltgen Date: Fri, 10 May 2024 16:49:48 -0700 Subject: [PATCH] Fix envconfig unit test --- server/envconfig/config_test.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/server/envconfig/config_test.go b/server/envconfig/config_test.go index b27602992b..bad7c4a731 100644 --- a/server/envconfig/config_test.go +++ b/server/envconfig/config_test.go @@ -1,20 +1,20 @@ 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) }