@@ -18,11 +18,15 @@ import (
18
18
19
19
func TestGetTemplateType (t * testing.T ) {
20
20
for expectedType , tmpl := range map [courier.TemplateType ]courier.EmailTemplate {
21
- courier .TypeRecoveryInvalid : & email.RecoveryInvalid {},
22
- courier .TypeRecoveryValid : & email.RecoveryValid {},
23
- courier .TypeVerificationInvalid : & email.VerificationInvalid {},
24
- courier .TypeVerificationValid : & email.VerificationValid {},
25
- courier .TypeTestStub : & email.TestStub {},
21
+ courier .TypeRecoveryInvalid : & email.RecoveryInvalid {},
22
+ courier .TypeRecoveryValid : & email.RecoveryValid {},
23
+ courier .TypeRecoveryCodeInvalid : & email.RecoveryCodeInvalid {},
24
+ courier .TypeRecoveryCodeValid : & email.RecoveryCodeValid {},
25
+ courier .TypeVerificationInvalid : & email.VerificationInvalid {},
26
+ courier .TypeVerificationValid : & email.VerificationValid {},
27
+ courier .TypeVerificationCodeInvalid : & email.VerificationCodeInvalid {},
28
+ courier .TypeVerificationCodeValid : & email.VerificationCodeValid {},
29
+ courier .TypeTestStub : & email.TestStub {},
26
30
} {
27
31
t .Run (fmt .Sprintf ("case=%s" , expectedType ), func (t * testing.T ) {
28
32
actualType , err := courier .GetEmailTemplateType (tmpl )
@@ -37,13 +41,15 @@ func TestNewEmailTemplateFromMessage(t *testing.T) {
37
41
ctx := context .Background ()
38
42
39
43
for tmplType , expectedTmpl := range map [courier.TemplateType ]courier.EmailTemplate {
40
- courier .TypeRecoveryInvalid : email .NewRecoveryInvalid (reg , & email.RecoveryInvalidModel {To : "foo" }),
41
- courier .TypeRecoveryValid : email .NewRecoveryValid (reg , & email.RecoveryValidModel {To : "bar" , RecoveryURL : "http://foo.bar" }),
42
- courier .TypeRecoveryCodeValid : email .NewRecoveryCodeValid (reg , & email.RecoveryCodeValidModel {To : "bar" , RecoveryCode : "12345678" }),
43
- courier .TypeRecoveryCodeInvalid : email .NewRecoveryCodeInvalid (reg , & email.RecoveryCodeInvalidModel {To : "bar" }),
44
- courier .TypeVerificationInvalid : email .NewVerificationInvalid (reg , & email.VerificationInvalidModel {To : "baz" }),
45
- courier .TypeVerificationValid : email .NewVerificationValid (reg , & email.VerificationValidModel {To : "faz" , VerificationURL : "http://bar.foo" }),
46
- courier .TypeTestStub : email .NewTestStub (reg , & email.TestStubModel {To : "far" , Subject : "test subject" , Body : "test body" }),
44
+ courier .TypeRecoveryInvalid : email .NewRecoveryInvalid (reg , & email.RecoveryInvalidModel {To : "foo" }),
45
+ courier .TypeRecoveryValid : email .NewRecoveryValid (reg , & email.RecoveryValidModel {To : "bar" , RecoveryURL : "http://foo.bar" }),
46
+ courier .TypeRecoveryCodeValid : email .NewRecoveryCodeValid (reg , & email.RecoveryCodeValidModel {To : "bar" , RecoveryCode : "12345678" }),
47
+ courier .TypeRecoveryCodeInvalid : email .NewRecoveryCodeInvalid (reg , & email.RecoveryCodeInvalidModel {To : "bar" }),
48
+ courier .TypeVerificationInvalid : email .NewVerificationInvalid (reg , & email.VerificationInvalidModel {To : "baz" }),
49
+ courier .TypeVerificationValid : email .NewVerificationValid (reg , & email.VerificationValidModel {To : "faz" , VerificationURL : "http://bar.foo" }),
50
+ courier .TypeVerificationCodeInvalid : email .NewVerificationCodeInvalid (reg , & email.VerificationCodeInvalidModel {To : "baz" }),
51
+ courier .TypeVerificationCodeValid : email .NewVerificationCodeValid (reg , & email.VerificationCodeValidModel {To : "faz" , VerificationURL : "http://bar.foo" , VerificationCode : "123456678" }),
52
+ courier .TypeTestStub : email .NewTestStub (reg , & email.TestStubModel {To : "far" , Subject : "test subject" , Body : "test body" }),
47
53
} {
48
54
t .Run (fmt .Sprintf ("case=%s" , tmplType ), func (t * testing.T ) {
49
55
tmplData , err := json .Marshal (expectedTmpl )
0 commit comments