# -*- coding: utf-8 -*-
from beehive import then
from beehive4cmd0.command_steps import \
    step_command_output_should_contain_text, \
    step_command_output_should_not_contain_text

# -- CONSTANTS:
# ANSI CONTROL SEQUENCE INTRODUCER (CSI).
CSI = u"\x1b["


@then(u'the command output should contain ANSI escape sequences')
def step_command_ouput_should_contain_ansi_sequences(context):
    step_command_output_should_contain_text(context, CSI)


@then(u'the command output should not contain any ANSI escape sequences')
def step_command_ouput_should_not_contain_ansi_sequences(context):
    step_command_output_should_not_contain_text(context, CSI)
