24c590040b
implements cli integration for configuring restore operation collision policy and restore destination. flags are hidden and will be made visible at a later time. --- #### Does this PR need a docs update or release note? - [x] 🕐 Yes, but in a later PR #### Type of change - [x] 🌻 Feature #### Issue(s) * #3562 #### Test Plan - [x] ⚡ Unit test - [x] 💚 E2E
141 lines
5.2 KiB
Go
141 lines
5.2 KiB
Go
package restore
|
|
|
|
import (
|
|
"bytes"
|
|
"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/cli/utils"
|
|
"github.com/alcionai/corso/src/cli/utils/testdata"
|
|
"github.com/alcionai/corso/src/internal/tester"
|
|
)
|
|
|
|
type ExchangeUnitSuite struct {
|
|
tester.Suite
|
|
}
|
|
|
|
func TestExchangeUnitSuite(t *testing.T) {
|
|
suite.Run(t, &ExchangeUnitSuite{Suite: tester.NewUnitSuite(t)})
|
|
}
|
|
|
|
func (suite *ExchangeUnitSuite) TestAddExchangeCommands() {
|
|
expectUse := exchangeServiceCommand + " " + exchangeServiceCommandUseSuffix
|
|
|
|
table := []struct {
|
|
name string
|
|
use string
|
|
expectUse string
|
|
expectShort string
|
|
expectRunE func(*cobra.Command, []string) error
|
|
}{
|
|
{"restore exchange", restoreCommand, expectUse, exchangeRestoreCmd().Short, restoreExchangeCmd},
|
|
}
|
|
for _, test := range table {
|
|
suite.Run(test.name, func() {
|
|
t := suite.T()
|
|
|
|
cmd := &cobra.Command{Use: test.use}
|
|
|
|
// normally a persistent flag from the root.
|
|
// required to ensure a dry run.
|
|
flags.AddRunModeFlag(cmd, true)
|
|
|
|
c := addExchangeCommands(cmd)
|
|
require.NotNil(t, c)
|
|
|
|
cmds := cmd.Commands()
|
|
require.Len(t, cmds, 1)
|
|
|
|
child := cmds[0]
|
|
assert.Equal(t, test.expectUse, child.Use)
|
|
assert.Equal(t, test.expectShort, child.Short)
|
|
tester.AreSameFunc(t, test.expectRunE, child.RunE)
|
|
|
|
// Test arg parsing for few args
|
|
cmd.SetArgs([]string{
|
|
"exchange",
|
|
"--" + flags.RunModeFN, flags.RunModeFlagTest,
|
|
"--" + flags.BackupFN, testdata.BackupInput,
|
|
|
|
"--" + flags.ContactFN, testdata.FlgInputs(testdata.ContactInput),
|
|
"--" + flags.ContactFolderFN, testdata.FlgInputs(testdata.ContactFldInput),
|
|
"--" + flags.ContactNameFN, testdata.ContactNameInput,
|
|
|
|
"--" + flags.EmailFN, testdata.FlgInputs(testdata.EmailInput),
|
|
"--" + flags.EmailFolderFN, testdata.FlgInputs(testdata.EmailFldInput),
|
|
"--" + flags.EmailReceivedAfterFN, testdata.EmailReceivedAfterInput,
|
|
"--" + flags.EmailReceivedBeforeFN, testdata.EmailReceivedBeforeInput,
|
|
"--" + flags.EmailSenderFN, testdata.EmailSenderInput,
|
|
"--" + flags.EmailSubjectFN, testdata.EmailSubjectInput,
|
|
|
|
"--" + flags.EventFN, testdata.FlgInputs(testdata.EventInput),
|
|
"--" + flags.EventCalendarFN, testdata.FlgInputs(testdata.EventCalInput),
|
|
"--" + flags.EventOrganizerFN, testdata.EventOrganizerInput,
|
|
"--" + flags.EventRecursFN, testdata.EventRecursInput,
|
|
"--" + flags.EventStartsAfterFN, testdata.EventStartsAfterInput,
|
|
"--" + flags.EventStartsBeforeFN, testdata.EventStartsBeforeInput,
|
|
"--" + flags.EventSubjectFN, testdata.EventSubjectInput,
|
|
|
|
"--" + flags.CollisionsFN, testdata.Collisions,
|
|
"--" + flags.DestinationFN, testdata.Destination,
|
|
|
|
"--" + flags.AWSAccessKeyFN, testdata.AWSAccessKeyID,
|
|
"--" + flags.AWSSecretAccessKeyFN, testdata.AWSSecretAccessKey,
|
|
"--" + flags.AWSSessionTokenFN, testdata.AWSSessionToken,
|
|
|
|
"--" + flags.AzureClientIDFN, testdata.AzureClientID,
|
|
"--" + flags.AzureClientTenantFN, testdata.AzureTenantID,
|
|
"--" + flags.AzureClientSecretFN, testdata.AzureClientSecret,
|
|
|
|
"--" + flags.CorsoPassphraseFN, testdata.CorsoPassphrase,
|
|
})
|
|
|
|
cmd.SetOut(new(bytes.Buffer)) // drop output
|
|
cmd.SetErr(new(bytes.Buffer)) // drop output
|
|
err := cmd.Execute()
|
|
assert.NoError(t, err, clues.ToCore(err))
|
|
|
|
opts := utils.MakeExchangeOpts(cmd)
|
|
assert.Equal(t, testdata.BackupInput, flags.BackupIDFV)
|
|
|
|
assert.ElementsMatch(t, testdata.ContactInput, opts.Contact)
|
|
assert.ElementsMatch(t, testdata.ContactFldInput, opts.ContactFolder)
|
|
assert.Equal(t, testdata.ContactNameInput, opts.ContactName)
|
|
|
|
assert.ElementsMatch(t, testdata.EmailInput, opts.Email)
|
|
assert.ElementsMatch(t, testdata.EmailFldInput, opts.EmailFolder)
|
|
assert.Equal(t, testdata.EmailReceivedAfterInput, opts.EmailReceivedAfter)
|
|
assert.Equal(t, testdata.EmailReceivedBeforeInput, opts.EmailReceivedBefore)
|
|
assert.Equal(t, testdata.EmailSenderInput, opts.EmailSender)
|
|
assert.Equal(t, testdata.EmailSubjectInput, opts.EmailSubject)
|
|
|
|
assert.ElementsMatch(t, testdata.EventInput, opts.Event)
|
|
assert.ElementsMatch(t, testdata.EventCalInput, opts.EventCalendar)
|
|
assert.Equal(t, testdata.EventOrganizerInput, opts.EventOrganizer)
|
|
assert.Equal(t, testdata.EventRecursInput, opts.EventRecurs)
|
|
assert.Equal(t, testdata.EventStartsAfterInput, opts.EventStartsAfter)
|
|
assert.Equal(t, testdata.EventStartsBeforeInput, opts.EventStartsBefore)
|
|
assert.Equal(t, testdata.EventSubjectInput, opts.EventSubject)
|
|
|
|
assert.Equal(t, testdata.Collisions, opts.RestoreCfg.Collisions)
|
|
assert.Equal(t, testdata.Destination, opts.RestoreCfg.Destination)
|
|
|
|
assert.Equal(t, testdata.AWSAccessKeyID, flags.AWSAccessKeyFV)
|
|
assert.Equal(t, testdata.AWSSecretAccessKey, flags.AWSSecretAccessKeyFV)
|
|
assert.Equal(t, testdata.AWSSessionToken, flags.AWSSessionTokenFV)
|
|
|
|
assert.Equal(t, testdata.AzureClientID, flags.AzureClientIDFV)
|
|
assert.Equal(t, testdata.AzureTenantID, flags.AzureClientTenantFV)
|
|
assert.Equal(t, testdata.AzureClientSecret, flags.AzureClientSecretFV)
|
|
|
|
assert.Equal(t, testdata.CorsoPassphrase, flags.CorsoPassphraseFV)
|
|
})
|
|
}
|
|
}
|