U ui'@sddlZddlZddlmZddlZddlmZddlm Z m Z ddl m Z ddl mZddlmZdd lmZdd lmZddlmm Zdd lmZd ejkZGd ddZGdddZddZdS)N)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}||ks|d}nd}z:dtjd<|d }d |ksdt |d}||kszt W5tjd=Xd S) zd SETUPTOOLS_EXT_SUFFIX variable always overrides default extension options. Zfor_abi3zs.cTrFzfor_abi3.test-suffixSETUPTOOLS_EXT_SUFFIXz .test-suffixnormalznormal.test-suffixN) rrrrZ_links_to_dynamicIS_PYPYrosenvironr)rrrextexpectrrrrtest_ext_suffix_override9s$     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%r4) script_namer package_dir)rrbuildr)rfilesr3Zext2r5rrrdist_with_exampleYs     zTestBuildExt.dist_with_examplecCs|dd|dd|}|d}d|_||jtj d}dd| D}||d |d |d gks~t | ikst | dd |_|d d| D}||d d|d d|d dikst dS)Nr+.mp3&setuptools.command.build_ext.use_stubsFr/cSsg|]}|tjdqSr=replacer.sep).0xrrr tsz1TestBuildExt.test_get_outputs.. /ext3.mp3/mypkg/ext1.mp3/mypkg/subpkg/ext2.mp3TcSs*i|]"\}}|tjd|tjdqSr>r?rBkvrrr s z1TestBuildExt.test_get_outputs.. src/ext3.mp3src/mypkg/ext1.mp3src/mypkg/subpkg/ext2.mp3)setenvsetattrr:get_command_obj editable_modeensure_finalized build_libr@r.rA get_outputsrget_output_mappingreinitialize_commanditems)r tmpdir_cwd monkeypatchrrrToutputsmappingrrrtest_get_outputsjs8      zTestBuildExt.test_get_outputsc CsL|dd|dd|}|d}d|_||jD]}||ddq>|jt j d}dd | D}d d }|||d |d |dd||d|d|dd||d|d|ddikst |d|kst |d|kst ||d} | |ks"t | |ds8t | dsHt dS)Nr+r;r<TrZ _needs_stubr=cSs*i|]"\}}|tjd|tjdqSr>r?rHrrrrKs zBTestBuildExt.test_get_output_mapping_with_stub..cSst|tjdS)zDMake it possible to do comparisons and tests in a OS-independent wayr=)_compiled_file_namer@r.rA)filerrrCsz9TestBuildExt.test_get_output_mapping_with_stub..Cz/ext3.pyz src/ext3.pyrErLz/mypkg/ext1.pyzsrc/mypkg/ext1.pyrFrMz/mypkg/subpkg/ext2.pyzsrc/mypkg/subpkg/ext2.pyrGrNz/mypkg/__pycache__/ext1z.pyc)rOrPr:rQrRrS extensionsrTr@r.rArVrXr startswithr&) rrYrZrrr0rTr\r`Z example_stubrrr!test_get_output_mapping_with_stubsB         z.TestBuildExt.test_get_output_mapping_with_stubN) __name__ __module__ __qualname__rr*r2r:r]rcrrrrr s    r c@s2eZdZedddZddZddZdd Zd S) TestBuildExtInplaceoptionalcKsndiidd}t|tddg|d}tt|gd}d|_t|}t|jfd d d || |S) Nz#include missingheader.h )libtmp)rz.buildrrrhrsetup.pyz .build/libz .build/tmp)rT build_temp) rr8rrr$r6rvarsupdaterS)rrioptsr9r)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|||Ds4tdS)z If optional extensions fail to build, setuptools should show the error in the logs but not fail to build TriZinplacecss|] }dVqdS)z0build_ext: building extension "spam.eggs" failedNr)rBmsgrrr sz4TestBuildExtInplace.test_optional..N)rqrunanyrxr)rrYrvrwrrrr test_optionals  z!TestBuildExtInplace.test_optionalc Cs0|jddd}tt|W5QRXdS)NFTry)rqpytestZraisesrr|)rrYrrrrtest_non_optionals z%TestBuildExtInplace.test_non_optionalN)rdrerfboolrqrxr~rrrrrrgs  rgcCsNtdtdtdd}t|tjdgdd\}}|dksJtd |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 )rlzfoo.cz setup.cfgr8)r)rZ data_streamrz STDOUT: %s STDERR: %s)r rr8r Z run_setup_pyr)rYr9codeoutputrrrtest_build_ext_config_handlings  +=  r)r.r'importlib.utilrr^rZjaracorZsetuptools.command.build_extrrZsetuptools.distrZsetuptools.errorsrZsetuptools.extensionrrr textwrapr Zdistutils.command.build_extcommandrdistutils.sysconfigr builtin_module_namesr-r rgrrrrrs"         ,