mirror of
https://github.com/project-zot/zot.git
synced 2024-12-23 22:27:35 -05:00
b997176363
Signed-off-by: Alex Stan <alexandrustan96@yahoo.ro> add page info to dynamo-> feat(repodb): add PageInfo to GlobalSearch and RepoListWithNewestImage results (cherry picked from commit 4fed42bb4bbc68199281d9d9a4e09b97fbd4759b) Signed-off-by: Andrei Aaron <aaaron@luxoft.com> Signed-off-by: Alex Stan <alexandrustan96@yahoo.ro> Signed-off-by: Andrei Aaron <aaaron@luxoft.com> Co-authored-by: Alex Stan <alexandrustan96@yahoo.ro>
240 lines
6.7 KiB
Go
240 lines
6.7 KiB
Go
package repodb_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
ispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
|
|
"zotregistry.io/zot/pkg/meta/repodb"
|
|
)
|
|
|
|
func TestPagination(t *testing.T) {
|
|
Convey("Repo Pagination", t, func() {
|
|
Convey("reset", func() {
|
|
pageFinder, err := repodb.NewBaseRepoPageFinder(1, 0, repodb.AlphabeticAsc)
|
|
So(err, ShouldBeNil)
|
|
So(pageFinder, ShouldNotBeNil)
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{})
|
|
pageFinder.Add(repodb.DetailedRepoMeta{})
|
|
pageFinder.Add(repodb.DetailedRepoMeta{})
|
|
|
|
pageFinder.Reset()
|
|
|
|
result, _ := pageFinder.Page()
|
|
So(result, ShouldBeEmpty)
|
|
})
|
|
})
|
|
|
|
Convey("Image Pagination", t, func() {
|
|
Convey("create new pageFinder errors", func() {
|
|
pageFinder, err := repodb.NewBaseImagePageFinder(-1, 10, repodb.AlphabeticAsc)
|
|
So(pageFinder, ShouldBeNil)
|
|
So(err, ShouldNotBeNil)
|
|
|
|
pageFinder, err = repodb.NewBaseImagePageFinder(2, -1, repodb.AlphabeticAsc)
|
|
So(pageFinder, ShouldBeNil)
|
|
So(err, ShouldNotBeNil)
|
|
|
|
pageFinder, err = repodb.NewBaseImagePageFinder(2, 1, "wrong sorting criteria")
|
|
So(pageFinder, ShouldBeNil)
|
|
So(err, ShouldNotBeNil)
|
|
})
|
|
|
|
Convey("Reset", func() {
|
|
pageFinder, err := repodb.NewBaseImagePageFinder(1, 0, repodb.AlphabeticAsc)
|
|
So(err, ShouldBeNil)
|
|
So(pageFinder, ShouldNotBeNil)
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{})
|
|
pageFinder.Add(repodb.DetailedRepoMeta{})
|
|
pageFinder.Add(repodb.DetailedRepoMeta{})
|
|
|
|
pageFinder.Reset()
|
|
|
|
result, _ := pageFinder.Page()
|
|
So(result, ShouldBeEmpty)
|
|
})
|
|
|
|
Convey("Page", func() {
|
|
Convey("no limit or offset", func() {
|
|
pageFinder, err := repodb.NewBaseImagePageFinder(0, 0, repodb.AlphabeticAsc)
|
|
So(err, ShouldBeNil)
|
|
So(pageFinder, ShouldNotBeNil)
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo1",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"tag1": {Digest: "dig1", MediaType: ispec.MediaTypeImageManifest},
|
|
},
|
|
},
|
|
})
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo2",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"Tag1": {Digest: "dig1", MediaType: ispec.MediaTypeImageManifest},
|
|
"Tag2": {Digest: "dig2", MediaType: ispec.MediaTypeImageManifest},
|
|
"Tag3": {Digest: "dig3", MediaType: ispec.MediaTypeImageManifest},
|
|
"Tag4": {Digest: "dig4", MediaType: ispec.MediaTypeImageManifest},
|
|
},
|
|
},
|
|
})
|
|
_, pageInfo := pageFinder.Page()
|
|
So(pageInfo.ItemCount, ShouldEqual, 5)
|
|
So(pageInfo.TotalCount, ShouldEqual, 5)
|
|
})
|
|
Convey("Test 1 limit < len(tags)", func() {
|
|
pageFinder, err := repodb.NewBaseImagePageFinder(5, 2, repodb.AlphabeticAsc)
|
|
So(err, ShouldBeNil)
|
|
So(pageFinder, ShouldNotBeNil)
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo1",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"tag1": {Digest: "dig1", MediaType: ispec.MediaTypeImageManifest},
|
|
},
|
|
},
|
|
})
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo2",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"Tag1": {Digest: "dig1", MediaType: ispec.MediaTypeImageManifest},
|
|
"Tag2": {Digest: "dig2", MediaType: ispec.MediaTypeImageManifest},
|
|
"Tag3": {Digest: "dig3", MediaType: ispec.MediaTypeImageManifest},
|
|
"Tag4": {Digest: "dig4", MediaType: ispec.MediaTypeImageManifest},
|
|
},
|
|
},
|
|
})
|
|
_, pageInfo := pageFinder.Page()
|
|
So(pageInfo.ItemCount, ShouldEqual, 3)
|
|
So(pageInfo.TotalCount, ShouldEqual, 5)
|
|
})
|
|
Convey("Test 2 limit < len(tags)", func() {
|
|
pageFinder, err := repodb.NewBaseImagePageFinder(5, 2, repodb.AlphabeticAsc)
|
|
So(err, ShouldBeNil)
|
|
So(pageFinder, ShouldNotBeNil)
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo1",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"tag1": {
|
|
Digest: "dig1",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo2",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"Tag1": {
|
|
Digest: "dig1",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
"Tag2": {
|
|
Digest: "dig2",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
"Tag3": {
|
|
Digest: "dig3",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
"Tag4": {
|
|
Digest: "dig4",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo3",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"Tag11": {
|
|
Digest: "dig11",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
"Tag12": {
|
|
Digest: "dig12",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
"Tag13": {
|
|
Digest: "dig13",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
"Tag14": {
|
|
Digest: "dig14",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
result, pageInfo := pageFinder.Page()
|
|
So(result[0].Tags, ShouldContainKey, "Tag2")
|
|
So(result[0].Tags, ShouldContainKey, "Tag3")
|
|
So(result[0].Tags, ShouldContainKey, "Tag4")
|
|
So(result[1].Tags, ShouldContainKey, "Tag11")
|
|
So(result[1].Tags, ShouldContainKey, "Tag12")
|
|
So(pageInfo.ItemCount, ShouldEqual, 5)
|
|
So(pageInfo.TotalCount, ShouldEqual, 9)
|
|
})
|
|
|
|
Convey("Test 2 limit > len(tags)", func() {
|
|
pageFinder, err := repodb.NewBaseImagePageFinder(3, 0, repodb.AlphabeticAsc)
|
|
So(err, ShouldBeNil)
|
|
So(pageFinder, ShouldNotBeNil)
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo1",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"tag1": {
|
|
Digest: "dig1",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo2",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"Tag1": {
|
|
Digest: "dig1",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
pageFinder.Add(repodb.DetailedRepoMeta{
|
|
RepoMeta: repodb.RepoMetadata{
|
|
Name: "repo3",
|
|
Tags: map[string]repodb.Descriptor{
|
|
"Tag11": {
|
|
Digest: "dig11",
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
result, _ := pageFinder.Page()
|
|
So(result[0].Tags, ShouldContainKey, "tag1")
|
|
So(result[1].Tags, ShouldContainKey, "Tag1")
|
|
So(result[2].Tags, ShouldContainKey, "Tag11")
|
|
})
|
|
})
|
|
})
|
|
}
|