支持提号完成后财务调整

2026-08-16 19:52:12 +08:00
parent 786d20e9cd
commit 80e55ba281
15 changed files with 1272 additions and 111 deletions
@@ -183,3 +183,60 @@ func TestDailyPickupCancelledByCancelledAt(t *testing.T) {
 		t.Fatalf("8月2日取消提号数 = %d, want 2", result[0].Count)
 	}
 }
+
+func TestPickupSummaryKeepsOriginalAndAdjustmentOnTheirOwnDates(t *testing.T) {
+	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
+	if err != nil {
+		t.Fatalf("打开测试数据库失败: %v", err)
+	}
+	if err := db.Exec(`CREATE TABLE admin_pickups (
+		id INTEGER PRIMARY KEY,
+		status TEXT NOT NULL,
+		settle_amount_cent INTEGER NOT NULL,
+		profit_amount_cent INTEGER NOT NULL,
+		completed_at DATETIME NULL
+	)`).Error; err != nil {
+		t.Fatalf("创建提号表失败: %v", err)
+	}
+	if err := db.Exec(`CREATE TABLE admin_pickup_financial_adjustments (
+		id INTEGER PRIMARY KEY,
+		pickup_id INTEGER NOT NULL,
+		settle_delta_cent INTEGER NOT NULL,
+		profit_delta_cent INTEGER NOT NULL,
+		created_at DATETIME NOT NULL
+	)`).Error; err != nil {
+		t.Fatalf("创建提号调整表失败: %v", err)
+	}
+	loc := timeutil.ShanghaiLocation()
+	completedAt := time.Date(2026, 8, 2, 10, 0, 0, 0, loc)
+	adjustedAt := time.Date(2026, 8, 4, 10, 0, 0, 0, loc)
+	if err := db.Exec(`INSERT INTO admin_pickups (id, status, settle_amount_cent, profit_amount_cent, completed_at)
+		VALUES (1, 'completed', 12000, 1000, ?)`, completedAt).Error; err != nil {
+		t.Fatalf("写入提号失败: %v", err)
+	}
+	if err := db.Exec(`INSERT INTO admin_pickup_financial_adjustments (id, pickup_id, settle_delta_cent, profit_delta_cent, created_at)
+		VALUES (1, 1, 1500, 500, ?)`, adjustedAt).Error; err != nil {
+		t.Fatalf("写入调整失败: %v", err)
+	}
+	repo := NewRepository(db)
+	dayQuery := func(day int) DashboardQuery {
+		return DashboardQuery{
+			StartDate: time.Date(2026, 8, day, 0, 0, 0, 0, loc),
+			EndDate:   time.Date(2026, 8, day, 23, 59, 59, 0, loc),
+		}
+	}
+	original, err := repo.pickupSummary(t.Context(), dayQuery(2))
+	if err != nil {
+		t.Fatalf("查询完成日提号汇总失败: %v", err)
+	}
+	if original.SettledAmountCent != 12000 || original.ProfitAmountCent != 1000 || original.CompletedCount != 1 {
+		t.Fatalf("完成日汇总 = %+v", original)
+	}
+	adjusted, err := repo.pickupSummary(t.Context(), dayQuery(4))
+	if err != nil {
+		t.Fatalf("查询调整日提号汇总失败: %v", err)
+	}
+	if adjusted.SettledAmountCent != 1500 || adjusted.ProfitAmountCent != 500 || adjusted.CompletedCount != 0 {
+		t.Fatalf("调整日汇总 = %+v", adjusted)
+	}
+}