This PR reworks the groups restore to and makes the site to restore to mandatory. This also updates some missing filtering capabilities in groups export. --- #### Does this PR need a docs update or release note? - [x] ✅ Yes, it's included - [ ] 🕐 Yes, but in a later PR - [ ] ⛔ No #### Type of change <!--- Please check the type of change your PR introduces: ---> - [x] 🌻 Feature - [ ] 🐛 Bugfix - [ ] 🗺️ Documentation - [ ] 🤖 Supportability/Tests - [ ] 💻 CI/Deployment - [ ] 🧹 Tech Debt/Cleanup #### Issue(s) <!-- Can reference multiple issues. Use one of the following "magic words" - "closes, fixes" to auto-close the Github issue. --> * closes https://github.com/alcionai/corso/issues/4462 #### Test Plan <!-- How will this be tested prior to merging.--> - [ ] 💪 Manual - [x] ⚡ Unit test - [ ] 💚 E2E
102 lines
3.5 KiB
Go
102 lines
3.5 KiB
Go
package restore
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/alcionai/corso/src/cli/flags"
|
|
flagsTD "github.com/alcionai/corso/src/cli/flags/testdata"
|
|
cliTD "github.com/alcionai/corso/src/cli/testdata"
|
|
"github.com/alcionai/corso/src/cli/utils"
|
|
"github.com/alcionai/corso/src/internal/tester"
|
|
)
|
|
|
|
type GroupsUnitSuite struct {
|
|
tester.Suite
|
|
}
|
|
|
|
func TestGroupsUnitSuite(t *testing.T) {
|
|
suite.Run(t, &GroupsUnitSuite{Suite: tester.NewUnitSuite(t)})
|
|
}
|
|
|
|
func (suite *GroupsUnitSuite) TestAddGroupsCommands() {
|
|
expectUse := groupsServiceCommand + " " + groupsServiceCommandUseSuffix
|
|
|
|
table := []struct {
|
|
name string
|
|
use string
|
|
expectUse string
|
|
expectShort string
|
|
expectRunE func(*cobra.Command, []string) error
|
|
}{
|
|
{"restore groups", restoreCommand, expectUse, groupsRestoreCmd().Short, restoreGroupsCmd},
|
|
}
|
|
for _, test := range table {
|
|
suite.Run(test.name, func() {
|
|
t := suite.T()
|
|
parent := &cobra.Command{Use: restoreCommand}
|
|
|
|
cmd := cliTD.SetUpCmdHasFlags(
|
|
t,
|
|
parent,
|
|
addGroupsCommands,
|
|
[]cliTD.UseCobraCommandFn{
|
|
flags.AddAllProviderFlags,
|
|
flags.AddAllStorageFlags,
|
|
},
|
|
flagsTD.WithFlags(
|
|
groupsServiceCommand,
|
|
[]string{
|
|
"--" + flags.RunModeFN, flags.RunModeFlagTest,
|
|
"--" + flags.BackupFN, flagsTD.BackupInput,
|
|
"--" + flags.SiteFN, flagsTD.SiteInput,
|
|
"--" + flags.LibraryFN, flagsTD.LibraryInput,
|
|
"--" + flags.FileFN, flagsTD.FlgInputs(flagsTD.FileNameInput),
|
|
"--" + flags.FolderFN, flagsTD.FlgInputs(flagsTD.FolderPathInput),
|
|
"--" + flags.FileCreatedAfterFN, flagsTD.FileCreatedAfterInput,
|
|
"--" + flags.FileCreatedBeforeFN, flagsTD.FileCreatedBeforeInput,
|
|
"--" + flags.FileModifiedAfterFN, flagsTD.FileModifiedAfterInput,
|
|
"--" + flags.FileModifiedBeforeFN, flagsTD.FileModifiedBeforeInput,
|
|
"--" + flags.ListItemFN, flagsTD.FlgInputs(flagsTD.ListItemInput),
|
|
"--" + flags.ListFolderFN, flagsTD.FlgInputs(flagsTD.ListFolderInput),
|
|
"--" + flags.PageFN, flagsTD.FlgInputs(flagsTD.PageInput),
|
|
"--" + flags.PageFolderFN, flagsTD.FlgInputs(flagsTD.PageFolderInput),
|
|
"--" + flags.CollisionsFN, flagsTD.Collisions,
|
|
"--" + flags.DestinationFN, flagsTD.Destination,
|
|
// "--" + flags.ToResourceFN, flagsTD.ToResource,
|
|
"--" + flags.NoPermissionsFN,
|
|
},
|
|
flagsTD.PreparedProviderFlags(),
|
|
flagsTD.PreparedStorageFlags()))
|
|
|
|
cliTD.CheckCmdChild(
|
|
t,
|
|
parent,
|
|
3,
|
|
test.expectUse,
|
|
test.expectShort,
|
|
test.expectRunE)
|
|
|
|
opts := utils.MakeGroupsOpts(cmd)
|
|
|
|
assert.Equal(t, flagsTD.BackupInput, flags.BackupIDFV)
|
|
assert.Equal(t, flagsTD.LibraryInput, opts.Library)
|
|
assert.ElementsMatch(t, flagsTD.FileNameInput, opts.FileName)
|
|
assert.ElementsMatch(t, flagsTD.FolderPathInput, opts.FolderPath)
|
|
assert.Equal(t, flagsTD.FileCreatedAfterInput, opts.FileCreatedAfter)
|
|
assert.Equal(t, flagsTD.FileCreatedBeforeInput, opts.FileCreatedBefore)
|
|
assert.Equal(t, flagsTD.FileModifiedAfterInput, opts.FileModifiedAfter)
|
|
assert.Equal(t, flagsTD.FileModifiedBeforeInput, opts.FileModifiedBefore)
|
|
assert.Equal(t, flagsTD.Collisions, opts.RestoreCfg.Collisions)
|
|
assert.Equal(t, flagsTD.Destination, opts.RestoreCfg.Destination)
|
|
// assert.Equal(t, flagsTD.ToResource, opts.RestoreCfg.ProtectedResource)
|
|
assert.True(t, flags.NoPermissionsFV)
|
|
flagsTD.AssertProviderFlags(t, cmd)
|
|
flagsTD.AssertStorageFlags(t, cmd)
|
|
})
|
|
}
|
|
}
|