mirror of https://github.com/goharbor/harbor.git
73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package manifest
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/distribution/distribution/v3/registry/storage/driver/inmemory"
|
|
"github.com/distribution/distribution/v3/testutil"
|
|
"github.com/gorilla/mux"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/goharbor/harbor/src/registryctl/api/registry/test"
|
|
)
|
|
|
|
func TestDeleteManifest(t *testing.T) {
|
|
ctx := context.Background()
|
|
inmemoryDriver := inmemory.New()
|
|
|
|
registry := test.CreateRegistry(t, inmemoryDriver)
|
|
repo := test.MakeRepository(t, registry, "mftest")
|
|
|
|
// Create random layers
|
|
randomLayers, err := testutil.CreateRandomLayers(3)
|
|
if err != nil {
|
|
t.Fatalf("failed to make layers: %v", err)
|
|
}
|
|
|
|
// Upload all layers
|
|
err = testutil.UploadBlobs(repo, randomLayers)
|
|
if err != nil {
|
|
t.Fatalf("failed to upload layers: %v", err)
|
|
}
|
|
|
|
sharedKey := test.GetAnyKey(randomLayers)
|
|
manifest, err := testutil.MakeSchema2Manifest(repo, append(test.GetKeys(randomLayers), sharedKey))
|
|
if err != nil {
|
|
t.Fatalf("failed to make manifest: %v", err)
|
|
}
|
|
|
|
manifestService := test.MakeManifestService(t, repo)
|
|
_, err = manifestService.Put(ctx, manifest)
|
|
if err != nil {
|
|
t.Fatalf("manifest upload failed: %v", err)
|
|
}
|
|
|
|
manifestDigest, err := manifestService.Put(ctx, manifest)
|
|
if err != nil {
|
|
t.Fatalf("manifest upload failed: %v", err)
|
|
}
|
|
|
|
req, err := http.NewRequest(http.MethodDelete, "http://api/registry/{name}/manifests/{reference}/?tags=1,2,3", nil)
|
|
varMap := make(map[string]string, 1)
|
|
varMap["reference"] = manifestDigest.String()
|
|
varMap["name"] = fmt.Sprintf("%v", repo.Named())
|
|
req = mux.SetURLVars(req, varMap)
|
|
|
|
manifestHandler := NewHandler(inmemoryDriver)
|
|
rec := httptest.NewRecorder()
|
|
manifestHandler.ServeHTTP(rec, req)
|
|
assert.True(t, rec.Result().StatusCode == 200)
|
|
|
|
// check that all of the layers of manifest are deleted.
|
|
blobs := test.AllBlobs(t, registry)
|
|
for dgst := range randomLayers {
|
|
if _, ok := blobs[dgst]; !ok {
|
|
t.Fatalf("random layer blob missing: %v", dgst)
|
|
}
|
|
}
|
|
}
|