Merge pull request #1619 from jmorganca/mxyng/fix-version-test

fix(test): use real version string for comparison
This commit is contained in:
Michael Yang 2023-12-19 15:48:52 -08:00 committed by GitHub
commit 23dc179350
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -16,6 +16,7 @@ import (
"github.com/jmorganca/ollama/api" "github.com/jmorganca/ollama/api"
"github.com/jmorganca/ollama/parser" "github.com/jmorganca/ollama/parser"
"github.com/jmorganca/ollama/version"
) )
func setupServer(t *testing.T) (*Server, error) { func setupServer(t *testing.T) (*Server, error) {
@ -71,7 +72,7 @@ func Test_Routes(t *testing.T) {
assert.Equal(t, contentType, "application/json; charset=utf-8") assert.Equal(t, contentType, "application/json; charset=utf-8")
body, err := io.ReadAll(resp.Body) body, err := io.ReadAll(resp.Body)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, `{"version":"0.0.0"}`, string(body)) assert.Equal(t, fmt.Sprintf(`{"version":"%s"}`, version.Version), string(body))
}, },
}, },
{ {