mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 08:24:40 +01:00
c2d25ac251
* chore: add pyupgrade * refactor: upgrade code to py3.7+ ran `pre-commit run pyupgrade -a` while excluding changes to ply * fix: flake errors
106 lines
2.7 KiB
Python
106 lines
2.7 KiB
Python
"""Tests foreign shells."""
|
|
import os
|
|
import subprocess
|
|
|
|
import pytest # noqa F401
|
|
from tools import skip_if_on_windows, skip_if_on_unix
|
|
|
|
from xonsh.foreign_shells import foreign_shell_data, parse_env, parse_aliases
|
|
|
|
|
|
def test_parse_env():
|
|
exp = {"X": "YES", "Y": "NO"}
|
|
s = (
|
|
"some garbage\n"
|
|
"__XONSH_ENV_BEG__\n"
|
|
"Y=NO\n"
|
|
"X=YES\n"
|
|
"__XONSH_ENV_END__\n"
|
|
"more filth"
|
|
)
|
|
obs = parse_env(s)
|
|
assert exp == obs
|
|
|
|
|
|
def test_parse_env_newline():
|
|
exp = {"X": "YES", "Y": "NO", "PROMPT": "why\nme "}
|
|
s = (
|
|
"some garbage\n"
|
|
"__XONSH_ENV_BEG__\n"
|
|
"Y=NO\n"
|
|
"PROMPT=why\nme \n"
|
|
"X=YES\n"
|
|
"__XONSH_ENV_END__\n"
|
|
"more filth"
|
|
)
|
|
obs = parse_env(s)
|
|
assert exp == obs
|
|
|
|
|
|
def test_parse_env_equals():
|
|
exp = {"X": "YES", "Y": "NO", "LS_COLORS": "*.tar=5"}
|
|
s = (
|
|
"some garbage\n"
|
|
"__XONSH_ENV_BEG__\n"
|
|
"Y=NO\n"
|
|
"LS_COLORS=*.tar=5\n"
|
|
"X=YES\n"
|
|
"__XONSH_ENV_END__\n"
|
|
"more filth"
|
|
)
|
|
obs = parse_env(s)
|
|
assert exp == obs
|
|
|
|
|
|
def test_parse_aliases():
|
|
exp = {"x": ["yes", "-1"], "y": ["echo", "no"]}
|
|
s = (
|
|
"some garbage\n"
|
|
"__XONSH_ALIAS_BEG__\n"
|
|
"alias x='yes -1'\n"
|
|
"alias y='echo no'\n"
|
|
"__XONSH_ALIAS_END__\n"
|
|
"more filth"
|
|
)
|
|
obs = parse_aliases(s, "bash")
|
|
assert exp == obs
|
|
|
|
|
|
@skip_if_on_windows
|
|
def test_foreign_bash_data():
|
|
expenv = {"EMERALD": "SWORD", "MIGHTY": "WARRIOR"}
|
|
expaliases = {"l": ["ls", "-CF"], "la": ["ls", "-A"], "ll": ["ls", "-a", "-lF"]}
|
|
rcfile = os.path.join(os.path.dirname(__file__), "bashrc.sh")
|
|
try:
|
|
obsenv, obsaliases = foreign_shell_data(
|
|
"bash", currenv=(), extra_args=("--rcfile", rcfile), safe=False
|
|
)
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
|
return
|
|
for key, expval in expenv.items():
|
|
assert expval == obsenv.get(key, False)
|
|
for key, expval in expaliases.items():
|
|
assert expval == obsaliases.get(key, False)
|
|
|
|
|
|
@skip_if_on_unix
|
|
def test_foreign_cmd_data():
|
|
env = (("ENV_TO_BE_REMOVED", "test"),)
|
|
batchfile = os.path.join(os.path.dirname(__file__), "batch.bat")
|
|
source_cmd = f'call "{batchfile}"\necho off'
|
|
try:
|
|
obsenv, _ = foreign_shell_data(
|
|
"cmd",
|
|
prevcmd=source_cmd,
|
|
currenv=env,
|
|
interactive=False,
|
|
sourcer="call",
|
|
envcmd="set",
|
|
use_tmpfile=True,
|
|
safe=False,
|
|
)
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
|
return
|
|
assert "ENV_TO_BE_ADDED" in obsenv
|
|
assert obsenv["ENV_TO_BE_ADDED"] == "Hallo world"
|
|
assert "ENV_TO_BE_REMOVED" not in obsenv
|