xonsh/tests/api/test_os.xsh

import os
import tempfile

from xonsh.api.os import indir, rmtree

import pytest

from xonsh.pytest.tools import ON_WINDOWS


def test_indir():
    if ON_WINDOWS:
        pytest.skip("On Windows")
    with tempfile.TemporaryDirectory() as tmpdir:
        assert $(pwd).strip() != tmpdir
        with indir(tmpdir):
            assert $(pwd).strip() == tmpdir
        assert $(pwd).strip() != tmpdir
        try:
            with indir(tmpdir):
                raise Exception
        except Exception:
            assert $(pwd).strip() != tmpdir


def test_rmtree():
    if ON_WINDOWS:
        pytest.skip("On Windows")
    with tempfile.TemporaryDirectory() as tmpdir:
        with indir(tmpdir):
            mkdir rmtree_test
            pushd rmtree_test
            git init
            git config user.email "test@example.com"
            git config user.name "Code Monkey"
            touch thing.txt
            git add thing.txt
            git commit -a --no-gpg-sign -m "add thing"
            popd
            assert os.path.exists('rmtree_test')
            assert os.path.exists('rmtree_test/thing.txt')
            rmtree('rmtree_test', force=True)
            assert not os.path.exists('rmtree_test')
            assert not os.path.exists('rmtree_test/thing.txt')