#### Does this PR need a docs update or release note? - [x] ⛔ No #### Type of change - [x] 🤖 Supportability/Tests #### Test Plan - [x] ⚡ Unit test
74 lines
2.1 KiB
Go
74 lines
2.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/alcionai/clues"
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/alcionai/corso/src/cli/flags"
|
|
"github.com/alcionai/corso/src/internal/tester"
|
|
)
|
|
|
|
type OptionsUnitSuite struct {
|
|
tester.Suite
|
|
}
|
|
|
|
func TestOptionsUnitSuite(t *testing.T) {
|
|
suite.Run(t, &OptionsUnitSuite{Suite: tester.NewUnitSuite(t)})
|
|
}
|
|
|
|
func (suite *OptionsUnitSuite) TestAddExchangeCommands() {
|
|
t := suite.T()
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "test",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
assert.True(t, flags.FailFastFV, flags.FailFastFN)
|
|
assert.True(t, flags.DisableIncrementalsFV, flags.DisableIncrementalsFN)
|
|
assert.True(t, flags.ForceItemDataDownloadFV, flags.ForceItemDataDownloadFN)
|
|
assert.True(t, flags.DisableDeltaFV, flags.DisableDeltaFN)
|
|
assert.True(t, flags.NoStatsFV, flags.NoStatsFN)
|
|
assert.True(t, flags.NoPermissionsFV, flags.NoPermissionsFN)
|
|
assert.True(t, flags.SkipReduceFV, flags.SkipReduceFN)
|
|
assert.Equal(t, 2, flags.FetchParallelismFV, flags.FetchParallelismFN)
|
|
assert.True(t, flags.DisableConcurrencyLimiterFV, flags.DisableConcurrencyLimiterFN)
|
|
assert.Equal(t, 499, flags.DeltaPageSizeFV, flags.DeltaPageSizeFN)
|
|
},
|
|
}
|
|
|
|
// adds no-stats
|
|
flags.AddGlobalOperationFlags(cmd)
|
|
|
|
flags.AddFailFastFlag(cmd)
|
|
flags.AddDisableIncrementalsFlag(cmd)
|
|
flags.AddForceItemDataDownloadFlag(cmd)
|
|
flags.AddDisableDeltaFlag(cmd)
|
|
flags.AddNoPermissionsFlag(cmd)
|
|
flags.AddSkipReduceFlag(cmd)
|
|
flags.AddFetchParallelismFlag(cmd)
|
|
flags.AddDisableConcurrencyLimiterFlag(cmd)
|
|
flags.AddDeltaPageSizeFlag(cmd)
|
|
|
|
// Test arg parsing for few args
|
|
cmd.SetArgs([]string{
|
|
"test",
|
|
"--" + flags.FailFastFN,
|
|
"--" + flags.DisableIncrementalsFN,
|
|
"--" + flags.ForceItemDataDownloadFN,
|
|
"--" + flags.DisableDeltaFN,
|
|
"--" + flags.NoStatsFN,
|
|
"--" + flags.NoPermissionsFN,
|
|
"--" + flags.SkipReduceFN,
|
|
"--" + flags.FetchParallelismFN, "2",
|
|
"--" + flags.DisableConcurrencyLimiterFN,
|
|
"--" + flags.DeltaPageSizeFN, "499",
|
|
})
|
|
|
|
err := cmd.Execute()
|
|
require.NoError(t, err, clues.ToCore(err))
|
|
}
|