将 cout 和 stdout 重定向到 C++ 中的字符串 用于单元测试

Question

我正在努力在单元测试下获取一些遗留代码，有时感知现有程序行为的唯一方法是从控制台输出。

我在网上看到很多关于如何将 stdout 重定向到 C++ 中的另一个文件的示例，但是有没有办法可以将其重定向到内存中的流，以便我的测试不必依赖于磁盘？

我希望将遗留代码发送到 stdout 的任何内容都转换为 std::string ，以便我可以轻松地在输出上查找。

编辑

遗留代码非常糟糕，以至于它使用了 cout << 的混合物。 .. 和 printf。 这是我到目前为止所得到的：

void TestSuite::setUp(void)
{
    oldStdoutBuf = std::cout.rdbuf();
    std::cout.rdbuf(consoleOutput.rdbuf());
}
void TestSuite::tearDown(void)
{
    std::cout.rdbuf(oldStdoutBuf);
}

问题是这不使用 printf 捕获输出。 我想要两者兼得的东西。 有任何想法吗？

Answer 1

std::stringstream 可能就是你的寻找。

更新
好吧，这有点 hack，但也许你可以这样做来获取 printf 输出：

char huge_string_buf[MASSIVE_SIZE];
freopen("NUL", "a", stdout);
setbuf(stdout, huge_string_buffer);

请注意，对于 linux，你应该使用“/dev/null”而不是“NUL”。 这将迅速开始填满巨大的字符串缓冲区。 如果您希望能够在缓冲区满后继续重定向输出，则必须调用 fflush()，否则它将引发错误。 有关详细信息，请参阅 std::setbuf 。

Answer 2

您可以使用 freopen(..., stdout) ，然后将文件转储到内存或 std::string 中。

Answer 3

这可能是一种替代方案：

char bigOutBuf[8192];
char savBuf[8192];

fflush(stdout);
setvbuf(stdout,bigOutBuf,IOFBF,8192);//stdout uses your buffer

//after each operation
strncpy(savBuf,bigOutBuf,8192);//won't flush until full or fflush called

//...

//at long last finished
setbuf(stdout,NULL);//reset to unnamed buffer

这只是拦截缓冲的输出，因此仍然会转到控制台或任何地方。

希望这可以帮助。

Answer 4

试试sprintf，效率更高。

int i;
char str[] = "asdf";
char output[256];
sprintf(output, "asdfasdf %s %d\n", str, i);