opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: extend banner guard coverage

Browse source
This commit is contained in:
XYSK-lilong007 2026-03-12 01:58:14 +08:00
parent df19b8c708
commit 73a1dbc8a6
2 changed files with 24 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cmd/picoclaw/main.go
View file

@ -54,7 +54,6 @@ func NewPicoclawCommand() *cobra.Command {
const (
colorBlue = "\033[1;38;2;62;93;185m"
colorRed = "\033[1;38;2;213;70;70m"
noBannerEnv = "PICOCLAW_NO_BANNER"
banner = "\r\n" +
colorBlue + "██████╗ ██╗ ██████╗ ██████╗ " + colorRed + " ██████╗██╗ █████╗ ██╗ ██╗\n" +
colorBlue + "██╔══██╗██║██╔════╝██╔═══██╗" + colorRed + "██╔════╝██║ ██╔══██╗██║ ██║\n" +
@ -65,6 +64,8 @@ const (
"\033[0m\r\n"
)
const noBannerEnv = "PICOCLAW_NO_BANNER"
func bannerDisabledByEnv() bool {
value := strings.TrimSpace(strings.ToLower(os.Getenv(noBannerEnv)))
switch value {

19
cmd/picoclaw/main_test.go
View file

@ -79,4 +79,23 @@ func TestShouldPrintBanner(t *testing.T) {
t.Setenv(noBannerEnv, "1")
assert.False(t, shouldPrintBanner([]string{"picoclaw", "agent"}, true))
})
t.Run("common truthy env values disable banner", func(t *testing.T) {
for _, value := range []string{"true", "yes", "on"} {
t.Run(value, func(t *testing.T) {
t.Setenv(noBannerEnv, value)
assert.False(t, shouldPrintBanner([]string{"picoclaw", "agent"}, true))
})
}
})
t.Run("root command still prints banner in terminal", func(t *testing.T) {
t.Setenv(noBannerEnv, "")
assert.True(t, shouldPrintBanner([]string{"picoclaw"}, true))
})
t.Run("unknown subcommand still prints banner in terminal", func(t *testing.T) {
t.Setenv(noBannerEnv, "")
assert.True(t, shouldPrintBanner([]string{"picoclaw", "unknown"}, true))
})
}