From e8788ae8dd16e591b51a3ec8e607faf3897ad78c Mon Sep 17 00:00:00 2001 From: Roy Han Date: Fri, 31 May 2024 09:45:47 -0700 Subject: [PATCH] Specify DNE error --- server/routes_test.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/server/routes_test.go b/server/routes_test.go index fe250b07c..650e50022 100644 --- a/server/routes_test.go +++ b/server/routes_test.go @@ -227,13 +227,14 @@ func Test_Routes(t *testing.T) { assert.Equal(t, resp.StatusCode, 200) _, err = GetModel("delete-model-1") - assert.NotNil(t, err) + assert.True(t, os.IsNotExist(err)) + model, err := GetModel("delete-model-2") assert.Nil(t, err) assert.Equal(t, "delete-model-2:latest", model.ShortName) _, err = GetBlobsPath(model.Digest) - assert.NotNil(t, err) + assert.Nil(t, err) }, }, { @@ -253,7 +254,7 @@ func Test_Routes(t *testing.T) { assert.Nil(t, err) _, err = GetModel("delete-model-2") - assert.Nil(t, err) + assert.True(t, os.IsNotExist(err)) // TODO: check if blob is deleted