def test_build_deactivate_shlvl_1(self):
with tempdir() as td:
mkdir_p(join(td, 'conda-meta'))
deactivate_d_dir = mkdir_p(join(td, 'etc', 'conda', 'deactivate.d'))
deactivate_d_1 = join(deactivate_d_dir, 'see-me-deactivate.sh')
deactivate_d_2 = join(deactivate_d_dir, 'dont-see-me.bat')
touch(join(deactivate_d_1))
touch(join(deactivate_d_2))
with env_var('CONDA_SHLVL', '1'):
with env_var('CONDA_PREFIX', td):
activator = Activator('posix')
original_path = tuple(activator._get_starting_path_list())
builder = activator.build_deactivate()
assert builder['unset_vars'] == (
'CONDA_PREFIX',
'CONDA_DEFAULT_ENV',
'CONDA_PYTHON_EXE',
'CONDA_PROMPT_MODIFIER',
)
new_path = activator.pathsep_join(activator.path_conversion(original_path))
assert builder['set_vars'] == {
'PATH': new_path,
'CONDA_SHLVL': 0,
}
assert builder['activate_scripts'] == ()
assert builder['deactivate_scripts'] == (activator.path_conversion(deactivate_d_1),)
def test_build_activate_shlvl_2(self):
with tempdir() as td:
mkdir_p(join(td, 'conda-meta'))
activate_d_dir = mkdir_p(join(td, 'etc', 'conda', 'activate.d'))
activate_d_1 = join(activate_d_dir, 'see-me.sh')
activate_d_2 = join(activate_d_dir, 'dont-see-me.bat')
touch(join(activate_d_1))
touch(join(activate_d_2))
old_prefix = join(td, 'old')
deactivate_d_dir = mkdir_p(join(old_prefix, 'etc', 'conda', 'deactivate.d'))
deactivate_d_1 = join(deactivate_d_dir, 'see-me.sh')
deactivate_d_2 = join(deactivate_d_dir, 'dont-see-me.bat')
touch(join(deactivate_d_1))
touch(join(deactivate_d_2))
with env_var('CONDA_SHLVL', '2'):
with env_var('CONDA_PREFIX', old_prefix):
activator = Activator('posix')
builder = activator.build_activate(td)
new_path = activator.pathsep_join(activator._add_prefix_to_path(td))
assert builder['unset_vars'] == ()
set_vars = {
'PATH': new_path,
'CONDA_PREFIX': td,
'CONDA_DEFAULT_ENV': td,
'CONDA_PROMPT_MODIFIER': "(%s) " % td,
}
assert builder['set_vars'] == set_vars
assert builder['activate_scripts'] == (activator.path_conversion(activate_d_1),)
assert builder['deactivate_scripts'] == (activator.path_conversion(deactivate_d_1),)
def test_activate_same_environment(self):
with tempdir() as td:
mkdir_p(join(td, 'conda-meta'))
activate_d_dir = mkdir_p(join(td, 'etc', 'conda', 'activate.d'))
activate_d_1 = join(activate_d_dir, 'see-me.sh')
activate_d_2 = join(activate_d_dir, 'dont-see-me.bat')
touch(join(activate_d_1))
touch(join(activate_d_2))
old_prefix = td
deactivate_d_dir = mkdir_p(join(old_prefix, 'etc', 'conda', 'deactivate.d'))
deactivate_d_1 = join(deactivate_d_dir, 'see-me.sh')
deactivate_d_2 = join(deactivate_d_dir, 'dont-see-me.bat')
touch(join(deactivate_d_1))
touch(join(deactivate_d_2))
with env_var('CONDA_SHLVL', '1'):
with env_var('CONDA_PREFIX', old_prefix):
activator = Activator('posix')
builder = activator.build_activate(td)
set_vars = {
'CONDA_PROMPT_MODIFIER': "(%s) " % td,
'CONDA_SHLVL': 1,
}
assert builder['unset_vars'] == ()
assert builder['set_vars'] == set_vars
assert builder['activate_scripts'] == (activator.path_conversion(activate_d_1),)
assert builder['deactivate_scripts'] == (activator.path_conversion(deactivate_d_1),)
def test_build_activate_shlvl_0(self):
with tempdir() as td:
mkdir_p(join(td, 'conda-meta'))
activate_d_dir = mkdir_p(join(td, 'etc', 'conda', 'activate.d'))
activate_d_1 = join(activate_d_dir, 'see-me.sh')
activate_d_2 = join(activate_d_dir, 'dont-see-me.bat')
touch(join(activate_d_1))
touch(join(activate_d_2))
with env_var('CONDA_SHLVL', '0'):
with env_var('CONDA_PREFIX', ''):
activator = Activator('posix')
builder = activator.build_activate(td)
new_path = activator.pathsep_join(activator._add_prefix_to_path(td))
conda_prompt_modifier = "(%s) " % td
ps1 = conda_prompt_modifier + os.environ.get('PS1', '')
assert builder['unset_vars'] == ()
set_vars = {
'PS1': ps1,
}
export_vars = {
'CONDA_PYTHON_EXE': activator.path_conversion(sys.executable),
'PATH': new_path,
'CONDA_PREFIX': td,
'CONDA_SHLVL': 1,
'CONDA_DEFAULT_ENV': td,
'CONDA_PROMPT_MODIFIER': conda_prompt_modifier,
}
assert builder['set_vars'] == set_vars
assert builder['export_vars'] == export_vars
assert builder['activate_scripts'] == (activator.path_conversion(activate_d_1),)
assert builder['deactivate_scripts'] == ()
def test_remove_prefix_from_path_1(self):
activator = Activator('posix')
original_path = tuple(activator._get_starting_path_list())
keep_path = activator.path_conversion('/keep/this/path')
final_path = (keep_path,) + original_path
final_path = activator.path_conversion(final_path)
test_prefix = join(os.getcwd(), 'mytestpath')
new_paths = tuple(activator._get_path_dirs(test_prefix))
prefix_added_path = (keep_path,) + new_paths + original_path
new_path = activator._remove_prefix_from_path(test_prefix, prefix_added_path)
assert final_path == new_path
def test_add_prefix_to_path(self):
activator = Activator('posix')
path_dirs = activator.path_conversion(['/path1/bin', '/path2/bin', '/usr/local/bin', '/usr/bin', '/bin'])
assert len(path_dirs) == 5
test_prefix = '/usr/mytest/prefix'
added_paths = activator.path_conversion(activator._get_path_dirs(test_prefix))
if isinstance(added_paths, string_types):
added_paths = added_paths,
new_path = activator._add_prefix_to_path(test_prefix, path_dirs)
assert new_path == added_paths + path_dirs
def test_remove_prefix_from_path_2(self):
# this time prefix doesn't actually exist in path
activator = Activator('posix')
original_path = tuple(activator._get_starting_path_list())
keep_path = activator.path_conversion('/keep/this/path')
final_path = (keep_path,) + original_path
final_path = activator.path_conversion(final_path)
test_prefix = join(os.getcwd(), 'mytestpath')
prefix_added_path = (keep_path,) + original_path
new_path = activator._remove_prefix_from_path(test_prefix, prefix_added_path)
assert final_path == new_path
def test_replace_prefix_in_path_1(self):
activator = Activator('posix')
original_path = tuple(activator._get_starting_path_list())
new_prefix = join(os.getcwd(), 'mytestpath-new')
new_paths = activator.path_conversion(activator._get_path_dirs(new_prefix))
if isinstance(new_paths, string_types):
new_paths = new_paths,
keep_path = activator.path_conversion('/keep/this/path')
final_path = (keep_path,) + new_paths + original_path
final_path = activator.path_conversion(final_path)
replace_prefix = join(os.getcwd(), 'mytestpath')
replace_paths = tuple(activator._get_path_dirs(replace_prefix))
prefix_added_path = (keep_path,) + replace_paths + original_path
new_path = activator._replace_prefix_in_path(replace_prefix, new_prefix, prefix_added_path)
assert final_path == new_path
def test_build_deactivate_shlvl_2(self):
with tempdir() as td:
mkdir_p(join(td, 'conda-meta'))
deactivate_d_dir = mkdir_p(join(td, 'etc', 'conda', 'deactivate.d'))
deactivate_d_1 = join(deactivate_d_dir, 'see-me-deactivate.sh')
deactivate_d_2 = join(deactivate_d_dir, 'dont-see-me.bat')
touch(join(deactivate_d_1))
touch(join(deactivate_d_2))
old_prefix = join(td, 'old')
activate_d_dir = mkdir_p(join(old_prefix, 'etc', 'conda', 'activate.d'))
activate_d_1 = join(activate_d_dir, 'see-me-activate.sh')
activate_d_2 = join(activate_d_dir, 'dont-see-me.bat')
touch(join(activate_d_1))
touch(join(activate_d_2))
with env_var('CONDA_SHLVL', '2'):
with env_var('CONDA_PREFIX_1', old_prefix):
with env_var('CONDA_PREFIX', td):
activator = Activator('posix')
original_path = tuple(activator._get_starting_path_list())
builder = activator.build_deactivate()
assert builder['unset_vars'] == ('CONDA_PREFIX_1',)
new_path = activator.pathsep_join(activator.path_conversion(original_path))
conda_prompt_modifier = "(%s) " % old_prefix
ps1 = conda_prompt_modifier + os.environ.get('PS1', '')
set_vars = {
'PS1': ps1,
}
export_vars = {
'PATH': new_path,
'CONDA_SHLVL': 1,
'CONDA_PREFIX': old_prefix,
'CONDA_DEFAULT_ENV': old_prefix,
'CONDA_PROMPT_MODIFIER': conda_prompt_modifier,
}
assert builder['set_vars'] == set_vars
assert builder['export_vars'] == export_vars
assert builder['activate_scripts'] == (activator.path_conversion(activate_d_1),)
assert builder['deactivate_scripts'] == (activator.path_conversion(deactivate_d_1),)
class InteractiveShell(object):
activator = None
init_command = None
print_env_var = None
shells = {
'posix': {
'activator': 'posix',
'init_command': 'set -u && . shell/etc/profile.d/conda.sh',
'print_env_var': 'echo $%s',
},
'bash': {
'base_shell': 'posix', # inheritance implemented in __init__
},
'dash': {
'base_shell': 'posix', # inheritance implemented in __init__
},
'zsh': {
'base_shell': 'posix', # inheritance implemented in __init__
},
'cmd.exe': {
'activator': 'cmd.exe',
'init_command': None,
'print_env_var': '@echo %%%s%%',
},
}
def __init__(self, shell_name):
self.shell_name = shell_name
base_shell = self.shells[shell_name].get('base_shell')
shell_vals = self.shells.get(base_shell, {})
shell_vals.update(self.shells[shell_name])
for key, value in iteritems(shell_vals):
setattr(self, key, value)
self.activator = Activator(shell_vals['activator'])
def __enter__(self):
from pexpect.popen_spawn import PopenSpawn
cwd = os.getcwd()
env = os.environ.copy()
env['PATH'] = self.activator.pathsep_join(self.activator.path_conversion(concatv(
self.activator._get_path_dirs(join(cwd, 'shell')),
(dirname(sys.executable),),
self.activator._get_starting_path_list(),
)))
env['PYTHONPATH'] = CONDA_PACKAGE_ROOT
env = {str(k): str(v) for k, v in iteritems(env)}
p = PopenSpawn(self.shell_name, timeout=6, maxread=2000, searchwindowsize=None,
logfile=sys.stdout, cwd=cwd, env=env, encoding=None,
codec_errors='strict')
if self.init_command:
p.sendline(self.init_command)
self.p = p
return self
def __exit__(self, exc_type, exc_val, exc_tb):
if self.p:
import signal
self.p.kill(signal.SIGINT)
def sendline(self, s):
return self.p.sendline(s)
def expect(self, pattern, timeout=-1, searchwindowsize=-1, async=False):
return self.p.expect(pattern, timeout, searchwindowsize, async)
def test_fish_basic(self):
activator = Activator('fish')
self.make_dot_d_files(activator.script_extension)
with captured() as c:
rc = activate_main(('', 'shell.fish', 'activate', self.prefix))
assert not c.stderr
assert rc == 0
activate_data = c.stdout
new_path_parts = activator._add_prefix_to_path(self.prefix)
assert activate_data == dals("""
set -gx CONDA_DEFAULT_ENV "%(native_prefix)s"
set -gx CONDA_PREFIX "%(native_prefix)s"
set -gx CONDA_PROMPT_MODIFIER "(%(native_prefix)s) "
set -gx CONDA_PYTHON_EXE "%(sys_executable)s"
set -gx CONDA_SHLVL "1"
set -gx PATH "%(new_path)s"
source "%(activate1)s"
""") % {
'converted_prefix': activator.path_conversion(self.prefix),
'native_prefix': self.prefix,
'new_path': activator.pathsep_join(new_path_parts),
'sys_executable': activator.path_conversion(sys.executable),
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.fish')),
}
with env_vars({
'CONDA_PREFIX': self.prefix,
'CONDA_SHLVL': '1',
'PATH': os.pathsep.join(concatv(new_path_parts, (os.environ['PATH'],))),
}):
with captured() as c:
rc = activate_main(('', 'shell.fish', 'reactivate'))
assert not c.stderr
assert rc == 0
reactivate_data = c.stdout
assert reactivate_data == dals("""
set -gx CONDA_PROMPT_MODIFIER "(%(native_prefix)s) "
set -gx CONDA_SHLVL "1"
source "%(deactivate1)s"
source "%(activate1)s"
""") % {
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.fish')),
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.fish')),
'native_prefix': self.prefix,
}
with captured() as c:
rc = activate_main(('', 'shell.fish', 'deactivate'))
assert not c.stderr
assert rc == 0
deactivate_data = c.stdout
new_path = activator.pathsep_join(activator._remove_prefix_from_path(self.prefix))
assert deactivate_data == dals("""
set -e CONDA_DEFAULT_ENV
set -e CONDA_PREFIX
set -e CONDA_PROMPT_MODIFIER
set -e CONDA_PYTHON_EXE
set -gx CONDA_SHLVL "0"
set -gx PATH "%(new_path)s"
source "%(deactivate1)s"
""") % {
'new_path': new_path,
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.fish')),
}
def test_xonsh_basic(self):
activator = Activator('xonsh')
self.make_dot_d_files(activator.script_extension)
with captured() as c:
rc = activate_main(('', 'shell.xonsh', 'activate', self.prefix))
assert not c.stderr
assert rc == 0
activate_result = c.stdout
with open(activate_result) as fh:
activate_data = fh.read()
rm_rf(activate_result)
new_path_parts = activator._add_prefix_to_path(self.prefix)
assert activate_data == dals("""
$CONDA_DEFAULT_ENV = "%(native_prefix)s"
$CONDA_PREFIX = "%(native_prefix)s"
$CONDA_PROMPT_MODIFIER = "(%(native_prefix)s) "
$CONDA_PYTHON_EXE = "%(sys_executable)s"
$CONDA_SHLVL = "1"
$PATH = "%(new_path)s"
source "%(activate1)s"
""") % {
'converted_prefix': activator.path_conversion(self.prefix),
'native_prefix': self.prefix,
'new_path': activator.pathsep_join(new_path_parts),
'sys_executable': activator.path_conversion(sys.executable),
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.xsh')),
}
with env_vars({
'CONDA_PREFIX': self.prefix,
'CONDA_SHLVL': '1',
'PATH': os.pathsep.join(concatv(new_path_parts, (os.environ['PATH'],))),
}):
with captured() as c:
assert activate_main(('', 'shell.xonsh', 'reactivate')) == 0
assert not c.stderr
reactivate_result = c.stdout
with open(reactivate_result) as fh:
reactivate_data = fh.read()
rm_rf(reactivate_result)
assert reactivate_data == dals("""
$CONDA_PROMPT_MODIFIER = "(%(native_prefix)s) "
$CONDA_SHLVL = "1"
source "%(deactivate1)s"
source "%(activate1)s"
""") % {
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.xsh')),
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.xsh')),
'native_prefix': self.prefix,
}
with captured() as c:
assert activate_main(('', 'shell.xonsh', 'deactivate')) == 0
assert not c.stderr
deactivate_result = c.stdout
with open(deactivate_result) as fh:
deactivate_data = fh.read()
rm_rf(deactivate_result)
new_path = activator.pathsep_join(activator._remove_prefix_from_path(self.prefix))
assert deactivate_data == dals("""
del $CONDA_DEFAULT_ENV
del $CONDA_PREFIX
del $CONDA_PROMPT_MODIFIER
del $CONDA_PYTHON_EXE
$CONDA_SHLVL = "0"
$PATH = "%(new_path)s"
source "%(deactivate1)s"
""") % {
'new_path': new_path,
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.xsh')),
}
def test_cmd_exe_basic(self):
activator = Activator('cmd.exe')
self.make_dot_d_files(activator.script_extension)
with captured() as c:
rc = activate_main(('', 'shell.cmd.exe', 'activate', '', self.prefix))
assert not c.stderr
assert rc == 0
activate_result = c.stdout
with open(activate_result) as fh:
activate_data = fh.read()
rm_rf(activate_result)
new_path_parts = activator._add_prefix_to_path(self.prefix)
assert activate_data == dals("""
@SET "CONDA_DEFAULT_ENV=%(native_prefix)s"
@SET "CONDA_PREFIX=%(converted_prefix)s"
@SET "CONDA_PROMPT_MODIFIER=(%(native_prefix)s) "
@SET "CONDA_PYTHON_EXE=%(sys_executable)s"
@SET "CONDA_SHLVL=1"
@SET "PATH=%(new_path)s"
@SET "PYTHONIOENCODING=%(PYTHONIOENCODING)s"
@CALL "%(activate1)s"
""") % {
'converted_prefix': activator.path_conversion(self.prefix),
'native_prefix': self.prefix,
'new_path': activator.pathsep_join(new_path_parts),
'sys_executable': activator.path_conversion(sys.executable),
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.bat')),
'PYTHONIOENCODING': PYTHONIOENCODING,
}
with env_vars({
'CONDA_PREFIX': self.prefix,
'CONDA_SHLVL': '1',
'PATH': os.pathsep.join(concatv(new_path_parts, (os.environ['PATH'],))),
}):
with captured() as c:
assert activate_main(('', 'shell.cmd.exe', 'reactivate')) == 0
assert not c.stderr
reactivate_result = c.stdout
with open(reactivate_result) as fh:
reactivate_data = fh.read()
rm_rf(reactivate_result)
assert reactivate_data == dals("""
@CALL "%(deactivate1)s"
@SET "CONDA_PROMPT_MODIFIER=(%(native_prefix)s) "
@SET "CONDA_SHLVL=1"
@CALL "%(activate1)s"
""") % {
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.bat')),
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.bat')),
'native_prefix': self.prefix,
}
with captured() as c:
assert activate_main(('', 'shell.cmd.exe', 'deactivate')) == 0
assert not c.stderr
deactivate_result = c.stdout
with open(deactivate_result) as fh:
deactivate_data = fh.read()
rm_rf(deactivate_result)
new_path = activator.pathsep_join(activator._remove_prefix_from_path(self.prefix))
assert deactivate_data == dals("""
@CALL "%(deactivate1)s"
@SET CONDA_DEFAULT_ENV=
@SET CONDA_PREFIX=
@SET CONDA_PROMPT_MODIFIER=
@SET CONDA_PYTHON_EXE=
@SET "CONDA_SHLVL=0"
@SET "PATH=%(new_path)s"
""") % {
'new_path': new_path,
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.bat')),
}
def test_posix_basic(self):
activator = Activator('posix')
self.make_dot_d_files(activator.script_extension)
with captured() as c:
rc = activate_main(('', 'shell.posix', 'activate', self.prefix))
assert not c.stderr
assert rc == 0
activate_data = c.stdout
new_path_parts = activator._add_prefix_to_path(self.prefix)
assert activate_data == dals("""
PS1='%(ps1)s'
export CONDA_DEFAULT_ENV='%(native_prefix)s'
export CONDA_PREFIX='%(native_prefix)s'
export CONDA_PROMPT_MODIFIER='(%(native_prefix)s) '
export CONDA_PYTHON_EXE='%(sys_executable)s'
export CONDA_SHLVL='1'
export PATH='%(new_path)s'
. "%(activate1)s"
""") % {
'converted_prefix': activator.path_conversion(self.prefix),
'native_prefix': self.prefix,
'new_path': activator.pathsep_join(new_path_parts),
'sys_executable': activator.path_conversion(sys.executable),
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.sh')),
'ps1': '(%s) ' % self.prefix + os.environ.get('PS1', '')
}
with env_vars({
'CONDA_PREFIX': self.prefix,
'CONDA_SHLVL': '1',
'PATH': os.pathsep.join(concatv(new_path_parts, (os.environ['PATH'],))),
}):
with captured() as c:
rc = activate_main(('', 'shell.posix', 'reactivate'))
assert not c.stderr
assert rc == 0
reactivate_data = c.stdout
assert reactivate_data == dals("""
. "%(deactivate1)s"
export CONDA_PROMPT_MODIFIER='(%(native_prefix)s) '
export CONDA_SHLVL='1'
. "%(activate1)s"
""") % {
'activate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'activate.d', 'activate1.sh')),
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.sh')),
'native_prefix': self.prefix,
}
with captured() as c:
rc = activate_main(('', 'shell.posix', 'deactivate'))
assert not c.stderr
assert rc == 0
deactivate_data = c.stdout
new_path = activator.pathsep_join(activator._remove_prefix_from_path(self.prefix))
assert deactivate_data == dals("""
. "%(deactivate1)s"
unset CONDA_DEFAULT_ENV
unset CONDA_PREFIX
unset CONDA_PROMPT_MODIFIER
unset CONDA_PYTHON_EXE
PS1='%(ps1)s'
export CONDA_SHLVL='0'
export PATH='%(new_path)s'
""") % {
'new_path': new_path,
'deactivate1': activator.path_conversion(join(self.prefix, 'etc', 'conda', 'deactivate.d', 'deactivate1.sh')),
'ps1': os.environ.get('PS1', ''),
}
