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