From ad222ed0acb42d8816c4ec93f1bdc65d87fe6f5c Mon Sep 17 00:00:00 2001 From: renzhiyuan <465386466@qq.com> Date: Mon, 27 Jul 2026 10:11:33 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=96=87=E4=BB=B6:=20xy=5Fsh?= =?UTF-8?q?/internal/test/example=5Ftest.go?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- xy_sh/internal/test/example_test.go | 248 ++++++++++++++++++++++++++++ 1 file changed, 248 insertions(+) create mode 100644 xy_sh/internal/test/example_test.go diff --git a/xy_sh/internal/test/example_test.go b/xy_sh/internal/test/example_test.go new file mode 100644 index 0000000..975ca2f --- /dev/null +++ b/xy_sh/internal/test/example_test.go @@ -0,0 +1,248 @@ +package test + +import ( + "bytes" + "encoding/json" + "io" + "net/http" + "net/http/httptest" + "strconv" + "testing" + "time" + + "xy_sh/internal/entities" + "xy_sh/internal/router" + "xy_sh/pkg/crypto" + + "github.com/gofiber/fiber/v2" +) + +func init() { + crypto.InitCrypto("test_salt_12345", "test_key_16bytes") +} + +func setupTestApp() *fiber.App { + app := fiber.New() + router.SetupRoutes(app) + return app +} + +func makeEncryptedRequest(t *testing.T, app *fiber.App, path string, bizData interface{}) (*http.Response, []byte) { + t.Helper() + + bizJSON, err := json.Marshal(bizData) + if err != nil { + t.Fatalf("业务数据序列化失败: %v", err) + } + + encryptedData, err := crypto.SM4CBCEncrypt(bizJSON) + if err != nil { + t.Fatalf("加密失败: %v", err) + } + + timestamp := strconv.FormatInt(time.Now().UnixMilli(), 10) + + sign := crypto.GenerateSign(timestamp, encryptedData) + + reqBody := entities.EncryptedRequest{ + EncryptedData: encryptedData, + } + reqJSON, _ := json.Marshal(reqBody) + + req := httptest.NewRequest(http.MethodPost, path, bytes.NewReader(reqJSON)) + req.Header.Set("Content-Type", "application/json") + req.Header.Set("timestamp", timestamp) + req.Header.Set("sign", sign) + + resp, err := app.Test(req) + if err != nil { + t.Fatalf("请求失败: %v", err) + } + + body, _ := io.ReadAll(resp.Body) + resp.Body.Close() + + return resp, body +} + +func TestCreateOrder(t *testing.T) { + app := setupTestApp() + + bizData := entities.OrderRequest{ + ActCode: "ACT001", + GoodsCode: "123456", + ActOrderNum: "TEST_ORDER_001", + Account: "19912345678", + CallbackUrl: "https://example.com/callback", + } + + resp, body := makeEncryptedRequest(t, app, "/api/v1/order/create", bizData) + + if resp.StatusCode != http.StatusOK { + t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body)) + } + + var commonResp entities.CommonResponse + if err := json.Unmarshal(body, &commonResp); err != nil { + t.Fatalf("响应解析失败: %v, 响应: %s", err, string(body)) + } + + if commonResp.Code != entities.CodeSuccess { + t.Errorf("期望code=0，实际: %d, msg: %s", commonResp.Code, commonResp.Msg) + } + + t.Logf("下单接口测试成功，响应: %s", string(body)) +} + +func TestQueryOrder(t *testing.T) { + app := setupTestApp() + + createBizData := entities.OrderRequest{ + ActCode: "ACT001", + GoodsCode: "123456", + ActOrderNum: "TEST_ORDER_002", + Account: "19912345678", + } + + _, createBody := makeEncryptedRequest(t, app, "/api/v1/order/create", createBizData) + + var createResp entities.CommonResponse + if err := json.Unmarshal(createBody, &createResp); err != nil { + t.Fatalf("创建订单响应解析失败: %v", err) + } + + if createResp.Code != entities.CodeSuccess { + t.Fatalf("创建订单失败: %s", createResp.Msg) + } + + encryptedData := createResp.Data.(string) + decryptedData, err := crypto.SM4CBCDecrypt(encryptedData) + if err != nil { + t.Fatalf("解密创建订单响应失败: %v", err) + } + + var orderResp entities.OrderResponseData + if err := json.Unmarshal(decryptedData, &orderResp); err != nil { + t.Fatalf("解析创建订单业务数据失败: %v", err) + } + + queryBizData := entities.QueryOrderRequest{ + ActCode: "ACT001", + OrderNo: orderResp.OrderNo, + } + + resp, body := makeEncryptedRequest(t, app, "/api/v1/order/query", queryBizData) + + if resp.StatusCode != http.StatusOK { + t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body)) + } + + var queryResp entities.CommonResponse + if err := json.Unmarshal(body, &queryResp); err != nil { + t.Fatalf("查询响应解析失败: %v", err) + } + + if queryResp.Code != entities.CodeSuccess { + t.Errorf("查询失败，code: %d, msg: %s", queryResp.Code, queryResp.Msg) + } + + t.Logf("查询接口测试成功，响应: %s", string(body)) +} + +func TestWxRecharge(t *testing.T) { + app := setupTestApp() + + bizData := entities.WxRechargeRequest{ + ActCode: "FBG6vdYqGE4mGX7EH/woEg==", + OrderNo: "WX_TEST_ORDER_001", + GoodsCode: "0001", + ActOrderNum: "WX_ORDER_001", + AppId: "wx1234567890", + OpenId: "oABC1234567890", + } + + resp, body := makeEncryptedRequest(t, app, "/api/v1/wx/recharge", bizData) + + if resp.StatusCode != http.StatusOK { + t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body)) + } + + var commonResp entities.CommonResponse + if err := json.Unmarshal(body, &commonResp); err != nil { + t.Fatalf("响应解析失败: %v", err) + } + + if commonResp.Code != entities.CodeSuccess { + t.Errorf("期望code=0，实际: %d, msg: %s", commonResp.Code, commonResp.Msg) + } + + t.Logf("微信立减金充值接口测试成功，响应: %s", string(body)) +} + +func TestCallback(t *testing.T) { + app := setupTestApp() + + bizData := entities.CallbackRequest{ + OrderNo: "HM202509291010001", + ActOrderNum: "XY2025092910100001", + Status: entities.StatusExpired, + Account: "19912345678", + } + + resp, body := makeEncryptedRequest(t, app, "/api/v1/callback/notify", bizData) + + if resp.StatusCode != http.StatusOK { + t.Errorf("期望状态码200，实际: %d, 响应: %s", resp.StatusCode, string(body)) + } + + if string(body) != "ok" { + t.Errorf("期望响应'ok'，实际: %s", string(body)) + } + + t.Logf("回调通知接口测试成功") +} + +func TestInvalidSign(t *testing.T) { + app := setupTestApp() + + bizData := entities.OrderRequest{ + ActCode: "ACT001", + GoodsCode: "123456", + ActOrderNum: "TEST_ORDER_003", + } + + bizJSON, _ := json.Marshal(bizData) + encryptedData, _ := crypto.SM4CBCEncrypt(bizJSON) + timestamp := strconv.FormatInt(time.Now().UnixMilli(), 10) + + wrongSign := "wrong_signature_12345" + + reqBody := entities.EncryptedRequest{EncryptedData: encryptedData} + reqJSON, _ := json.Marshal(reqBody) + + req := httptest.NewRequest(http.MethodPost, "/api/v1/order/create", bytes.NewReader(reqJSON)) + req.Header.Set("Content-Type", "application/json") + req.Header.Set("timestamp", timestamp) + req.Header.Set("sign", wrongSign) + + resp, _ := app.Test(req) + + if resp.StatusCode != http.StatusUnauthorized { + t.Errorf("期望状态码401（签名验证失败），实际: %d", resp.StatusCode) + } + + t.Logf("签名验证失败测试成功") +} + +func TestHealthCheck(t *testing.T) { + app := setupTestApp() + + req := httptest.NewRequest(http.MethodGet, "/health", nil) + resp, body := app.Test(req) + + if resp.StatusCode != http.StatusOK { + t.Errorf("期望状态码200，实际: %d", resp.StatusCode) + } + + t.Logf("健康检查接口测试成功，响应: %s", string(body)) +} \ No newline at end of file