diff --git a/src/cli/restore/exchange_test.go b/src/cli/restore/exchange_test.go index 5a8651230..d2d6c5d4d 100644 --- a/src/cli/restore/exchange_test.go +++ b/src/cli/restore/exchange_test.go @@ -11,15 +11,15 @@ import ( "github.com/alcionai/corso/src/internal/tester" ) -type ExchangeSuite struct { +type ExchangeUnitSuite struct { tester.Suite } -func TestExchangeSuite(t *testing.T) { - suite.Run(t, &ExchangeSuite{Suite: tester.NewUnitSuite(t)}) +func TestExchangeUnitSuite(t *testing.T) { + suite.Run(t, &ExchangeUnitSuite{Suite: tester.NewUnitSuite(t)}) } -func (suite *ExchangeSuite) TestAddExchangeCommands() { +func (suite *ExchangeUnitSuite) TestAddExchangeCommands() { expectUse := exchangeServiceCommand + " " + exchangeServiceCommandUseSuffix table := []struct { diff --git a/src/cli/restore/onedrive.go b/src/cli/restore/onedrive.go index 54ef1d409..a5fa84975 100644 --- a/src/cli/restore/onedrive.go +++ b/src/cli/restore/onedrive.go @@ -36,6 +36,9 @@ func addOneDriveCommands(cmd *cobra.Command) *cobra.Command { utils.AddBackupIDFlag(c, true) utils.AddOneDriveDetailsAndRestoreFlags(c) + // restore permissions + options.AddRestorePermissionsFlag(c) + // others options.AddOperationFlags(c) } diff --git a/src/cli/restore/onedrive_test.go b/src/cli/restore/onedrive_test.go index 01c0e4220..05a2a8cc5 100644 --- a/src/cli/restore/onedrive_test.go +++ b/src/cli/restore/onedrive_test.go @@ -8,18 +8,19 @@ import ( "github.com/stretchr/testify/require" "github.com/stretchr/testify/suite" + "github.com/alcionai/corso/src/cli/utils" "github.com/alcionai/corso/src/internal/tester" ) -type OneDriveSuite struct { +type OneDriveUnitSuite struct { tester.Suite } -func TestOneDriveSuite(t *testing.T) { - suite.Run(t, &OneDriveSuite{Suite: tester.NewUnitSuite(t)}) +func TestOneDriveUnitSuite(t *testing.T) { + suite.Run(t, &OneDriveUnitSuite{Suite: tester.NewUnitSuite(t)}) } -func (suite *OneDriveSuite) TestAddOneDriveCommands() { +func (suite *OneDriveUnitSuite) TestAddOneDriveCommands() { expectUse := oneDriveServiceCommand + " " + oneDriveServiceCommandUseSuffix table := []struct { @@ -47,6 +48,9 @@ func (suite *OneDriveSuite) TestAddOneDriveCommands() { assert.Equal(t, test.expectUse, child.Use) assert.Equal(t, test.expectShort, child.Short) tester.AreSameFunc(t, test.expectRunE, child.RunE) + + assert.NotNil(t, c.Flag(utils.BackupFN)) + assert.NotNil(t, c.Flag("restore-permissions")) }) } } diff --git a/src/cli/restore/sharepoint_test.go b/src/cli/restore/sharepoint_test.go index 1dd7e9475..413090231 100644 --- a/src/cli/restore/sharepoint_test.go +++ b/src/cli/restore/sharepoint_test.go @@ -11,15 +11,15 @@ import ( "github.com/alcionai/corso/src/internal/tester" ) -type SharePointSuite struct { +type SharePointUnitSuite struct { tester.Suite } -func TestSharePointSuite(t *testing.T) { - suite.Run(t, &SharePointSuite{Suite: tester.NewUnitSuite(t)}) +func TestSharePointUnitSuite(t *testing.T) { + suite.Run(t, &SharePointUnitSuite{Suite: tester.NewUnitSuite(t)}) } -func (suite *SharePointSuite) TestAddSharePointCommands() { +func (suite *SharePointUnitSuite) TestAddSharePointCommands() { expectUse := sharePointServiceCommand + " " + sharePointServiceCommandUseSuffix table := []struct {