115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package service
|
|
|
|
import (
|
|
"flag"
|
|
"go-common/app/job/main/up/conf"
|
|
"go-common/app/job/main/up/dao/upcrm"
|
|
"go-common/app/job/main/up/model/signmodel"
|
|
"html/template"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
flag.Set("conf", "../cmd/up-job.toml")
|
|
flag.Parse()
|
|
if err := conf.Init(); err != nil {
|
|
panic(err)
|
|
}
|
|
m.Run()
|
|
os.Exit(0)
|
|
}
|
|
|
|
func TestTemplateSign(t *testing.T) {
|
|
var data = &dueData{
|
|
Signs: []*upcrm.SignWithName{
|
|
{Name: "test", SignUp: signmodel.SignUp{Mid: 123, EndDate: 1540901779}},
|
|
{Name: "tes2t", SignUp: signmodel.SignUp{Mid: 1234, EndDate: 1540902000}},
|
|
},
|
|
}
|
|
tmpl, err := template.New("signTitle").Funcs(funcHelper).Parse(conf.Conf.MailTemplateConf.SignTmplTitle)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
err = tmpl.Execute(os.Stdout, data.Signs)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
tmpl, err = template.New("sign").Funcs(funcHelper).Parse(conf.Conf.MailTemplateConf.SignTmplContent)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
err = tmpl.Execute(os.Stdout, data.Signs)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
}
|
|
|
|
func TestTemplatePay(t *testing.T) {
|
|
var data = &dueData{
|
|
Pays: []*upcrm.PayWithAdmin{
|
|
{Name: "test", SignPay: signmodel.SignPay{Mid: 123, DueDate: 1540901779, PayValue: 10000}},
|
|
{Name: "test", SignPay: signmodel.SignPay{Mid: 123, DueDate: 1540901779, PayValue: 10000}},
|
|
{Name: "test", SignPay: signmodel.SignPay{Mid: 123, DueDate: 1540901779, PayValue: 10000}},
|
|
},
|
|
}
|
|
tmpl, err := template.New("payTitle").Funcs(funcHelper).Parse(conf.Conf.MailTemplateConf.PayTmplTitle)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
err = tmpl.Execute(os.Stdout, data.Pays)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
tmpl, err = template.New("pay").Funcs(funcHelper).Parse(conf.Conf.MailTemplateConf.PayTmplContent)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
err = tmpl.Execute(os.Stdout, data.Pays)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
}
|
|
|
|
func TestTemplateTask(t *testing.T) {
|
|
var data = &dueData{
|
|
Tasks: []*upcrm.TaskWithAdmin{
|
|
{Name: "test", SignTaskHistory: signmodel.SignTaskHistory{Mid: 123, GenerateDate: 1540901779, TaskType: 2, TaskCounter: 1, TaskCondition: 10}},
|
|
{Name: "test", SignTaskHistory: signmodel.SignTaskHistory{Mid: 123, GenerateDate: 1540901779, TaskType: 3, TaskCounter: 1, TaskCondition: 10}},
|
|
{Name: "test", SignTaskHistory: signmodel.SignTaskHistory{Mid: 123, GenerateDate: 1540901779, TaskType: 0, TaskCounter: 1, TaskCondition: 10}},
|
|
{Name: "test", SignTaskHistory: signmodel.SignTaskHistory{Mid: 123, GenerateDate: 1540901779, TaskType: 1, TaskCounter: 1, TaskCondition: 10}},
|
|
}}
|
|
|
|
tmpl, err := template.New("task").Funcs(funcHelper).Parse(conf.Conf.MailTemplateConf.TaskTmplTitle)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
err = tmpl.Execute(os.Stdout, data.Tasks)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
tmpl, err = template.New("task").Funcs(funcHelper).Parse(conf.Conf.MailTemplateConf.TaskTmplContent)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
err = tmpl.Execute(os.Stdout, data.Tasks)
|
|
if err != nil {
|
|
t.Errorf("err=%v", err)
|
|
t.FailNow()
|
|
}
|
|
}
|