package sendemailcode

import (
	"context"
	"errors"
	"testing"
	"time"

	stubmail "galaxy/authsession/internal/adapters/mail"
	"galaxy/authsession/internal/domain/challenge"
	"galaxy/authsession/internal/domain/common"
	"galaxy/authsession/internal/service/shared"
	"galaxy/authsession/internal/testkit"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestExecuteWithStubSender(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name                string
		sender              *stubmail.StubSender
		wantStatus          challenge.Status
		wantDeliveryState   challenge.DeliveryState
		wantErrorCode       string
		wantRecordedAttempt int
	}{
		{
			name:                "sent",
			sender:              &stubmail.StubSender{},
			wantStatus:          challenge.StatusSent,
			wantDeliveryState:   challenge.DeliverySent,
			wantRecordedAttempt: 1,
		},
		{
			name: "suppressed",
			sender: &stubmail.StubSender{
				DefaultMode: stubmail.StubModeSuppressed,
			},
			wantStatus:          challenge.StatusDeliverySuppressed,
			wantDeliveryState:   challenge.DeliverySuppressed,
			wantRecordedAttempt: 1,
		},
		{
			name: "failed",
			sender: &stubmail.StubSender{
				DefaultMode:  stubmail.StubModeFailed,
				DefaultError: errors.New("stub delivery failed"),
			},
			wantStatus:          challenge.StatusFailed,
			wantDeliveryState:   challenge.DeliveryFailed,
			wantErrorCode:       shared.ErrorCodeServiceUnavailable,
			wantRecordedAttempt: 1,
		},
	}

	for _, tt := range tests {
		tt := tt

		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			challengeStore := &testkit.InMemoryChallengeStore{}
			service, err := New(
				challengeStore,
				&testkit.InMemoryUserDirectory{},
				&testkit.SequenceIDGenerator{ChallengeIDs: []common.ChallengeID{"challenge-1"}},
				testkit.FixedCodeGenerator{Code: "654321"},
				testkit.DeterministicCodeHasher{},
				tt.sender,
				testkit.FixedClock{Time: time.Unix(10, 0).UTC()},
			)
			require.NoError(t, err)

			result, err := service.Execute(context.Background(), Input{Email: "pilot@example.com"})
			if tt.wantErrorCode == "" {
				require.NoError(t, err)
				assert.Equal(t, "challenge-1", result.ChallengeID)
			} else {
				require.Error(t, err)
				assert.Equal(t, tt.wantErrorCode, shared.CodeOf(err))
				assert.Equal(t, Result{}, result)
			}

			record, getErr := challengeStore.Get(context.Background(), common.ChallengeID("challenge-1"))
			require.NoError(t, getErr)
			assert.Equal(t, tt.wantStatus, record.Status)
			assert.Equal(t, tt.wantDeliveryState, record.DeliveryState)

			attempts := tt.sender.RecordedAttempts()
			require.Len(t, attempts, tt.wantRecordedAttempt)
			assert.Equal(t, common.Email("pilot@example.com"), attempts[0].Input.Email)
			assert.Equal(t, "654321", attempts[0].Input.Code)
			assert.Equal(t, "en", attempts[0].Input.Locale)
		})
	}
}