Navigation Menu

Skip to content

Commit

Permalink
bpo-34530: Fix distutils find_executable() (GH-9049)
Browse files Browse the repository at this point in the history
distutils.spawn.find_executable() now falls back on os.defpath if the
PATH environment variable is not set.
  • Loading branch information
vstinner committed Sep 4, 2018
1 parent ec74d18 commit 3948719
Show file tree
Hide file tree
Showing 3 changed files with 50 additions and 3 deletions.
2 changes: 1 addition & 1 deletion Lib/distutils/spawn.py
Expand Up @@ -173,7 +173,7 @@ def find_executable(executable, path=None):
os.environ['PATH']. Returns the complete filename or None if not found.
"""
if path is None:
path = os.environ['PATH']
path = os.environ.get('PATH', os.defpath)

paths = path.split(os.pathsep)
base, ext = os.path.splitext(executable)
Expand Down
49 changes: 47 additions & 2 deletions Lib/distutils/tests/test_spawn.py
@@ -1,9 +1,13 @@
"""Tests for distutils.spawn."""
import unittest
import sys
import os
import stat
import sys
import unittest
from unittest import mock
from test.support import run_unittest, unix_shell
from test import support as test_support

from distutils.spawn import find_executable
from distutils.spawn import _nt_quote_args
from distutils.spawn import spawn
from distutils.errors import DistutilsExecError
Expand Down Expand Up @@ -51,6 +55,47 @@ def test_spawn(self):
os.chmod(exe, 0o777)
spawn([exe]) # should work without any error

def test_find_executable(self):
with test_support.temp_dir() as tmp_dir:
# use TESTFN to get a pseudo-unique filename
program_noeext = test_support.TESTFN
# Give the temporary program an ".exe" suffix for all.
# It's needed on Windows and not harmful on other platforms.
program = program_noeext + ".exe"

filename = os.path.join(tmp_dir, program)
with open(filename, "wb"):
pass
os.chmod(filename, stat.S_IXUSR)

# test path parameter
rv = find_executable(program, path=tmp_dir)
self.assertEqual(rv, filename)

if sys.platform == 'win32':
# test without ".exe" extension
rv = find_executable(program_noeext, path=tmp_dir)
self.assertEqual(rv, filename)

# test find in the current directory
with test_support.change_cwd(tmp_dir):
rv = find_executable(program)
self.assertEqual(rv, program)

# test non-existent program
dont_exist_program = "dontexist_" + program
rv = find_executable(dont_exist_program , path=tmp_dir)
self.assertIsNone(rv)

# test os.defpath: missing PATH environment variable
with test_support.EnvironmentVarGuard() as env:
with mock.patch('distutils.spawn.os.defpath', tmp_dir):
env.pop('PATH')

rv = find_executable(program)
self.assertEqual(rv, filename)


def test_suite():
return unittest.makeSuite(SpawnTestCase)

Expand Down
@@ -0,0 +1,2 @@
``distutils.spawn.find_executable()`` now falls back on :data:`os.defpath`
if the ``PATH`` environment variable is not set.

0 comments on commit 3948719

Please sign in to comment.