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

Check Blob

Browse Source
This commit is contained in:
Roy Han 2024-05-31 12:07:52 -07:00
parent e8788ae8dd
commit c62df6b3bf
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server/routes_test.go
View File

@ -65,6 +65,8 @@ func Test_Routes(t *testing.T) {
assert.Nil(t, err)
}
var blobDigest string
testCases := []testCase{
{
Name: "Version Handler",
@ -229,12 +231,12 @@ func Test_Routes(t *testing.T) {
_, err = GetModel("delete-model-1")
assert.True(t, os.IsNotExist(err))
model, err := GetModel("delete-model-2")
assert.Nil(t, err)
model, _ := GetModel("delete-model-2")
assert.Equal(t, "delete-model-2:latest", model.ShortName)
_, err = GetBlobsPath(model.Digest)
assert.Nil(t, err)
blobDigest = model.Digest
_, err = GetBlobsPath(blobDigest)
assert.False(t, os.IsNotExist(err))
},
},
{
@ -256,8 +258,8 @@ func Test_Routes(t *testing.T) {
_, err = GetModel("delete-model-2")
assert.True(t, os.IsNotExist(err))
// TODO: check if blob is deleted
_, err = GetBlobsPath(blobDigest)
assert.True(t, os.IsNotExist(err))
},
},
}