From 8c56f954116d68d2560bf5bb3ee2e6a2e3a09536 Mon Sep 17 00:00:00 2001 From: renzhiyuan <465386466@qq.com> Date: Tue, 21 Jul 2026 14:47:09 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=96=87=E4=BB=B6:=20ymt=5Fv?= =?UTF-8?q?3=5Fapi/example=5Ftest.go?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ymt_v3_api/example_test.go | 239 +++++++++++++++++++++++++++++++++++++ 1 file changed, 239 insertions(+) create mode 100644 ymt_v3_api/example_test.go diff --git a/ymt_v3_api/example_test.go b/ymt_v3_api/example_test.go new file mode 100644 index 0000000..61aaab0 --- /dev/null +++ b/ymt_v3_api/example_test.go @@ -0,0 +1,239 @@ +package ymt_v3_api + +import ( + "context" + "crypto/rand" + "crypto/rsa" + "crypto/x509" + "encoding/json" + "encoding/pem" + "net/http" + "net/http/httptest" + "testing" + "time" +) + +// 生成测试用的 RSA 密钥对和 AES 密钥 +func generateTestKeys() (appID, privateKeyPEM, publicKeyPEM, aesKey string, err error) { + appID = "test_app_id" + priv, err := rsa.GenerateKey(rand.Reader, 2048) + if err != nil { + return + } + privBytes, err := x509.MarshalPKCS8PrivateKey(priv) + if err != nil { + return + } + privateKeyPEM = string(pem.EncodeToMemory(&pem.Block{Type: "PRIVATE KEY", Bytes: privBytes})) + + pubBytes, err := x509.MarshalPKIXPublicKey(&priv.PublicKey) + if err != nil { + return + } + publicKeyPEM = string(pem.EncodeToMemory(&pem.Block{Type: "PUBLIC KEY", Bytes: pubBytes})) + + aesKey = "1234567890123456" // 16 字节 + return +} + +// 启动一个模拟服务器，返回加密的成功响应 +func setupMockServer(t *testing.T, client *Client, bizResp interface{}) *httptest.Server { + t.Helper() + server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + // 验证必要 Header + if r.Header.Get("Appid") == "" || r.Header.Get("Timestamp") == "" || r.Header.Get("Sign") == "" { + w.WriteHeader(http.StatusBadRequest) + json.NewEncoder(w).Encode(apiResponse{Code: 400, Message: "missing header"}) + return + } + // 解密请求体，验证 ciphertext 存在 + var reqBody cipherRequest + if err := json.NewDecoder(r.Body).Decode(&reqBody); err != nil || reqBody.Ciphertext == "" { + w.WriteHeader(http.StatusBadRequest) + json.NewEncoder(w).Encode(apiResponse{Code: 400, Message: "invalid body"}) + return + } + // 构造加密响应 + ciphertext, err := client.encryptPayload(bizResp) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + return + } + resp := apiResponse{ + Code: 200, + Message: "成功", + Data: &cipherData{Ciphertext: ciphertext}, + } + w.Header().Set("Content-Type", "application/json") + json.NewEncoder(w).Encode(resp) + })) + return server +} + +func TestOrderKey(t *testing.T) { + appID, privPEM, pubPEM, aesKey, err := generateTestKeys() + if err != nil { + t.Fatal(err) + } + client, err := NewClient(appID, privPEM, pubPEM, aesKey, WithEncryptType("AES")) + if err != nil { + t.Fatal(err) + } + expectedResp := &OrderKeyResponse{ + OutBizNo: "order_001", + TradeNo: "7251449503000383488", + Key: "aZKdU9BymzR6qGRzJM", + ValidBeginTime: "2026-06-22 15:30:00", + ValidEndTime: "2026-12-31 23:59:59", + UsableNum: 1, + UsageNum: 0, + Status: 1, + SettlementPrice: 9.9, + Account: "18666666666", + } + server := setupMockServer(t, client, expectedResp) + defer server.Close() + client.baseURL = server.URL + + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) + defer cancel() + + req := &OrderKeyRequest{ + OutBizNo: "order_001", + ActivityNo: "ACT20260622001", + Account: "18666666666", + NotifyURL: "https://notify.example.com/openapi", + } + resp, err := client.OrderKey(ctx, req) + if err != nil { + t.Fatalf("OrderKey failed: %v", err) + } + if resp.TradeNo != expectedResp.TradeNo { + t.Errorf("expected trade_no %s, got %s", expectedResp.TradeNo, resp.TradeNo) + } + if resp.Key != expectedResp.Key { + t.Errorf("expected key %s, got %s", expectedResp.Key, resp.Key) + } +} + +func TestQueryKey(t *testing.T) { + appID, privPEM, pubPEM, aesKey, err := generateTestKeys() + if err != nil { + t.Fatal(err) + } + client, err := NewClient(appID, privPEM, pubPEM, aesKey) + if err != nil { + t.Fatal(err) + } + expectedResp := &QueryKeyResponse{ + OutBizNo: "order_001", + TradeNo: "7251449503000383488", + Status: 1, + } + server := setupMockServer(t, client, expectedResp) + defer server.Close() + client.baseURL = server.URL + + ctx := context.Background() + req := &QueryKeyRequest{OutBizNo: "order_001"} + resp, err := client.QueryKey(ctx, req) + if err != nil { + t.Fatalf("QueryKey failed: %v", err) + } + if resp.OutBizNo != expectedResp.OutBizNo { + t.Errorf("expected out_biz_no %s, got %s", expectedResp.OutBizNo, resp.OutBizNo) + } +} + +func TestDiscardKey(t *testing.T) { + appID, privPEM, pubPEM, aesKey, err := generateTestKeys() + if err != nil { + t.Fatal(err) + } + client, err := NewClient(appID, privPEM, pubPEM, aesKey) + if err != nil { + t.Fatal(err) + } + expectedResp := &DiscardKeyResponse{ + OutBizNo: "order_001", + TradeNo: "7251449503000383488", + Status: 3, + } + server := setupMockServer(t, client, expectedResp) + defer server.Close() + client.baseURL = server.URL + + ctx := context.Background() + req := &DiscardKeyRequest{TradeNo: "7251449503000383488"} + resp, err := client.DiscardKey(ctx, req) + if err != nil { + t.Fatalf("DiscardKey failed: %v", err) + } + if resp.Status != 3 { + t.Errorf("expected status 3, got %d", resp.Status) + } +} + +func TestBatchOrderKey(t *testing.T) { + appID, privPEM, pubPEM, aesKey, err := generateTestKeys() + if err != nil { + t.Fatal(err) + } + client, err := NewClient(appID, privPEM, pubPEM, aesKey) + if err != nil { + t.Fatal(err) + } + expectedResp := &BatchOrderResponse{ + OutBizNo: "batch_001", + TradeNo: "7251449503000383499", + Status: "processing", + } + server := setupMockServer(t, client, expectedResp) + defer server.Close() + client.baseURL = server.URL + + ctx := context.Background() + req := &BatchOrderRequest{ + OutBizNo: "batch_001", + ActivityNo: "ACT20260622001", + Number: 100, + } + resp, err := client.BatchOrderKey(ctx, req) + if err != nil { + t.Fatalf("BatchOrderKey failed: %v", err) + } + if resp.Status != "processing" { + t.Errorf("expected status processing, got %s", resp.Status) + } +} + +func TestBatchQueryKey(t *testing.T) { + appID, privPEM, pubPEM, aesKey, err := generateTestKeys() + if err != nil { + t.Fatal(err) + } + client, err := NewClient(appID, privPEM, pubPEM, aesKey) + if err != nil { + t.Fatal(err) + } + expectedResp := &BatchQueryResponse{ + OutBizNo: "batch_001", + TradeNo: "7251449503000383499", + Status: "success", + DownloadURL: "https://oss.example.com/openapi_7251449503000383499.zip", + ZipPassword: "123456", + } + server := setupMockServer(t, client, expectedResp) + defer server.Close() + client.baseURL = server.URL + + ctx := context.Background() + req := &BatchQueryRequest{TradeNo: "7251449503000383499"} + resp, err := client.BatchQueryKey(ctx, req) + if err != nil { + t.Fatalf("BatchQueryKey failed: %v", err) + } + if resp.DownloadURL != expectedResp.DownloadURL { + t.Errorf("expected download_url %s, got %s", expectedResp.DownloadURL, resp.DownloadURL) + } +} \ No newline at end of file