Ilia Denisov
99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
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)
|
|
})
|
|
}
|
|
}
|