fix(cov): coverage boltdb+dynamo (#2018)

Signed-off-by: Laurentiu Niculae <niculae.laurentiu1@gmail.com>
2026-06-17 12:58:02 +08:00 · 2023-11-08 23:37:52 +02:00
parent c9cc5b9acb
commit 2db6e86fb5
22 changed files with 1796 additions and 100 deletions
@@ -0,0 +1,115 @@
+package ociutils_test
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	. "github.com/smartystreets/goconvey/convey"
+
+	"zotregistry.io/zot/pkg/meta/types"
+	"zotregistry.io/zot/pkg/test/image-utils"
+	"zotregistry.io/zot/pkg/test/mocks"
+	ociutils "zotregistry.io/zot/pkg/test/oci-utils"
+)
+
+var ErrTestFail = fmt.Errorf("fail")
+
+func TestInitializeMetaDBErrors(t *testing.T) {
+	ctx := context.Background()
+
+	Convey("InitializeTestMetaDB", t, func() {
+		metaDB := mocks.MetaDBMock{
+			GetRepoMetaFn: func(ctx context.Context, repo string) (types.RepoMeta, error) {
+				return types.RepoMeta{
+					Statistics: map[string]types.DescriptorStatistics{},
+					Signatures: map[string]types.ManifestSignatures{},
+					Referrers:  map[string][]types.ReferrerInfo{},
+				}, nil
+			},
+		}
+
+		Convey("Multiple repos same name", func() {
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB, ociutils.Repo{Name: "repo"}, ociutils.Repo{Name: "repo"})
+			So(err, ShouldNotBeNil)
+		})
+		Convey("Set Repo Ref fails", func() {
+			metaDB.SetRepoReferenceFn = func(ctx context.Context, repo, reference string, imageMeta types.ImageMeta) error {
+				return ErrTestFail
+			}
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{Name: "repo", Images: []ociutils.RepoImage{{}}},
+			)
+			So(err, ShouldNotBeNil)
+		})
+		Convey("Set Repo Ref fails for manifest in index", func() {
+			metaDB.SetRepoReferenceFn = func(ctx context.Context, repo, reference string, imageMeta types.ImageMeta) error {
+				return ErrTestFail
+			}
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{
+					Name:            "repo",
+					MultiArchImages: []ociutils.RepoMultiArchImage{{MultiarchImage: image.CreateRandomMultiarch()}},
+				},
+			)
+			So(err, ShouldNotBeNil)
+		})
+		Convey("Set Repo Ref fails for index", func() {
+			count := 0
+			metaDB.SetRepoReferenceFn = func(ctx context.Context, repo, reference string, imageMeta types.ImageMeta) error {
+				if count == 1 {
+					return ErrTestFail
+				}
+
+				count++
+
+				return nil
+			}
+
+			multiarch := image.CreateMultiarchWith().Images([]image.Image{{}}).Build()
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{
+					Name:            "repo",
+					MultiArchImages: []ociutils.RepoMultiArchImage{{MultiarchImage: multiarch}},
+				},
+			)
+			So(err, ShouldNotBeNil)
+		})
+		Convey("Get repo meta errors", func() {
+			metaDB.GetRepoMetaFn = func(ctx context.Context, repo string) (types.RepoMeta, error) {
+				return types.RepoMeta{}, ErrTestFail
+			}
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{Name: "repo", Images: []ociutils.RepoImage{{}}},
+			)
+			So(err, ShouldNotBeNil)
+		})
+		Convey("Set repo meta errors", func() {
+			metaDB.SetRepoMetaFn = func(repo string, repoMeta types.RepoMeta) error {
+				return ErrTestFail
+			}
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{Name: "repo", Images: []ociutils.RepoImage{{}}},
+			)
+			So(err, ShouldNotBeNil)
+		})
+		Convey("ToggleBookmarkRepo errors", func() {
+			metaDB.ToggleBookmarkRepoFn = func(ctx context.Context, repo string) (types.ToggleState, error) {
+				return types.NotChanged, ErrTestFail
+			}
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{Name: "repo", Images: []ociutils.RepoImage{{}}, IsBookmarked: true},
+			)
+			So(err, ShouldNotBeNil)
+		})
+		Convey("ToggleStarRepo errors", func() {
+			metaDB.ToggleStarRepoFn = func(ctx context.Context, repo string) (types.ToggleState, error) {
+				return types.NotChanged, ErrTestFail
+			}
+			_, err := ociutils.InitializeTestMetaDB(ctx, metaDB,
+				ociutils.Repo{Name: "repo", Images: []ociutils.RepoImage{{}}, IsStarred: true},
+			)
+			So(err, ShouldNotBeNil)
+		})
+	})
+}