jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adjust Response and Blob Check

Browse Source
This commit is contained in:
Roy Han 2024-05-31 13:08:59 -07:00
parent c62df6b3bf
commit a946b6f020
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/routes_test.go
View File

@ -65,7 +65,7 @@ func Test_Routes(t *testing.T) {
assert.Nil(t, err)
}
var blobDigest string
var blob string
testCases := []testCase{
{
@ -222,8 +222,6 @@ func Test_Routes(t *testing.T) {
req.Body = io.NopCloser(bytes.NewReader(jsonData))
},
Expected: func(t *testing.T, resp *http.Response) {
contentType := resp.Header.Get("Content-Type")
assert.Equal(t, "application/json", contentType)
_, err := io.ReadAll(resp.Body)
assert.Nil(t, err)
assert.Equal(t, resp.StatusCode, 200)
@ -234,8 +232,8 @@ func Test_Routes(t *testing.T) {
model, _ := GetModel("delete-model-2")
assert.Equal(t, "delete-model-2:latest", model.ShortName)
blobDigest = model.Digest
_, err = GetBlobsPath(blobDigest)
blob, err = GetBlobsPath(model.Digest)
_, err = os.Stat(blob)
assert.False(t, os.IsNotExist(err))
},
},
@ -250,15 +248,13 @@ func Test_Routes(t *testing.T) {
req.Body = io.NopCloser(bytes.NewReader(jsonData))
},
Expected: func(t *testing.T, resp *http.Response) {
contentType := resp.Header.Get("Content-Type")
assert.Equal(t, "application/json", contentType)
_, err := io.ReadAll(resp.Body)
assert.Nil(t, err)
_, err = GetModel("delete-model-2")
assert.True(t, os.IsNotExist(err))
_, err = GetBlobsPath(blobDigest)
_, err = os.Stat(blob)
assert.True(t, os.IsNotExist(err))
},
},