1f4a490c33
## Description Adds a flag and options control for disabling metrics. ## Type of change - [x] 🌻 Feature ## Issue(s) * #741 ## Test Plan - [x] 💪 Manual
168 lines
3.8 KiB
Go
168 lines
3.8 KiB
Go
package repo_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/alcionai/corso/src/cli"
|
|
"github.com/alcionai/corso/src/cli/config"
|
|
"github.com/alcionai/corso/src/internal/tester"
|
|
"github.com/alcionai/corso/src/pkg/account"
|
|
"github.com/alcionai/corso/src/pkg/control"
|
|
"github.com/alcionai/corso/src/pkg/repository"
|
|
)
|
|
|
|
type S3IntegrationSuite struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func TestS3IntegrationSuite(t *testing.T) {
|
|
if err := tester.RunOnAny(
|
|
tester.CorsoCITests,
|
|
tester.CorsoCLITests,
|
|
tester.CorsoCLIRepoTests,
|
|
); err != nil {
|
|
t.Skip(err)
|
|
}
|
|
|
|
suite.Run(t, new(S3IntegrationSuite))
|
|
}
|
|
|
|
func (suite *S3IntegrationSuite) SetupSuite() {
|
|
_, err := tester.GetRequiredEnvSls(
|
|
tester.AWSStorageCredEnvs,
|
|
tester.M365AcctCredEnvs)
|
|
require.NoError(suite.T(), err)
|
|
}
|
|
|
|
func (suite *S3IntegrationSuite) TestInitS3Cmd() {
|
|
ctx := tester.NewContext()
|
|
t := suite.T()
|
|
|
|
st := tester.NewPrefixedS3Storage(t)
|
|
cfg, err := st.S3Config()
|
|
require.NoError(t, err)
|
|
|
|
vpr, configFP, err := tester.MakeTempTestConfigClone(t, nil)
|
|
require.NoError(t, err)
|
|
|
|
ctx = config.SetViper(ctx, vpr)
|
|
|
|
cmd := tester.StubRootCmd(
|
|
"repo", "init", "s3",
|
|
"--config-file", configFP,
|
|
"--bucket", cfg.Bucket,
|
|
"--prefix", cfg.Prefix)
|
|
cli.BuildCommandTree(cmd)
|
|
|
|
// run the command
|
|
require.NoError(t, cmd.ExecuteContext(ctx))
|
|
}
|
|
|
|
func (suite *S3IntegrationSuite) TestInitS3Cmd_missingBucket() {
|
|
ctx := tester.NewContext()
|
|
t := suite.T()
|
|
|
|
st := tester.NewPrefixedS3Storage(t)
|
|
cfg, err := st.S3Config()
|
|
require.NoError(t, err)
|
|
|
|
vpr, configFP, err := tester.MakeTempTestConfigClone(t, nil)
|
|
require.NoError(t, err)
|
|
|
|
ctx = config.SetViper(ctx, vpr)
|
|
|
|
cmd := tester.StubRootCmd(
|
|
"repo", "init", "s3",
|
|
"--config-file", configFP,
|
|
"--prefix", cfg.Prefix)
|
|
cli.BuildCommandTree(cmd)
|
|
|
|
// run the command
|
|
require.Error(t, cmd.ExecuteContext(ctx))
|
|
}
|
|
|
|
func (suite *S3IntegrationSuite) TestConnectS3Cmd() {
|
|
ctx := tester.NewContext()
|
|
t := suite.T()
|
|
|
|
st := tester.NewPrefixedS3Storage(t)
|
|
cfg, err := st.S3Config()
|
|
require.NoError(t, err)
|
|
|
|
force := map[string]string{
|
|
tester.TestCfgAccountProvider: "M365",
|
|
tester.TestCfgStorageProvider: "S3",
|
|
tester.TestCfgPrefix: cfg.Prefix,
|
|
}
|
|
vpr, configFP, err := tester.MakeTempTestConfigClone(t, force)
|
|
require.NoError(t, err)
|
|
|
|
ctx = config.SetViper(ctx, vpr)
|
|
|
|
// init the repo first
|
|
_, err = repository.Initialize(ctx, account.Account{}, st, control.Options{})
|
|
require.NoError(t, err)
|
|
|
|
// then test it
|
|
cmd := tester.StubRootCmd(
|
|
"repo", "connect", "s3",
|
|
"--config-file", configFP,
|
|
"--bucket", cfg.Bucket,
|
|
"--prefix", cfg.Prefix)
|
|
cli.BuildCommandTree(cmd)
|
|
|
|
// run the command
|
|
require.NoError(t, cmd.ExecuteContext(ctx))
|
|
}
|
|
|
|
func (suite *S3IntegrationSuite) TestConnectS3Cmd_BadBucket() {
|
|
ctx := tester.NewContext()
|
|
t := suite.T()
|
|
|
|
st := tester.NewPrefixedS3Storage(t)
|
|
cfg, err := st.S3Config()
|
|
require.NoError(t, err)
|
|
|
|
vpr, configFP, err := tester.MakeTempTestConfigClone(t, nil)
|
|
require.NoError(t, err)
|
|
|
|
ctx = config.SetViper(ctx, vpr)
|
|
|
|
cmd := tester.StubRootCmd(
|
|
"repo", "connect", "s3",
|
|
"--config-file", configFP,
|
|
"--bucket", "wrong",
|
|
"--prefix", cfg.Prefix)
|
|
cli.BuildCommandTree(cmd)
|
|
|
|
// run the command
|
|
require.Error(t, cmd.ExecuteContext(ctx))
|
|
}
|
|
|
|
func (suite *S3IntegrationSuite) TestConnectS3Cmd_BadPrefix() {
|
|
ctx := tester.NewContext()
|
|
t := suite.T()
|
|
|
|
st := tester.NewPrefixedS3Storage(t)
|
|
cfg, err := st.S3Config()
|
|
require.NoError(t, err)
|
|
|
|
vpr, configFP, err := tester.MakeTempTestConfigClone(t, nil)
|
|
require.NoError(t, err)
|
|
|
|
ctx = config.SetViper(ctx, vpr)
|
|
|
|
cmd := tester.StubRootCmd(
|
|
"repo", "connect", "s3",
|
|
"--config-file", configFP,
|
|
"--bucket", cfg.Bucket,
|
|
"--prefix", "wrong")
|
|
cli.BuildCommandTree(cmd)
|
|
|
|
// run the command
|
|
require.Error(t, cmd.ExecuteContext(ctx))
|
|
}
|