baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #4331 from dhiltgen/fix_unit

Browse source 
Fix envconfig unit test
This commit is contained in:
Daniel Hiltgen 2024-05-11 09:16:28 -07:00 committed by GitHub
commit c60a086635
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
server/envconfig/config_test.go
View file

@ -1,20 +1,20 @@
package envconfig package envconfig
import ( import (
"os"
"testing" "testing"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
func TestConfig(t *testing.T) { func TestConfig(t *testing.T) {
os.Setenv("OLLAMA_DEBUG", "") Debug = false // Reset whatever was loaded in init()
t.Setenv("OLLAMA_DEBUG", "")
LoadConfig() LoadConfig()
require.False(t, Debug) require.False(t, Debug)
os.Setenv("OLLAMA_DEBUG", "false") t.Setenv("OLLAMA_DEBUG", "false")
LoadConfig() LoadConfig()
require.False(t, Debug) require.False(t, Debug)
os.Setenv("OLLAMA_DEBUG", "1") t.Setenv("OLLAMA_DEBUG", "1")
LoadConfig() LoadConfig()
require.True(t, Debug) require.True(t, Debug)
} }