xonsh/tests/test_execer.py

"""Tests the xonsh lexer."""
from __future__ import unicode_literals, print_function
import os
import sys
import ast

from xonsh.execer import Execer
from xonsh.tools import ON_WINDOWS

from tools import mock_xonsh_env

DEBUG_LEVEL = 0
EXECER = None

#
# Helpers
#

def setup():
    # only setup one parser
    global EXECER
    EXECER = Execer(debug_level=DEBUG_LEVEL)

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

def check_eval(input):
    with mock_xonsh_env(None):
        EXECER.debug_level = DEBUG_LEVEL
        EXECER.eval(input)

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

#
# Tests
#

if ON_WINDOWS:
    def test_win_ipconfig():
        yield (check_eval,
               os.environ['SYSTEMROOT'] + '\\System32\\ipconfig.exe /all')

    def test_ipconfig():
        yield check_eval, 'ipconfig /all'

else:
    def test_bin_ls():
        yield check_eval, '/bin/ls -l'

    def test_ls_dashl():
        yield check_eval, 'ls -l'

    def test_which_ls():
        yield check_eval, 'which ls'


def test_simple_func():
    code = ('def prompt():\n'
            "    return '{user}'.format(user='me')\n")
    yield check_parse, code

def test_simple_func_broken():
    code = ('def prompt():\n'
            "    return '{user}'.format(\n"
            "       user='me')\n")
    yield check_parse, code




if __name__ == '__main__':
    nose.runmodule()