xonsh/tests/aliases/test_source.py
Gil Forsyth fde07cda5f
refactor(aliases): source foreign shell funcs without interactive mode (#5344)
Co-authored-by: Noortheen Raja <jnoortheen@gmail.com>
2024-04-18 19:51:39 +05:30

82 lines
2 KiB
Python

import builtins
import os.path
from contextlib import contextmanager
from unittest.mock import MagicMock
import pytest
from xonsh.aliases import make_default_aliases, source_alias
@pytest.fixture
def mockopen(xession, monkeypatch):
@contextmanager
def mocked_open(fpath, *args, **kwargs):
yield MagicMock(read=lambda: fpath)
monkeypatch.setattr(builtins, "open", mocked_open)
@pytest.fixture
def mocked_execx_checker(xession, monkeypatch):
checker = []
def mocked_execx(src, *args, **kwargs):
checker.append(src.strip())
monkeypatch.setattr(xession.builtins, "execx", mocked_execx)
return checker
def test_source_current_dir(mockopen, monkeypatch, mocked_execx_checker):
monkeypatch.setattr(os.path, "isfile", lambda x: True)
source_alias(["foo", "bar"])
assert mocked_execx_checker == ["foo", "bar"]
def test_source_path(mockopen, mocked_execx_checker, patch_locate_binary, xession):
patch_locate_binary(xession.commands_cache)
source_alias(["foo", "bar"])
path_foo = os.path.join("bin", "foo")
path_bar = os.path.join("bin", "bar")
assert mocked_execx_checker[0].endswith(path_foo)
assert mocked_execx_checker[1].endswith(path_bar)
@pytest.mark.parametrize(
"alias",
[
"source-bash",
"source-zsh",
],
)
def test_source_foreign_fn_parser(alias, xession):
aliases = make_default_aliases()
source_bash = aliases[alias]
positionals = [act.dest for act in source_bash.parser._get_positional_actions()]
options = [act.dest for act in source_bash.parser._get_optional_actions()]
assert positionals == ["files_or_code"]
assert options == [
"help",
"interactive",
"login",
"envcmd",
"aliascmd",
"extra_args",
"safe",
"prevcmd",
"postcmd",
"funcscmd",
"sourcer",
"use_tmpfile",
"seterrprevcmd",
"seterrpostcmd",
"overwrite_aliases",
"suppress_skip_message",
"show",
"dryrun",
"interactive",
]