实现分销网后台基础功能与独立前端

2026-09-05 10:52:36 +08:00
parent 85332df2bd
commit 2ffa7a89b1
24 changed files with 6122 additions and 25 deletions
@@ -0,0 +1,55 @@
+package distribution
+
+import "testing"
+
+func TestDefaultRules(t *testing.T) {
+	cases := []struct {
+		mode  string
+		check func(Rule) bool
+	}{
+		{ModeTwoPartyDistributorReceive, func(r Rule) bool { return r.DistributorReceiveRatio == 3.5 && r.PlatformSellRatio == 4 }},
+		{ModeTwoPartyPlatformReceive, func(r Rule) bool { return r.PlatformReceiveRatio == 3.5 && r.DistributorSellRatio == 4 }},
+		{ModeThreeParty, func(r Rule) bool { return r.DistributorARatio == 3.5 && r.DistributorBRatio == 4 }},
+	}
+	for _, item := range cases {
+		rule, err := DefaultRule(item.mode)
+		if err != nil || !item.check(rule) || rule.ServiceFeeValue != 2.5 || rule.PaymentFeeBPS != 50 {
+			t.Fatalf("default rule %s = %#v, err=%v", item.mode, rule, err)
+		}
+	}
+}
+
+func TestCalculateSettlementDeductsFeeBeforeSplit(t *testing.T) {
+	rule := Rule{ServiceFeeValue: 0, ServiceFeeUnit: ServiceFeeUnitFixedAmount, PaymentFeeBPS: 50}
+	got, err := CalculateSettlement(rule, SettlementInput{GrossPaidAmountCent: 10001, RefundAmountCent: 1000, MarkupAmountCent: 300})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if got.NetPaidAmountCent != 9001 || got.PaymentFeeCent != 45 || got.SplitBaseAmountCent != 8956 {
+		t.Fatalf("unexpected settlement: %#v", got)
+	}
+}
+
+func TestCalculateSettlementTaxesEachPositivePartyProfit(t *testing.T) {
+	rule := Rule{ServiceFeeUnit: ServiceFeeUnitFixedAmount, PaymentFeeBPS: 0, TaxEnabled: true, TaxRateBPS: 100}
+	got, err := CalculateSettlement(rule, SettlementInput{
+		GrossPaidAmountCent: 10000,
+		GrossProfitByParty: []PartyProfitInput{
+			{PartyRole: "distributor_a", IncomeAmountCent: 1000, CostAmountCent: 400},
+			{PartyRole: "distributor_b", IncomeAmountCent: 500, CostAmountCent: 700},
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if got.TaxAmountCent != 6 || len(got.TaxByParty) != 2 || got.TaxByParty[0].TaxAmountCent != 6 || got.TaxByParty[1].TaxAmountCent != 0 {
+		t.Fatalf("unexpected tax result: %#v", got)
+	}
+}
+
+func TestCalculateSettlementRejectsRefundOverPayment(t *testing.T) {
+	_, err := CalculateSettlement(Rule{ServiceFeeUnit: ServiceFeeUnitFixedAmount}, SettlementInput{GrossPaidAmountCent: 1, RefundAmountCent: 2})
+	if err != ErrInvalidAmount {
+		t.Fatalf("err = %v", err)
+	}
+}