pocketbase/tests/mailer.go

package tests

import (
	"io"
	"net/mail"

	"github.com/pocketbase/pocketbase/tools/mailer"
)

var _ mailer.Mailer = (*TestMailer)(nil)

// TestMailer is a mock `mailer.Mailer` implementation.
type TestMailer struct {
	TotalSend       int
	LastFromAddress mail.Address
	LastToAddress   mail.Address
	LastHtmlSubject string
	LastHtmlBody    string
}

// Reset clears any previously test collected data.
func (m *TestMailer) Reset() {
	m.TotalSend = 0
	m.LastFromAddress = mail.Address{}
	m.LastToAddress = mail.Address{}
	m.LastHtmlSubject = ""
	m.LastHtmlBody = ""
}

// Send implements `mailer.Mailer` interface.
func (m *TestMailer) Send(fromEmail mail.Address, toEmail mail.Address, subject string, html string, attachments map[string]io.Reader) error {
	m.LastFromAddress = fromEmail
	m.LastToAddress = toEmail
	m.LastHtmlSubject = subject
	m.LastHtmlBody = html
	m.TotalSend++
	return nil
}