xonsh/tests/aliases/test_source.py

import os.path
import pytest

from contextlib import contextmanager
from unittest.mock import MagicMock
from xonsh.aliases import source_alias, builtins


@pytest.fixture
def mockopen(xonsh_builtins, monkeypatch):
    @contextmanager
    def mocked_open(fpath, *args, **kwargs):
        yield MagicMock(read=lambda: fpath)
    monkeypatch.setattr(builtins, 'open', mocked_open)


def test_source_current_dir(mockopen, monkeypatch):
    checker = []

    def mocked_execx(src, *args, **kwargs):
        checker.append(src.strip())
    monkeypatch.setattr(builtins, 'execx', mocked_execx)
    monkeypatch.setattr(os.path, 'isfile', lambda x: True)
    source_alias(['foo', 'bar'])
    assert checker == ['foo', 'bar']


def test_source_path(mockopen, monkeypatch):
    checker = []

    def mocked_execx(src, *args, **kwargs):
        checker.append(src.strip())
    monkeypatch.setattr(builtins, 'execx', mocked_execx)
    source_alias(['foo', 'bar'])
    path_foo = os.path.join('tests', 'bin', 'foo')
    path_bar = os.path.join('tests', 'bin', 'bar')
    assert checker[0].endswith(path_foo)
    assert checker[1].endswith(path_bar)