weijuesen
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package model
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestValidConsultationTransition(t *testing.T) {
|
|
ok := [][2]string{
|
|
{"unassigned", "assigned"},
|
|
{"assigned", "accepted"},
|
|
{"assigned", "needs_info"},
|
|
{"accepted", "resolved"},
|
|
{"needs_info", "resolved"},
|
|
{"resolved", "archived"},
|
|
}
|
|
for _, c := range ok {
|
|
if !ValidConsultationTransition(c[0], c[1]) {
|
|
t.Errorf("%s → %s 应合法", c[0], c[1])
|
|
}
|
|
}
|
|
bad := [][2]string{
|
|
{"unassigned", "archived"},
|
|
{"unassigned", "resolved"},
|
|
{"resolved", "resolved"},
|
|
{"archived", "pending"},
|
|
{"", "resolved"},
|
|
}
|
|
for _, c := range bad {
|
|
if ValidConsultationTransition(c[0], c[1]) {
|
|
t.Errorf("%s → %s 不应合法", c[0], c[1])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestConsultationRejectsArchiveBeforeResolution(t *testing.T) {
|
|
if ValidConsultationTransition("accepted", "archived") {
|
|
t.Fatal("accepted 不能直接归档")
|
|
}
|
|
if ValidConsultationTransition("unassigned", "archived") {
|
|
t.Fatal("unassigned 不能直接归档")
|
|
}
|
|
}
|
|
|
|
func TestConsultationSLAMarksOverdue(t *testing.T) {
|
|
now := time.Now()
|
|
deadline := now.Add(-time.Hour)
|
|
task := Consultation{Status: "assigned", SLADeadline: &deadline}
|
|
if state := ConsultationSLAState(task, now); state != "overdue" {
|
|
t.Fatalf("SLA 状态 = %s, want overdue", state)
|
|
}
|
|
future := now.Add(time.Hour)
|
|
task.SLADeadline = &future
|
|
if state := ConsultationSLAState(task, now); state != "on_time" {
|
|
t.Fatalf("SLA 状态 = %s, want on_time", state)
|
|
}
|
|
if state := ConsultationSLAState(Consultation{}, now); state != "unscheduled" {
|
|
t.Fatalf("SLA 状态 = %s, want unscheduled", state)
|
|
}
|
|
}
|
|
|
|
func TestConsultationSnapshotJSON(t *testing.T) {
|
|
now := time.Now()
|
|
s := ConsultationSnapshot{
|
|
RoomName: "蚕房1#",
|
|
LampTest: &LampTest{ID: "lamp-1", Status: "resulted"},
|
|
Inspection: &InspectionRecord{ID: "insp-1", AIStatus: "done"},
|
|
Batch: &Batch{ID: "batch-1", Name: "批次A"},
|
|
WeatherAlerts: []WeatherAlert{{Disease: "白僵病", Level: "orange"}},
|
|
CreatedAt: now,
|
|
}
|
|
raw, err := json.Marshal(s)
|
|
if err != nil {
|
|
t.Fatalf("快照序列化失败: %v", err)
|
|
}
|
|
var m map[string]any
|
|
if err := json.Unmarshal(raw, &m); err != nil {
|
|
t.Fatalf("快照反序列化失败: %v", err)
|
|
}
|
|
for _, key := range []string{"roomName", "lampTest", "inspection", "batch", "weatherAlerts", "createdAt"} {
|
|
if _, ok := m[key]; !ok {
|
|
t.Errorf("快照缺少字段 %s", key)
|
|
}
|
|
}
|
|
}
|