25

我正在尝试测试一个从 获取输入的函数,stdin我目前正在使用以下内容进行测试:

cat /usr/share/dict/words | ./spellchecker.py

以测试自动化的名义,有什么方法pyunit可以伪造输入raw_input()

4

5 回答 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 回答