mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-05 17:00:58 +01:00
2abb24f4f6
A number of common spelling typos found within various docstrings and comments found within source code files have been identified and resolved.
294 lines
11 KiB
Python
294 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Testing dirstack"""
|
|
#from __future__ import unicode_literals, print_function
|
|
|
|
from contextlib import contextmanager
|
|
from functools import wraps
|
|
import os
|
|
import os.path
|
|
import subprocess
|
|
|
|
import builtins
|
|
import pytest
|
|
from xonsh import dirstack
|
|
from xonsh.environ import Env
|
|
from xonsh.built_ins import load_builtins
|
|
from xonsh.dirstack import DIRSTACK
|
|
from xonsh.platform import ON_WINDOWS
|
|
from xonsh.dirstack import _unc_tempDrives
|
|
|
|
HERE = os.path.abspath(os.path.dirname(__file__))
|
|
PARENT = os.path.dirname(HERE)
|
|
|
|
|
|
def drive_in_use(letter):
|
|
return ON_WINDOWS and os.system('vol {}: 2>nul>nul'.format(letter)) == 0
|
|
|
|
MAX_TEMP_DRIVES = 4
|
|
TEMP_DRIVE = []
|
|
|
|
for d in 'zyxwvuts':
|
|
if not drive_in_use(d):
|
|
TEMP_DRIVE.append(d + ':')
|
|
pytestmark = pytest.mark.skipif(len(TEMP_DRIVE) < MAX_TEMP_DRIVES,
|
|
reason='Too many drive letters are already used by Windows to run the tests.')
|
|
|
|
|
|
@pytest.yield_fixture(scope="module")
|
|
def shares_setup(tmpdir_factory):
|
|
"""create some shares to play with on current machine.
|
|
|
|
Yield (to test case) array of structs: [uncPath, driveLetter, equivLocalPath]
|
|
|
|
Side effect: `os.chdir(TEST_WORK_DIR)`
|
|
"""
|
|
|
|
if not ON_WINDOWS:
|
|
return []
|
|
|
|
shares = [[r'uncpushd_test_HERE', TEMP_DRIVE[1], HERE]
|
|
, [r'uncpushd_test_PARENT', TEMP_DRIVE[3], PARENT]]
|
|
|
|
for s, d, l in shares: # set up some shares on local machine. dirs already exist test case must invoke wd_setup.
|
|
rtn = subprocess.call(['NET', 'SHARE', s, '/delete'], universal_newlines=True) # clean up from previous run after good, long wait.
|
|
if rtn != 0:
|
|
yield None
|
|
return
|
|
rtn = subprocess.call(['NET', 'SHARE', s + '=' + l], universal_newlines=True)
|
|
if rtn != 0:
|
|
yield None
|
|
return
|
|
rtn = subprocess.call(['NET', 'USE', d, r"\\localhost" + '\\' + s], universal_newlines=True)
|
|
if rtn != 0:
|
|
yield None
|
|
return
|
|
|
|
yield [[r"\\localhost" + '\\' + s[0], s[1], s[2]] for s in shares]
|
|
|
|
# we want to delete the test shares we've created, but can't do that if unc shares in DIRSTACK
|
|
# (left over from assert fail aborted test)
|
|
os.chdir(HERE)
|
|
for dl in _unc_tempDrives:
|
|
rtn = subprocess.call(['net', 'use', dl, '/delete'], universal_newlines=True)
|
|
for s, d, l in shares:
|
|
rtn = subprocess.call(['net', 'use', d, '/delete'], universal_newlines=True)
|
|
# subprocess.call(['net', 'share', s, '/delete'], universal_newlines=True) # fails with access denied,
|
|
# unless I wait > 10 sec. see http://stackoverflow.com/questions/38448413/access-denied-in-net-share-delete
|
|
|
|
|
|
def test_pushdpopd(xonsh_builtins):
|
|
"""Simple non-UNC push/pop to verify we didn't break nonUNC case.
|
|
"""
|
|
xonsh_builtins.__xonsh_env__ = Env(CDPATH=PARENT, PWD=HERE)
|
|
|
|
dirstack.cd([PARENT])
|
|
owd = os.getcwd()
|
|
assert owd.casefold() == xonsh_builtins.__xonsh_env__['PWD'].casefold()
|
|
dirstack.pushd([HERE])
|
|
wd = os.getcwd()
|
|
assert wd.casefold() == HERE.casefold()
|
|
dirstack.popd([])
|
|
assert owd.casefold() == os.getcwd().casefold(), "popd returned cwd to expected dir"
|
|
|
|
|
|
def test_cd_dot(xonsh_builtins):
|
|
xonsh_builtins.__xonsh_env__ = Env(PWD=os.getcwd())
|
|
|
|
owd = os.getcwd().casefold()
|
|
dirstack.cd(['.'])
|
|
assert owd == os.getcwd().casefold()
|
|
|
|
|
|
@pytest.mark.skipif( not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_simple_push_pop(xonsh_builtins, shares_setup):
|
|
if shares_setup is None:
|
|
return
|
|
xonsh_builtins.__xonsh_env__ = Env(CDPATH=PARENT, PWD=HERE)
|
|
dirstack.cd([PARENT])
|
|
owd = os.getcwd()
|
|
assert owd.casefold() == xonsh_builtins.__xonsh_env__['PWD'].casefold()
|
|
dirstack.pushd([r'\\localhost\uncpushd_test_HERE'])
|
|
wd = os.getcwd()
|
|
assert os.path.splitdrive(wd)[0].casefold() == TEMP_DRIVE[0]
|
|
assert os.path.splitdrive(wd)[1].casefold() == '\\'
|
|
dirstack.popd([])
|
|
assert owd.casefold() == os.getcwd().casefold(), "popd returned cwd to expected dir"
|
|
assert len(_unc_tempDrives) == 0
|
|
|
|
|
|
@pytest.mark.skipif(not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_push_to_same_share(xonsh_builtins, shares_setup):
|
|
if shares_setup is None:
|
|
return
|
|
xonsh_builtins.__xonsh_env__ = Env(CDPATH=PARENT, PWD=HERE)
|
|
|
|
dirstack.cd([PARENT])
|
|
owd = os.getcwd()
|
|
assert owd.casefold() == xonsh_builtins.__xonsh_env__['PWD'].casefold()
|
|
dirstack.pushd([r'\\localhost\uncpushd_test_HERE'])
|
|
wd = os.getcwd()
|
|
assert os.path.splitdrive(wd)[0].casefold() == TEMP_DRIVE[0]
|
|
assert os.path.splitdrive(wd)[1].casefold() == '\\'
|
|
assert len(_unc_tempDrives) == 1
|
|
assert len(DIRSTACK) == 1
|
|
|
|
dirstack.pushd([r'\\localhost\uncpushd_test_HERE'])
|
|
wd = os.getcwd()
|
|
assert os.path.splitdrive(wd)[0].casefold() == TEMP_DRIVE[0]
|
|
assert os.path.splitdrive(wd)[1].casefold() == '\\'
|
|
assert len(_unc_tempDrives) == 1
|
|
assert len(DIRSTACK) == 2
|
|
|
|
dirstack.popd([])
|
|
assert os.path.isdir(TEMP_DRIVE[0] + '\\'), "Temp drive not unmapped till last reference removed"
|
|
dirstack.popd([])
|
|
assert owd.casefold() == os.getcwd().casefold(), "popd returned cwd to expected dir"
|
|
assert len(_unc_tempDrives) == 0
|
|
|
|
|
|
@pytest.mark.skipif( not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_push_other_push_same(xonsh_builtins, shares_setup):
|
|
"""push to a, then to b. verify drive letter is TEMP_DRIVE[2], skipping already used TEMP_DRIVE[1]
|
|
Then push to a again. Pop (check b unmapped and a still mapped), pop, pop (check a is unmapped)"""
|
|
if shares_setup is None:
|
|
return
|
|
xonsh_builtins.__xonsh_env__ = Env(CDPATH=PARENT, PWD=HERE)
|
|
|
|
dirstack.cd([PARENT])
|
|
owd = os.getcwd()
|
|
assert owd.casefold() == xonsh_builtins.__xonsh_env__['PWD'].casefold()
|
|
dirstack.pushd([r'\\localhost\uncpushd_test_HERE'])
|
|
assert os.getcwd().casefold() == TEMP_DRIVE[0] + '\\'
|
|
assert len(_unc_tempDrives) == 1
|
|
assert len(DIRSTACK) == 1
|
|
|
|
dirstack.pushd([r'\\localhost\uncpushd_test_PARENT'])
|
|
wd = os.getcwd()
|
|
assert os.getcwd().casefold() == TEMP_DRIVE[2] + '\\'
|
|
assert len(_unc_tempDrives) == 2
|
|
assert len(DIRSTACK) == 2
|
|
|
|
dirstack.pushd([r'\\localhost\uncpushd_test_HERE'])
|
|
assert os.getcwd().casefold() == TEMP_DRIVE[0] + '\\'
|
|
assert len(_unc_tempDrives) == 2
|
|
assert len(DIRSTACK) == 3
|
|
|
|
dirstack.popd([])
|
|
assert os.getcwd().casefold() == TEMP_DRIVE[2] + '\\'
|
|
assert len(_unc_tempDrives) == 2
|
|
assert len(DIRSTACK) == 2
|
|
assert os.path.isdir(TEMP_DRIVE[2] + '\\')
|
|
assert os.path.isdir(TEMP_DRIVE[0] + '\\')
|
|
|
|
dirstack.popd([])
|
|
assert os.getcwd().casefold() == TEMP_DRIVE[0] + '\\'
|
|
assert len(_unc_tempDrives) == 1
|
|
assert len(DIRSTACK) == 1
|
|
assert not os.path.isdir(TEMP_DRIVE[2] + '\\')
|
|
assert os.path.isdir(TEMP_DRIVE[0] + '\\')
|
|
|
|
dirstack.popd([])
|
|
assert os.getcwd().casefold() == owd.casefold()
|
|
assert len(_unc_tempDrives) == 0
|
|
assert len(DIRSTACK) == 0
|
|
assert not os.path.isdir(TEMP_DRIVE[2] + '\\')
|
|
assert not os.path.isdir(TEMP_DRIVE[0] + '\\')
|
|
|
|
|
|
@pytest.mark.skipif(not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_push_base_push_rempath(xonsh_builtins):
|
|
"""push to subdir under share, verify mapped path includes subdir"""
|
|
pass
|
|
|
|
|
|
#really? Need to cut-and-paste 2 flavors of this? yield_fixture requires yield in defined function body, not callee
|
|
@pytest.yield_fixture()
|
|
def with_unc_check_enabled():
|
|
if not ON_WINDOWS:
|
|
return
|
|
|
|
import winreg
|
|
|
|
old_wval = 0
|
|
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'software\microsoft\command processor', access=winreg.KEY_WRITE)
|
|
try:
|
|
wval, wtype = winreg.QueryValueEx(key, 'DisableUNCCheck')
|
|
old_wval = wval # if values was defined at all
|
|
except OSError as e:
|
|
pass
|
|
winreg.SetValueEx(key, 'DisableUNCCheck', None, winreg.REG_DWORD, 0)
|
|
winreg.CloseKey(key)
|
|
|
|
yield old_wval
|
|
|
|
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'software\microsoft\command processor', access=winreg.KEY_WRITE)
|
|
winreg.SetValueEx(key, 'DisableUNCCheck', None, winreg.REG_DWORD, old_wval)
|
|
winreg.CloseKey(key)
|
|
|
|
|
|
@pytest.yield_fixture()
|
|
def with_unc_check_disabled(): # just like the above, but value is 1 to *disable* unc check
|
|
if not ON_WINDOWS:
|
|
return
|
|
|
|
import winreg
|
|
|
|
old_wval = 0
|
|
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'software\microsoft\command processor', access=winreg.KEY_WRITE)
|
|
try:
|
|
wval, wtype = winreg.QueryValueEx(key, 'DisableUNCCheck')
|
|
old_wval = wval # if values was defined at all
|
|
except OSError as e:
|
|
pass
|
|
winreg.SetValueEx(key, 'DisableUNCCheck', None, winreg.REG_DWORD, 1)
|
|
winreg.CloseKey(key)
|
|
|
|
yield old_wval
|
|
|
|
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'software\microsoft\command processor', access=winreg.KEY_WRITE)
|
|
winreg.SetValueEx(key, 'DisableUNCCheck', None, winreg.REG_DWORD, old_wval)
|
|
winreg.CloseKey(key)
|
|
|
|
|
|
@pytest.fixture()
|
|
def xonsh_builtins_cd(xonsh_builtins):
|
|
xonsh_builtins.__xonsh_env__['CDPATH'] = PARENT
|
|
xonsh_builtins.__xonsh_env__['PWD'] = os.getcwd()
|
|
xonsh_builtins.__xonsh_env__['DIRSTACK_SIZE'] = 20
|
|
return xonsh_builtins
|
|
|
|
|
|
@pytest.mark.skipif(not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_cd_unc_auto_pushd(xonsh_builtins_cd, with_unc_check_enabled):
|
|
xonsh_builtins_cd.__xonsh_env__['AUTO_PUSHD'] = True
|
|
so, se, rc = dirstack.cd([r'\\localhost\uncpushd_test_PARENT'])
|
|
if rc != 0:
|
|
return
|
|
assert os.getcwd().casefold() == TEMP_DRIVE[0] + '\\'
|
|
assert len(DIRSTACK) == 1
|
|
assert os.path.isdir(TEMP_DRIVE[0] + '\\')
|
|
|
|
|
|
@pytest.mark.skipif(not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_cd_unc_nocheck(xonsh_builtins_cd, with_unc_check_disabled):
|
|
if with_unc_check_disabled == 0:
|
|
return
|
|
dirstack.cd([r'\\localhost\uncpushd_test_HERE'])
|
|
assert os.getcwd().casefold() == r'\\localhost\uncpushd_test_here'
|
|
|
|
|
|
@pytest.mark.skipif(not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_cd_unc_no_auto_pushd(xonsh_builtins_cd, with_unc_check_enabled):
|
|
if with_unc_check_enabled == 0:
|
|
return
|
|
so, se, rc = dirstack.cd([r'\\localhost\uncpushd_test_PARENT'])
|
|
assert rc != 0
|
|
assert so is None or len(so) == 0
|
|
assert 'disableunccheck' in se.casefold() and 'auto_pushd' in se.casefold()
|
|
|
|
|
|
@pytest.mark.skipif(not ON_WINDOWS, reason="Windows-only UNC functionality")
|
|
def test_uncpushd_unc_check():
|
|
# emminently suited to mocking, but I don't know how
|
|
# need to verify unc_check_enabled correct whether values set in HKCU or HKLM
|
|
pass
|