mirror of
https://github.com/goharbor/harbor
synced 2025-04-20 19:18:02 +00:00
bad913cf6d
* use lib trace helper function * add gracefull shutdown * Add commens for new added exposed function * Add licence on top of new created files * Update trace library * Update configs * Add attribute and namespance in config Signed-off-by: Qian Deng <dengq@vmware.com>
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package blob
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/docker/distribution/registry/storage/driver/inmemory"
|
|
"github.com/docker/distribution/testutil"
|
|
"github.com/goharbor/harbor/src/registryctl/api/registry/test"
|
|
"github.com/gorilla/mux"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDeletionBlob(t *testing.T) {
|
|
inmemoryDriver := inmemory.New()
|
|
|
|
registry := test.CreateRegistry(t, inmemoryDriver)
|
|
repo := test.MakeRepository(t, registry, "blobdeletion")
|
|
|
|
// Create random layers
|
|
randomLayers1, err := testutil.CreateRandomLayers(3)
|
|
if err != nil {
|
|
t.Fatalf("failed to make layers: %v", err)
|
|
}
|
|
|
|
randomLayers2, err := testutil.CreateRandomLayers(3)
|
|
if err != nil {
|
|
t.Fatalf("failed to make layers: %v", err)
|
|
}
|
|
|
|
// Upload all layers
|
|
err = testutil.UploadBlobs(repo, randomLayers1)
|
|
if err != nil {
|
|
t.Fatalf("failed to upload layers: %v", err)
|
|
}
|
|
|
|
err = testutil.UploadBlobs(repo, randomLayers2)
|
|
if err != nil {
|
|
t.Fatalf("failed to upload layers: %v", err)
|
|
}
|
|
|
|
req, err := http.NewRequest(http.MethodDelete, "", nil)
|
|
varMap := make(map[string]string, 1)
|
|
varMap["reference"] = test.GetKeys(randomLayers1)[0].String()
|
|
req = mux.SetURLVars(req, varMap)
|
|
|
|
blobHandler := NewHandler(inmemoryDriver)
|
|
rec := httptest.NewRecorder()
|
|
blobHandler.ServeHTTP(rec, req)
|
|
assert.True(t, rec.Result().StatusCode == 200)
|
|
|
|
// layer1 is deleted and layer2 is still there
|
|
blobs := test.AllBlobs(t, registry)
|
|
for dgst := range randomLayers1 {
|
|
if _, ok := blobs[dgst]; !ok {
|
|
t.Logf("random layer 1 blob missing is correct as it has been deleted: %v", dgst)
|
|
}
|
|
}
|
|
for dgst := range randomLayers2 {
|
|
if _, ok := blobs[dgst]; !ok {
|
|
t.Fatalf("random layer 2 blob missing: %v", dgst)
|
|
}
|
|
}
|
|
}
|