144 lines
3.7 KiB
Go
144 lines
3.7 KiB
Go
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)
|
|
}
|
|
}
|