如何为Python单元测试提供标准输入，文件和环境变量输入？

Question

在出现以下情况时如何编写测试：

1. 测试用户输入。
2. 测试从文件读取的输入。
3. 测试从环境变量读取的输入。

如果有人可以告诉我如何解决上述情况，那就太好了；如果您能指出一些我可以阅读的docs / articles / blog帖子，那还是很棒的。

Answer 1

您所描述的所有三种情况都需要专门解决，以确保在设计中使用松散耦合。

您是否 真的 需要对Python的raw_input方法进行单元测试？该open方法？ os.environ.get？没有。

您需要设置您的设计，以便您可以替代其他方法来检索该输入。然后，在单元测试期间，您将抛出实际上并没有调用raw_input或的某种类型的存根open。

例如，您的常规代码可能类似于：

import os
def say_hello(input_func):
    name = input_func()
    return "Hello " + name

def prompt_for_name():
    return raw_input("What is your name? ")

print say_hello(prompt_for_name)
# Normally would pass in methods, but lambdas can be used for brevity
print say_hello(lambda: open("a.txt").readline())
print say_hello(lambda: os.environ.get("USER"))

会话看起来像：

请问你贵姓大名？有人
你好
您好[部分文字]

你好标记

然后您的测试将像：

def test_say_hello():
    output = say_hello(lambda: "test")
    assert(output == "Hello test")

请记住，您不必测试一种语言的IO功能（除非您是设计该语言的人，这完全是另一回事）。