使用gtest在C单元测试中仅Mock函数内变量测试整数溢出

嘿，这个问题我之前也碰到过——为了测试溢出硬蹭计数器到最大值确实太浪费时间了，而且还要保留初始化函数的其他逻辑，完全没必要Mock整个初始化流程。下面给你几个实用的方案，根据你的计数器的存储方式选就行：

1. 如果计数器是全局/外部可见变量 #

这是最直接的情况，你只需要在测试用例里直接赋值就行：

// 假设被测模块里的计数器是全局变量
extern uint32_t g_module_counter;

TEST(CounterOverflowTest, IncrementFromMaxTriggersOverflow) {
    // 先正常调用初始化函数，保留其他必要数据的初始化
    module_init();
    
    // 直接把计数器拉到最大值
    g_module_counter = UINT32_MAX;
    
    // 触发递增事件
    trigger_counter_increment();
    
    // 验证溢出后的预期行为（比如无符号整数溢出会回到0）
    EXPECT_EQ(g_module_counter, 0);
    // 如果有溢出告警逻辑，也可以在这里验证
}

2. 如果计数器是模块内部结构体的成员 #

如果计数器封装在模块内的结构体里（对外不可见），可以给被测代码加个测试专用的访问接口，用条件编译控制，不会污染生产代码：

第一步：在被测模块的.c文件里加测试接口 #

// module.c
#include <stdint.h>

// 模块内部状态结构体
typedef struct {
    uint32_t counter;
    // 其他需要初始化的重要成员
} ModuleState;

static ModuleState s_module_state;

// 正常的初始化函数
void module_init() {
    s_module_state.counter = 0;
    // 初始化其他成员...
}

// 触发计数器递增的函数
void trigger_counter_increment() {
    s_module_state.counter++;
}

// 测试专用：暴露状态结构体指针（仅测试编译时生效）
#ifdef TEST_BUILD
ModuleState* get_module_state() {
    return &s_module_state;
}
#endif

第二步：在测试用例里调用接口修改计数器 #

// 测试文件
#include "gtest/gtest.h"
#include "module.h"

// 声明测试专用接口（或者在测试头文件里声明）
#ifdef TEST_BUILD
extern ModuleState* get_module_state();
#endif

TEST(CounterOverflowTest, StructMemberCounterOverflows) {
    module_init();
    
    // 获取模块状态指针，直接修改计数器
    ModuleState* state = get_module_state();
    state->counter = UINT32_MAX;
    
    trigger_counter_increment();
    
    EXPECT_EQ(state->counter, 0);
}

编译测试代码的时候，加上-DTEST_BUILD宏就行，生产编译时这个接口会被忽略。

3. 如果计数器是函数内的静态变量 #

这种情况稍微麻烦一点，但还是可以用类似的思路——把静态变量的定义抽出来，用测试接口暴露：

// module.c
#include <stdint.h>

// 测试专用：暴露静态计数器的指针
#ifdef TEST_BUILD
static uint32_t s_event_counter = 0;
uint32_t* get_event_counter() {
    return &s_event_counter;
}
#endif

void trigger_counter_increment() {
#ifdef TEST_BUILD
    static uint32_t* counter = get_event_counter();
#else
    static uint32_t counter = 0;
#endif
    (*counter)++;
}

然后测试用例里就可以通过get_event_counter()拿到指针，直接赋值为最大值了。

几个关键注意事项 #

• 测试隔离：记得在SetUp()或TearDown()里重置计数器的值，避免不同测试用例之间互相干扰。
• 类型匹配：一定要用对应类型的最大值宏，比如uint32_t用UINT32_MAX，int用INT_MAX，别搞错了。
• 溢出行为的确定性：C语言中带符号整数溢出是未定义行为，如果你的计数器是带符号类型，最好改用无符号类型（溢出行为是定义好的模2^n），或者给编译器加参数（比如GCC的-fwrapv）让溢出行为可预测。
• 不要修改生产逻辑：所有测试专用的代码都要用条件编译包起来，确保生产版本不会包含这些内容。

这样既保留了初始化函数的完整逻辑，又能瞬间把计数器拉到最大值，完美解决耗时问题～

内容的提问来源于stack exchange，提问作者kreso