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 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()
			parent := &cobra.Command{Use: restoreCommand}

			cmd := cliTD.SetUpCmdHasFlags(
				t,
				parent,
				addExchangeCommands,
				[]cliTD.UseCobraCommandFn{
					flags.AddAllProviderFlags,
					flags.AddAllStorageFlags,
				},
				flagsTD.WithFlags(
					exchangeServiceCommand,
					[]string{
						"--" + flags.RunModeFN, flags.RunModeFlagTest,
						"--" + flags.BackupFN, flagsTD.BackupInput,
						"--" + flags.ContactFN, flagsTD.FlgInputs(flagsTD.ContactInput),
						"--" + flags.ContactFolderFN, flagsTD.FlgInputs(flagsTD.ContactFldInput),
						"--" + flags.ContactNameFN, flagsTD.ContactNameInput,
						"--" + flags.EmailFN, flagsTD.FlgInputs(flagsTD.EmailInput),
						"--" + flags.EmailFolderFN, flagsTD.FlgInputs(flagsTD.EmailFldInput),
						"--" + flags.EmailReceivedAfterFN, flagsTD.EmailReceivedAfterInput,
						"--" + flags.EmailReceivedBeforeFN, flagsTD.EmailReceivedBeforeInput,
						"--" + flags.EmailSenderFN, flagsTD.EmailSenderInput,
						"--" + flags.EmailSubjectFN, flagsTD.EmailSubjectInput,
						"--" + flags.EventFN, flagsTD.FlgInputs(flagsTD.EventInput),
						"--" + flags.EventCalendarFN, flagsTD.FlgInputs(flagsTD.EventCalInput),
						"--" + flags.EventOrganizerFN, flagsTD.EventOrganizerInput,
						"--" + flags.EventRecursFN, flagsTD.EventRecursInput,
						"--" + flags.EventStartsAfterFN, flagsTD.EventStartsAfterInput,
						"--" + flags.EventStartsBeforeFN, flagsTD.EventStartsBeforeInput,
						"--" + flags.EventSubjectFN, flagsTD.EventSubjectInput,
						"--" + flags.CollisionsFN, flagsTD.Collisions,
						"--" + flags.DestinationFN, flagsTD.Destination,
						"--" + flags.ToResourceFN, flagsTD.ToResource,
					},
					flagsTD.PreparedProviderFlags(),
					flagsTD.PreparedStorageFlags()))

			cliTD.CheckCmdChild(
				t,
				parent,
				3,
				test.expectUse,
				test.expectShort,
				test.expectRunE)

			opts := utils.MakeExchangeOpts(cmd)

			assert.Equal(t, flagsTD.BackupInput, flags.BackupIDFV)
			assert.ElementsMatch(t, flagsTD.ContactInput, opts.Contact)
			assert.ElementsMatch(t, flagsTD.ContactFldInput, opts.ContactFolder)
			assert.Equal(t, flagsTD.ContactNameInput, opts.ContactName)
			assert.ElementsMatch(t, flagsTD.EmailInput, opts.Email)
			assert.ElementsMatch(t, flagsTD.EmailFldInput, opts.EmailFolder)
			assert.Equal(t, flagsTD.EmailReceivedAfterInput, opts.EmailReceivedAfter)
			assert.Equal(t, flagsTD.EmailReceivedBeforeInput, opts.EmailReceivedBefore)
			assert.Equal(t, flagsTD.EmailSenderInput, opts.EmailSender)
			assert.Equal(t, flagsTD.EmailSubjectInput, opts.EmailSubject)
			assert.ElementsMatch(t, flagsTD.EventInput, opts.Event)
			assert.ElementsMatch(t, flagsTD.EventCalInput, opts.EventCalendar)
			assert.Equal(t, flagsTD.EventOrganizerInput, opts.EventOrganizer)
			assert.Equal(t, flagsTD.EventRecursInput, opts.EventRecurs)
			assert.Equal(t, flagsTD.EventStartsAfterInput, opts.EventStartsAfter)
			assert.Equal(t, flagsTD.EventStartsBeforeInput, opts.EventStartsBefore)
			assert.Equal(t, flagsTD.EventSubjectInput, opts.EventSubject)
			assert.Equal(t, flagsTD.Collisions, opts.RestoreCfg.Collisions)
			assert.Equal(t, flagsTD.Destination, opts.RestoreCfg.Destination)
			assert.Equal(t, flagsTD.ToResource, opts.RestoreCfg.ProtectedResource)
			flagsTD.AssertProviderFlags(t, cmd)
			flagsTD.AssertStorageFlags(t, cmd)
		})
	}
}