Python 2 unittest模块中的模拟raw_input（）

Question

我试图理解如何使用python的unittest框架

我有一段代码，看起来像这样--

while True:
    filename = raw_input('Enter file')
    if os.path.exists(filename):
        break
    else:
        print "That file does not exist"
return filename

有人能帮我开发unittest模块来测试这个吗？我问这个问题是为了学习如何使用单元测试（我试图学习TTD：测试驱动开发）

到目前为止，我想出了这个...导入单元测试导入os.path

类TestFunctions（unittest.TestCase）：

def setUp(self):
    self.prompt = 'Enter filename: '

def test_get_file(self):

    # TODO make sure empty filename argument requests for new filename
    filename = find_author.get_valid_filename(self.prompt)
    self.assertTrue(<EXPRESSION?>)

    # TODO make sure valid filename returns the "filename"
    # TODO make sure invalid filename prompts that file does not exit and requests new filename 

如果name==“main”：unittest。main（）

Answer 1

扩展vgel对这个问题的回答，我需要模拟不同的连续输入，所以总是返回相同的字符串是不够的。我是这样解决的：

import module_being_tested

mock_raw_input_counter = 0
mock_raw_input_values = []
def mock_raw_input(s):
    global mock_raw_input_counter
    global mock_raw_input_values
    mock_raw_input_counter += 1
    return mock_raw_input_values[mock_raw_input_counter - 1]

module_being_tested.raw_input = mock_raw_input

然后在测试中，您希望使用此多输入功能：

def test_interactive_sequence_selection_dummy_2_99_0_Z_1(self):
    global mock_raw_input_counter
    global mock_raw_input_values
    mock_raw_input_counter = 0
    mock_raw_input_values = ["2", "99", "0", "Z", "1", ""]
    new_sequences = module_being_tested.interactive_sequence_selection(["A", "B", "C"], [None])
    self.assertEqual(new_sequences, ["C", "A", "B"])

这将模拟输入以下值：

2[ENTER]
99[ENTER]
0[ENTER]
Z[ENTER]
1[ENTER]
[ENTER] (no input, just pressing Enter)

（interactive_sequence_selection中的代码使用而s！= "":循环，要求用户输入各种值，并以空Enter按结束）

Answer 2

一种简单的方法是使用monkey patchraw\u input。

例如，在您的测试模块中（因为您应该将受测试者和测试者拆分为单独的文件），您可能有：

import module_being_tested
... run tests ...

在运行测试之前，您可以简单地执行以下操作：

import module_being_tested

def mock_raw_input(s):
    return 'data.txt'

module_being_tested.raw_input = mock_raw_input

... run tests ....

现在，当您的测试对象模块调用raw\u input时，它实际上将调用mock\u raw\u input，并且将始终获取数据。txt'返回。