feat(expiry): 完成 #21 物品与设置模型实现

2026-03-04 17:07:30 +08:00
parent f092019c61
commit 83fbbe7189
2 changed files with 162 additions and 10 deletions
@@ -0,0 +1,80 @@
+package expiry
+
+import (
+	"testing"
+	"time"
+)
+
+func TestCalculateDaysLeft(t *testing.T) {
+	base := dateOnly(time.Now())
+	item := &ExpiryItem{
+		ExpiryDate: base.AddDate(0, 0, 5),
+	}
+
+	if daysLeft := item.CalculateDaysLeft(); daysLeft != 5 {
+		t.Fatalf("expected daysLeft=5, got %d", daysLeft)
+	}
+}
+
+func TestCalculateStatus(t *testing.T) {
+	base := dateOnly(time.Now())
+	tests := []struct {
+		name       string
+		item       ExpiryItem
+		wantStatus string
+	}{
+		{
+			name: "已过期",
+			item: ExpiryItem{
+				ExpiryDate: base.AddDate(0, 0, -1),
+			},
+			wantStatus: StatusExpired,
+		},
+		{
+			name: "今天过期",
+			item: ExpiryItem{
+				ExpiryDate: base,
+			},
+			wantStatus: StatusExpiring,
+		},
+		{
+			name: "7天内即将过期",
+			item: ExpiryItem{
+				ExpiryDate: base.AddDate(0, 0, 7),
+			},
+			wantStatus: StatusExpiring,
+		},
+		{
+			name: "正常状态",
+			item: ExpiryItem{
+				ExpiryDate: base.AddDate(0, 0, 8),
+			},
+			wantStatus: StatusNormal,
+		},
+		{
+			name: "保留已使用状态",
+			item: ExpiryItem{
+				ExpiryDate: base.AddDate(0, 0, -10),
+				Status:     StatusUsed,
+			},
+			wantStatus: StatusUsed,
+		},
+		{
+			name: "保留已丢弃状态",
+			item: ExpiryItem{
+				ExpiryDate: base.AddDate(0, 0, -10),
+				Status:     StatusDiscarded,
+			},
+			wantStatus: StatusDiscarded,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := tt.item.CalculateStatus()
+			if got != tt.wantStatus {
+				t.Fatalf("expected status=%s, got %s", tt.wantStatus, got)
+			}
+		})
+	}
+}