如何对涉及IO的c函数进行单元测试？

Question

我在将单元测试写入涉及IO操作的C函数时面临问题.例如,下面是我编写的代码,用于从控制台获取用户的输入字符串.我不知道如何使用getchar()函数自动测试用户输入.

char * GetStringFromConsole()
{

    char *strToReturn = NULL;
    int len = 128;

    strToReturn = (char*)malloc(len);
    if (strToReturn) 
    {
        int ch;
        char *ptr  = strToReturn;
        int counter = 0;
        for (; ;) 
        {
            ch = getchar();
            counter++;

            if (counter == len)
            {
                strToReturn = realloc(strToReturn, len*=2 );
                ptr = strToReturn + counter-1;
            }

            if ((ch != EOF) && (ch != '\n') && (counter < len))
            {
                *ptr++ = ch;
            }
            else 
            {
                break;
            }

        }
        *ptr = '\0';
    }
    return strToReturn;
}   

Answer 1

模拟getchar:

1. 利用预处理器,例如在测试文件中.

   #define getchar mock_getchar
   #include "GetStringFromConsole.h"
   ...
   const char* mock_getchar_data_ptr;
   char mock_getchar()
   {
       return *mock_getchar_data_ptr++;
   }
   ...
   // in test function
   mock_getchar_data_ptr = "hello!\n";
   YourAssertEquals("hello", GetStringFromConsole());
   

2. 链接器的替换符号(在我看来更难),例如getchar在源.c文件中的某个地方定义自己的符号,而不是链接到stdlib(例如windows上的msvcrt)

3. 修改测试中的函数以接受函数返回char,最佳选择(恕我直言) - 与stdlib无冲突.并通过mock_getchar从测试代码中的第1点传递类似的东西来设置测试.

   typedef char (*getchartype)();
   char * GetStringFromConsole(getchartype mygetchar)
   {
       ...
       c  = mygetchar()
   

对于第1点和第2点,我建议使用您自己的函数而不是getchar(例如mygetchar) - 这样您就可以模拟/替换它而不会遇到与std includes/libs的冲突.