添加文件: ymt_v3_api/example_test.go
This commit is contained in:
parent
58b5061bdf
commit
8c56f95411
|
|
@ -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)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue