115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
package email
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
// MockSender records calls and returns configurable results (for tests and local dev).
|
|
type MockSender struct {
|
|
name string
|
|
sort int
|
|
mu sync.Mutex
|
|
calls []*Message
|
|
|
|
Err error
|
|
MessageID string
|
|
SendHook func(ctx context.Context, msg *Message) (string, error)
|
|
}
|
|
|
|
type MockSenderOption func(*MockSender)
|
|
|
|
func WithMockName(name string) MockSenderOption {
|
|
return func(m *MockSender) { m.name = name }
|
|
}
|
|
|
|
func WithMockSort(sort int) MockSenderOption {
|
|
return func(m *MockSender) { m.sort = sort }
|
|
}
|
|
|
|
func WithMockError(err error) MockSenderOption {
|
|
return func(m *MockSender) { m.Err = err }
|
|
}
|
|
|
|
func WithMockMessageID(id string) MockSenderOption {
|
|
return func(m *MockSender) { m.MessageID = id }
|
|
}
|
|
|
|
func NewMockSender(opts ...MockSenderOption) *MockSender {
|
|
m := &MockSender{
|
|
name: "mock",
|
|
sort: 0,
|
|
MessageID: "mock-email-id",
|
|
}
|
|
for _, opt := range opts {
|
|
opt(m)
|
|
}
|
|
return m
|
|
}
|
|
|
|
func (m *MockSender) Name() string { return m.name }
|
|
func (m *MockSender) Sort() int { return m.sort }
|
|
|
|
func (m *MockSender) Send(ctx context.Context, msg *Message) (string, error) {
|
|
m.mu.Lock()
|
|
m.calls = append(m.calls, msg)
|
|
m.mu.Unlock()
|
|
|
|
if m.SendHook != nil {
|
|
return m.SendHook(ctx, msg)
|
|
}
|
|
if m.Err != nil {
|
|
return "", m.Err
|
|
}
|
|
if msg != nil {
|
|
logx.Infof("[notification mock email] from=%s to=%s subject=%q body=%s message_id=%s",
|
|
msg.From, strings.Join(msg.To, ","), msg.Subject, truncateForLog(msg.Body, 500), m.MessageID)
|
|
}
|
|
return m.MessageID, nil
|
|
}
|
|
|
|
func truncateForLog(s string, max int) string {
|
|
if max <= 0 || len(s) <= max {
|
|
return s
|
|
}
|
|
return s[:max] + "…(truncated)"
|
|
}
|
|
|
|
func (m *MockSender) Calls() []*Message {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
out := make([]*Message, len(m.calls))
|
|
copy(out, m.calls)
|
|
return out
|
|
}
|
|
|
|
func (m *MockSender) Reset() {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.calls = nil
|
|
}
|
|
|
|
// ErrSender is a Sender that always fails (helper for tests).
|
|
func ErrSender(name string, sort int, err error) Sender {
|
|
return &staticErrSender{name: name, sort: sort, err: err}
|
|
}
|
|
|
|
type staticErrSender struct {
|
|
name string
|
|
sort int
|
|
err error
|
|
}
|
|
|
|
func (s *staticErrSender) Name() string { return s.name }
|
|
func (s *staticErrSender) Sort() int { return s.sort }
|
|
func (s *staticErrSender) Send(context.Context, *Message) (string, error) {
|
|
if s.err != nil {
|
|
return "", s.err
|
|
}
|
|
return "", fmt.Errorf("%s: send failed", s.name)
|
|
}
|