mirror of
https://github.com/project-zot/zot.git
synced 2026-06-18 05:28:07 +08:00
Andrei Aaron
9dfa7c3ae6
Replace MakeTempFile usage with MakeTempFilePath and MakeTempFileWithContent helpers that automatically handle file lifecycle. This prevents resource leaks by ensuring temporary files are properly closed. Shoudld also make the tests easier to read. Signed-off-by: Andrei Aaron <andreifdaaron@gmail.com>
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
//go:build search
|
|
|
|
package client_test
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
|
|
"zotregistry.dev/zot/v2/pkg/api"
|
|
"zotregistry.dev/zot/v2/pkg/api/config"
|
|
"zotregistry.dev/zot/v2/pkg/cli/client"
|
|
test "zotregistry.dev/zot/v2/pkg/test/common"
|
|
. "zotregistry.dev/zot/v2/pkg/test/image-utils"
|
|
)
|
|
|
|
func TestReposCommand(t *testing.T) {
|
|
Convey("repos", t, func() {
|
|
port := test.GetFreePort()
|
|
baseURL := test.GetBaseURL(port)
|
|
conf := config.New()
|
|
conf.HTTP.Port = port
|
|
|
|
ctlr := api.NewController(conf)
|
|
ctlr.Config.Storage.RootDirectory = t.TempDir()
|
|
cm := test.NewControllerManager(ctlr)
|
|
|
|
cm.StartAndWait(conf.HTTP.Port)
|
|
defer cm.StopServer()
|
|
|
|
err := UploadImage(CreateRandomImage(), baseURL, "repo1", "tag1")
|
|
So(err, ShouldBeNil)
|
|
err = UploadImage(CreateRandomImage(), baseURL, "repo2", "tag2")
|
|
So(err, ShouldBeNil)
|
|
|
|
_ = makeConfigFile(t, fmt.Sprintf(`{"configs":[{"_name":"repostest","url":"%s","showspinner":false}]}`,
|
|
baseURL))
|
|
|
|
args := []string{"list", "--config", "repostest"}
|
|
cmd := client.NewRepoCommand(client.NewSearchService())
|
|
buff := bytes.NewBufferString("")
|
|
cmd.SetOut(buff)
|
|
cmd.SetErr(buff)
|
|
cmd.SetArgs(args)
|
|
err = cmd.Execute()
|
|
So(err, ShouldBeNil)
|
|
space := regexp.MustCompile(`\s+`)
|
|
str := space.ReplaceAllString(buff.String(), " ")
|
|
actual := strings.TrimSpace(str)
|
|
So(actual, ShouldContainSubstring, "repo1")
|
|
So(actual, ShouldContainSubstring, "repo2")
|
|
|
|
args = []string{"list", "--sort-by", "alpha-dsc", "--config", "repostest"}
|
|
cmd = client.NewRepoCommand(client.NewSearchService())
|
|
buff = bytes.NewBufferString("")
|
|
cmd.SetOut(buff)
|
|
cmd.SetErr(buff)
|
|
cmd.SetArgs(args)
|
|
err = cmd.Execute()
|
|
So(err, ShouldBeNil)
|
|
space = regexp.MustCompile(`\s+`)
|
|
str = space.ReplaceAllString(buff.String(), " ")
|
|
actual = strings.TrimSpace(str)
|
|
So(actual, ShouldContainSubstring, "repo1")
|
|
So(actual, ShouldContainSubstring, "repo2")
|
|
So(strings.Index(actual, "repo2"), ShouldBeLessThan, strings.Index(actual, "repo1"))
|
|
|
|
args = []string{"list", "--sort-by", "alpha-asc", "--config", "repostest"}
|
|
cmd = client.NewRepoCommand(client.NewSearchService())
|
|
buff = bytes.NewBufferString("")
|
|
cmd.SetOut(buff)
|
|
cmd.SetErr(buff)
|
|
cmd.SetArgs(args)
|
|
err = cmd.Execute()
|
|
So(err, ShouldBeNil)
|
|
space = regexp.MustCompile(`\s+`)
|
|
str = space.ReplaceAllString(buff.String(), " ")
|
|
actual = strings.TrimSpace(str)
|
|
So(actual, ShouldContainSubstring, "repo1")
|
|
So(actual, ShouldContainSubstring, "repo2")
|
|
So(strings.Index(actual, "repo1"), ShouldBeLessThan, strings.Index(actual, "repo2"))
|
|
})
|
|
}
|
|
|
|
func TestSuggestions(t *testing.T) {
|
|
Convey("Suggestions", t, func() {
|
|
space := regexp.MustCompile(`\s+`)
|
|
suggestion := client.ShowSuggestionsIfUnknownCommand(
|
|
client.NewRepoCommand(client.NewSearchService()), []string{"bad-command"})
|
|
str := space.ReplaceAllString(suggestion.Error(), " ")
|
|
So(str, ShouldContainSubstring, "unknown cli subcommand")
|
|
|
|
suggestion = client.ShowSuggestionsIfUnknownCommand(
|
|
client.NewRepoCommand(client.NewSearchService()), []string{"listt"})
|
|
str = space.ReplaceAllString(suggestion.Error(), " ")
|
|
So(str, ShouldContainSubstring, "Did you mean this? list")
|
|
})
|
|
}
|