实现资产双向换货链路查询
This commit is contained in:
@@ -14,7 +14,7 @@ func registerAssetRoutes(router fiber.Router, handler *admin.AssetHandler, walle
|
||||
|
||||
Register(assets, doc, groupPath, "GET", "/resolve/:identifier", handler.Resolve, RouteSpec{
|
||||
Summary: "解析资产",
|
||||
Description: "通过虚拟号/ICCID/IMEI/SN/MSISDN 解析设备或卡的完整详情。企业账号禁止调用。",
|
||||
Description: "通过虚拟号/ICCID/IMEI/SN/MSISDN 解析设备或卡的完整详情。exchange_trace 始终存在，previous_asset/next_asset 无关系时为 null；关联项仅在 can_view=true 且 asset_id 非空时允许跳转。企业账号禁止调用。",
|
||||
Tags: []string{"资产管理"},
|
||||
Input: new(dto.AssetResolveRequest),
|
||||
Output: new(dto.AssetResolveResponse),
|
||||
|
||||
268
internal/routes/asset_exchange_trace_integration_test.go
Normal file
268
internal/routes/asset_exchange_trace_integration_test.go
Normal file
@@ -0,0 +1,268 @@
|
||||
package routes
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
"strconv"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/bytedance/sonic"
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/redis/go-redis/v9"
|
||||
"go.uber.org/zap"
|
||||
"gorm.io/gorm"
|
||||
|
||||
"github.com/break/junhong_cmp_fiber/internal/bootstrap"
|
||||
"github.com/break/junhong_cmp_fiber/internal/handler/admin"
|
||||
internalMiddleware "github.com/break/junhong_cmp_fiber/internal/middleware"
|
||||
"github.com/break/junhong_cmp_fiber/internal/model"
|
||||
assetQuery "github.com/break/junhong_cmp_fiber/internal/query/asset"
|
||||
assetService "github.com/break/junhong_cmp_fiber/internal/service/asset"
|
||||
"github.com/break/junhong_cmp_fiber/internal/store/postgres"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/auth"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/config"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/database"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
||||
pkgMiddleware "github.com/break/junhong_cmp_fiber/pkg/middleware"
|
||||
)
|
||||
|
||||
// TestAssetResolveHTTPIntegratesAuthenticationPermissionsAndExchangeTrace 验证真实认证、权限、资产解析和双向换货投影。
|
||||
func TestAssetResolveHTTPIntegratesAuthenticationPermissionsAndExchangeTrace(t *testing.T) {
|
||||
env := newAssetTraceHTTPEnv(t)
|
||||
visibleShop := env.createShop(t, "UR86 可见店铺", nil, 1)
|
||||
hiddenShop := env.createShop(t, "UR86 不可见店铺", nil, 1)
|
||||
oldCard := env.createCard(t, 1, &hiddenShop.ID)
|
||||
middleCard := env.createCard(t, 2, &visibleShop.ID)
|
||||
nextCard := env.createCard(t, 3, &visibleShop.ID)
|
||||
env.createOrder(t, "UR86-HTTP-PREV", oldCard, middleCard, "HTTP 历史前代快照", "HTTP 中间快照", time.Now().Add(-time.Minute))
|
||||
env.createOrder(t, "UR86-HTTP-NEXT", middleCard, nextCard, "HTTP 中间快照", "HTTP 历史后代快照", time.Now())
|
||||
token := env.newAgentToken(t, visibleShop.ID)
|
||||
|
||||
status, body := env.request(t, "/api/admin/assets/resolve/"+middleCard.ICCID, token)
|
||||
if status != http.StatusOK {
|
||||
t.Fatalf("查询中间资产失败，状态 %d：%s", status, body)
|
||||
}
|
||||
var success struct {
|
||||
Code int `json:"code"`
|
||||
Data struct {
|
||||
AssetID uint `json:"asset_id"`
|
||||
ExchangeTrace struct {
|
||||
PreviousAsset *struct {
|
||||
AssetID *uint `json:"asset_id"`
|
||||
Identifier string `json:"identifier"`
|
||||
ExchangeNo string `json:"exchange_no"`
|
||||
CanView bool `json:"can_view"`
|
||||
} `json:"previous_asset"`
|
||||
NextAsset *struct {
|
||||
AssetID *uint `json:"asset_id"`
|
||||
Identifier string `json:"identifier"`
|
||||
ExchangeNo string `json:"exchange_no"`
|
||||
CanView bool `json:"can_view"`
|
||||
} `json:"next_asset"`
|
||||
} `json:"exchange_trace"`
|
||||
} `json:"data"`
|
||||
}
|
||||
if err := sonic.Unmarshal(body, &success); err != nil {
|
||||
t.Fatalf("解析资产详情响应失败：%v", err)
|
||||
}
|
||||
if success.Code != errors.CodeSuccess || success.Data.AssetID != middleCard.ID || success.Data.ExchangeTrace.PreviousAsset == nil || success.Data.ExchangeTrace.NextAsset == nil {
|
||||
t.Fatalf("双向换货响应不完整：%s", body)
|
||||
}
|
||||
previous := success.Data.ExchangeTrace.PreviousAsset
|
||||
if previous.CanView || previous.AssetID != nil || previous.Identifier != "HTTP 历史前代快照" || previous.ExchangeNo != "UR86-HTTP-PREV" {
|
||||
t.Fatalf("不可见前代未按契约隐藏 ID：%s", body)
|
||||
}
|
||||
next := success.Data.ExchangeTrace.NextAsset
|
||||
if !next.CanView || next.AssetID == nil || *next.AssetID != nextCard.ID || next.Identifier != "HTTP 历史后代快照" || next.ExchangeNo != "UR86-HTTP-NEXT" {
|
||||
t.Fatalf("可见后代未按契约返回：%s", body)
|
||||
}
|
||||
|
||||
noTraceCard := env.createCard(t, 4, &visibleShop.ID)
|
||||
status, body = env.request(t, "/api/admin/assets/resolve/"+noTraceCard.ICCID, token)
|
||||
assertHTTPTraceDirections(t, status, body, false, false)
|
||||
|
||||
previousOnlyOld := env.createCard(t, 5, &visibleShop.ID)
|
||||
previousOnlyNew := env.createCard(t, 6, &visibleShop.ID)
|
||||
env.createOrder(t, "UR86-HTTP-PREV-ONLY", previousOnlyOld, previousOnlyNew, "仅前代旧快照", "仅前代新快照", time.Now())
|
||||
status, body = env.request(t, "/api/admin/assets/resolve/"+previousOnlyNew.ICCID, token)
|
||||
assertHTTPTraceDirections(t, status, body, true, false)
|
||||
|
||||
nextOnlyOld := env.createCard(t, 7, &visibleShop.ID)
|
||||
nextOnlyNew := env.createCard(t, 8, &visibleShop.ID)
|
||||
env.createOrder(t, "UR86-HTTP-NEXT-ONLY", nextOnlyOld, nextOnlyNew, "仅后代旧快照", "仅后代新快照", time.Now())
|
||||
status, body = env.request(t, "/api/admin/assets/resolve/"+nextOnlyOld.ICCID, token)
|
||||
assertHTTPTraceDirections(t, status, body, false, true)
|
||||
|
||||
status, body = env.request(t, "/api/admin/assets/resolve/"+oldCard.ICCID, token)
|
||||
if status != http.StatusNotFound {
|
||||
t.Fatalf("当前资产无权限时应维持防枚举响应，状态 %d：%s", status, body)
|
||||
}
|
||||
var denied struct {
|
||||
Code int `json:"code"`
|
||||
Data any `json:"data"`
|
||||
}
|
||||
if err := sonic.Unmarshal(body, &denied); err != nil || denied.Code != errors.CodeNotFound || denied.Data != nil {
|
||||
t.Fatalf("当前资产无权限响应不符合原契约：%s", body)
|
||||
}
|
||||
}
|
||||
|
||||
func assertHTTPTraceDirections(t *testing.T, status int, body []byte, hasPrevious, hasNext bool) {
|
||||
t.Helper()
|
||||
if status != http.StatusOK {
|
||||
t.Fatalf("资产详情状态错误，实际 %d：%s", status, body)
|
||||
}
|
||||
var response struct {
|
||||
Data struct {
|
||||
ExchangeTrace struct {
|
||||
PreviousAsset any `json:"previous_asset"`
|
||||
NextAsset any `json:"next_asset"`
|
||||
} `json:"exchange_trace"`
|
||||
} `json:"data"`
|
||||
}
|
||||
if err := sonic.Unmarshal(body, &response); err != nil {
|
||||
t.Fatalf("解析换货方向响应失败：%v", err)
|
||||
}
|
||||
if (response.Data.ExchangeTrace.PreviousAsset != nil) != hasPrevious || (response.Data.ExchangeTrace.NextAsset != nil) != hasNext {
|
||||
t.Fatalf("换货方向空值语义错误：previous=%v next=%v body=%s", hasPrevious, hasNext, body)
|
||||
}
|
||||
}
|
||||
|
||||
type assetTraceHTTPEnv struct {
|
||||
app *fiber.App
|
||||
db *gorm.DB
|
||||
redis *redis.Client
|
||||
tokenManager *auth.TokenManager
|
||||
}
|
||||
|
||||
func newAssetTraceHTTPEnv(t *testing.T) *assetTraceHTTPEnv {
|
||||
t.Helper()
|
||||
if os.Getenv("JUNHONG_DATABASE_HOST") == "" || os.Getenv("JUNHONG_REDIS_ADDRESS") == "" {
|
||||
t.Skip("未加载 .env.local，跳过依赖真实 PostgreSQL 和 Redis 的资产换货链 HTTP 集成测试")
|
||||
}
|
||||
cfg, err := config.Load()
|
||||
if err != nil {
|
||||
t.Fatalf("加载测试配置失败：%v", err)
|
||||
}
|
||||
log := zap.NewNop()
|
||||
db, err := database.InitPostgreSQL(&cfg.Database, log)
|
||||
if err != nil {
|
||||
t.Fatalf("连接 PostgreSQL 失败：%v", err)
|
||||
}
|
||||
tx := db.Begin()
|
||||
if tx.Error != nil {
|
||||
t.Fatalf("开启资产换货链测试事务失败：%v", tx.Error)
|
||||
}
|
||||
redisClient, err := database.NewRedisClient(database.RedisConfig{
|
||||
Address: cfg.Redis.Address + ":" + strconv.Itoa(cfg.Redis.Port), Password: cfg.Redis.Password, DB: cfg.Redis.DB,
|
||||
PoolSize: cfg.Redis.PoolSize, MinIdleConns: cfg.Redis.MinIdleConns,
|
||||
DialTimeout: cfg.Redis.DialTimeout, ReadTimeout: cfg.Redis.ReadTimeout, WriteTimeout: cfg.Redis.WriteTimeout,
|
||||
}, log)
|
||||
if err != nil {
|
||||
t.Fatalf("连接 Redis 失败：%v", err)
|
||||
}
|
||||
t.Cleanup(func() {
|
||||
_ = tx.Rollback().Error
|
||||
_ = redisClient.Close()
|
||||
if sqlDB, dbErr := db.DB(); dbErr == nil {
|
||||
_ = sqlDB.Close()
|
||||
}
|
||||
})
|
||||
|
||||
shopStore := postgres.NewShopStore(tx, redisClient)
|
||||
deviceStore := postgres.NewDeviceStore(tx, redisClient)
|
||||
cardStore := postgres.NewIotCardStore(tx, redisClient)
|
||||
assetSvc := assetService.New(
|
||||
tx, deviceStore, cardStore,
|
||||
postgres.NewPackageUsageStore(tx, redisClient), postgres.NewPackageStore(tx), postgres.NewPackageSeriesStore(tx),
|
||||
postgres.NewDeviceSimBindingStore(tx, redisClient), shopStore, redisClient, nil, nil,
|
||||
postgres.NewAssetIdentifierStore(tx), postgres.NewOrderStore(tx, redisClient), postgres.NewOrderItemStore(tx, redisClient),
|
||||
postgres.NewExchangeOrderStore(tx), nil,
|
||||
)
|
||||
handler := admin.NewAssetHandler(assetSvc, nil, nil, nil, nil, nil, assetQuery.NewExchangeTraceQuery(tx, log))
|
||||
tokenManager := auth.NewTokenManager(redisClient, cfg.JWT.AccessTokenTTL, cfg.JWT.RefreshTokenTTL)
|
||||
authMiddleware := pkgMiddleware.Auth(pkgMiddleware.AuthConfig{
|
||||
TokenValidator: func(token string) (*pkgMiddleware.UserContextInfo, error) {
|
||||
info, validateErr := tokenManager.ValidateAccessToken(context.Background(), token)
|
||||
if validateErr != nil {
|
||||
return nil, errors.New(errors.CodeInvalidToken, "认证令牌无效或已过期")
|
||||
}
|
||||
return &pkgMiddleware.UserContextInfo{UserID: info.UserID, UserType: info.UserType, Username: info.Username, ShopID: info.ShopID}, nil
|
||||
},
|
||||
ShopStore: shopStore,
|
||||
})
|
||||
app := fiber.New(fiber.Config{JSONEncoder: sonic.Marshal, JSONDecoder: sonic.Unmarshal, ErrorHandler: internalMiddleware.ErrorHandler(log)})
|
||||
RegisterAdminRoutes(app.Group("/api/admin"), &bootstrap.Handlers{Asset: handler}, &bootstrap.Middlewares{AdminAuth: authMiddleware}, nil, "/api/admin")
|
||||
return &assetTraceHTTPEnv{app: app, db: tx, redis: redisClient, tokenManager: tokenManager}
|
||||
}
|
||||
|
||||
func (e *assetTraceHTTPEnv) createShop(t *testing.T, name string, parentID *uint, level int) *model.Shop {
|
||||
t.Helper()
|
||||
shop := &model.Shop{ShopName: name, ShopCode: fmt.Sprintf("UR86-%d", time.Now().UnixNano()), ParentID: parentID, Level: level, Status: constants.ShopStatusEnabled}
|
||||
if err := e.db.Create(shop).Error; err != nil {
|
||||
t.Fatalf("创建资产换货链测试店铺失败：%v", err)
|
||||
}
|
||||
return shop
|
||||
}
|
||||
|
||||
func (e *assetTraceHTTPEnv) createCard(t *testing.T, suffix int, shopID *uint) *model.IotCard {
|
||||
t.Helper()
|
||||
iccid := fmt.Sprintf("8986333333333333%04d", suffix)
|
||||
card := &model.IotCard{ICCID: iccid, ICCID19: iccid[:19], VirtualNo: fmt.Sprintf("UR86-HTTP-CARD-%04d", suffix), ShopID: shopID, Status: constants.IotCardStatusInStock, AssetStatus: constants.AssetStatusInStock}
|
||||
if err := e.db.Create(card).Error; err != nil {
|
||||
t.Fatalf("创建资产换货链 HTTP 测试卡失败：%v", err)
|
||||
}
|
||||
return card
|
||||
}
|
||||
|
||||
func (e *assetTraceHTTPEnv) createOrder(t *testing.T, exchangeNo string, oldCard, newCard *model.IotCard, oldSnapshot, newSnapshot string, completedAt time.Time) {
|
||||
t.Helper()
|
||||
newID := newCard.ID
|
||||
order := &model.ExchangeOrder{
|
||||
ExchangeNo: exchangeNo, FlowType: constants.ExchangeFlowTypeDirect,
|
||||
OldAssetType: constants.ExchangeAssetTypeIotCard, OldAssetID: oldCard.ID, OldAssetIdentifier: oldSnapshot,
|
||||
NewAssetType: constants.ExchangeAssetTypeIotCard, NewAssetID: &newID, NewAssetIdentifier: newSnapshot,
|
||||
ExchangeReason: "UR86 HTTP 测试", Status: constants.ExchangeStatusCompleted, CompletedAt: &completedAt,
|
||||
}
|
||||
if err := e.db.Create(order).Error; err != nil {
|
||||
t.Fatalf("创建资产换货链 HTTP 测试单失败：%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func (e *assetTraceHTTPEnv) newAgentToken(t *testing.T, shopID uint) string {
|
||||
t.Helper()
|
||||
info := &auth.TokenInfo{UserID: uint(time.Now().UnixNano() % 1_000_000_000), UserType: constants.UserTypeAgent, ShopID: shopID, Username: "UR86测试代理"}
|
||||
accessToken, refreshToken, err := e.tokenManager.GenerateTokenPair(context.Background(), info)
|
||||
if err != nil {
|
||||
t.Fatalf("创建资产换货链测试令牌失败：%v", err)
|
||||
}
|
||||
t.Cleanup(func() {
|
||||
_ = e.tokenManager.RevokeToken(context.Background(), accessToken)
|
||||
_ = e.tokenManager.RevokeToken(context.Background(), refreshToken)
|
||||
_ = e.redis.Del(context.Background(), constants.RedisUserTokensKey(info.UserID)).Err()
|
||||
})
|
||||
return accessToken
|
||||
}
|
||||
|
||||
func (e *assetTraceHTTPEnv) request(t *testing.T, path, token string) (int, []byte) {
|
||||
t.Helper()
|
||||
request, err := http.NewRequest(http.MethodGet, path, nil)
|
||||
if err != nil {
|
||||
t.Fatalf("创建资产换货链 HTTP 请求失败：%v", err)
|
||||
}
|
||||
request.Header.Set(fiber.HeaderAuthorization, "Bearer "+token)
|
||||
response, err := e.app.Test(request, -1)
|
||||
if err != nil {
|
||||
t.Fatalf("执行资产换货链 HTTP 请求失败：%v", err)
|
||||
}
|
||||
defer response.Body.Close()
|
||||
body, err := io.ReadAll(response.Body)
|
||||
if err != nil {
|
||||
t.Fatalf("读取资产换货链 HTTP 响应失败：%v", err)
|
||||
}
|
||||
return response.StatusCode, body
|
||||
}
|
||||
Reference in New Issue
Block a user