Files
hotime/docs/Testing_API测试框架.md
T
hoteas bd20af0c89 refactor(db): 增强测试事务支持和保存点管理
- 在 HoTimeDB 中新增测试事务功能,允许在测试模式下使用事务进行操作
- 实现 BeginTestTx 和 RollbackTestTx 方法,支持测试事务的开启和回滚
- 在 Action 方法中集成保存点管理,确保在测试模式下的嵌套事务处理
- 更新 README.md,添加 API 测试框架的相关说明,提升文档完整性
2026-03-14 10:19:57 +08:00

18 KiB
Raw Blame History

HoTime API 测试框架使用说明

不启动 HTTP 服务、自动事务回滚、链式 API、覆盖率报告、API 调试控制台生成。

目录


概述

HoTime 内置 API 测试框架,核心能力:

能力 说明
零启动 基于 net/http/httptest,不需要启动 HTTP 服务器
事务回滚 每个接口方法独立开启数据库事务,测试结束自动回滚,数据库不留任何痕迹
SAVEPOINT 业务代码中的 that.Db.Action() 在测试模式下自动用 SAVEPOINT 替代真事务,嵌套事务完整可用
链式 API a.JSON(...).Post("描述", 期望status) 一行完成:数据组装 + 请求 + 断言
覆盖率 自动对比路由注册表与测试定义,输出哪些接口已覆盖、哪些未覆盖
调试控制台 根据测试用例自动生成交互式 API 调试控制台,支持在线测试、认证管理、参数编辑

快速开始

第一步:创建测试定义文件(推荐)

推荐方式:为每个 handler 创建对应的 _test.go 文件,测试定义与业务代码分离。_test.go 文件仅在 go test 时编译,不影响生产构建。

// app/user.go — 只有业务代码
package app

import . "code.hoteas.com/golang/hotime"
import . "code.hoteas.com/golang/hotime/common"

var UserCtr = Ctr{
    "login":  func(that *Context) { /* ... */ },
    "info":   func(that *Context) { /* ... */ },
    "create": func(that *Context) { /* ... */ },
}
// app/user_test.go — 测试定义
package app

import . "code.hoteas.com/golang/hotime"
import . "code.hoteas.com/golang/hotime/common"

var UserTest = CtrTest{
    "login": {Desc: "用户登录", Func: func(a *Api) {
        a.JSON(Map{"name": "13800138000", "password": "123456"}).Post("正常登录", 0)
        a.JSON(Map{"name": "13800138000", "password": ""}).Post("密码为空", 4, "用户名或密码不能为空")
        a.JSON(Map{"name": "1380013", "password": "123456"}).Post("手机号格式错误", 4, "手机号码格式错误")
    }},
    "create": {Desc: "用户注册", Func: func(a *Api) {
        phone := "138" + ObjToStr(RandX(10000000, 99999999))
        a.JSON(Map{"phone": phone, "password": "123456"}).Post("正常注册", 0)
        a.JSON(Map{"phone": phone, "password": "123456"}).Post("重复注册", 3, "该手机号已被注册")
    }},
}

为什么要分离? _test.go 文件仅在 go test 时编译,不会进入生产二进制,保持业务代码干净。同时符合 Go 语言惯例:测试相关代码放在 _test.go 文件中。

第二步:注册路由和测试

路由在 init.go 中注册,测试注册放在 app_test.go 中(因为 _test.go 中的变量只在测试编译时可见):

// app/init.go — 只注册路由
var Project = Proj{
    "user": UserCtr,
    // ...
}
// app/app_test.go — 注册测试 + 运行入口
package app

import (
    "os"
    "testing"
    . "code.hoteas.com/golang/hotime"
)

var ProjectTest = ProjTest{
    "user": UserTest,
}

var testApp *TestApp

func TestMain(m *testing.M) {
    os.Chdir("..") // 确保工作目录为项目根目录
    testApp = NewTestApp("config/config.json", TestProj{
        "app": {Proj: Project, Tests: ProjectTest},
    })
    code := m.Run()
    testApp.PrintCoverage()
    testApp.GenerateSwagger("My API", "1.0.0", testApp.Config.GetString("tpt"))
    os.Exit(code)
}

func TestApi(t *testing.T) {
    testApp.RunTests(t)
}

推荐目录结构

app/
├── user.go          ← 业务代码 (UserCtr)
├── user_test.go     ← 测试定义 (UserTest)
├── tags.go          ← 业务代码 (TagsCtr)
├── tags_test.go     ← 测试定义 (TagsTest)
├── init.go          ← 路由注册 (Project)
└── app_test.go      ← 测试注册 (ProjectTest) + TestMain + TestApi

