• Search by Module
• Search by Word
• Project Search
• Top Python APIs
• Popular Projects

• Java
• C++
• Python
• Scala
• Blog

• clikit-master
  • .github
    • workflows
      • push.yml
  • src
    • clikit
      • resolver
        • default_resolver.py
        • resolve_result.py
        • help_resolver.py
        • __init__.py
      • utils
        • terminal.py
        • string.py
        • _compat.py
        • __init__.py
        • command.py
        • time.py
      • console_application.py
      • formatter
        • null_formatter.py
        • ansi_formatter.py
        • plain_formatter.py
        • __init__.py
        • default_style_set.py
      • args
        • argv_args.py
        • token_parser.py
        • inputs
          • __init__.py
        • default_args_parser.py
        • __init__.py
        • string_args.py
      • ui
        • help
          • abstract_help.py
          • command_help.py
          • __init__.py
          • application_help.py
        • style
          • alignment.py
          • __init__.py
          • table_style.py
          • border_style.py
        • layout
          • __init__.py
          • block_layout.py
        • components
          • empty_line.py
          • paragraph.py
          • question.py
          • progress_indicator.py
          • table.py
          • exception_trace.py
          • name_version.py
          • progress_bar.py
          • confirmation_question.py
          • border_util.py
          • __init__.py
          • cell_wrapper.py
          • choice_question.py
          • labeled_paragraph.py
        • alignment
          • __init__.py
          • label_alignment.py
        • rectangle.py
        • component.py
        • __init__.py
      • api
        • resolver
          • command_resolver.py
          • exceptions.py
          • __init__.py
          • resolved_command.py
        • exceptions.py
        • formatter
          • formatter.py
          • style_set.py
          • __init__.py
          • style.py
        • args
          • raw_args.py
          • exceptions.py
          • format
            • args_format_builder.py
            • command_option.py
            • option.py
            • argument.py
            • abstract_option.py
            • args_format.py
            • __init__.py
            • command_name.py
          • args_parser.py
          • __init__.py
          • args.py
        • application
          • application.py
          • __init__.py
        • config
          • application_config.py
          • config.py
          • command_config.py
          • __init__.py
        • event
          • pre_handle_event.py
          • config_event.py
          • pre_resolve_event.py
          • console_events.py
          • event_dispatcher.py
          • event.py
          • __init__.py
        • io
          • output.py
          • flags.py
          • input_stream.py
          • input.py
          • io.py
          • __init__.py
          • output_stream.py
          • section_output.py
          • io_exception.py
        • __init__.py
        • command
          • exceptions.py
          • __init__.py
          • command.py
          • command_collection.py
      • config
        • default_application_config.py
        • __init__.py
      • adapter
        • __init__.py
        • style_converter.py
      • io
        • null_io.py
        • output_stream
          • error_output_stream.py
          • standard_output_stream.py
          • null_output_stream.py
          • buffered_output_stream.py
          • __init__.py
          • stream_output_stream.py
        • console_io.py
        • __init__.py
        • buffered_io.py
        • input_stream
          • stream_input_stream.py
          • standard_input_stream.py
          • string_input_stream.py
          • null_input_stream.py
          • __init__.py
      • __init__.py
      • handler
        • help
          • __init__.py
          • help_text_handler.py
          • help_handler.py
        • callback_handler.py
        • __init__.py
  • .pre-commit-config.yaml
  • LICENSE
  • CHANGELOG.md
  • README.md
  • tests
    • utils
      • __init__.py
      • test_command.py
      • test_terminal.py
    • args
      • test_argv_args.py
      • test_string_args.py
      • __init__.py
      • test_default_args_parser.py
    • ui
      • help
        • test_application_help.py
        • test_command_help.py
        • __init__.py
      • components
        • test_exception_trace.py
        • test_question.py
        • test_confirmation_question.py
        • test_progress_indicator.py
        • test_choice_question.py
        • test_progress_bar.py
        • test_table.py
        • __init__.py
        • helpers.py
      • __init__.py
    • test_console_aplication.py
    • api
      • args
        • format
          • test_args_format_builder.py
          • test_option.py
          • __init__.py
          • test_argument.py
        • __init__.py
      • event
        • test_event_dispatcher.py
        • __init__.py
        • test_event.py
      • io
        • test_section_output.py
        • test_input.py
        • __init__.py
      • __init__.py
    • __init__.py
    • handler
      • help
        • test_help_option.py
        • __init__.py
        • test_help_text_handler.py
      • __init__.py
    • conftest.py
  • pyproject.toml
  • .gitignore
  • tox.ini

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import os
import subprocess

import pytest

from clikit.ui.components import Question


def has_tty_available():
    devnull = open(os.devnull, "w")
    exit_code = subprocess.call(["stty", "2"], stdout=devnull, stderr=devnull)

    return exit_code == 0


TTY_AVAILABLE = has_tty_available()


def test_ask(io):
    question = Question("What time is it?", "2PM")
    io.set_input("\n8AM\n")

    assert "2PM" == question.ask(io)

    io.clear_error()

    assert "8AM" == question.ask(io)
    assert "What time is it? " == io.fetch_error()


@pytest.mark.skipif(
    not TTY_AVAILABLE, reason="`stty` is required to test hidden response functionality"
)
def test_ask_hidden_response(io):
    question = Question("What time is it?", "2PM")
    question.hide()
    io.set_input("8AM\n")

    assert "8AM" == question.ask(io)
    assert "What time is it? " == io.fetch_error()


def test_ask_and_validate(io):
    error = "This is not a color!"

    def validator(color):
        if color not in ["white", "black"]:
            raise Exception(error)

        return color

    question = Question("What color was the white horse of Henry IV?", "white")
    question.set_validator(validator)
    question.set_max_attempts(2)

    io.set_input("\nblack\n")
    assert "white" == question.ask(io)
    assert "black" == question.ask(io)

    io.set_input("green\nyellow\norange\n")

    with pytest.raises(Exception) as e:
        question.ask(io)

    assert error == str(e.value)


def test_no_interaction(io):
    io.set_interactive(False)

    question = Question("Do you have a job?", "not yet")
    assert "not yet" == question.ask(io)


def test_ask_question_with_special_characters(io):
    question = Question("What time is it, Sébastien?", "2PMë")
    io.set_input("\n")

    assert "2PMë" == question.ask(io)
    assert "What time is it, Sébastien? " == io.fetch_error()