c - 在 C 中自动注册单元测试

Question

考虑这个用于运行单元测试的最小框架:

#include <stdio.h>

#define TEST(name) int test_##name()
#define RUN_TEST(name) \
    do {printf("%s\t%s\n", #name, test_##name() ? "OK" : "FAIL");} while(0)
#define ASSERT_EQ(left, right) \
    do {return (left) == (right);} while(0)

TEST(addition) {
    ASSERT_EQ(1 + 2, 3);
}

TEST(mult_fail) {
    ASSERT_EQ(42, 6 * 9);
}

int main() {
    RUN_TEST(addition);
    RUN_TEST(mult_fail);
}

它的使用方式如下:

> gcc test_runner.c && ./a.out
addition        OK
mult_fail       FAIL

如何才能让 TEST 宏自动注册测试？我希望主函数包含单个 RUN_ALL_TESTS 语句，而不必按名称引用所有测试。

我知道这可以在 C++ 中轻松完成，但我更愿意看到纯 C 解决方案。

Answer 1

由于您开始使用基于宏的方法，因此您可能会对围绕 X 宏构建的解决方案感兴趣。它会有点丑陋，而且很奇怪，但它可以做你想做的事:

#include <stdio.h>

// Test definitions
#define TESTS \
  X(addition,  ASSERT_EQ(1 + 2, 3) ) \
  X(mult_fail, ASSERT_EQ(42, 6 * 9))

// Helper macros    
#define ASSERT_EQ(left, right) \
    do {return (left) == (right);} while(0)

// declare test functions
#define X(name, body) int test_##name(void) { body; }
TESTS
#undef X

int main(void) {
    // run tests
    #define X(name, ignored) \
      printf("%s\t%s\n", #name, test_##name() ? "OK" : "FAIL");
    TESTS
    #undef X
}

要添加、删除或修改测试，只需修改宏TESTS的定义即可；预处理器从中生成您需要的所有其他内容。