Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/payment/provider/easypay_query_test.go

@@ -0,0 +1,131 @@
+package provider
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"testing"
+
+	"github.com/Wei-Shaw/sub2api/internal/payment"
+)
+
+func TestEasyPayQueryOrderStatusMapping(t *testing.T) {
+	t.Parallel()
+
+	const orderID = "order-123"
+	tests := []struct {
+		name        string
+		body        string
+		wantStatus  string
+		wantTradeNo string
+		wantAmount  float64
+	}{
+		{
+			name:        "top level trade success is paid",
+			body:        `{"code":1,"trade_status":"TRADE_SUCCESS","status":0,"money":"12.34","trade_no":"gateway-123"}`,
+			wantStatus:  payment.ProviderStatusPaid,
+			wantTradeNo: "gateway-123",
+			wantAmount:  12.34,
+		},
+		{
+			name:        "waiting trade status with paid numeric status stays pending",
+			body:        `{"code":1,"trade_status":"WAITING","status":1,"money":"12.34","trade_no":"gateway-123"}`,
+			wantStatus:  payment.ProviderStatusPending,
+			wantTradeNo: "gateway-123",
+			wantAmount:  12.34,
+		},
+		{
+			name:        "empty trade status with paid numeric status stays pending",
+			body:        `{"code":1,"trade_status":"","status":1,"money":"12.34"}`,
+			wantStatus:  payment.ProviderStatusPending,
+			wantTradeNo: orderID,
+			wantAmount:  12.34,
+		},
+		{
+			name:        "nested data trade success is paid",
+			body:        `{"code":1,"data":{"trade_status":"TRADE_SUCCESS","status":0,"money":"9.99","trade_no":"data-456"}}`,
+			wantStatus:  payment.ProviderStatusPaid,
+			wantTradeNo: "data-456",
+			wantAmount:  9.99,
+		},
+		{
+			name:        "legacy numeric paid status remains compatible",
+			body:        `{"code":1,"status":1,"money":"3.21"}`,
+			wantStatus:  payment.ProviderStatusPaid,
+			wantTradeNo: orderID,
+			wantAmount:  3.21,
+		},
+		{
+			name:        "legacy numeric non paid status is pending",
+			body:        `{"code":1,"status":0,"money":"3.21"}`,
+			wantStatus:  payment.ProviderStatusPending,
+			wantTradeNo: orderID,
+			wantAmount:  3.21,
+		},
+		{
+			name:        "query failure with missing status is pending",
+			body:        `{"code":0,"msg":"订单不存在"}`,
+			wantStatus:  payment.ProviderStatusPending,
+			wantTradeNo: orderID,
+		},
+		{
+			name:        "missing fields are pending",
+			body:        `{}`,
+			wantStatus:  payment.ProviderStatusPending,
+			wantTradeNo: orderID,
+		},
+	}
+
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			var gotForm url.Values
+			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				if r.Method != http.MethodPost {
+					t.Errorf("method = %q, want %q", r.Method, http.MethodPost)
+				}
+				if r.URL.Path != "/api.php" {
+					t.Errorf("path = %q, want /api.php", r.URL.Path)
+				}
+				if err := r.ParseForm(); err != nil {
+					t.Errorf("ParseForm: %v", err)
+				}
+				gotForm = make(url.Values, len(r.PostForm))
+				for key, values := range r.PostForm {
+					gotForm[key] = append([]string(nil), values...)
+				}
+				w.Header().Set("Content-Type", "application/json")
+				_, _ = w.Write([]byte(tt.body))
+			}))
+			defer server.Close()
+
+			provider := newTestEasyPay(t, server.URL)
+			resp, err := provider.QueryOrder(context.Background(), orderID)
+			if err != nil {
+				t.Fatalf("QueryOrder returned error: %v", err)
+			}
+			if resp.Status != tt.wantStatus {
+				t.Fatalf("status = %q, want %q (response=%+v)", resp.Status, tt.wantStatus, resp)
+			}
+			if resp.TradeNo != tt.wantTradeNo {
+				t.Fatalf("trade_no = %q, want %q", resp.TradeNo, tt.wantTradeNo)
+			}
+			if resp.Amount != tt.wantAmount {
+				t.Fatalf("amount = %v, want %v", resp.Amount, tt.wantAmount)
+			}
+			for key, want := range map[string]string{
+				"act":          "order",
+				"pid":          "pid-1",
+				"key":          "pkey-1",
+				"out_trade_no": orderID,
+			} {
+				if got := gotForm.Get(key); got != want {
+					t.Fatalf("form[%s] = %q, want %q (form=%v)", key, got, want, gotForm)
+				}
+			}
+		})
+	}
+}