web-dev-qa-db-ja.com

pyunitのstdinへの入力をシミュレートするにはどうすればよいですか？

stdinから入力を受け取る関数をテストしようとしています。これは、現在次のようなものでテストしています。

_cat /usr/share/dict/words | ./spellchecker.py
_

テスト自動化の名の下に、pyunitがraw_input()への入力を偽造できる方法はありますか？

pythonlinuxpython-unittest
19
bejar37

簡単な答えは、 モンキーパッチraw_input()です。

Pythonでリダイレクトされたstdinを表示する方法 への回答にはいくつかの良い例があります

これは、プロンプトを破棄して必要なものを返すlambdaを使用した単純で簡単な例です。

テスト中のシステム

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
22
Johnsyweb

更新-unittest.mock.patchを使用

python 3.3以降、必要なことを正確に実行するmockと呼ばれるunittestの新しいサブモジュールがあります。python 2.6以上を使用している場合mockのバックポートが見つかりました ここ 。

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は、呼び出されるたびにsimulatedInput.txtから読み取るようになりました。 simulatedInputの内容が

hello
bob

次に、raw_inputの最初の呼び出しは「hello」を返し、2番目の「bob」と3番目の呼び出しは、読み取るテキストがなくなったためにEOFErrorをスローします。

12
Dunes

sys.stdinをStringIOのインスタンスに置き換え、StringIOインスタンスにsys.stdinを介して返されるデータをロードします。また、sys.__stdin__には元のsys.stdinオブジェクトが含まれているため、テスト後にsys.stdinを復元するのはsys.stdin = sys.__stdin__と同じくらい簡単です。

Fudge は素晴らしいpythonモックモジュールで、このようなパッチを適用するための便利なデコレータと自動クリーンアップを備えています。チェックしてください。

3
Michael Kent

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セント。

2
Pavel Repin

モックモジュール （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**
1
Serge