feat: 迁移注册充值明细和筛选导出

2026-11-04 00:32:49 +08:00
parent f5b7ce21b9
commit 6703a66f0d
34 changed files with 1964 additions and 9 deletions
@@ -0,0 +1,58 @@
+package business
+
+import (
+	"bytes"
+	"context"
+	"testing"
+
+	"github.com/xuri/excelize/v2"
+	"kra/internal/biz/business"
+	"kra/internal/service/dto"
+)
+
+type ledgerExportFake struct{ LedgerUsecase }
+
+func (ledgerExportFake) ExportRegistrations(context.Context, business.LedgerFilter) ([]*business.Registration, error) {
+	amount := "9007199254740993.10"
+	stamp := int64(1793577600)
+	return []*business.Registration{{ChannelKey: business.ChannelKey{SourceDB: "lx", ID: 7}, RegisterTime: &stamp, FirstGameName: "游戏", TotalPayAmount: &amount, LedgerNames: business.LedgerNames{PromoteAccount: "=1+1"}}}, nil
+}
+func (ledgerExportFake) ExportSpends(context.Context, business.LedgerFilter) ([]*business.Spend, error) {
+	amount := "123.45"
+	return []*business.Spend{{ChannelKey: business.ChannelKey{SourceDB: "lx", ID: 8}, OrderNumber: "000000000000000001", PayAmount: &amount}}, nil
+}
+func TestLedgerWorkbookPreservesNumbersAndText(t *testing.T) {
+	s := &LedgerService{uc: ledgerExportFake{}}
+	data, err := s.ExportRegistrations(context.Background(), &dto.LedgerQuery{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	f, err := excelize.OpenReader(bytes.NewReader(data))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer f.Close()
+	amount, _ := f.GetCellValue("注册明细", "H2", excelize.Options{RawCellValue: true})
+	formula, _ := f.GetCellFormula("注册明细", "K2")
+	text, _ := f.GetCellValue("注册明细", "K2")
+	if amount != "9007199254740993.10" || formula != "" || text != "=1+1" {
+		t.Fatalf("导出丢失金额精度或把账号当成公式: %s %s %s", amount, formula, text)
+	}
+	rows, err := f.GetRows("注册明细")
+	if err != nil || len(rows) != 2 || rows[0][0] != "sourceDb" || rows[0][12] != "所属商务" {
+		t.Fatal("原导出字段与顺序丢失")
+	}
+	data, err = s.ExportSpends(context.Background(), &dto.LedgerQuery{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	pay, err := excelize.OpenReader(bytes.NewReader(data))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer pay.Close()
+	order, _ := pay.GetCellValue("充值明细", "E2")
+	if order != "000000000000000001" {
+		t.Fatal("订单号前导零丢失")
+	}
+}