mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 16:34:47 +01:00
113 lines
3.4 KiB
Python
113 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Tests sample inputs to PTK multiline and checks parser response"""
|
|
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from prompt_toolkit.interface import CommandLineInterface
|
|
from prompt_toolkit.document import Document
|
|
from prompt_toolkit.buffer import Buffer, AcceptAction
|
|
from xonsh.environ import Env
|
|
from xonsh.tools import ON_WINDOWS
|
|
|
|
def setup_module():
|
|
global indent_
|
|
global buffer
|
|
global bufaccept
|
|
global cli
|
|
global carriage_return
|
|
global builtins
|
|
|
|
import builtins
|
|
|
|
builtins.__xonsh_env__ = Env()
|
|
builtins.__xonsh_env__['INDENT'] = ' '
|
|
|
|
from xonsh.ptk.key_bindings import carriage_return
|
|
|
|
indent_ = ' '
|
|
buffer = Buffer()
|
|
bufaccept = MagicMock(name='accept', spec=AcceptAction)
|
|
cli = MagicMock(name='cli', spec=CommandLineInterface)
|
|
buffer.accept_action = bufaccept
|
|
|
|
def teardown_module():
|
|
global indent_
|
|
global buffer
|
|
global bufaccept
|
|
global cli
|
|
global carriage_return
|
|
global builtins
|
|
|
|
del indent_
|
|
del buffer
|
|
del bufaccept
|
|
del cli
|
|
del carriage_return
|
|
del builtins
|
|
|
|
def test_colon_indent():
|
|
document = Document('for i in range(5):')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert buffer.document.current_line == indent_
|
|
|
|
def test_dedent():
|
|
document = Document('\n'+indent_+'pass')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert buffer.document.current_line == ''
|
|
|
|
document = Document('\n'+2*indent_+'continue')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert buffer.document.current_line == indent_
|
|
|
|
def test_nodedent():
|
|
'''don't dedent if first line of buffer'''
|
|
mock = MagicMock(return_value = True)
|
|
with patch('xonsh.ptk.key_bindings.can_compile', mock):
|
|
document = Document('pass')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert bufaccept.mock_calls is not None
|
|
|
|
|
|
mock = MagicMock(return_value = True)
|
|
with patch('xonsh.ptk.key_bindings.can_compile', mock):
|
|
document = Document(indent_+'pass')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert bufaccept.mock_calls is not None
|
|
|
|
def test_continuation_line():
|
|
document = Document('\nsecond line')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert buffer.document.current_line == ''
|
|
|
|
def test_trailing_slash():
|
|
mock = MagicMock(return_value = True)
|
|
with patch('xonsh.ptk.key_bindings.can_compile', mock):
|
|
document = Document('this line will \\')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
if not ON_WINDOWS:
|
|
assert buffer.document.current_line == ''
|
|
else:
|
|
assert bufaccept.mock_calls is not None
|
|
|
|
def test_cant_compile_newline():
|
|
mock = MagicMock(return_value = False)
|
|
with patch('xonsh.ptk.key_bindings.can_compile', mock):
|
|
document = Document('for i in (1, 2, ')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert buffer.document.current_line == ''
|
|
|
|
def test_can_compile_and_executes():
|
|
mock = MagicMock(return_value = True)
|
|
with patch('xonsh.ptk.key_bindings.can_compile', mock):
|
|
document = Document('ls')
|
|
buffer.set_document(document)
|
|
carriage_return(buffer, cli)
|
|
assert bufaccept.mock_calls is not None
|