openbilibili/app/job/main/up/service/service_test.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()
	}
}