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))
}