o 5hs'@sddlmZddlZddlZddlmZddlZddlm Z ddl m Z m Z ddl mZddlmZddlmZd d lmZd d lmZddlmm Zdd lmZd ejvZGdddZGdddZddZ dS)) annotationsN)cache_from_source)path) build_extget_abi3_suffix) Distribution) CompileError) Extension) environment)DALS)get_config_varZ__pypy__c@s<eZdZddZddZddZddZd d Zd d Zd S) TestBuildExtcCs@t}t|}d|jd<|d}tj|d}||ksJdS)z Setuptools needs to give back the same result as distutils, even if the fullname is not in ext_map. zfoo/barZfooN)rrext_mapget_ext_filenameorig)selfdistcmdresZwantedr?lib/python3.10/site-packages/setuptools/tests/test_build_ext.pytest_get_ext_filenames   z"TestBuildExt.test_get_ext_filenamecCstttddgdd}tt|gd}t|}|d|jvs$J|d}ts7| t ds5Jd St j dkrE| dsCJd Sd |vsKJd S) z} Filename needs to be loadable by several versions of Python 3 if 'is_abi3' is truthy on Extension() spam.eggseggs.cTZpy_limited_api ext_modulesZ EXT_SUFFIXwin32zeggs.pydZabi3N) printrr rdictrZfinalize_optionsrrendswithr sysplatform)r extensionrrrrrrtest_abi3_filename&s   zTestBuildExt.test_abi3_filenamecCst}t|}tddgdd|jd<}d|_ts|d}nd}z"dtjd<|d }d |ks2J|d}||ks=JWtjd=d Stjd=w) zd SETUPTOOLS_EXT_SUFFIX variable always overrides default extension options. Zfor_abi3zs.cTrFzfor_abi3.test-suffixz .test-suffixSETUPTOOLS_EXT_SUFFIXZnormalznormal.test-suffixN) rrr rZ_links_to_dynamicIS_PYPYrosenviron)rrrextexpectrrrrtest_ext_suffix_override;s$     z%TestBuildExt.test_ext_suffix_overridecCsfddddiiidddiid}tdd g}td d g}td d g}t|td|||gddidS)NZmypkgZsubpkgzext2.crext1zmain.c)srcz c-extensionsz mypkg.ext1zc-extensions/ext1/main.czmypkg.subpkg.ext2zsrc/mypkg/subpkg/ext2.cext3zc-extension/ext3.cz%test%r/) script_namerZ package_dir)r rbuildr)rfilesr.Zext2r0rrrdist_with_example[s     zTestBuildExt.dist_with_examplecCs|dd|dd|}|d}d|_||jtj d}dd| D}||d |d |d gks?J| iksGJ| dd |_|d d| D}||d d|d d|d dikssJdS)Nr'.mp3&setuptools.command.build_ext.use_stubsFr/cSsg|] }|tjdqSr7replacer)sep).0xrrr vsz1TestBuildExt.test_get_outputs.. /ext3.mp3/mypkg/ext1.mp3/mypkg/subpkg/ext2.mp3TcS*i|]\}}|tjd|tjdqSr8r9r<kvrrr z1TestBuildExt.test_get_outputs.. src/ext3.mp3src/mypkg/ext1.mp3src/mypkg/subpkg/ext2.mp3)setenvsetattrr4get_command_obj editable_modeensure_finalized build_libr:r)r;Z get_outputsget_output_mappingZreinitialize_commanditems)r tmpdir_cwd monkeypatchrrrPZoutputsmappingrrrtest_get_outputsls2         zTestBuildExt.test_get_outputsc CsD|dd|dd|}|d}d|_||jD] }||ddq|jt j d}dd | D}d d }|||d |d |dd||d|d|dd||d|d|ddikspJ|d|vsyJ|d|vsJ||d} | |vsJ| |dsJ| dsJdS)Nr'r5r6TrZ _needs_stubr7cSrBr8r9rCrrrrFrGzBTestBuildExt.test_get_output_mapping_with_stub..cSst|tjdS)zDMake it possible to do comparisons and tests in a OS-independent wayr7)_compiled_file_namer:r)r;)filerrrCsz9TestBuildExt.test_get_output_mapping_with_stub..Cz/ext3.pyz src/ext3.pyr?rHz/mypkg/ext1.pyzsrc/mypkg/ext1.pyr@rIz/mypkg/subpkg/ext2.pyzsrc/mypkg/subpkg/ext2.pyrArJz/mypkg/__pycache__/ext1z.pyc)rKrLr4rMrNrO extensionsrPr:r)r;rQrR startswithr") rrSrTrrr+rPrUrYZ example_stubrrr!test_get_output_mapping_with_stubs6           z.TestBuildExt.test_get_output_mapping_with_stubN) __name__ __module__ __qualname__rr&r-r4rVr\rrrrrs   rc@s.eZdZdddZddZd d Zd d Zd S)TestBuildExtInplaceoptionalboolreturnrcKsndiidd}t|tddg|d}tt|gd}d|_t|}t|jd d d d || |S) Nz#include missingheader.h )libtmp)rz.buildrr)rarsetup.pyz .build/libz .build/tmp)rPZ build_tempr) rr2r rr!r1rvarsupdaterO)rraZoptsr3r%rrrrrget_build_ext_cmds z%TestBuildExtInplace.get_build_ext_cmdcCs"|}|j|j|jS)z Historically, distutils "logged" by printing to sys.std*. Later versions adopted the logging framework. Grab messages regardless of how they were captured. )Z readouterrout splitlineserrmessages)rcaplogcapsysZstdrrrget_log_messagessz$TestBuildExtInplace.get_log_messagescCs8|jddd}|tdd|||DsJdS)z If optional extensions fail to build, setuptools should show the error in the logs but not fail to build TraZinplacecss|]}dVqdS)z0build_ext: building extension "spam.eggs" failedNr)r<msgrrr s  z4TestBuildExtInplace.test_optional..N)rirunanyrp)rrSrnrorrrr test_optionals  z!TestBuildExtInplace.test_optionalcCsF|jddd}tt |WddS1swYdS)NFTrq)ripytestZraisesrrt)rrSrrrrtest_non_optionals  "z%TestBuildExtInplace.test_non_optionalN)rarbrcr)r]r^r_rirprvrxrrrrr`s   r`cCsZtdtdtdd}t|tjdgdd\}\}}|dks+Jd |d |dS) Nz from setuptools import Extension, setup setup( name='foo', version='0.0.0', ext_modules=[Extension('foo', ['foo.c'])], ) a #include "Python.h" #if PY_MAJOR_VERSION >= 3 static struct PyModuleDef moduledef = { PyModuleDef_HEAD_INIT, "foo", NULL, 0, NULL, NULL, NULL, NULL, NULL }; #define INITERROR return NULL PyMODINIT_FUNC PyInit_foo(void) #else #define INITERROR return void initfoo(void) #endif { #if PY_MAJOR_VERSION >= 3 PyObject *module = PyModule_Create(&moduledef); #else PyObject *module = Py_InitModule("extension", NULL); #endif if (module == NULL) INITERROR; #if PY_MAJOR_VERSION >= 3 return module; #endif } zD [build] build_base = foo_build )rfzfoo.cz setup.cfgr2)r)rZ data_streamrz STDOUT: z STDERR: )r rr2r Z run_setup_py)rSr3codestdoutstderrrrrtest_build_ext_config_handlings  + = r})!Z __future__rr)r#importlib.utilrrWrwZjaracorZsetuptools.command.build_extrrZsetuptools.distrZsetuptools.errorsrZsetuptools.extensionr rr textwrapr Zdistutils.command.build_extZcommandrZdistutils.sysconfigr builtin_module_namesr(rr`r}rrrrs&           ,