Python单元测试If语句的预期输出

Question

我目前正在创建一些单元测试。我对他们来说相当陌生，只是想弄湿我的脚。因此，我试图运行的当前测试是根据用户输入检查预期输出。因此，我会用某种类型的值修补输入，然后检查是否在最后收到stdout消息。听起来有点混乱，但我希望有人能帮忙。这是我的运行代码。

def main():


  Attack = input("Are we being attacked?!")

  if(Attack == "yes"):
    print("We are being attacked! Attack Back!")

因此，在上面的示例中，我将测试print语句，因为我将使用yes值修补用户输入。这是我的测试套件

import unittest
from unittest.mock import patch
import io
import sys

from RunFile import main

class GetInputTest(unittest.TestCase):

  @patch('builtins.input', return_value='yes')
  def test_output(self):
      saved_stdout = sys.stdout
      try:
          out = io.StringIO()
          sys.stdout = out
          main()
          output = out.getvalue().strip()
          self.assertEqual(output, "We are being attacked! Attack Back!")
      finally:
          sys.stdout = saved_stdout


if __name__ == "__main__":
  unittest.main()

所以这显然行不通。那我错过了什么？提前谢谢大家！

编辑：这是我运行测试时收到的错误消息。我理解这个错误，只是不知道该如何去修复它。

Error
Traceback (most recent call last):
  File "C:\Python33\lib\unittest\mock.py", line 1087, in patched
    return func(*args, **keywargs)
TypeError: test_output() takes 1 positional argument but 2 were given

Answer 1

除了@chepner的答案，您还需要使用unittest。TestCase的断言方法而不是断言自己（双关语）

class TestStuff(unittest.TestCase):
    @patch('builtins.input', return_value='yes')
    def test_output(self, new_input):
        try:
            out = io.StringIO()
            sys.stdout = out
            main()
            output = out.getvalue().strip()
            self.assertEqual(output, "We are being attacked! Attack Back!")
        finally:
            sys.stdout = saved_stdout

然而，这可能不是做你想做的事情的最佳方式。你可以修补多个内置，你知道！

class TestStuff(unittest.TestCase):
    @patch('builtins.input', return_value='yes')
    @patch('builtins.print')
    def test_output(self, new_print, new_input):
        # the mocked functions are passed in opposite order
        # to where they're decorated
        main()
        new_print.assert_called_with("We are being attacked! Attack Back!")

如果装饰师很吓人，你甚至可以：

class TestStuff(unittest.TestCase):
    def test_output(self):
        with patch('builtins.input', return_value='yes'), \
             patch('builtins.print') as new_print:
            main()
            new_print.assert_called_with("We are being attacked! Attack Back!")

Answer 2

由patch修饰的函数将使用Mock作为附加参数。你需要

@patch('builtins.input', return_value='yes')
def test_output(self, m):

其中第二个参数m将引用调用test\u output时替换input的Mock对象。

从pydocunittest.mock下面的补丁：

如果patch用作修饰符，并且省略了new，则创建的模拟将作为额外参数传递给修饰函数。