mirror of
https://github.com/kopia/kopia.git
synced 2026-03-05 23:08:11 -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
51 lines
1.8 KiB
Go
51 lines
1.8 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/alecthomas/kingpin"
|
|
|
|
"github.com/kopia/kopia/repo/blob"
|
|
"github.com/kopia/kopia/repo/blob/throttling"
|
|
)
|
|
|
|
// StorageProviderServices is implemented by the cli App that allows the cli
|
|
// and tests to mutate the default storage providers.
|
|
type StorageProviderServices interface {
|
|
EnvName(s string) string
|
|
setPasswordFromToken(pwd string)
|
|
storageProviders() []StorageProvider
|
|
}
|
|
|
|
// StorageFlags is implemented by cli storage providers which need to support a
|
|
// particular backend. This requires the common setup and connection methods
|
|
// implemented by all the cli storage providers.
|
|
type StorageFlags interface {
|
|
Setup(sps StorageProviderServices, cmd *kingpin.CmdClause)
|
|
Connect(ctx context.Context, isCreate bool, formatVersion int) (blob.Storage, error)
|
|
}
|
|
|
|
// StorageProvider is a CLI provider for storage options and allows the CLI to
|
|
// multiplex between various provider CLI flag constructors.
|
|
type StorageProvider struct {
|
|
Name string
|
|
Description string
|
|
NewFlags func() StorageFlags
|
|
}
|
|
|
|
func commonThrottlingFlags(cmd *kingpin.CmdClause, limits *throttling.Limits) {
|
|
cmd.Flag("max-download-speed", "Limit the download speed.").PlaceHolder("BYTES_PER_SEC").FloatVar(&limits.DownloadBytesPerSecond)
|
|
cmd.Flag("max-upload-speed", "Limit the upload speed.").PlaceHolder("BYTES_PER_SEC").FloatVar(&limits.UploadBytesPerSecond)
|
|
}
|
|
|
|
// AddStorageProvider adds a new StorageProvider at runtime after the App has
|
|
// been initialized with the default providers. This is used in tests which
|
|
// require custom storage providers to simulate various edge cases.
|
|
func (c *App) AddStorageProvider(p StorageProvider) {
|
|
c.cliStorageProviders = append(c.cliStorageProviders, p)
|
|
}
|
|
|
|
func (c *App) storageProviders() []StorageProvider {
|
|
return c.cliStorageProviders
|
|
}
|