增加task单元测试文件

zhangqian

2023-10-09 80c82619642eccd2509154aa918b1dcecbe76514

增加task单元测试文件

 test/task_test.go

New file
@@ -0,0 +1,31 @@
package test

import (
    "apsClient/conf"
    "apsClient/constvar"
    "apsClient/model/common"
    "apsClient/nsq"
    "apsClient/pkg/logx"
    "apsClient/service"
    "fmt"
    "testing"
)

func TestTaskStatusUpdate(t *testing.T) {
    Init()
    procedureId := 1
    taskService := service.NewTaskService()
    procedure, _ := taskService.GetProcedureById(procedureId)
    msg := &common.MsgTaskStatusUpdate{
        WorkOrderId:  procedure.WorkOrderID,
        ProcedureID:  procedure.ProceduresInfo.ProcedureID,
        DeviceId:     procedure.ProceduresInfo.DeviceID,
        IsProcessing: false,
        IsFinish:     true,
    }
    caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicTaskProcedureStatusUpdate, conf.Conf.NsqConf.NodeId), "")
    err := caller.Send(msg)
    if err != nil {
        logx.Errorf("send task status update msg error:%v", err.Error())
    }
}

 test/test.go

@@ -10,7 +10,7 @@
func Init() {
    logx.Init(conf.Conf.Log)
    defer logx.Sync()

    conf.Conf.Sqlite.Dsn = "../aps.db"
    if err := model.Init(); err != nil {
        logx.Errorf("model Init err:%v", err)
        return