From aec06fbb0df14f1611bc0bf404f2904a178d567c Mon Sep 17 00:00:00 2001 From: Andres Martinez Gotor Date: Thu, 28 Nov 2019 10:40:28 +0100 Subject: [PATCH] Delete repository from all collections (#655) Signed-off-by: Andres Martinez Gotor --- cmd/chart-repo/utils.go | 7 +++++++ cmd/chart-repo/utils_test.go | 3 +++ 2 files changed, 10 insertions(+) diff --git a/cmd/chart-repo/utils.go b/cmd/chart-repo/utils.go index c5ba0791c..79b17c6db 100644 --- a/cmd/chart-repo/utils.go +++ b/cmd/chart-repo/utils.go @@ -212,6 +212,13 @@ func deleteRepo(dbSession datastore.Session, repoName string) error { _, err = db.C(chartFilesCollection).RemoveAll(bson.M{ "repo.name": repoName, }) + if err != nil { + return err + } + + _, err = db.C(repositoryCollection).RemoveAll(bson.M{ + "_id": repoName, + }) return err } diff --git a/cmd/chart-repo/utils_test.go b/cmd/chart-repo/utils_test.go index 58b32de36..0adc27918 100644 --- a/cmd/chart-repo/utils_test.go +++ b/cmd/chart-repo/utils_test.go @@ -336,6 +336,9 @@ func Test_DeleteRepo(t *testing.T) { m.On("RemoveAll", bson.M{ "repo.name": "test", }) + m.On("RemoveAll", bson.M{ + "_id": "test", + }) dbSession := mockstore.NewMockSession(m) err := deleteRepo(dbSession, "test")