File: test_confirmation_question.py

package info (click to toggle)
python-cleo 2.2.1-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,120 kB
  • sloc: python: 8,293; makefile: 22; sh: 2
file content (38 lines) | stat: -rw-r--r-- 1,017 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from __future__ import annotations

from typing import TYPE_CHECKING

import pytest

from cleo.ui.confirmation_question import ConfirmationQuestion


if TYPE_CHECKING:
    from cleo.io.buffered_io import BufferedIO


@pytest.mark.parametrize(
    ("input", "expected", "default"),
    [
        ("", True, True),
        ("", False, False),
        ("y", True, True),
        ("yes", True, True),
        ("n", False, True),
        ("no", False, True),
    ],
)
def test_ask(io: BufferedIO, input: str, expected: bool, default: bool) -> None:
    io.set_user_input(f"{input}\n")
    question = ConfirmationQuestion("Do you like French fries?", default)
    assert question.ask(io) == expected


def test_ask_with_custom_answer(io: BufferedIO) -> None:
    io.set_user_input("j\ny\n")

    question = ConfirmationQuestion("Do you like French fries?", False, r"(?i)^(j|y)")
    assert question.ask(io)

    question = ConfirmationQuestion("Do you like French fries?", False, r"(?i)^(j|y)")
    assert question.ask(io)