如何在C++代码中设置用户输入以实现自动化测试

方法一：替换std::cin的缓冲区（streambuf） #

无需修改待测试函数代码，直接通过替换std::cin的底层缓冲区，让它从预设字符串中读取内容。测试完成后记得恢复原缓冲区，避免影响后续操作。

示例代码：

#include <iostream>
#include <sstream>
#include <string>

// 待测试函数
std::string get_user_input() {
    std::string input;
    std::cin >> input;
    return input;
}

int main() {
    // 预设测试输入
    std::string test_input = "TestInput123";
    std::stringstream ss(test_input);
    
    // 保存原cin缓冲区
    std::streambuf* original_buf = std::cin.rdbuf();
    // 替换为自定义流的缓冲区
    std::cin.rdbuf(ss.rdbuf());
    
    // 执行测试
    std::string result = get_user_input();
    if (result == test_input) {
        std::cout << "测试通过" << std::endl;
    } else {
        std::cout << "测试失败" << std::endl;
    }
    
    // 恢复原缓冲区
    std::cin.rdbuf(original_buf);
    return 0;
}

方法二：依赖注入输入流（推荐） #

如果允许修改待测试函数，可重构函数让其接受std::istream参数，默认使用std::cin，测试时传入预设好的流。这种方式更灵活，也符合模块化设计思想。

示例代码：

#include <iostream>
#include <sstream>
#include <string>

// 重构后的待测试函数
std::string get_user_input(std::istream& in = std::cin) {
    std::string input;
    in >> input;
    return input;
}

int main() {
    std::string test_input = "TestInput123";
    std::stringstream ss(test_input);
    
    // 测试时传入自定义流
    std::string result = get_user_input(ss);
    if (result == test_input) {
        std::cout << "测试通过" << std::endl;
    } else {
        std::cout << "测试失败" << std::endl;
    }
    return 0;
}

额外提示 #

• 若需模拟多行输入，只需在预设字符串中加入换行符即可，比如std::string test_input = "FirstLine\nSecondLine";
• 若使用std::getline读取整行，要注意处理流中残留的换行符，避免读取空内容。

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