mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-02 09:21:40 +02:00
1de876ed4d
The menuFromCommand option is a little complicated, so I'm adding an easy way to just use the command output directly, where each line becomes a suggestion, as-is. Now that we support suggestions in the input prompt, there's less of a need for menuFromCommand, but it probably still serves some purpose. In future I want to support this filter/valueFormat/labelFormat thing for suggestions too. I would like to think a little more about the interface though: is using a regex like we currently do really the simplest approach?
90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package custom_commands
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMenuGenerator(t *testing.T) {
|
|
type scenario struct {
|
|
testName string
|
|
cmdOut string
|
|
filter string
|
|
valueFormat string
|
|
labelFormat string
|
|
test func([]*commandMenuItem, error)
|
|
}
|
|
|
|
scenarios := []scenario{
|
|
{
|
|
"Extract remote branch name",
|
|
"upstream/pr-1",
|
|
"(?P<remote>[a-z_]+)/(?P<branch>.*)",
|
|
"{{ .branch }}",
|
|
"Remote: {{ .remote }}",
|
|
func(actualEntry []*commandMenuItem, err error) {
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "pr-1", actualEntry[0].value)
|
|
assert.EqualValues(t, "Remote: upstream", actualEntry[0].label)
|
|
},
|
|
},
|
|
{
|
|
"Multiple named groups with empty labelFormat",
|
|
"upstream/pr-1",
|
|
"(?P<remote>[a-z]*)/(?P<branch>.*)",
|
|
"{{ .branch }}|{{ .remote }}",
|
|
"",
|
|
func(actualEntry []*commandMenuItem, err error) {
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "pr-1|upstream", actualEntry[0].value)
|
|
assert.EqualValues(t, "pr-1|upstream", actualEntry[0].label)
|
|
},
|
|
},
|
|
{
|
|
"Multiple named groups with group ids",
|
|
"upstream/pr-1",
|
|
"(?P<remote>[a-z]*)/(?P<branch>.*)",
|
|
"{{ .group_2 }}|{{ .group_1 }}",
|
|
"Remote: {{ .group_1 }}",
|
|
func(actualEntry []*commandMenuItem, err error) {
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "pr-1|upstream", actualEntry[0].value)
|
|
assert.EqualValues(t, "Remote: upstream", actualEntry[0].label)
|
|
},
|
|
},
|
|
{
|
|
"No named groups",
|
|
"upstream/pr-1",
|
|
"([a-z]*)/(.*)",
|
|
"{{ .group_2 }}|{{ .group_1 }}",
|
|
"Remote: {{ .group_1 }}",
|
|
func(actualEntry []*commandMenuItem, err error) {
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "pr-1|upstream", actualEntry[0].value)
|
|
assert.EqualValues(t, "Remote: upstream", actualEntry[0].label)
|
|
},
|
|
},
|
|
{
|
|
"No filter",
|
|
"upstream/pr-1",
|
|
"",
|
|
"",
|
|
"",
|
|
func(actualEntry []*commandMenuItem, err error) {
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "upstream/pr-1", actualEntry[0].value)
|
|
assert.EqualValues(t, "upstream/pr-1", actualEntry[0].label)
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, s := range scenarios {
|
|
s := s
|
|
t.Run(s.testName, func(t *testing.T) {
|
|
s.test(NewMenuGenerator(utils.NewDummyCommon()).call(s.cmdOut, s.filter, s.valueFormat, s.labelFormat))
|
|
})
|
|
}
|
|
}
|