feat(t222): add freight ingestion admin flow

2026-07-28 23:26:34 +08:00
parent c524e734fb
commit 6d052b0462
37 changed files with 3397 additions and 101 deletions
@@ -0,0 +1,143 @@
+package erpconnector
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+func TestQueryOrderAcceptsAllowlistResponse(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(
+		writer http.ResponseWriter,
+		request *http.Request,
+	) {
+		if request.Header.Get("X-API-Key") != "12345678901234567890123456789012" {
+			t.Fatal("service key was not forwarded")
+		}
+		writer.Header().Set("Content-Type", "application/json")
+		_, _ = writer.Write([]byte(`{
+			"schema_version":1,
+			"query":{"mode":"ORDER_NUMBER"},
+			"orders":[{
+				"external_stock_id":"12",
+				"source_code":"SOURCE-12",
+				"platform_order_no":null,
+				"shop_name":"测试店铺",
+				"source_created_at":"2026-07-28 08:00:00",
+				"order_status":"0",
+				"purchase_status":"1",
+				"is_canceled":false,
+				"items":[{
+					"external_item_id":"88",
+					"title":"商品",
+					"product_spec":"黑色,L",
+					"sku":"BLACK-L",
+					"quantity":2,
+					"product_thumb_ref":"190",
+					"purchase_status":"0"
+				}]
+			}]
+		}`))
+	}))
+	defer server.Close()
+	client, err := New(
+		server.URL,
+		"12345678901234567890123456789012",
+		time.Second,
+	)
+	if err != nil {
+		t.Fatalf("New() error = %v", err)
+	}
+	result, err := client.QueryOrder(context.Background(), "SOURCE-12")
+	if err != nil {
+		t.Fatalf("QueryOrder() error = %v", err)
+	}
+	if len(result.Orders) != 1 || len(result.Orders[0].Items) != 1 ||
+		result.Orders[0].Items[0].SKU != "BLACK-L" {
+		t.Fatalf("result = %+v", result)
+	}
+}
+
+func TestQueryOrderRejectsUnexpectedPIIField(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(
+		writer http.ResponseWriter,
+		_ *http.Request,
+	) {
+		writer.Header().Set("Content-Type", "application/json")
+		_, _ = writer.Write([]byte(`{
+			"schema_version":1,
+			"query":{"mode":"ORDER_NUMBER"},
+			"orders":[],
+			"receiverTel":"private"
+		}`))
+	}))
+	defer server.Close()
+	client, _ := New(server.URL, "12345678901234567890123456789012", time.Second)
+	_, err := client.QueryOrder(context.Background(), "SOURCE-12")
+	if !errors.Is(err, ErrProtocol) {
+		t.Fatalf("QueryOrder() error = %v, want protocol error", err)
+	}
+}
+
+func TestQueryOrderMapsStableErrorsWithoutReadingDetails(t *testing.T) {
+	tests := []struct {
+		status int
+		want   error
+	}{
+		{http.StatusUnauthorized, ErrSessionRequired},
+		{http.StatusNotFound, ErrNotFound},
+		{http.StatusBadGateway, ErrUnavailable},
+	}
+	for _, test := range tests {
+		t.Run(http.StatusText(test.status), func(t *testing.T) {
+			server := httptest.NewServer(http.HandlerFunc(func(
+				writer http.ResponseWriter,
+				_ *http.Request,
+			) {
+				writer.WriteHeader(test.status)
+				_, _ = writer.Write([]byte(`{"detail":"private upstream text"}`))
+			}))
+			defer server.Close()
+			client, _ := New(
+				server.URL,
+				"12345678901234567890123456789012",
+				time.Second,
+			)
+			_, err := client.QueryOrder(context.Background(), "SOURCE-12")
+			if !errors.Is(err, test.want) {
+				t.Fatalf("QueryOrder() error = %v, want %v", err, test.want)
+			}
+		})
+	}
+}
+
+func TestQueryOrderDoesNotFollowRedirects(t *testing.T) {
+	followed := false
+	target := httptest.NewServer(http.HandlerFunc(func(
+		http.ResponseWriter,
+		*http.Request,
+	) {
+		followed = true
+	}))
+	defer target.Close()
+	redirect := httptest.NewServer(http.HandlerFunc(func(
+		writer http.ResponseWriter,
+		_ *http.Request,
+	) {
+		writer.Header().Set("Location", target.URL)
+		writer.WriteHeader(http.StatusTemporaryRedirect)
+	}))
+	defer redirect.Close()
+	client, _ := New(
+		redirect.URL,
+		"12345678901234567890123456789012",
+		time.Second,
+	)
+	_, err := client.QueryOrder(context.Background(), "SOURCE-12")
+	if !errors.Is(err, ErrUnavailable) || followed {
+		t.Fatalf("redirect error/followed = %v / %t", err, followed)
+	}
+}