运行测试

go test ./app/... -v

注意TestMain 中的 os.Chdir("..") 确保 go test ./app/ 的工作目录为项目根目录,使配置文件路径和模板输出路径正确。


核心类型

注册类型

// 顶层:多项目集合(项目名 → 定义)
type TestProj map[string]TestProjDef

// 单项目:路由 + 测试绑定在一起
type TestProjDef struct {
    Proj  Proj     // 路由定义(与 Run 传入的 Router 保持一致)
    Tests ProjTest // 测试定义
}

// 控制器级测试(控制器名 → 控制器测试集合)
type ProjTest map[string]CtrTest

// 方法级测试(方法名 → 单条用例定义)
type CtrTest map[string]ApiTestDef

// 单个接口的用例定义
type ApiTestDef struct {
    Desc string       // 接口描述,用于 Swagger 说明和 t.Run 层级名称
    Func func(a *Api) // 测试函数体
}

TestApp

// NewTestApp 创建测试应用
// configPath: 配置文件路径
// projects:   TestProj 注册所有项目的路由和测试
// listeners:  可选,与 SetConnectListener 相同的请求拦截器
func NewTestApp(configPath string, projects TestProj, listeners ...func(*Context) bool) *TestApp

// RunTests 运行所有注册的测试(每个方法级别自动事务隔离)
func (app *TestApp) RunTests(t *testing.T)

// PrintCoverage 输出覆盖率报告,返回报告结构体
func (app *TestApp) PrintCoverage() CoverageReport

// GenerateSwagger 生成 API 调试控制台(api-spec.json + index.html
func (app *TestApp) GenerateSwagger(title, version, outputDir string) error

// DB 获取测试应用的数据库实例(在事务内)
func (app *TestApp) DB() *HoTimeDB

链式 API 参考

测试函数接收 *Api 作为入口,调用数据设置方法返回 *ApiCase,再调用终端方法发出请求并断言。

Api — 测试入口

方法 说明 返回
a.JSON(body) 设置 JSON body *ApiCase
a.Query(params) 设置 URL 查询参数 *ApiCase
a.Form(body) 设置 form 表单 body *ApiCase
a.File(field, name, content) 设置上传文件 *ApiCase
a.WithSession(s) 返回携带 session 的新 Api *Api
a.Get(desc, status, msg...) 直接 GET 请求(无数据场景) *ApiResponse
a.Post(desc, status, msg...) 直接 POST 请求(无数据场景) *ApiResponse
a.Put(desc, status, msg...) 直接 PUT 请求 *ApiResponse
a.Delete(desc, status, msg...) 直接 DELETE 请求 *ApiResponse
a.DB() 获取数据库实例(在事务内,可用于准备/清查数据) *HoTimeDB

ApiCase — 链式构建器

Api 的数据设置方法返回 *ApiCase,支持继续叠加数据:

方法 说明 返回
c.JSON(body) 追加 JSON body *ApiCase
c.Query(params) 追加 URL 查询参数 *ApiCase
c.Form(body) 追加 form 字段 *ApiCase
c.File(field, name, content) 设置上传文件 *ApiCase
c.WithSession(s) 覆盖本次请求的 session *ApiCase
c.Get(desc, status, msg...) 终端:发送 GET 请求并断言 *ApiResponse
c.Post(desc, status, msg...) 终端:发送 POST 请求并断言 *ApiResponse
c.Put(desc, status, msg...) 终端:发送 PUT 请求并断言 *ApiResponse
c.Delete(desc, status, msg...) 终端:发送 DELETE 请求并断言 *ApiResponse

终端方法参数说明:

参数 类型 说明
desc string 用例描述,作为 t.Run 的子测试名称
status int 期望的 JSON 响应体中 status 字段值(0=成功)
msg ...string 可选,期望的 msg 字段值(传空则不校验)

ApiResponse — 响应对象

终端方法返回 *ApiResponse,可用于进一步的自定义断言:

resp := a.JSON(Map{"name": "user", "password": "123"}).Post("正常登录", 0)

// 获取响应字段
resp.GetStatus()          // int:业务 status
resp.GetMsg()             // string:业务 msg
resp.GetResult()          // interface{}result 字段
resp.GetBody()            // Map:完整响应体

// 自定义断言
token := resp.GetBody().GetMap("result").GetString("token")
if token == "" {
    resp.Fail("缺少 token 字段")
}

完整场景对照

以下示例均写在 _test.go 文件的 CtrTest 定义中(如 user_test.go),a*Api 参数:

