From 80c82619642eccd2509154aa918b1dcecbe76514 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期一, 09 十月 2023 15:05:27 +0800
Subject: [PATCH] 增加task单元测试文件

---
 test/test.go      |    2 +-
 test/task_test.go |   31 +++++++++++++++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletions(-)

diff --git a/test/task_test.go b/test/task_test.go
new file mode 100644
index 0000000..caa2a00
--- /dev/null
+++ b/test/task_test.go
@@ -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())
+	}
+}
diff --git a/test/test.go b/test/test.go
index 62ec57a..c5dd986 100644
--- a/test/test.go
+++ b/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

--
Gitblit v1.8.0