shanshanzhong
a01570b59d
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 8m21s
29 lines
1.0 KiB
Go
29 lines
1.0 KiB
Go
package emailLogic
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/perfect-panel/server/pkg/constant"
|
|
"github.com/perfect-panel/server/pkg/email"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSelectVerifyTemplate(t *testing.T) {
|
|
sceneTemplates := map[string]string{
|
|
"default": "DEFAULT_TEMPLATE",
|
|
"register": "REGISTER_TEMPLATE",
|
|
"delete_account": "DELETE_TEMPLATE",
|
|
}
|
|
|
|
assert.Equal(t, "REGISTER_TEMPLATE", selectVerifyTemplate(sceneTemplates, "LEGACY_TEMPLATE", "register"))
|
|
assert.Equal(t, "DEFAULT_TEMPLATE", selectVerifyTemplate(sceneTemplates, "LEGACY_TEMPLATE", "security"))
|
|
assert.Equal(t, "LEGACY_TEMPLATE", selectVerifyTemplate(nil, "LEGACY_TEMPLATE", "security"))
|
|
assert.Equal(t, email.DefaultEmailVerifyTemplate, selectVerifyTemplate(nil, "", "security"))
|
|
}
|
|
|
|
func TestResolveVerifyScene(t *testing.T) {
|
|
assert.Equal(t, "register", resolveVerifyScene("register", 0))
|
|
assert.Equal(t, constant.DeleteAccount.String(), resolveVerifyScene("", int(constant.DeleteAccount)))
|
|
assert.Equal(t, "unknown", resolveVerifyScene("", 99))
|
|
}
|