6a8bd92189
Move pkg/cli from core/go to core/cli. Includes Frame AppShell, Stream, TaskTracker, Tree, Rich Table. Update imports to v0.0.1 tagged deps and fix openpgp import path for go-crypt split. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package cli
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTree_Good(t *testing.T) {
|
|
t.Run("single root", func(t *testing.T) {
|
|
tree := NewTree("root")
|
|
assert.Equal(t, "root\n", tree.String())
|
|
})
|
|
|
|
t.Run("flat children", func(t *testing.T) {
|
|
tree := NewTree("root")
|
|
tree.Add("alpha")
|
|
tree.Add("beta")
|
|
tree.Add("gamma")
|
|
|
|
expected := "root\n" +
|
|
"├── alpha\n" +
|
|
"├── beta\n" +
|
|
"└── gamma\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
|
|
t.Run("nested children", func(t *testing.T) {
|
|
tree := NewTree("core-php")
|
|
tree.Add("core-tenant").Add("core-bio")
|
|
tree.Add("core-admin")
|
|
tree.Add("core-api")
|
|
|
|
expected := "core-php\n" +
|
|
"├── core-tenant\n" +
|
|
"│ └── core-bio\n" +
|
|
"├── core-admin\n" +
|
|
"└── core-api\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
|
|
t.Run("deep nesting", func(t *testing.T) {
|
|
tree := NewTree("a")
|
|
tree.Add("b").Add("c").Add("d")
|
|
|
|
expected := "a\n" +
|
|
"└── b\n" +
|
|
" └── c\n" +
|
|
" └── d\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
|
|
t.Run("mixed depth", func(t *testing.T) {
|
|
tree := NewTree("root")
|
|
a := tree.Add("a")
|
|
a.Add("a1")
|
|
a.Add("a2")
|
|
tree.Add("b")
|
|
|
|
expected := "root\n" +
|
|
"├── a\n" +
|
|
"│ ├── a1\n" +
|
|
"│ └── a2\n" +
|
|
"└── b\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
|
|
t.Run("AddTree composes subtrees", func(t *testing.T) {
|
|
sub := NewTree("sub-root")
|
|
sub.Add("child")
|
|
|
|
tree := NewTree("main")
|
|
tree.AddTree(sub)
|
|
|
|
expected := "main\n" +
|
|
"└── sub-root\n" +
|
|
" └── child\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
|
|
t.Run("styled nodes", func(t *testing.T) {
|
|
SetColorEnabled(false)
|
|
defer SetColorEnabled(true)
|
|
|
|
tree := NewTree("root")
|
|
tree.AddStyled("green", SuccessStyle)
|
|
tree.Add("plain")
|
|
|
|
expected := "root\n" +
|
|
"├── green\n" +
|
|
"└── plain\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
|
|
t.Run("WithStyle on root", func(t *testing.T) {
|
|
SetColorEnabled(false)
|
|
defer SetColorEnabled(true)
|
|
|
|
tree := NewTree("root").WithStyle(ErrorStyle)
|
|
tree.Add("child")
|
|
|
|
expected := "root\n" +
|
|
"└── child\n"
|
|
assert.Equal(t, expected, tree.String())
|
|
})
|
|
}
|
|
|
|
func TestTree_Bad(t *testing.T) {
|
|
t.Run("empty label", func(t *testing.T) {
|
|
tree := NewTree("")
|
|
assert.Equal(t, "\n", tree.String())
|
|
})
|
|
}
|