Commit f0434e64 authored by Brett Cannon's avatar Brett Cannon

Issue #14599: Generalize a test for ImportError.path and add support

in Python/dynload_shlibs.c.

This should fix the remaining importlib test failure on Windows.
Support in AIX and HP-UX will be in a separate checkin.
parent 3c23a87e
...@@ -179,6 +179,17 @@ class ImportTests(unittest.TestCase): ...@@ -179,6 +179,17 @@ class ImportTests(unittest.TestCase):
self.assertRaises(SyntaxError, self.assertRaises(SyntaxError,
imp.find_module, "badsyntax_pep3120", [path]) imp.find_module, "badsyntax_pep3120", [path])
def test_load_dynamic_ImportError_path(self):
# Issue #1559549 added `name` and `path` attributes to ImportError
# in order to provide better detail. Issue #10854 implemented those
# attributes on import failures of extensions on Windows.
path = 'bogus file path'
name = 'extension'
with self.assertRaises(ImportError) as err:
imp.load_dynamic(name, path)
self.assertIn(path, err.exception.path)
self.assertEqual(name, err.exception.name)
class ReloadTests(unittest.TestCase): class ReloadTests(unittest.TestCase):
......
...@@ -337,28 +337,6 @@ class ImportTests(unittest.TestCase): ...@@ -337,28 +337,6 @@ class ImportTests(unittest.TestCase):
del sys.path[0] del sys.path[0]
remove_files(TESTFN) remove_files(TESTFN)
@unittest.skipUnless(sys.platform == "win32", "Windows-specific")
def test_extension_import_fail(self):
# Issue 1559549 added `name` and `path` attributes to ImportError
# in order to provide better detail. Issue 10854 implemented those
# attributes on import failures of extensions on Windows.
debug = True if sys.executable[-6:] == "_d.exe" else False
pkg_name = "extension"
pkg_file = pkg_name + "{}".format("_d.pyd" if debug else ".pyd")
with open(pkg_file, "w"): pass
importlib.invalidate_caches()
try:
with self.assertRaises(ImportError) as err:
import extension
self.assertEqual(err.exception.name, pkg_name)
# The path we get back has the dot-slash, e.g., ".\\extension.pyd"
self.assertIsNotNone(err.exception.path,
'unexpected None for ImportError.path: '
'{!r}'.format(err.exception))
self.assertEqual(os.path.relpath(err.exception.path), pkg_file)
finally:
unlink(pkg_file)
class PycRewritingTests(unittest.TestCase): class PycRewritingTests(unittest.TestCase):
# Test that the `co_filename` attribute on code objects always points # Test that the `co_filename` attribute on code objects always points
......
...@@ -129,10 +129,19 @@ dl_funcptr _PyImport_GetDynLoadFunc(const char *shortname, ...@@ -129,10 +129,19 @@ dl_funcptr _PyImport_GetDynLoadFunc(const char *shortname,
handle = dlopen(pathname, dlopenflags); handle = dlopen(pathname, dlopenflags);
if (handle == NULL) { if (handle == NULL) {
PyObject *mod_name = NULL;
PyObject *path = NULL;
PyObject *error_ob = NULL;
const char *error = dlerror(); const char *error = dlerror();
if (error == NULL) if (error == NULL)
error = "unknown dlopen() error"; error = "unknown dlopen() error";
PyErr_SetString(PyExc_ImportError, error); error_ob = PyUnicode_FromString(error);
path = PyUnicode_FromString(pathname);
mod_name = PyUnicode_FromString(shortname);
PyErr_SetImportError(error_ob, mod_name, path);
Py_DECREF(error_ob);
Py_DECREF(path);
Py_DECREF(mod_name);
return NULL; return NULL;
} }
if (fp != NULL && nhandles < 128) if (fp != NULL && nhandles < 128)
......
...@@ -74,10 +74,11 @@ _PyImport_LoadDynamicModule(PyObject *name, PyObject *path, FILE *fp) ...@@ -74,10 +74,11 @@ _PyImport_LoadDynamicModule(PyObject *name, PyObject *path, FILE *fp)
if (PyErr_Occurred()) if (PyErr_Occurred())
goto error; goto error;
if (p == NULL) { if (p == NULL) {
PyErr_Format(PyExc_ImportError, PyObject *msg = PyUnicode_FromFormat("dynamic module does not define "
"dynamic module does not define init function" "init function (PyInit_%s)",
" (PyInit_%s)",
shortname); shortname);
PyErr_SetImportError(msg, name, path);
Py_DECREF(msg);
goto error; goto error;
} }
oldcontext = _Py_PackageContext; oldcontext = _Py_PackageContext;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment