CMocka实战：手把手教你用Mock和断言，给老旧C库写“安全隔离”测试

CMocka实战：构建C语言模块的“安全隔离”测试体系

在嵌入式系统和底层开发领域，C语言模块经常需要与硬件驱动、网络接口或第三方闭源库交互。这种强依赖关系使得传统单元测试难以实施——你无法在持续集成环境中接入真实硬件，也不该为测试而修改生产代码。本文将展示如何用CMocka框架为这类"顽固"模块建立完全隔离的测试环境，通过一个配置文件解析器的完整案例，演示Mock技术如何替代真实依赖。

1. 理解测试隔离的价值

假设我们有一个读取JSON配置文件的模块，核心函数load_config()依赖fopen()、fread()等系统调用。直接测试这个模块会遇到几个典型问题：

• 测试机器上可能不存在特定路径的配置文件
• 文件系统操作会拖慢测试速度（特别是大量用例时）
• 无法模拟文件损坏、权限错误等边界情况
• 测试会留下临时文件需要清理

通过CMocka的Mock功能，我们可以拦截所有文件系统调用，让测试在内存中完成。下面是对比实验数据：

这种隔离带来的优势在持续集成(CI)环境中尤为明显。测试不再受外部系统状态影响，可以百分百重复执行，且能在资源受限的嵌入式编译服务器上运行。

2. 构建Mock测试环境

2.1 基础测试框架搭建

首先建立基本的测试项目结构：

config_parser/ ├── src/ │ ├── config_loader.c # 被测模块 │ └── config.h └── tests/ ├── test_config.c # 测试套件 └── CMakeLists.txt

测试文件需要包含CMocka头文件并链接库：

#include <stdarg.h> #include <stddef.h> #include <setjmp.h> #include <cmocka.h> // 测试用例放在此处 int main(void) { const struct CMUnitTest tests[] = { cmocka_unit_test(test_load_config), }; return cmocka_run_group_tests(tests, NULL, NULL); }

2.2 拦截系统调用

关键步骤是替换原始的文件操作函数。在测试文件中创建Mock版本：

FILE *__wrap_fopen(const char *path, const char *mode) { check_expected(path); // 验证输入参数 check_expected(mode); return (FILE *)mock(); // 返回预设值 } size_t __wrap_fread(void *ptr, size_t size, size_t nmemb, FILE *stream) { check_expected(size); check_expected(nmemb); return mock(); // 返回实际读取的条目数 }

注意：使用GCC时通过--wrap符号实现函数替换，其他编译器需使用不同机制

3. 设计隔离测试用例

3.1 正常流程测试

模拟配置文件成功加载的场景：

void test_load_config_success(void **state) { // 设置fopen返回非NULL文件指针 expect_string(__wrap_fopen, path, "config.json"); expect_string(__wrap_fopen, mode, "r"); will_return(__wrap_fopen, (FILE *)0x1234); // 设置fread行为：首次读取返回完整数据 char mock_data[] = "{\"timeout\": 100}"; expect_value(__wrap_fread, size, 1); expect_value(__wrap_fread, nmemb, sizeof(mock_data)); will_return(__wrap_fread, sizeof(mock_data)); will_return(__wrap_fread, mock_data); // 设置fread行为：第二次返回0表示EOF will_return(__wrap_fread, 0); // 执行测试 Config config; assert_int_equal(load_config("config.json", &config), 0); assert_int_equal(config.timeout, 100); }

3.2 异常场景覆盖

利用Mock模拟各种错误条件：

void test_load_config_file_not_found(void **state) { expect_string(__wrap_fopen, path, "missing.json"); expect_string(__wrap_fopen, mode, "r"); will_return(__wrap_fopen, NULL); // 模拟文件不存在 Config config; assert_int_equal(load_config("missing.json", &config), ERR_FILE_NOT_FOUND); } void test_load_config_invalid_json(void **state) { expect_string(__wrap_fopen, path, "bad.json"); expect_string(__wrap_fopen, mode, "r"); will_return(__wrap_fopen, (FILE *)0x1234); char mock_data[] = "{invalid: json}"; expect_value(__wrap_fread, size, 1); expect_value(__wrap_fread, nmemb, sizeof(mock_data)); will_return(__wrap_fread, sizeof(mock_data)); will_return(__wrap_fread, mock_data); will_return(__wrap_fread, 0); Config config; assert_int_equal(load_config("bad.json", &config), ERR_PARSE_FAILED); }

4. 高级Mock技巧

4.1 参数动态验证

对于复杂参数，可以使用自定义检查函数：

static int check_fwrite_buffer(const void *actual, const void *expected) { const char *buf = (const char *)actual; return strstr(buf, "expected_pattern") != NULL; } void test_save_config(void **state) { expect_string(__wrap_fopen, path, "output.json"); expect_string(__wrap_fopen, mode, "w"); will_return(__wrap_fopen, (FILE *)0x1234); // 使用自定义检查器验证写入内容 expect_check(__wrap_fwrite, ptr, check_fwrite_buffer, NULL); expect_value(__wrap_fwrite, size, 1); expect_value(__wrap_fwrite, nmemb, 128); will_return(__wrap_fwrite, 128); Config config = {.timeout = 200}; assert_int_equal(save_config("output.json", &config), 0); }

4.2 调用顺序验证

确保函数按预期顺序被调用：

void test_config_reload_sequence(void **state) { // 预期调用顺序：fopen -> fread -> fclose -> fopen -> fwrite -> fclose expect_function_call(__wrap_fopen); expect_function_call(__wrap_fread); expect_function_call(__wrap_fclose); expect_function_call(__wrap_fopen); expect_function_call(__wrap_fwrite); expect_function_call(__wrap_fclose); // 设置各Mock函数的返回值 will_return(__wrap_fopen, (FILE *)0x1234); will_return(__wrap_fread, 0); // 立即EOF will_return(__wrap_fopen, (FILE *)0x1234); will_return(__wrap_fwrite, 128); assert_int_equal(reload_config("config.json"), 0); }

5. 嵌入式环境适配

在资源受限的嵌入式平台上，需注意：

1. 内存管理：替换标准库的内存操作

#define malloc test_malloc #define free test_free

2. 平台头文件：创建cmocka_platform.h解决类型定义问题

// cmocka_platform.h typedef int pid_t; #define HAVE_STRUCT_TIMESPEC 1

3. 线程安全：设置环境变量避免线程问题

export CMOCKA_TEST_ABORT=1

4. 输出控制：使用TAP格式节省空间

setenv("CMOCKA_MESSAGE_OUTPUT", "TAP", 1);

6. 测试策略建议

根据项目特点选择适当的Mock粒度：

在持续集成流水线中，建议组合使用：

1. 单元测试：完全Mock，快速验证逻辑
2. 集成测试：部分Mock，验证真实交互
3. 硬件在环：在最终阶段运行

通过这种分层策略，既能保证开发效率，又能确保最终产品质量。