如何为需要输入的函数编写单元测试并因此调用其他函数？

【问题标题】：How to write unit tests for Functions requiring input and as a result, calling other functions?如何为需要输入的函数编写单元测试并因此调用其他函数？
【发布时间】：2021-08-01 20:43:50
【问题描述】：

假设我有这个代码。真的不确定如何对此进行单元测试，因为它是一个 void 方法，需要用户输入，并且其中一种情况的有效答案是调用另一种方法。比如我如何测试这个？

def start():
    user_Response = input(
        '\n' "○ Press 1 to Enter the Main Menu" '\n'
        "○ Press 2 or type 'exit' to Exit \n ")

    if user_Response == str(1):
        mainMenu()

    elif user_Response == str(2) or user_Response.lower() == "exit":
        print(
            'Thank you, good bye.
        exit()

    else:
        "Please enter a valid response."
        start() ```

【问题讨论】：

This answer 可能对模拟标准输入和标准输出有用。

标签： python unit-testing testing

【解决方案1】：

我会重写函数。我不会调用input，而是将一个类似文件的对象作为参数传递，您可以对其进行迭代。调用者可以负责传递sys.stdin 或其他一些类似文件的对象（例如io.StringIO 的实例）以便于测试。

我也会放弃递归，转而使用简单的循环。

def start(input_stream):
    while True:
        print("...")
        user_response = input_stream.readline().rstrip('\n')

        if user_response == "1":
            return mainMenu()

        elif user_response == "2" or user_response.lower() == "exit":
            print('Thank you, good bye.')
            exit()

        else:
            print("Please enter a valid response.")

样本测试：

def test_loop():
    try:
        start(io.StringIO("exit"))
    except SystemExit:
        pass
    else:
        assert False, "start did not exit"

【讨论】：