From c62df6b3bf22b0d035d40c29981487d828328dd7 Mon Sep 17 00:00:00 2001 From: Roy Han Date: Fri, 31 May 2024 12:07:52 -0700 Subject: [PATCH] Check Blob --- server/routes_test.go | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/server/routes_test.go b/server/routes_test.go index 650e50022..424dea963 100644 --- a/server/routes_test.go +++ b/server/routes_test.go @@ -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)) }, }, }