更改预处理器符号的值以进行测试

时间:2010-09-14 17:50:14

标签: c++ c-preprocessor

我正在编写一个动态增长的字符串缓冲区。我在.c文件中有以下内容。

#ifndef STRBUF_GROWTH_SIZE
#define STRBUF_GROWTH_SIZE 4096
#endif

我的代码使用此常量来重新分配缓冲区。现在在测试中,我需要将此值设置为小值,以便我可以检查重新分配。我尝试在tests.cpp中定义它(所有测试都在C ++中使用UnitTest ++)。

#define STRBUF_GROWTH_SIZE 10
TEST(StringBuffer)
{
    struct strbuf *string = strbuf_init();

    strbuf_add(string, "first");
    CHECK_EQUAL("first", string->buffer);
    CHECK_EQUAL(5, string->length);
    CHECK_EQUAL(10, string->allocated);   /* this fails */

    strbuf_add(string, " second");
    CHECK_EQUAL("first second", string->buffer);
    CHECK_EQUAL(12, string->length);
    CHECK_EQUAL(20, string->allocated); /* this fails */

    strbuf_destroy(string);
}

我想知道为什么价值没有变成10?我该如何解决这个问题?

有什么想法吗?

2 个答案:

答案 0 :(得分:4)

您的测试中的#define在其他.c文件中的代码中看不到。但是,您可以在构建中注入宏。 gcc是-D argument。 msvc的cl为/D argument

答案 1 :(得分:3)

预处理是在逐个源文件的基础上完成的(好吧,不完全,但它足够接近)。一个源文件中的#define不会影响另一个源文件中的任何内容。

你需要在标题中#define这个,你可以在测试过程中换掉另一个标题,或者你需要参数化你的库(例如strbuf_init(int growth_size)

或者,为什么不用长度为4096的字符串进行测试?然后你将测试你的实际生产代码。