MiFaZhan
78 lines
3.5 KiB
Go
78 lines
3.5 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gogf/gf/v2/encoding/gjson"
|
|
"testing"
|
|
"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/batchmetadata"
|
|
"git.5iot.com/iotlcm-public/iotlcm-app/internal/model"
|
|
"git.5iot.com/iotlcm-public/iotlcm-app/internal/model/entity"
|
|
)
|
|
|
|
func TestAppAllocateParentMatchesWorkerContract(t *testing.T) {
|
|
batch := &appAllocateBatchContext{Target: &entity.SysDept{DeptId: 2}, Cards: []*appAllocateCardContext{{Card: &appCardQueryRow{TSimCardInfo: entity.TSimCardInfo{Iccid: "A", DeptId: 1}}}}}
|
|
info := appAllocateParentInfo(batch, "test")
|
|
if info.Get("jobRole").String() != "allocate_parent" || !info.Get("splitCompleted").Bool() || info.Get("chunkCount").Int() != 1 || info.Get("targetBatchType").Int() != consts.BatchTypeAllocate {
|
|
t.Fatal("worker parent contract mismatch", info)
|
|
}
|
|
}
|
|
|
|
func TestAppAllocateRejectsCanceledCard(t *testing.T) {
|
|
card := &appCardQueryRow{TSimCardInfo: entity.TSimCardInfo{Iccid: "A", BossCardStatus: 9}}
|
|
if err := NewAppCard().validateAllocateCardBusinessRules(context.Background(), nil, card); err == nil {
|
|
t.Fatal("canceled card accepted")
|
|
}
|
|
}
|
|
|
|
func TestAppAllocateQueuedResponseRetainsFinalStatus(t *testing.T) {
|
|
batch := &entity.BatchMain{BatchNo: "ALC1", BatchType: 3, BatchStatus: 3, BatchAsyncStatus: 3, TotalCount: 2}
|
|
if res := appAllocateQueuedResponse(batch); res.Status != "success" || res.BatchNo != batch.BatchNo || res.Quantity != 2 {
|
|
t.Fatal("retry lost final result", res)
|
|
}
|
|
}
|
|
|
|
func TestAppAllocateRetryRejectsChangedSelection(t *testing.T) {
|
|
batch := &appAllocateBatchContext{Target: &entity.SysDept{DeptId: 2}, Cards: []*appAllocateCardContext{{Card: &appCardQueryRow{TSimCardInfo: entity.TSimCardInfo{Iccid: "A"}}}}}
|
|
main := &entity.BatchMain{TargetDeptId: 2, BatchInfo: appAllocateParentInfo(batch, "")}
|
|
for _, param := range []*model.AppCardAllocateBatchSubmitParam{{TargetDeptId: 3}, {CardValues: []string{"B"}}} {
|
|
if appValidateAllocateRetry(main, param) == nil {
|
|
t.Fatal("changed request accepted")
|
|
}
|
|
}
|
|
if err := appValidateAllocateRetry(main, &model.AppCardAllocateBatchSubmitParam{CardValues: []string{"A"}, TargetDeptId: 2}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if appAllocateExecutionMode(0) != batchmetadata.Bulk || appAllocateExecutionMode(2) != batchmetadata.Bulk || appAllocateExecutionMode(1) != batchmetadata.Single {
|
|
t.Fatal("request shape lost")
|
|
}
|
|
}
|
|
|
|
func TestAppAllocateQueuedResponsePreservesFailure(t *testing.T) {
|
|
batch := &entity.BatchMain{BatchNo: "ALC1", BatchAsyncStatus: consts.BatchAsyncStatusFailed, FailReason: "卡已停机，不能划拨", TotalCount: 1, FailCount: 1}
|
|
res := appAllocateQueuedResponse(batch)
|
|
if res.Status != "failed" || res.FailReason != batch.FailReason || res.FailCount != 1 {
|
|
t.Fatal("retry hid failure", res)
|
|
}
|
|
}
|
|
|
|
func TestAppAllocateRequestRejectsChangedHashAndScope(t *testing.T) {
|
|
user := &entity.SysUser{DeptId: 1}
|
|
batch := &entity.BatchMain{BatchType: 3, CreatedDept: 1, BatchInfo: gjson.New(map[string]any{"jobRole": "allocate_parent", "requestHash": "hash"})}
|
|
if err := appValidateAllocateRequest(batch, user, "hash"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if appValidateAllocateRequest(batch, user, "changed") == nil {
|
|
t.Fatal("changed request accepted")
|
|
}
|
|
user.DeptId = 2
|
|
if appValidateAllocateRequest(batch, user, "hash") == nil {
|
|
t.Fatal("changed department accepted")
|
|
}
|
|
user.DeptId = 1
|
|
batch.SourceBatchNo = "parent"
|
|
if appValidateAllocateRequest(batch, user, "hash") == nil {
|
|
t.Fatal("child accepted")
|
|
}
|
|
}
|