一个简易c++单元测试框架可通过注册测试函数、自定义断言宏实现;使用std::function存储测试用例,通过全局构造对象注册,每个测试在try-catch中执行,断言失败时输出信息并计数,确保所有用例运行完毕后统一报告结果。

写C++程序时,手动验证函数行为既费时又容易出错。一个简单的单元测试框架能自动运行测试用例、检查结果并报告失败。从零开始构建一个支持断言的测试工具并不复杂,关键在于组织测试用例、捕获断言结果和输出清晰信息。
设计基本结构:测试用例与运行器
先定义测试的基本单元——测试用例。每个测试是一个函数,注册到全局管理器中,在程序启动时统一执行。
使用函数指针或std::function来存储测试函数,配合一个注册机制:
- 定义 Test 类型:using Test = std::function
; - 维护一个全局 vector 存放所有测试用例
- 提供宏 REGISTER_TEST(name) 来注册函数
通过构造全局对象触发注册,比如在 main 之前完成收集。
立即学习“C++免费学习笔记(深入)”;
实现断言机制:捕获失败但不停止
标准 assert 在失败时终止程序,不适合批量测试。需要自定义 ASSERT_EQ、ASSERT_TRUE 等宏,记录错误但继续执行。
思路是抛异常或设置标志位。这里推荐使用局部 try-catch 配合布尔标记:
- 每个测试函数运行在一个作用域内
- 断言失败时输出错误信息,增加失败计数
- 不中断当前测试,允许执行完所有检查点
例如:
#define ASSERT_EQ(actual, expected) do { if ((actual) != (expected)) { std::cerr << "FaiL: " << #actual << " == " << #expected << " in " << __FILE__ << ":" << __LINE__ << "n"; test_failed = true; } } while(0)
运行测试并输出结果
main 函数调用测试运行器,遍历所有注册的测试,逐个执行并统计结果。
struct TestCase { std::string name; Test func; };
运行时打印测试名,捕获异常(如断言抛出),汇总通过/失败数量。
示例输出:
Running test_addition... OK Running test_subtraction... FAILED (1 assertion failed)
简化测试编写:宏定义接口
让用户用类似 TEST(TestCaseName, TestName) 定义测试,提升可读性。
实现方式:
- 定义宏展开为一个函数和一个注册语句
- 利用静态变量确保只注册一次
例如:
#define TEST(suite_name, test_name) void test_##suite_name##_##test_name(); struct Register_##suite_name##_##test_name { Register_##suite_name##_##test_name() { add_test(#suite_name "." #test_name, test_##suite_name##_##test_name); } }; static Register_##suite_name##_##test_name reg_##suite_name##_##test_name; void test_##suite_name##_##test_name()
用户只需写 TEST(math, Addition) { … },即可自动注册。
基本上就这些。这个轻量框架没有外部依赖,编译即用,适合嵌入小型项目。随着需求增长,可扩展超时控制、参数化测试等功能。关键是理解测试生命周期:注册、执行、断言、报告。