// POST JSON(最常见)
a.JSON(Map{"name": "138", "password": "123"}).Post("正常登录", 0)

// GET 无参数
a.Get("未登录", 2, "请先登录")

// GET + URL 参数
a.Query(Map{"shop_id": "1", "page": "1"}).Get("查询列表", 0)

// 需登录 + GET
a.WithSession(Map{"user_id": int64(1)}).Get("已登录", 0)

// 需登录 + GET + URL 参数
a.WithSession(Map{"user_id": int64(1)}).Query(Map{"shop_id": "1"}).Get("带参查询", 0)

// 需登录 + POST JSON
a.WithSession(Map{"user_id": int64(1)}).JSON(Map{"name": "新名字"}).Post("更新信息", 0)

// 混合:URL 参数 + JSON body
a.Query(Map{"token": "abc"}).JSON(Map{"phone": "138"}).Post("混合传参", 3, "异常")

// POST Form 表单
a.Form(Map{"shop_id": "1", "id": "5"}).Post("表单提交", 0)

// 文件上传
a.File("file", "photo.png", imgBytes).Post("上传文件", 0)

// 文件 + 表单字段混合
a.File("file", "photo.png", imgBytes).Form(Map{"type": "avatar"}).Post("上传头像", 0)

// PUT
a.JSON(Map{"name": "新名字"}).Put("更新", 0)

// DELETE + URL 参数
a.Query(Map{"id": "5"}).Delete("删除", 0)

// 校验响应中的具体字段
resp := a.JSON(Map{"name": "138", "password": "123"}).Post("正常登录", 0)
if resp.GetBody().GetMap("result").GetString("token") == "" {
    resp.Fail("响应中缺少 token 字段")
}

事务隔离机制

工作原理

RunTests 在每个方法级测试前开启一个数据库事务(testTx),测试结束后无论成功与否一律回滚:

TestApi
└── app(项目名)
    └── user(控制器名)
        └── create(方法名)← 在这一级 BEGIN → 运行测试用例 → ROLLBACK
            └── 正常注册(用例)
            └── 重复注册(用例)

所有测试用例共享同一个 testTx,因此"正常注册"写入的数据对"重复注册"可见,模拟了真实的并发隔离场景。

嵌套事务 (SAVEPOINT)

业务代码中常见的 that.Db.Action() 在测试模式下自动切换为 SAVEPOINT 机制,不会脱离外层测试事务:

