添加文件: ymt_v3/example_test.go
This commit is contained in:
parent
09d4868e6a
commit
cf1ab094f6
|
|
@ -0,0 +1,324 @@
|
||||||
|
package ymt_v3
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 测试参数（请替换为实际值）
|
||||||
|
const (
|
||||||
|
testAppID = "your_app_id"
|
||||||
|
testPrivateKey = `-----BEGIN PRIVATE KEY-----
|
||||||
|
your_private_key_here
|
||||||
|
-----END PRIVATE KEY-----`
|
||||||
|
testPublicKey = `-----BEGIN PUBLIC KEY-----
|
||||||
|
your_public_key_here
|
||||||
|
-----END PUBLIC KEY-----`
|
||||||
|
testKey = "your_encryption_key_16"
|
||||||
|
testActivityNo = "your_activity_no"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ExampleClient_KeyOrder() {
|
||||||
|
// 创建客户端
|
||||||
|
client, err := NewClient(&ClientConfig{
|
||||||
|
AppID: testAppID,
|
||||||
|
PrivateKey: testPrivateKey,
|
||||||
|
PublicKey: testPublicKey,
|
||||||
|
Key: testKey,
|
||||||
|
EncryptType: EncryptTypeAES,
|
||||||
|
Env: EnvTest,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("创建客户端失败: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取券码
|
||||||
|
resp, err := client.KeyOrder(&KeyOrderRequest{
|
||||||
|
OutBizNo: "order_001",
|
||||||
|
ActivityNo: testActivityNo,
|
||||||
|
Account: "18666666666",
|
||||||
|
NotifyURL: "https://notify.example.com/openapi",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("获取券码失败: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("交易号: %s\n", resp.TradeNo)
|
||||||
|
fmt.Printf("卡密: %s\n", resp.Key)
|
||||||
|
fmt.Printf("状态: %d\n", resp.Status)
|
||||||
|
|
||||||
|
// Output:
|
||||||
|
// 交易号: 7251449503000383488
|
||||||
|
// 卡密: aZKdU9BymzR6qGRzJM
|
||||||
|
// 状态: 1
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExampleClient_KeyQuery() {
|
||||||
|
client, _ := NewClient(&ClientConfig{
|
||||||
|
AppID: testAppID,
|
||||||
|
PrivateKey: testPrivateKey,
|
||||||
|
PublicKey: testPublicKey,
|
||||||
|
Key: testKey,
|
||||||
|
EncryptType: EncryptTypeAES,
|
||||||
|
Env: EnvTest,
|
||||||
|
})
|
||||||
|
|
||||||
|
// 按外部业务号查询
|
||||||
|
resp, err := client.KeyQuery(&KeyQueryRequest{
|
||||||
|
OutBizNo: "order_001",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("查询失败: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("交易号: %s\n", resp.TradeNo)
|
||||||
|
fmt.Printf("状态: %d\n", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExampleClient_KeyDiscard() {
|
||||||
|
client, _ := NewClient(&ClientConfig{
|
||||||
|
AppID: testAppID,
|
||||||
|
PrivateKey: testPrivateKey,
|
||||||
|
PublicKey: testPublicKey,
|
||||||
|
Key: testKey,
|
||||||
|
EncryptType: EncryptTypeAES,
|
||||||
|
Env: EnvTest,
|
||||||
|
})
|
||||||
|
|
||||||
|
// 按交易号作废
|
||||||
|
resp, err := client.KeyDiscard(&KeyDiscardRequest{
|
||||||
|
TradeNo: "7251449503000383488",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("作废失败: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("交易号: %s\n", resp.TradeNo)
|
||||||
|
fmt.Printf("状态: %d\n", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExampleClient_BatchOrder() {
|
||||||
|
client, _ := NewClient(&ClientConfig{
|
||||||
|
AppID: testAppID,
|
||||||
|
PrivateKey: testPrivateKey,
|
||||||
|
PublicKey: testPublicKey,
|
||||||
|
Key: testKey,
|
||||||
|
EncryptType: EncryptTypeAES,
|
||||||
|
Env: EnvTest,
|
||||||
|
})
|
||||||
|
|
||||||
|
// 批量发卡
|
||||||
|
resp, err := client.BatchOrder(&BatchOrderRequest{
|
||||||
|
OutBizNo: "batch_001",
|
||||||
|
ActivityNo: testActivityNo,
|
||||||
|
Number: 100,
|
||||||
|
NotifyURL: "https://notify.example.com/openapi",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("批量发卡失败: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("交易号: %s\n", resp.TradeNo)
|
||||||
|
fmt.Printf("状态: %s\n", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExampleClient_BatchQuery() {
|
||||||
|
client, _ := NewClient(&ClientConfig{
|
||||||
|
AppID: testAppID,
|
||||||
|
PrivateKey: testPrivateKey,
|
||||||
|
PublicKey: testPublicKey,
|
||||||
|
Key: testKey,
|
||||||
|
EncryptType: EncryptTypeAES,
|
||||||
|
Env: EnvTest,
|
||||||
|
})
|
||||||
|
|
||||||
|
// 查询批量任务
|
||||||
|
resp, err := client.BatchQuery(&BatchQueryRequest{
|
||||||
|
TradeNo: "7251449503000383499",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("查询失败: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("状态: %s\n", resp.Status)
|
||||||
|
if resp.DownloadURL != "" {
|
||||||
|
fmt.Printf("下载地址: %s\n", resp.DownloadURL)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarshalWithoutZero(t *testing.T) {
|
||||||
|
req := KeyOrderRequest{
|
||||||
|
OutBizNo: "order_001",
|
||||||
|
ActivityNo: "ACT20260622001",
|
||||||
|
// Account和NotifyURL为空，应该被去掉
|
||||||
|
}
|
||||||
|
|
||||||
|
result, err := marshalWithoutZero(req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("marshalWithoutZero失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
expected := `{"activity_no":"ACT20260622001","out_biz_no":"order_001"}`
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("期望 %s, 得到 %s", expected, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildSignStr(t *testing.T) {
|
||||||
|
appID := "123456"
|
||||||
|
timestamp := "2026-06-22 15:30:00"
|
||||||
|
ciphertext := "encrypted_data"
|
||||||
|
|
||||||
|
signStr := BuildSignStr(appID, timestamp, ciphertext)
|
||||||
|
expected := "1234562026-06-22 15:30:00encrypted_data"
|
||||||
|
if signStr != expected {
|
||||||
|
t.Errorf("期望 %s, 得到 %s", expected, signStr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGenerateTimestamp(t *testing.T) {
|
||||||
|
ts := GenerateTimestamp()
|
||||||
|
// 验证格式 yyyy-MM-dd HH:mm:ss
|
||||||
|
if len(ts) != 19 {
|
||||||
|
t.Errorf("时间戳格式错误: %s", ts)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEncryptDecryptAES(t *testing.T) {
|
||||||
|
key := []byte("1234567890123456") // 16字节
|
||||||
|
plaintext := `{"out_biz_no":"order_001","activity_no":"ACT001"}`
|
||||||
|
|
||||||
|
ciphertext, err := AESECBEncrypt([]byte(plaintext), key)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("加密失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
decrypted, err := AESECBDecrypt(ciphertext, key)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("解密失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if string(decrypted) != plaintext {
|
||||||
|
t.Errorf("解密结果不匹配: 期望 %s, 得到 %s", plaintext, string(decrypted))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEncryptDecryptSM4(t *testing.T) {
|
||||||
|
key := []byte("1234567890123456") // 16字节
|
||||||
|
plaintext := `{"out_biz_no":"order_001","activity_no":"ACT001"}`
|
||||||
|
|
||||||
|
ciphertext, err := SM4CBCEncrypt([]byte(plaintext), key)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SM4加密失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
decrypted, err := SM4CBCDecrypt(ciphertext, key)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SM4解密失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if string(decrypted) != plaintext {
|
||||||
|
t.Errorf("SM4解密结果不匹配: 期望 %s, 得到 %s", plaintext, string(decrypted))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEncryptBizParams(t *testing.T) {
|
||||||
|
key := []byte("1234567890123456")
|
||||||
|
req := KeyOrderRequest{
|
||||||
|
OutBizNo: "order_001",
|
||||||
|
ActivityNo: "ACT20260622001",
|
||||||
|
}
|
||||||
|
|
||||||
|
ciphertext, err := EncryptBizParams(req, key, EncryptTypeAES)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("加密业务参数失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if ciphertext == "" {
|
||||||
|
t.Fatal("加密结果为空")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证可以解密
|
||||||
|
decrypted, err := DecryptBizParams(ciphertext, key, EncryptTypeAES)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("解密失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var result KeyOrderRequest
|
||||||
|
if err := json.Unmarshal(decrypted, &result); err != nil {
|
||||||
|
t.Fatalf("解析失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if result.OutBizNo != "order_001" {
|
||||||
|
t.Errorf("OutBizNo不匹配: 期望 order_001, 得到 %s", result.OutBizNo)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNewClient(t *testing.T) {
|
||||||
|
_, err := NewClient(&ClientConfig{
|
||||||
|
AppID: testAppID,
|
||||||
|
PrivateKey: testPrivateKey,
|
||||||
|
PublicKey: testPublicKey,
|
||||||
|
Key: testKey,
|
||||||
|
EncryptType: EncryptTypeAES,
|
||||||
|
Env: EnvTest,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("创建客户端失败: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNewClientMissingFields(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
config *ClientConfig
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{"空AppID", &ClientConfig{PrivateKey: "pk", PublicKey: "pub", Key: "key"}, true},
|
||||||
|
{"空PrivateKey", &ClientConfig{AppID: "app", PublicKey: "pub", Key: "key"}, true},
|
||||||
|
{"空PublicKey", &ClientConfig{AppID: "app", PrivateKey: "pk", Key: "key"}, true},
|
||||||
|
{"空Key", &ClientConfig{AppID: "app", PrivateKey: "pk", PublicKey: "pub"}, true},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
_, err := NewClient(tt.config)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("NewClient() error = %v, wantErr = %v", err, tt.wantErr)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIError(t *testing.T) {
|
||||||
|
err := NewAPIError(401, "Signature verification failed", "INVALID_SIGNATURE")
|
||||||
|
if err.Code != 401 {
|
||||||
|
t.Errorf("Code不匹配: 期望 401, 得到 %d", err.Code)
|
||||||
|
}
|
||||||
|
if err.Message != "Signature verification failed" {
|
||||||
|
t.Errorf("Message不匹配")
|
||||||
|
}
|
||||||
|
if err.Reason != "INVALID_SIGNATURE" {
|
||||||
|
t.Errorf("Reason不匹配")
|
||||||
|
}
|
||||||
|
if err.Error() != "API error: code=401, message=Signature verification failed, reason=INVALID_SIGNATURE" {
|
||||||
|
t.Errorf("Error()不匹配: %s", err.Error())
|
||||||
|
}
|
||||||
|
if err.IsSuccess() {
|
||||||
|
t.Error("401应该不是成功")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSDKError(t *testing.T) {
|
||||||
|
err := NewSDKError("测试错误", nil)
|
||||||
|
if err.Error() != "SDK error: 测试错误" {
|
||||||
|
t.Errorf("Error()不匹配: %s", err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue