diff --git a/auth/handler_test.go b/auth/handler_test.go index c4af628..0a1bfb7 100644 --- a/auth/handler_test.go +++ b/auth/handler_test.go @@ -15,7 +15,6 @@ import ( "github.com/dhax/go-base/email" "github.com/dhax/go-base/logging" - "github.com/dhax/go-base/testing/mock" "github.com/go-chi/chi" "github.com/go-chi/jwtauth" "github.com/spf13/viper" @@ -24,7 +23,7 @@ import ( var ( auth *Resource authstore MockStorer - mailer mock.Mailer + mailer email.MockMailer ts *httptest.Server ) diff --git a/email/mockMailer.go b/email/mockMailer.go new file mode 100644 index 0000000..f70d778 --- /dev/null +++ b/email/mockMailer.go @@ -0,0 +1,13 @@ +package email + +// MockMailer is a mock Mailer +type MockMailer struct { + LoginTokenFn func(name, email string, c ContentLoginToken) error + LoginTokenInvoked bool +} + +// LoginToken is a mock for LoginToken +func (s *MockMailer) LoginToken(n, e string, c ContentLoginToken) error { + s.LoginTokenInvoked = true + return s.LoginTokenFn(n, e, c) +} diff --git a/testing/mock/mailer.go b/testing/mock/mailer.go deleted file mode 100644 index 9f07bc4..0000000 --- a/testing/mock/mailer.go +++ /dev/null @@ -1,13 +0,0 @@ -package mock - -import "github.com/dhax/go-base/email" - -type Mailer struct { - LoginTokenFn func(name, email string, c email.ContentLoginToken) error - LoginTokenInvoked bool -} - -func (s *Mailer) LoginToken(n, e string, c email.ContentLoginToken) error { - s.LoginTokenInvoked = true - return s.LoginTokenFn(n, e, c) -}