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

Clean Up Delete Tests

Browse Source
This commit is contained in:
Roy Han 2024-05-31 16:44:54 -07:00
parent 77487ada72
commit 5a3cb1064a
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
server/routes_test.go
View File

@ -65,7 +65,12 @@ func Test_Routes(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
} }
var blob string // Test Model Digests
blobs := []string{
"sha256:a4e5e156ddec27e286f75328784d7106b60a4eb1d246e950a001a3f944fbda99",
"sha256:4f9d252f34ae677363956ffc6dd2d10918a539c5c91f5ee2fe889d9178be6ae3",
"sha256:0f239b83e9e2aad7cd997a5bb44124937a32ac1f4e98e95a2f46e7b966bfc878",
}
testCases := []testCase{ testCases := []testCase{
{ {
@ -214,9 +219,11 @@ func Test_Routes(t *testing.T) {
Method: http.MethodDelete, Method: http.MethodDelete,
Path: "/api/delete", Path: "/api/delete",
Setup: func(t *testing.T, req *http.Request) { Setup: func(t *testing.T, req *http.Request) {
createTestModel(t, "delete-model-1") err := DeleteModel("test-model")
createTestModel(t, "delete-model-2") assert.Nil(t, err)
deleteReq := api.DeleteRequest{Model: "delete-model-1"} err = DeleteModel("hamshank")
assert.Nil(t, err)
deleteReq := api.DeleteRequest{Model: "beefsteak"}
jsonData, err := json.Marshal(deleteReq) jsonData, err := json.Marshal(deleteReq)
assert.Nil(t, err) assert.Nil(t, err)
req.Body = io.NopCloser(bytes.NewReader(jsonData)) req.Body = io.NopCloser(bytes.NewReader(jsonData))
@ -226,17 +233,17 @@ func Test_Routes(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, resp.StatusCode, 200) assert.Equal(t, resp.StatusCode, 200)
_, err = GetModel("delete-model-1") _, err = GetModel("beefsteak")
assert.True(t, os.IsNotExist(err)) assert.True(t, os.IsNotExist(err))
model, _ := GetModel("delete-model-2") model, _ := GetModel("show-model")
assert.Equal(t, "delete-model-2:latest", model.ShortName) assert.Equal(t, "show-model:latest", model.ShortName)
blob, err = GetBlobsPath(model.Digest) for _, blob := range blobs {
assert.Nil(t, err) blob, _ = GetBlobsPath(blob)
_, err := os.Stat(blob)
_, err = os.Stat(blob) assert.False(t, os.IsNotExist(err))
assert.False(t, os.IsNotExist(err)) }
}, },
}, },
{ {
@ -244,7 +251,7 @@ func Test_Routes(t *testing.T) {
Method: http.MethodDelete, Method: http.MethodDelete,
Path: "/api/delete", Path: "/api/delete",
Setup: func(t *testing.T, req *http.Request) { Setup: func(t *testing.T, req *http.Request) {
deleteReq := api.DeleteRequest{Model: "delete-model-2"} deleteReq := api.DeleteRequest{Model: "show-model"}
jsonData, err := json.Marshal(deleteReq) jsonData, err := json.Marshal(deleteReq)
assert.Nil(t, err) assert.Nil(t, err)
req.Body = io.NopCloser(bytes.NewReader(jsonData)) req.Body = io.NopCloser(bytes.NewReader(jsonData))
@ -253,11 +260,13 @@ func Test_Routes(t *testing.T) {
_, err := io.ReadAll(resp.Body) _, err := io.ReadAll(resp.Body)
assert.Nil(t, err) assert.Nil(t, err)
_, err = GetModel("delete-model-2") _, err = GetModel("show-model")
assert.True(t, os.IsNotExist(err)) assert.True(t, os.IsNotExist(err))
_, err = os.Stat(blob) for _, blob := range blobs {
assert.True(t, os.IsNotExist(err)) _, err := os.Stat(blob)
assert.True(t, os.IsNotExist(err))
}
}, },
}, },
} }