// 业务代码(无需修改)
that.Db.Action(func(db HoTimeDB) bool {
    db.Insert("order", orderData)   // 在 SAVEPOINT sp_N 内执行
    db.Update("customer", balanceData, where)
    return true  // → RELEASE SAVEPOINT sp_N(提交到外层 testTx
    // 返回 false → ROLLBACK TO SAVEPOINT sp_N(只回滚这一层)
})
// 测试结束 → ROLLBACK testTx(回滚一切)
机制 生产模式 测试模式
Action() BEGIN / COMMIT / ROLLBACK SAVEPOINT / RELEASE / ROLLBACK TO
数据持久化 持久化到数据库 测试结束全部回滚
对现有代码影响 无(testTx 默认 nil

一次性数据的处理

对于需要随机数据的测试(如注册),直接用 RandX 生成随机值即可,无需手动清理

// app/user_test.go 中的 UserTest 片段
"create": {Desc: "用户注册", Func: func(a *Api) {
    // 生成随机手机号,测试完成后事务自动回滚,数据库中不留任何记录
    phone := "138" + ObjToStr(RandX(10000000, 99999999))
    a.JSON(Map{"phone": phone, "password": "123456"}).Post("正常注册", 0)
    a.JSON(Map{"phone": phone, "password": "123456"}).Post("重复注册", 3, "该手机号已被注册")
}},

覆盖率报告

PrintCoverage() 自动对比 Proj(所有路由)和 ProjTest(有测试的路由),在 TestMain 中调用:

func TestMain(m *testing.M) {
    // ...
    code := m.Run()
    testApp.PrintCoverage()   // 在测试运行完成后输出报告
    os.Exit(code)
}

输出示例:

========== API 测试覆盖率报告 ==========
总接口: 42 | 已覆盖: 6 | 覆盖率: 14.3%

已覆盖的接口:
  + /app/user/login      4 用例 (4 通过, 0 失败) 12ms
  + /app/user/info       2 用例 (2 通过, 0 失败) 5ms
  + /app/user/token      2 用例 (2 通过, 0 失败) 4ms
  + /app/user/create     4 用例 (4 通过, 0 失败) 18ms
  + /app/user/forget     2 用例 (2 通过, 0 失败) 6ms
  + /app/user/file       2 用例 (2 通过, 0 失败) 9ms

未覆盖的接口:
  - app/goods/list
  - app/goods/create
  - app/order/list
  - ... (共36个未覆盖)
========================================

API 调试控制台

GenerateSwagger 遍历 TestProj 中所有项目的路由和测试用例,生成一个交互式 API 调试控制台(暗色高对比度主题),支持在线调试、参数编辑、认证管理。

testApp.GenerateSwagger(
    "My API",                              // 文档标题
    "2.1.0",                               // 版本号
    testApp.Config.GetString("tpt"),       // 输出目录(如 "tpt"
)

生成文件:

{outputDir}/swagger/
├── api-spec.json   ← 自定义 API 规范文件(包含路由、测试用例、请求/响应数据)
└── index.html      ← 交互式调试控制台

访问地址:http://localhost:8081/swagger/

控制台功能

功能 说明
三级导航 左侧侧边栏按 项目 > 控制器 > 方法 三级树形展开
搜索和筛选 支持关键字搜索,按 全部/已测试/未测试 筛选
测试用例查看 每个接口的测试用例以手风琴形式展示(第一个默认展开),包含请求头、请求参数(Query/JSON/Form/文件)、响应结果
接口调试 可编辑的请求头、Query 参数、文件上传、请求体(表单/JSON 互斥切换),直接发送请求
预填用例 下拉选择测试用例一键填充请求参数,方便快速调试
全局认证 顶部认证栏支持四种模式:无认证、Header (Authorization)、URL (?token=)、Cookie,修改后实时同步到各参数区域
Cookie 隔离 Header/URL 认证模式下自动阻止浏览器发送 Cookie(credentials: 'omit'),避免意外自动授权
cURL 生成 每次请求自动生成对应的 cURL 命令,支持一键复制
响应格式化 JSON 响应自动格式化显示,支持复制
覆盖率可视化 已测试接口显示通过/失败计数徽章,未测试接口标记"未测试"

多项目支持

所有注册项目的路由和测试用例合并到同一份 api-spec.json,在侧边栏按项目名分组,不会互相覆盖:

侧边栏结构:
├── app(项目)
│   ├── user(控制器)
│   │   ├── POST login 用户登录          ✅ 4/4
│   │   ├── POST info                    未测试
│   │   └── POST create 用户注册         ✅ 4/4
│   └── goods(控制器)
│       └── ...
└── customer(项目)
    └── pay(控制器)
        └── ...

指定运行范围

RunTests 使用 Go 标准 t.Run() 子测试机制,天然支持 -run 参数过滤:

# 运行全部测试
go test ./app/... -v

# 只跑 user 控制器的所有接口
go test ./app/... -v -run TestApi/app/user

# 只跑 user/login 接口
go test ./app/... -v -run TestApi/app/user/login

# 只跑 login 接口中"密码为空"这一个用例
go test ./app/... -v -run "TestApi/app/user/login/密码为空"

# 同时跑 user 和 goods 两个控制器
go test ./app/... -v -run "TestApi/app/(user|goods)"

# 跑所有控制器中包含"未登录"的用例
go test ./app/... -v -run "TestApi/.*/.*未登录"

子测试层级结构:

TestApi
└── {项目名}         ← -run TestApi/app
    └── {控制器名}   ← -run TestApi/app/user
        └── {方法名} ← -run TestApi/app/user/login
            └── {ApiTestDef.Desc}(接口描述)
                └── {用例描述}  ← -run TestApi/app/user/login/密码为空

框架改动说明

测试框架对 hotimev1.5 共改动 6 个文件,其中 3 个为修改、3 个为新增:

修改的文件

文件 改动 生产影响
db/db.go 新增 testTx *sql.Tx 字段 + BeginTestTx()/RollbackTestTx() 方法 无(testTx 默认 nil
db/query.go QueryExec 各增加一个 if testTx != nil 分支 无(分支默认跳过)
db/transaction.go Action() 增加 SAVEPOINT 分支 无(testTx 为 nil 时走原有逻辑)

新增的文件

文件 内容
testing_helper.go TestAppNewTestAppSetupForTestRunTests(事务隔离)、PrintCoverage(覆盖率)
testing_api.go 注册类型 + Api/ApiCase/ApiResponse 链式 API + TestCollector
testing_swagger.go GenerateSwagger:合并所有项目生成 API 调试控制台(api-spec.json + index.html

所有新增文件仅在 go test 环境下被使用,生产构建不引入 testing 包依赖,完全不影响线上服务。