Add C API tests

Author: DinoV

Lib/test/test_import/test_lazy_imports.py

@@ -9,6 +9,11 @@
 import types
 import unittest
 
+try:
+    import _testcapi
+except ImportError:
+    _testcapi = None
+
 
 class LazyImportTests(unittest.TestCase):
     """Tests for basic lazy import functionality."""
@@ -1630,6 +1635,30 @@ def test_normal_import_dis(self):
             self.assertFail("IMPORT_NAME not found")
 
 
+@unittest.skipIf(_testcapi is None, 'need the _testcapi module')
+class LazyCApiTests(unittest.TestCase):
+    def tearDown(self):
+        sys.set_lazy_imports("normal")
+        sys.set_lazy_imports_filter(None)
+
+    def test_set_matches_sys(self):
+        self.assertEqual(_testcapi.PyImport_GetLazyImportsMode(), sys.get_lazy_imports())
+        for mode in ("normal", "all", "none"):
+            _testcapi.PyImport_SetLazyImportsMode(mode)
+            self.assertEqual(_testcapi.PyImport_GetLazyImportsMode(), sys.get_lazy_imports())
+
+    def test_filter_matches_sys(self):
+        self.assertEqual(_testcapi.PyImport_GetLazyImportsFilter(), sys.get_lazy_imports_filter())
+
+        def filter(*args):
+            pass
+
+        _testcapi.PyImport_SetLazyImportsFilter(filter)
+        self.assertEqual(_testcapi.PyImport_GetLazyImportsFilter(), sys.get_lazy_imports_filter())
+
+    def test_set_bad_filter(self):
+        self.assertRaises(ValueError, _testcapi.PyImport_SetLazyImportsFilter, 42)
+
 
 if __name__ == '__main__':
     unittest.main()

Modules/_testcapi/import.c

@@ -30,9 +30,75 @@ pyimport_importmoduleattrstring(PyObject *self, PyObject *args)
 }
 
 
+static PyObject *
+pyimport_setlazyimportsmode(PyObject *self, PyObject *args)
+{
+    PyObject *mode;
+    if (!PyArg_ParseTuple(args, "U", &mode)) {
+        return NULL;
+    }
+    if (strcmp(PyUnicode_AsUTF8(mode), "normal") == 0) {
+        PyImport_SetLazyImportsMode(PyImport_LAZY_NORMAL);
+    } else if (strcmp(PyUnicode_AsUTF8(mode), "all") == 0) {
+        PyImport_SetLazyImportsMode(PyImport_LAZY_ALL);
+    } else if (strcmp(PyUnicode_AsUTF8(mode), "none") == 0) {
+        PyImport_SetLazyImportsMode(PyImport_LAZY_NONE);
+    } else {
+        PyErr_SetString(PyExc_ValueError, "invalid mode");
+        return NULL;
+    }
+
+    Py_RETURN_NONE;
+}
+
+static PyObject *
+pyimport_getlazyimportsmode(PyObject *self, PyObject *args)
+{
+    switch (PyImport_GetLazyImportsMode()) {
+        case PyImport_LAZY_NORMAL:
+            return PyUnicode_FromString("normal");
+        case PyImport_LAZY_ALL:
+            return PyUnicode_FromString("all");
+        case PyImport_LAZY_NONE:
+            return PyUnicode_FromString("none");
+        default:
+            PyErr_SetString(PyExc_ValueError, "unknown mode");
+            return NULL;
+    }
+}
+
+static PyObject *
+pyimport_setlazyimportsfilter(PyObject *self, PyObject *args)
+{
+    PyObject *filter;
+    if (!PyArg_ParseTuple(args, "O", &filter)) {
+        return NULL;
+    }
+
+    if (PyImport_SetLazyImportsFilter(filter) < 0) {
+        return NULL;
+    }
+
+    Py_RETURN_NONE;
+}
+
+static PyObject *
+pyimport_getlazyimportsfilter(PyObject *self, PyObject *args)
+{
+    PyObject *res = PyImport_GetLazyImportsFilter();
+    if (res == NULL) {
+        Py_RETURN_NONE;
+    }
+    return res;
+}
+
 static PyMethodDef test_methods[] = {
     {"PyImport_ImportModuleAttr", pyimport_importmoduleattr, METH_VARARGS},
     {"PyImport_ImportModuleAttrString", pyimport_importmoduleattrstring, METH_VARARGS},
+    {"PyImport_SetLazyImportsMode", pyimport_setlazyimportsmode, METH_VARARGS},
+    {"PyImport_GetLazyImportsMode", pyimport_getlazyimportsmode, METH_NOARGS},
+    {"PyImport_SetLazyImportsFilter", pyimport_setlazyimportsfilter, METH_VARARGS},
+    {"PyImport_GetLazyImportsFilter", pyimport_getlazyimportsfilter, METH_NOARGS},
     {NULL},
 };
 
@@ -41,4 +107,3 @@ _PyTestCapi_Init_Import(PyObject *m)
 {
     return PyModule_AddFunctions(m, test_methods);
 }
-