diff --git a/pkg/email/email_test.go b/pkg/email/email_test.go index 34cdf88..f70ed58 100755 --- a/pkg/email/email_test.go +++ b/pkg/email/email_test.go @@ -43,7 +43,7 @@ func TestNewConfig_MissingEnvFile(t *testing.T) { } func TestMockSendEmail(t *testing.T) { - service := NewMockMailService() + service := NewMockMailService(func(params ...interface{}) {}) emailData := EmailMessage{ To: "test@example.com", diff --git a/pkg/email/mock.go b/pkg/email/mock.go index 3920661..c7944bb 100644 --- a/pkg/email/mock.go +++ b/pkg/email/mock.go @@ -6,7 +6,9 @@ import ( "net/smtp" ) -func NewMockMailService() *EmailService { +type MockCallbackFn func(params ...interface{}) + +func NewMockMailService(callbackFn MockCallbackFn, params ...interface{}) *EmailService { return &EmailService{ auth: smtp.PlainAuth("", "", "", ""), host: "", @@ -17,6 +19,7 @@ func NewMockMailService() *EmailService { ServerName: "", }, dial: func(hostPort string) (SMTPClientIface, error) { + callbackFn(params) return &mockSMTP{}, nil }, }