mirror of
https://github.com/kopia/kopia.git
synced 2026-03-06 15:28:08 -05:00
8515d050e5
* feat(infra): improved support for in-process testing * support for killing of a running server using simulated Ctrl-C * support for overriding os.Stdin * migrated many tests from the exe runner to in-process runner * added required indirection when defining Envar() so we can later override it in tests * refactored CLI runners by moving environment overrides to CLITestEnv
87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package testenv
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/alecthomas/kingpin"
|
|
|
|
"github.com/kopia/kopia/cli"
|
|
"github.com/kopia/kopia/internal/testlogging"
|
|
)
|
|
|
|
var envPrefixCounter = new(int32)
|
|
|
|
// CLIInProcRunner is a CLIRunner that invokes provided commands in the current process.
|
|
type CLIInProcRunner struct {
|
|
mu sync.Mutex
|
|
|
|
// +checklocks:mu
|
|
nextCommandStdin io.Reader // this is used for stdin source tests
|
|
|
|
CustomizeApp func(a *cli.App, kp *kingpin.Application)
|
|
}
|
|
|
|
// Start implements CLIRunner.
|
|
func (e *CLIInProcRunner) Start(t *testing.T, args []string, env map[string]string) (stdout, stderr io.Reader, wait func() error, kill func()) {
|
|
t.Helper()
|
|
|
|
ctx := testlogging.Context(t)
|
|
|
|
a := cli.NewApp()
|
|
a.AdvancedCommands = "enabled"
|
|
|
|
envPrefix := fmt.Sprintf("T%v_", atomic.AddInt32(envPrefixCounter, 1))
|
|
a.SetEnvNamePrefixForTesting(envPrefix)
|
|
|
|
kpapp := kingpin.New("test", "test")
|
|
|
|
if e.CustomizeApp != nil {
|
|
e.CustomizeApp(a, kpapp)
|
|
}
|
|
|
|
e.mu.Lock()
|
|
stdin := e.nextCommandStdin
|
|
e.nextCommandStdin = nil
|
|
e.mu.Unlock()
|
|
|
|
for k, v := range env {
|
|
os.Setenv(envPrefix+k, v)
|
|
}
|
|
|
|
return a.RunSubcommand(ctx, kpapp, stdin, args)
|
|
}
|
|
|
|
// SetNextStdin sets the stdin to be used on next command execution.
|
|
func (e *CLIInProcRunner) SetNextStdin(stdin io.Reader) {
|
|
e.mu.Lock()
|
|
defer e.mu.Unlock()
|
|
|
|
e.nextCommandStdin = stdin
|
|
}
|
|
|
|
// NewInProcRunner returns a runner that executes CLI subcommands in the current process using cli.RunSubcommand().
|
|
func NewInProcRunner(t *testing.T) *CLIInProcRunner {
|
|
t.Helper()
|
|
|
|
if os.Getenv("KOPIA_EXE") != "" && os.Getenv("KOPIA_RUN_ALL_INTEGRATION_TESTS") == "" {
|
|
t.Skip("not running test since it's also included in the unit tests")
|
|
}
|
|
|
|
return &CLIInProcRunner{
|
|
CustomizeApp: func(a *cli.App, kp *kingpin.Application) {
|
|
a.AddStorageProvider(cli.StorageProvider{
|
|
Name: "in-memory",
|
|
Description: "in-memory storage backend",
|
|
NewFlags: func() cli.StorageFlags { return &storageInMemoryFlags{} },
|
|
})
|
|
},
|
|
}
|
|
}
|
|
|
|
var _ CLIRunner = (*CLIInProcRunner)(nil)
|