xonsh/tests/test_foreign_shells.py
pre-commit-ci[bot] 66c0490d37
[pre-commit.ci] pre-commit autoupdate (#5271)
* [pre-commit.ci] pre-commit autoupdate

updates:
- [github.com/psf/black: 23.12.1 → 24.1.1](https://github.com/psf/black/compare/23.12.1...24.1.1)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2024-01-30 12:23:50 +01:00

112 lines
2.9 KiB
Python

"""Tests foreign shells."""
import os
import subprocess
import pytest # noqa F401
from xonsh.foreign_shells import foreign_shell_data, parse_aliases, parse_env
from xonsh.pytest.tools import skip_if_on_unix, skip_if_on_windows
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"],
"z": ["echo", "True", "&&", "echo", "Next", "||", "echo", "False"],
}
s = (
"some garbage\n"
"__XONSH_ALIAS_BEG__\n"
"alias x='yes -1'\n"
"alias y='echo no'\n"
"alias z='echo True && \\\n echo Next || \\\n echo False'\n" # noqa: E261,W605
"__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