How can I mock user input (from “input” in 3.x, or “raw_input” in 2.x) for a unit test?

by

You can’t patch input but you can wrap it to use mock.patch(). Here is a solution:

from unittest.mock import patch
from unittest import TestCase


def get_input(text):
    return input(text)


def answer():
    ans = get_input('enter yes or no')
    if ans == 'yes':
        return 'you entered yes'
    if ans == 'no':
        return 'you entered no'


class Test(TestCase):

    # get_input will return 'yes' during this test
    @patch('yourmodule.get_input', return_value="yes")
    def test_answer_yes(self, input):
        self.assertEqual(answer(), 'you entered yes')

    @patch('yourmodule.get_input', return_value="no")
    def test_answer_no(self, input):
        self.assertEqual(answer(), 'you entered no')

Keep in mind that this snippet will only work in Python versions 3.3+

More Related Contents:

  1. python mocking raw input in unittests
  2. Can I patch a Python decorator before it wraps a function?
  3. Why python mock patch doesn’t work?
  4. Python mock multiple return values
  5. Python Mock object with method called multiple times
  6. How can one mock/stub python module like urllib
  7. Mocking a class: Mock() or patch()?
  8. Mock attributes in Python mock?
  9. Mock Python’s built in print function
  10. How to test or mock “if __name__ == ‘__main__'” contents
  11. Mocking Functions Using Python Mock
  12. Python mock Patch os.environ and return value
  13. Customizing unittest.mock.mock_open for iteration
  14. Unit testing a python app that uses the requests library
  15. Django: How to create a model dynamically just for testing
  16. Can I “fake” a package (or at least a module) in python for testing purposes?
  17. How to mock a readonly property with mock?
  18. How do I mock an open used in a with statement (using the Mock framework in Python)?
  19. Pytest fixture for a class through self not as method argument
  20. How to run Django’s test database only in memory?
  21. How to test Python 3.4 asyncio code?
  22. Continuing in Python’s unittest when an assertion fails
  23. Mocking boto3 S3 client method Python
  24. What is your favorite Python mocking library? [closed]
  25. Can Python’s unittest test in parallel, like nose can?
  26. Using mock patch to mock an instance method
  27. Running a specific test case in Django when your app has a tests directory
  28. Python Unit Testing: Automatically Running the Debugger when a test fails
  29. python: mock a module
  30. How can I specify a database for Django Tests to use instead of having it build it every time?

Leave a Comment