我正在尝试测试一个从 获取输入的函数，stdin我目前正在使用以下内容进行测试：
cat /usr/share/dict/words | ./spellchecker.py
以测试自动化的名义，有什么方法pyunit可以伪造输入raw_input()？
问问题
24677 次
5 回答
23
简短的回答是猴子补丁 raw_input()。
如何在 Python 中显示重定向的标准输入的答案中有一些很好的例子？
这是一个简单的例子，它使用 alambda丢弃提示并返回我们想要的内容。
被测系统
cat ./name_getter.py
#!/usr/bin/env python
class NameGetter(object):
def get_name(self):
self.name = raw_input('What is your name? ')
def greet(self):
print 'Hello, ', self.name, '!'
def run(self):
self.get_name()
self.greet()
if __name__ == '__main__':
ng = NameGetter()
ng.run()
$ echo Derek | ./name_getter.py
What is your name? Hello, Derek !
测试用例：
$ cat ./t_name_getter.py
#!/usr/bin/env python
import unittest
import name_getter
class TestNameGetter(unittest.TestCase):
def test_get_alice(self):
name_getter.raw_input = lambda _: 'Alice'
ng = name_getter.NameGetter()
ng.get_name()
self.assertEquals(ng.name, 'Alice')
def test_get_bob(self):
name_getter.raw_input = lambda _: 'Bob'
ng = name_getter.NameGetter()
ng.get_name()
self.assertEquals(ng.name, 'Bob')
if __name__ == '__main__':
unittest.main()
$ ./t_name_getter.py -v
test_get_alice (__main__.TestNameGetter) ... ok
test_get_bob (__main__.TestNameGetter) ... ok
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
于 2011-06-07T21:47:49.673 回答
18
更新——使用 unittest.mock.patch
从 python 3.3 开始，有一个unittest名为 mock 的新子模块可以完全满足您的需要。对于那些使用 python 2.6 或更高版本的人，有一个 backport of mockfound here。
import unittest
from unittest.mock import patch
import module_under_test
class MyTestCase(unittest.TestCase):
def setUp(self):
# raw_input is untouched before test
assert module_under_test.raw_input is __builtins__.raw_input
def test_using_with(self):
input_data = "123"
expected = int(input_data)
with patch.object(module_under_test, "raw_input", create=True,
return_value=expected):
# create=True is needed as raw_input is not in the globals of
# module_under_test, but actually found in __builtins__ .
actual = module_under_test.function()
self.assertEqual(expected, actual)
@patch.object(module_under_test, "raw_input", create=True)
def test_using_decorator(self, raw_input):
raw_input.return_value = input_data = "123"
expected = int(input_data)
actual = module_under_test.function()
self.assertEqual(expected, actual)
def tearDown(self):
# raw input is restored after test
assert module_under_test.raw_input is __builtins__.raw_input
if __name__ == "__main__":
unittest.main()
# where module_under_test.function is:
def function():
return int(raw_input("prompt> "))
上一个答案——替换 sys.stdin
我认为 sys 模块可能是您正在寻找的。
你可以做类似的事情
import sys
# save actual stdin in case we need it again later
stdin = sys.stdin
sys.stdin = open('simulatedInput.txt','r')
# or whatever else you want to provide the input eg. StringIO
每当调用 raw_input 时，它都会从模拟输入.txt 中读取。如果模拟输入的内容是
hello
bob
那么对 raw_input 的第一次调用将返回“hello”，第二个“bob”和第三个将抛出 EOFError，因为没有更多的文本可供阅读。
于 2011-06-07T21:59:05.457 回答
3
替换sys.stdin为 的实例，并使用您希望通过返回的数据StringIO加载该实例。此外，包含原始对象，因此在测试后恢复就像.StringIOsys.stdinsys.__stdin__sys.stdinsys.stdinsys.stdin = sys.__stdin__
Fudge是一个很棒的 python 模拟模块，带有方便的装饰器，可以为您进行这样的修补，并具有自动清理功能。你应该检查一下。
于 2011-06-07T23:30:57.910 回答
3
您没有描述其中的代码类型spellchecker.py，这让我可以自由推测。
假设它是这样的：
import sys
def check_stdin():
# some code that uses sys.stdin
为了提高check_stdin函数的可测试性，我建议像这样重构它：
def check_stdin():
return check(sys.stdin)
def check(input_stream):
# same as original code, but instead of
# sys.stdin it is written it terms of input_stream.
现在你的大部分逻辑都在check起作用，你可以手工制作你能想象到的任何输入来正确测试它，而无需处理stdin.
我的 2 美分。
于 2011-06-08T03:42:21.107 回答
2
如果您使用模拟模块（由 Michael Foord 编写），为了模拟raw_input函数，您可以使用如下语法：
@patch('src.main.raw_input', create=True, new=MagicMock(return_value='y'))
def test_1(self):
method_we_try_to_test(); # method or function that calls **raw_input**
于 2013-09-04T10:47:43.707 回答