iframe-proxy | Sunbelt Computer Software

History

154 lines (109 loc) · 4.56 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

from __future__ import annotations

import contextlib

import os.path

import shutil

import sys

import pytest

from pre_commit import parse_shebang

from pre_commit.envcontext import envcontext

from pre_commit.envcontext import Var

from pre_commit.util import make_executable

def _echo_exe() -> str:

exe = shutil.which('echo')

assert exe is not None

return exe

def test_file_doesnt_exist():

assert parse_shebang.parse_filename('herp derp derp') == ()

def test_simple_case(tmpdir):

x = tmpdir.join('f')

x.write('#!/usr/bin/env echo')

make_executable(x.strpath)

assert parse_shebang.parse_filename(x.strpath) == ('echo',)

def test_find_executable_full_path():

assert parse_shebang.find_executable(sys.executable) == sys.executable

def test_find_executable_on_path():

assert parse_shebang.find_executable('echo') == _echo_exe()

def test_find_executable_not_found_none():

assert parse_shebang.find_executable('not-a-real-executable') is None

def write_executable(shebang, filename='run'):

os.mkdir('bin')

path = os.path.join('bin', filename)

with open(path, 'w') as f:

f.write(f'#!{shebang}')

make_executable(path)

return path

@contextlib.contextmanager

def bin_on_path():

bindir = os.path.join(os.getcwd(), 'bin')

with envcontext((('PATH', (bindir, os.pathsep, Var('PATH'))),)):

yield

def test_find_executable_path_added(in_tmpdir):

path = os.path.abspath(write_executable('/usr/bin/env sh'))

assert parse_shebang.find_executable('run') is None

with bin_on_path():

assert parse_shebang.find_executable('run') == path

def test_find_executable_path_ext(in_tmpdir):

"""Windows exports PATHEXT as a list of extensions to automatically add

to executables when doing PATH searching.

"""

exe_path = os.path.abspath(

write_executable('/usr/bin/env sh', filename='run.myext'),

)

env_path = {'PATH': os.path.dirname(exe_path)}

env_path_ext = dict(env_path, PATHEXT=os.pathsep.join(('.exe', '.myext')))

assert parse_shebang.find_executable('run') is None

assert parse_shebang.find_executable('run', env=env_path) is None

ret = parse_shebang.find_executable('run.myext', env=env_path)

assert ret == exe_path

ret = parse_shebang.find_executable('run', env=env_path_ext)

assert ret == exe_path

def test_normexe_does_not_exist():

with pytest.raises(OSError) as excinfo:

parse_shebang.normexe('i-dont-exist-lol')

assert excinfo.value.args == ('Executable `i-dont-exist-lol` not found',)

def test_normexe_does_not_exist_sep():

with pytest.raises(OSError) as excinfo:

parse_shebang.normexe('./i-dont-exist-lol')

assert excinfo.value.args == ('Executable `./i-dont-exist-lol` not found',)

@pytest.mark.xfail(sys.platform == 'win32', reason='posix only')

def test_normexe_not_executable(tmpdir): # pragma: win32 no cover

tmpdir.join('exe').ensure()

with tmpdir.as_cwd(), pytest.raises(OSError) as excinfo:

parse_shebang.normexe('./exe')

assert excinfo.value.args == ('Executable `./exe` is not executable',)

def test_normexe_is_a_directory(tmpdir):

with tmpdir.as_cwd():

tmpdir.join('exe').ensure_dir()

exe = os.path.join('.', 'exe')

with pytest.raises(OSError) as excinfo:

parse_shebang.normexe(exe)

msg, = excinfo.value.args

assert msg == f'Executable `{exe}` is a directory'

def test_normexe_already_full_path():

assert parse_shebang.normexe(sys.executable) == sys.executable

def test_normexe_gives_full_path():

assert parse_shebang.normexe('echo') == _echo_exe()

assert os.sep in _echo_exe()

def test_normalize_cmd_trivial():

cmd = (_echo_exe(), 'hi')

assert parse_shebang.normalize_cmd(cmd) == cmd

def test_normalize_cmd_PATH():

cmd = ('echo', '--version')

expected = (_echo_exe(), '--version')

assert parse_shebang.normalize_cmd(cmd) == expected

def test_normalize_cmd_shebang(in_tmpdir):

us = sys.executable.replace(os.sep, '/')

path = write_executable(us)

assert parse_shebang.normalize_cmd((path,)) == (us, path)

def test_normalize_cmd_PATH_shebang_full_path(in_tmpdir):

us = sys.executable.replace(os.sep, '/')

path = write_executable(us)

with bin_on_path():

ret = parse_shebang.normalize_cmd(('run',))

assert ret == (us, os.path.abspath(path))

def test_normalize_cmd_PATH_shebang_PATH(in_tmpdir):

echo = _echo_exe()

path = write_executable('/usr/bin/env echo')

with bin_on_path():

ret = parse_shebang.normalize_cmd(('run',))

assert ret == (echo, os.path.abspath(path))

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

parse_shebang_test.py

Latest commit

History

parse_shebang_test.py

File metadata and controls