我正在尝试测试一个从 获取输入的函数,stdin
我目前正在使用以下内容进行测试:
cat /usr/share/dict/words | ./spellchecker.py
以测试自动化的名义,有什么方法pyunit
可以伪造输入raw_input()
?
我正在尝试测试一个从 获取输入的函数,stdin
我目前正在使用以下内容进行测试:
cat /usr/share/dict/words | ./spellchecker.py
以测试自动化的名义,有什么方法pyunit
可以伪造输入raw_input()
?
简短的回答是猴子补丁 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
从 python 3.3 开始,有一个unittest
名为 mock 的新子模块可以完全满足您的需要。对于那些使用 python 2.6 或更高版本的人,有一个 backport of mock
found 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 模块可能是您正在寻找的。
你可以做类似的事情
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,因为没有更多的文本可供阅读。
替换sys.stdin
为 的实例,并使用您希望通过返回的数据StringIO
加载该实例。此外,包含原始对象,因此在测试后恢复就像.StringIO
sys.stdin
sys.__stdin__
sys.stdin
sys.stdin
sys.stdin = sys.__stdin__
Fudge是一个很棒的 python 模拟模块,带有方便的装饰器,可以为您进行这样的修补,并具有自动清理功能。你应该检查一下。
您没有描述其中的代码类型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 美分。
如果您使用模拟模块(由 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**