mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-14 11:23:09 +02:00
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package clients
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/integration/components"
|
|
"github.com/jesseduffield/lazygit/pkg/integration/tests"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
// see pkg/integration/README.md
|
|
|
|
// The purpose of this program is to run integration tests. It does this by
|
|
// building our injector program (in the sibling injector directory) and then for
|
|
// each test we're running, invoke the injector program with the test's name as
|
|
// an environment variable. Then the injector finds the test and passes it to
|
|
// the lazygit startup code.
|
|
|
|
// If invoked directly, you can specify tests to run by passing their names as positional arguments
|
|
|
|
func RunCLI(testNames []string, slow bool, sandbox bool) {
|
|
inputDelay := tryConvert(os.Getenv("INPUT_DELAY"), 0)
|
|
if slow {
|
|
inputDelay = SLOW_INPUT_DELAY
|
|
}
|
|
|
|
err := components.RunTests(
|
|
getTestsToRun(testNames),
|
|
log.Printf,
|
|
runCmdInTerminal,
|
|
runAndPrintFatalError,
|
|
sandbox,
|
|
inputDelay,
|
|
1,
|
|
)
|
|
if err != nil {
|
|
log.Print(err.Error())
|
|
}
|
|
}
|
|
|
|
func runAndPrintFatalError(test *components.IntegrationTest, f func() error) {
|
|
if err := f(); err != nil {
|
|
log.Fatalf(err.Error())
|
|
}
|
|
}
|
|
|
|
func getTestsToRun(testNames []string) []*components.IntegrationTest {
|
|
allIntegrationTests := tests.GetTests()
|
|
var testsToRun []*components.IntegrationTest
|
|
|
|
if len(testNames) == 0 {
|
|
return allIntegrationTests
|
|
}
|
|
|
|
testNames = lo.Map(testNames, func(name string, _ int) string {
|
|
// allowing full test paths to be passed for convenience
|
|
return strings.TrimSuffix(
|
|
regexp.MustCompile(`.*pkg/integration/tests/`).ReplaceAllString(name, ""),
|
|
".go",
|
|
)
|
|
})
|
|
|
|
if lo.SomeBy(testNames, func(name string) bool {
|
|
return strings.HasSuffix(name, "/shared")
|
|
}) {
|
|
log.Fatalf("'shared' is a reserved name for tests that are shared between multiple test files. Please rename your test.")
|
|
}
|
|
|
|
outer:
|
|
for _, testName := range testNames {
|
|
// check if our given test name actually exists
|
|
for _, test := range allIntegrationTests {
|
|
if test.Name() == testName {
|
|
testsToRun = append(testsToRun, test)
|
|
continue outer
|
|
}
|
|
}
|
|
log.Fatalf("test %s not found. Perhaps you forgot to add it to `pkg/integration/integration_tests/test_list.go`? This can be done by running `go generate ./...` from the Lazygit root. You'll need to ensure that your test name and the file name match (where the test name is in PascalCase and the file name is in snake_case).", testName)
|
|
}
|
|
|
|
return testsToRun
|
|
}
|
|
|
|
func runCmdInTerminal(cmd *exec.Cmd) error {
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stderr = os.Stderr
|
|
|
|
return cmd.Run()
|
|
}
|
|
|
|
func tryConvert(numStr string, defaultVal int) int {
|
|
num, err := strconv.Atoi(numStr)
|
|
if err != nil {
|
|
return defaultVal
|
|
}
|
|
|
|
return num
|
|
}
|