diff options
Diffstat (limited to 'tests/cli/test_input.py')
| -rw-r--r-- | tests/cli/test_input.py | 99 |
1 files changed, 52 insertions, 47 deletions
diff --git a/tests/cli/test_input.py b/tests/cli/test_input.py index 8efb1f5..2f15f25 100644 --- a/tests/cli/test_input.py +++ b/tests/cli/test_input.py @@ -9,12 +9,11 @@ from snakeoil.test.argparse_helpers import FakeStreamFormatter @pytest.fixture def mocked_input(): - with mock.patch('builtins.input') as mocked_input: + with mock.patch("builtins.input") as mocked_input: yield mocked_input class TestUserQuery: - @pytest.fixture(autouse=True) def __setup(self): self.out = FakeStreamFormatter() @@ -22,98 +21,104 @@ class TestUserQuery: self.query = partial(userquery, out=self.out, err=self.err) def test_default_answer(self, mocked_input): - mocked_input.return_value = '' - assert self.query('foo') == True + mocked_input.return_value = "" + assert self.query("foo") == True def test_tuple_prompt(self, mocked_input): - mocked_input.return_value = '' - prompt = 'perhaps a tuple' + mocked_input.return_value = "" + prompt = "perhaps a tuple" assert self.query(tuple(prompt.split())) == True - output = ''.join(prompt.split()) - assert self.out.get_text_stream().strip().split('\n')[0][:len(output)] == output + output = "".join(prompt.split()) + assert ( + self.out.get_text_stream().strip().split("\n")[0][: len(output)] == output + ) def test_no_default_answer(self, mocked_input): responses = { - 'a': ('z', 'Yes'), - 'b': ('y', 'No'), + "a": ("z", "Yes"), + "b": ("y", "No"), } # no default answer returns None for empty input - mocked_input.return_value = '' - assert self.query('foo', responses=responses) == None - mocked_input.return_value = 'a' - assert self.query('foo', responses=responses) == 'z' - mocked_input.return_value = 'b' - assert self.query('foo', responses=responses) == 'y' + mocked_input.return_value = "" + assert self.query("foo", responses=responses) == None + mocked_input.return_value = "a" + assert self.query("foo", responses=responses) == "z" + mocked_input.return_value = "b" + assert self.query("foo", responses=responses) == "y" def test_ambiguous_input(self, mocked_input): responses = { - 'a': ('z', 'Yes'), - 'A': ('y', 'No'), + "a": ("z", "Yes"), + "A": ("y", "No"), } - mocked_input.return_value = 'a' + mocked_input.return_value = "a" with pytest.raises(NoChoice): - self.query('foo', responses=responses) - error_output = self.err.get_text_stream().strip().split('\n')[1] - expected = 'Response %r is ambiguous (%s)' % ( - mocked_input.return_value, ', '.join(sorted(responses.keys()))) + self.query("foo", responses=responses) + error_output = self.err.get_text_stream().strip().split("\n")[1] + expected = "Response %r is ambiguous (%s)" % ( + mocked_input.return_value, + ", ".join(sorted(responses.keys())), + ) assert error_output == expected def test_default_correct_input(self, mocked_input): - for input, output in (('no', False), - ('No', False), - ('yes', True), - ('Yes', True)): + for input, output in ( + ("no", False), + ("No", False), + ("yes", True), + ("Yes", True), + ): mocked_input.return_value = input - assert self.query('foo') == output + assert self.query("foo") == output def test_default_answer_no_matches(self, mocked_input): - mocked_input.return_value = '' + mocked_input.return_value = "" with pytest.raises(ValueError): - self.query('foo', default_answer='foo') + self.query("foo", default_answer="foo") assert self.out.stream == [] def test_custom_default_answer(self, mocked_input): - mocked_input.return_value = '' - assert self.query('foo', default_answer=False) == False + mocked_input.return_value = "" + assert self.query("foo", default_answer=False) == False def test_eof_nochoice(self, mocked_input): # user hits ctrl-d mocked_input.side_effect = EOFError with pytest.raises(NoChoice): - self.query('foo') - output = self.out.get_text_stream().strip().split('\n')[1] - expected = 'Not answerable: EOF on STDIN' + self.query("foo") + output = self.out.get_text_stream().strip().split("\n")[1] + expected = "Not answerable: EOF on STDIN" assert output == expected def test_stdin_closed_nochoice(self, mocked_input): - mocked_input.side_effect = IOError(errno.EBADF, '') + mocked_input.side_effect = IOError(errno.EBADF, "") with pytest.raises(NoChoice): - self.query('foo') - output = self.out.get_text_stream().strip().split('\n')[1] - expected = 'Not answerable: STDIN is either closed, or not readable' + self.query("foo") + output = self.out.get_text_stream().strip().split("\n")[1] + expected = "Not answerable: STDIN is either closed, or not readable" assert output == expected def test_unhandled_ioerror(self, mocked_input): - mocked_input.side_effect = IOError(errno.ENODEV, '') + mocked_input.side_effect = IOError(errno.ENODEV, "") with pytest.raises(IOError): - self.query('foo') + self.query("foo") def test_bad_choice_limit(self, mocked_input): # user hits enters a bad choice 3 times in a row - mocked_input.return_value = 'bad' + mocked_input.return_value = "bad" with pytest.raises(NoChoice): - self.query('foo') + self.query("foo") assert mocked_input.call_count == 3 - output = self.err.get_text_stream().strip().split('\n')[1] + output = self.err.get_text_stream().strip().split("\n")[1] expected = "Sorry, response %r not understood." % (mocked_input.return_value,) assert output == expected def test_custom_choice_limit(self, mocked_input): # user hits enters a bad choice 5 times in a row - mocked_input.return_value = 'haha' + mocked_input.return_value = "haha" with pytest.raises(NoChoice): - self.query('foo', limit=5) + self.query("foo", limit=5) assert mocked_input.call_count == 5 - output = self.err.get_text_stream().strip().split('\n')[1] + output = self.err.get_text_stream().strip().split("\n")[1] expected = "Sorry, response %r not understood." % (mocked_input.return_value,) assert output == expected |