mirror of
https://github.com/grdl/git-get.git
synced 2026-02-04 19:44:41 +00:00
32211e6408
- Use strings.Builder instead of concatenation - Move smart tree to a separate file - Add missing docstrings
109 lines
1.9 KiB
Go
109 lines
1.9 KiB
Go
package print
|
|
|
|
import (
|
|
"fmt"
|
|
"git-get/pkg/repo"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestTree(t *testing.T) {
|
|
var tests = []struct {
|
|
paths []string
|
|
want string
|
|
}{
|
|
{
|
|
[]string{
|
|
"root/github.com/grdl/repo1",
|
|
}, `
|
|
root/
|
|
github.com/grdl/repo1
|
|
`,
|
|
},
|
|
{
|
|
[]string{
|
|
"root/github.com/grdl/repo1",
|
|
"root/github.com/grdl/repo2",
|
|
}, `
|
|
root/
|
|
github.com/grdl/
|
|
repo1
|
|
repo2
|
|
`,
|
|
},
|
|
{
|
|
[]string{
|
|
"root/gitlab.com/grdl/repo1",
|
|
"root/github.com/grdl/repo1",
|
|
}, `
|
|
root/
|
|
gitlab.com/grdl/repo1
|
|
github.com/grdl/repo1
|
|
`,
|
|
},
|
|
{
|
|
[]string{
|
|
"root/gitlab.com/grdl/repo1",
|
|
"root/gitlab.com/grdl/repo2",
|
|
"root/gitlab.com/other/repo1",
|
|
"root/github.com/grdl/repo1",
|
|
"root/github.com/grdl/nested/repo2",
|
|
}, `
|
|
root/
|
|
gitlab.com/
|
|
grdl/
|
|
repo1
|
|
repo2
|
|
other/repo1
|
|
github.com/grdl/
|
|
repo1
|
|
nested/repo2
|
|
`,
|
|
},
|
|
{
|
|
[]string{
|
|
"root/gitlab.com/grdl/nested/repo1",
|
|
"root/gitlab.com/grdl/nested/repo2",
|
|
"root/gitlab.com/other/repo1",
|
|
}, `
|
|
root/
|
|
gitlab.com/
|
|
grdl/nested/
|
|
repo1
|
|
repo2
|
|
other/repo1
|
|
`,
|
|
},
|
|
{
|
|
[]string{
|
|
"root/gitlab.com/grdl/double/nested/repo1",
|
|
"root/gitlab.com/grdl/nested/repo2",
|
|
"root/gitlab.com/other/repo1",
|
|
}, `
|
|
root/
|
|
gitlab.com/
|
|
grdl/
|
|
double/nested/repo1
|
|
nested/repo2
|
|
other/repo1
|
|
`,
|
|
},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
var repos []*repo.Repo
|
|
for _, path := range test.paths {
|
|
repos = append(repos, repo.New(nil, path)) //&Repo{path: path})
|
|
}
|
|
|
|
printer := SmartPrinter{}
|
|
// Leading and trailing newlines are added to test cases for readability. We also need to add them to the rendering result.
|
|
got := fmt.Sprintf("\n%s\n", printer.Print("root", repos))
|
|
|
|
// Rendered tree uses spaces for indentation but the test cases use tabs.
|
|
if got != strings.ReplaceAll(test.want, "\t", " ") {
|
|
t.Errorf("Failed test case %d, got: %+v; want: %+v", i, got, test.want)
|
|
}
|
|
}
|
|
}
|