mirror of
https://github.com/kennethreitz/pipenv.git
synced 2026-06-05 22:50:18 +00:00
Matt Davis
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from pipenv.project import Project
|
|
from pipenv.utils import subprocess_run, temp_environ
|
|
|
|
|
|
@pytest.mark.run
|
|
@pytest.mark.dotenv
|
|
def test_env(PipenvInstance):
|
|
with PipenvInstance(pipfile=False, chdir=True) as p:
|
|
with open(os.path.join(p.path, ".env"), "w") as f:
|
|
f.write("HELLO=WORLD")
|
|
c = subprocess_run(['pipenv', 'run', 'python', '-c', "import os; print(os.environ['HELLO'])"], env=p.env)
|
|
assert c.returncode == 0
|
|
assert 'WORLD' in c.stdout
|
|
|
|
|
|
@pytest.mark.run
|
|
def test_scripts(PipenvInstance):
|
|
with PipenvInstance(chdir=True) as p:
|
|
with open(p.pipfile_path, 'w') as f:
|
|
f.write(r"""
|
|
[scripts]
|
|
printfoo = "python -c \"print('foo')\""
|
|
notfoundscript = "randomthingtotally"
|
|
appendscript = "cmd arg1"
|
|
multicommand = "bash -c \"cd docs && make html\""
|
|
""")
|
|
if os.name == "nt":
|
|
f.write('scriptwithenv = "echo %HELLO%"\n')
|
|
else:
|
|
f.write('scriptwithenv = "echo $HELLO"\n')
|
|
c = p.pipenv('install')
|
|
assert c.returncode == 0
|
|
c = p.pipenv('run printfoo')
|
|
assert c.returncode == 0
|
|
assert c.stdout.splitlines()[1] == 'foo'
|
|
assert not c.stderr.strip()
|
|
|
|
c = p.pipenv('run notfoundscript')
|
|
assert c.returncode != 0
|
|
assert c.stdout == 'Loading .env environment variables...\n'
|
|
if os.name != 'nt': # TODO: Implement this message for Windows.
|
|
assert 'not found' in c.stderr
|
|
|
|
project = Project()
|
|
|
|
script = project.build_script('multicommand')
|
|
assert script.command == 'bash'
|
|
assert script.args == ['-c', 'cd docs && make html']
|
|
|
|
script = project.build_script('appendscript', ['a', 'b'])
|
|
assert script.command == 'cmd'
|
|
assert script.args == ['arg1', 'a', 'b']
|
|
|
|
with temp_environ():
|
|
os.environ['HELLO'] = 'WORLD'
|
|
c = p.pipenv("run scriptwithenv")
|
|
assert c.returncode == 0
|
|
if os.name != "nt": # This doesn't work on CI windows.
|
|
assert c.stdout.splitlines()[1] == "WORLD"
|
|
|
|
|
|
@pytest.mark.run
|
|
@pytest.mark.skip_windows
|
|
def test_run_with_usr_env_shebang(PipenvInstance):
|
|
with PipenvInstance(chdir=True) as p:
|
|
p.pipenv('install')
|
|
script_path = os.path.join(p.path, "test_script")
|
|
with open(script_path, "w") as f:
|
|
f.write(
|
|
"#!/usr/bin/env python\n"
|
|
"import sys, os\n\n"
|
|
"print(sys.prefix)\n"
|
|
"print(os.getenv('VIRTUAL_ENV'))\n"
|
|
)
|
|
os.chmod(script_path, 0o700)
|
|
c = p.pipenv("run ./test_script")
|
|
assert c.returncode == 0
|
|
project = Project()
|
|
lines = [line.strip() for line in c.stdout.splitlines()[1:]]
|
|
assert all(line == project.virtualenv_location for line in lines)
|