use testing tempdirs

This commit is contained in:
Michael Yang 2024-08-02 15:55:34 -07:00
parent 77ccbf04dc
commit a091fadfda

View file

@ -2,6 +2,7 @@ package server
import (
"bytes"
"cmp"
"encoding/json"
"fmt"
"io"
@ -53,6 +54,8 @@ func (t *responseRecorder) CloseNotify() <-chan bool {
func createRequest(t *testing.T, fn func(*gin.Context), body any) *httptest.ResponseRecorder {
t.Helper()
// if OLLAMA_MODELS is not set, set it to the temp directory
t.Setenv("OLLAMA_MODELS", cmp.Or(os.Getenv("OLLAMA_MODELS"), t.TempDir()))
w := NewRecorder()
c, _ := gin.CreateTestContext(w)