xonsh/tests/tools.py

# -*- coding: utf-8 -*-
"""Tests the xonsh lexer."""
from __future__ import unicode_literals, print_function
import sys
import glob
import builtins
import platform
import subprocess
from collections import defaultdict
from contextlib import contextmanager

import pytest

from xonsh.built_ins import ensure_list_of_strs
from xonsh.environ import Env
builtins.__xonsh_env__ = Env()
from xonsh.base_shell import BaseShell
from xonsh.execer import Execer
from xonsh.tools import XonshBlockError


VER_3_4 = (3, 4)
VER_3_5 = (3, 5)
VER_MAJOR_MINOR = sys.version_info[:2]
VER_FULL = sys.version_info[:3]
ON_DARWIN = (platform.system() == 'Darwin')
ON_WINDOWS = (platform.system() == 'Windows')


skip_if_py34 = pytest.mark.skipif(VER_MAJOR_MINOR < VER_3_5,
                                   reason="Py3.5+ only test")


skip_if_on_windows = pytest.mark.skipif(ON_WINDOWS, reason='Unix stuff')


skip_if_on_unix = pytest.mark.skipif(not ON_WINDOWS, reason='Windows stuff')


def sp(cmd):
    return subprocess.check_output(cmd, universal_newlines=True)

class DummyStyler():
    styles = defaultdict(None.__class__)

class DummyBaseShell(BaseShell):

    def __init__(self):
        self.styler = DummyStyler()


class DummyShell:
    def settitle(self):
        pass

    _shell = None

    @property
    def shell(self):
        if self._shell is None:
            self._shell = DummyBaseShell()
        return self._shell


#
# Execer tools
#

DEBUG_LEVEL = 0
EXECER = None

def execer_setup():
    # only setup one parser
    global EXECER
    if EXECER is None:
        EXECER = Execer(debug_level=DEBUG_LEVEL, login=False)

def check_exec(input, xonsh_env, **kwargs):
    # with mock_xonsh_env(None):
    if not input.endswith('\n'):
        input += '\n'
    EXECER.debug_level = DEBUG_LEVEL
    EXECER.exec(input, **kwargs)

def check_eval(input):
    env = {'AUTO_CD': False, 'XONSH_ENCODING' :'utf-8',
           'XONSH_ENCODING_ERRORS': 'strict', 'PATH': []}
    if ON_WINDOWS:
        env['PATHEXT'] = ['.COM', '.EXE', '.BAT', '.CMD']
    with mock_xonsh_env(env):
        EXECER.debug_level = DEBUG_LEVEL
        EXECER.eval(input)

def check_parse(input):
    with mock_xonsh_env(None):
        EXECER.debug_level = DEBUG_LEVEL
        tree = EXECER.parse(input, ctx=None)
    return tree