import sys

from unittest.mock import patch
from pytest_relaxed import trap

from invocations.console import confirm


class confirm_:
    @patch("invocations.console.input", return_value="yes")
    def displays_question_with_yes_no_suffix(self, mock_input):
        confirm("Are you sure?")
        assert mock_input.call_args[0][0] == "Are you sure? [Y/n] "

    @patch("invocations.console.input")
    def returns_True_for_yeslike_responses(self, mock_input):
        for value in ("y", "Y", "yes", "YES", "yES", "Yes"):
            mock_input.return_value = value
            assert confirm("Meh") is True

    @patch("invocations.console.input")
    def returns_False_for_nolike_responses(self, mock_input):
        for value in ("n", "N", "no", "NO", "nO", "No"):
            mock_input.return_value = value
            assert confirm("Meh") is False

    @trap
    @patch("invocations.console.input", side_effect=["wat", "y"])
    def reprompts_on_bad_input(self, mock_input):
        assert confirm("O rly?") is True
        assert "I didn't understand you" in sys.stderr.getvalue()

    @patch("invocations.console.input", return_value="y")
    def suffix_changes_when_assume_yes_False(self, mock_input):
        confirm("Are you sure?", assume_yes=False)
        assert mock_input.call_args[0][0] == "Are you sure? [y/N] "

    @patch("invocations.console.input", return_value="")
    def default_on_empty_response_is_True_by_default(self, mock_input):
        assert confirm("Are you sure?") is True

    @patch("invocations.console.input", return_value="")
    def default_on_empty_response_is_False_if_assume_yes_False(
        self, mock_input
    ):
        assert confirm("Are you sure?", assume_yes=False) is False

    @patch("invocations.console.input", return_value=" y ")
    def whitespace_is_trimmed(self, mock_input):
        assert confirm("Are you sure?") is True