Your IP : 216.73.216.52
��Yf�D � @ s> d d l m Z m Z m Z m Z d d l Z d d l Z d d l Z d d l m Z d d l
Z
d d l Z d d l Z d d l
Z
d d l Z d d l Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z Gd
d � d e j � Z Gd d
� d
e j � Z d d � Z e d k r:e � d S)� )� run_unittest�unload�check_warnings� CleanImportN)�spec_from_file_locationc @ sd e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)� PkgutilTestsc C s? t j � | _ | j t j | j � t j j d | j � d S)Nr ) �tempfile�mkdtemp�dirname�
addCleanup�shutil�rmtree�sys�path�insert)�self� r �6/opt/alt/python35/lib64/python3.5/test/test_pkgutil.py�setUp s zPkgutilTests.setUpc C s t j d =d S)Nr )r r )r r r r �tearDown s zPkgutilTests.tearDownc C s4 d } d } t j j | j | � } t j | � t t j j | d � d � } | j � t t j j | d � d � } | j | � | j � t j t j j | d � � t t j j | d d � d � } | j | � | j � t j | d � } | j
| | � t j | d � } | j
| | � t j | =d S)N�test_getdata_filesyss% Hello, world!
Second line
Third linez __init__.py�wbzres.txt�subz sub/res.txt)
�osr �joinr
�mkdir�open�close�write�pkgutil�get_data� assertEqualr �modules)r �pkg�
RESOURCE_DATAZ package_dir�f�res1�res2r r r r s$
!
z!PkgutilTests.test_getdata_filesysc C s= d } d } d } t j j | j | � } t j | d � } | j | d d � | j | d | � | j | d | � | j � t j j d | � t
j | d
� } | j | | � t
j | d � } | j | | � g } x0 t
j
| g � D] \ } }
} | j |
� q� W| j | d g � t j d =t j | =d S) Nztest_getdata_zipfile.zip�test_getdata_zipfiles% Hello, world!
Second line
Third line�wz /__init__.py� z/res.txtz /sub/res.txtr zres.txtz sub/res.txt)r r r r
�zipfile�ZipFile�writestrr r r r r r! Z iter_modules�appendr" ) r �zipr# r$ �zip_file�zr& r' �names�loader�nameZispkgr r r r( : s(
"
z!PkgutilTests.test_getdata_zipfilec C su d } t j j | j | � } t j | d � | j t j | � x- t j d | j g � D] } | j d � qZ Wd S)NZunreadable_packager r zunexpected package found)
r r r r
r r �rmdirr �
walk_packagesZfail)r Z package_name�d�tr r r � test_unreadable_dir_on_syspath\ s z+PkgutilTests.test_unreadable_dir_on_syspathc C s d } t j j | j | � } t j | � t t j j | d � d � } | j � t j t j j | d � � t t j j | d d � d � } | j � t t j j | d d � d � } | j � d } t j j | j | � } t j | � t t j j | d � d � } | j � t j t j j | d � � t t j j | d d � d � } | j � t t j j | d d � d � } | j � d d d d d d g } d
d � t j | j g � D� } | j | | � x* | D]" } | j
d � r�q�t j | =q�Wd S)
N�test_walkpackages_filesysz __init__.pyr r zmod.pyz sub.test_walkpackages_filesysz!sub.test_walkpackages_filesys.modz test_walkpackages_filesys.subz!test_walkpackages_filesys.sub.modc S s g | ] } | d � q S)� r )�.0�er r r �
<listcomp>� s z:PkgutilTests.test_walkpackages_filesys.<locals>.<listcomp>�mod)
r r r r
r r r r r6 r! �endswithr r" ) r �pkg1Zpkg1_dirr% �pkg2Zpkg2_dir�expected�actualr# r r r r: g s@
!
!
!
!
"
z&PkgutilTests.test_walkpackages_filesysc C ss d } d } d } t j j | j | � } t j | d � } | j | d d � | j | d | d d � | j | d | d d � | j | d d � | j | d | d d � | j | d | d d � | j � t j j d | � d d
d d d d
g } d d � t
j | g � D� } | j | | � t j d =x* | D]" } | j
d � raqIt j | =qIWd S)zBTests the same as test_walkpackages_filesys, only with a zip file.z test_walkpackages_zipfile.zip�test_walkpackages_zipfiler r) z /__init__.pyr* �/z/mod.pyr z sub.test_walkpackages_zipfilez!sub.test_walkpackages_zipfile.modz test_walkpackages_zipfile.subz!test_walkpackages_zipfile.sub.modc S s g | ] } | d � q S)r; r )r< r= r r r r> � s z:PkgutilTests.test_walkpackages_zipfile.<locals>.<listcomp>r? N)r r r r
r+ r, r- r r r r r6 r! r@ r" ) r r/ rA rB r0 r1 rC rD r# r r r rE � s4
z&PkgutilTests.test_walkpackages_zipfileN)
�__name__�
__module__� __qualname__r r r r( r9 r: rE r r r r r s " (r c @ sl e Z d Z Gd d � d e � Z Gd d � d e � Z d d � Z d d � Z d d
� Z d d � Z d
S)�PkgutilPEP302Testsc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)z PkgutilPEP302Tests.MyTestLoaderc C s d S)Nr )r �specr r r �
create_module� s z-PkgutilPEP302Tests.MyTestLoader.create_modulec C s$ | j j d d � d | j d <d S)N�loadsr r; )�__dict__�get)r r? r r r � exec_module� s z+PkgutilPEP302Tests.MyTestLoader.exec_modulec C s d S)Nz
Hello, world!r )r r r r r r � s z(PkgutilPEP302Tests.MyTestLoader.get_dataN)rG rH rI rL rP r r r r r � MyTestLoader� s rQ c @ s" e Z d Z d d d d � Z d S)z!PkgutilPEP302Tests.MyTestImporterNc C s/ t j � } t | d | j j d | d g �S)Nz<%s>r3 �submodule_search_locations)rJ rQ r � __class__rG )r �fullnamer �targetr3 r r r � find_spec� s
z+PkgutilPEP302Tests.MyTestImporter.find_spec)rG rH rI rV r r r r �MyTestImporter� s rW c C s t j j d | j � � d S)Nr )r � meta_pathr rW )r r r r r � s zPkgutilPEP302Tests.setUpc C s t j d =d S)Nr )r rX )r r r r r � s zPkgutilPEP302Tests.tearDownc C s* | j t j d d � d � t j d =d S)N�foo�dummyz
Hello, world!)r! r r r r" )r r r r �test_getdata_pep302� s z&PkgutilPEP302Tests.test_getdata_pep302c C s\ d d l } | j | j d � | j t j d d � d � | j | j d � t j d =d S)Nr r; rY rZ z
Hello, world!)rY r! rM r r r r" )r rY r r r �test_alreadyloaded� s
z%PkgutilPEP302Tests.test_alreadyloadedN)
rG rH rI �objectrQ rW r r r[ r\ r r r r rJ � s rJ c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �ExtendPathTestsc C sz t j � } t j j d | � t j j | | � } t j | � t t j j | d � d � � } | j d � Wd QRX| S)Nr z __init__.pyr) zKfrom pkgutil import extend_path
__path__ = extend_path(__path__, __name__)
)
r r r r r r r r r r )r �pkgnamer
Zpkgdir�flr r r � create_init� s
!zExtendPathTests.create_initc C sR t j j | | | d � } t | d � � } t d j | � d | �Wd QRXd S)Nz.pyr) zvalue={}�file)r r r r �print�format)r r
r_ Zsubmodule_name�valueZ module_namer` r r r �create_submodule� s z ExtendPathTests.create_submodulec C s' d } | j | � } | j | � } | j | | d d � | j | | d d � d d l } d d l } | j | j j d � | j | j j d � | j t | j � t t
j j | | � t
j j | | � g � � t
j | � t
j | � t j d =t j d =t j d =t j d =t j d =d S)NrY �barr �bazr; zfoo.barzfoo.baz)ra rf �foo.bar�foo.bazr! rg re rh �sorted�__path__r r r r r
r r" )r r_ � dirname_0� dirname_1rY r r r � test_simple� s&
zExtendPathTests.test_simplec
C s� t j } t j } d } d } | j | � } t j j | | � } d j | | � } t j j
| d � t j j
| d � z6| j | | | d � t j
| � t | | � � } | | � } x� | D]� }
t j |
| � } | j } y
| j } Wn t k
rYn X| j |
t j j � | j |
| � | j | t j j � | j t j |
| � � q� W| j t � � t | d � � Wd QRX| j t � � t | d � � Wd QRXWd t j | � t j d =y t j d =t j d =Wn t k
r�Yn XXd S)N�spamZeggsz{}.{}r zinvalid.modulez.spamz spam.eggs) r �iter_importers� get_importerra r r r rd r r" �poprf � importlib�
import_module�list� _get_specr3 �AttributeErrorZassertIsInstance� machinery�
FileFinderr! �SourceFileLoader� assertIsNoneZ assertRaises� ImportErrorr r
�KeyError)
r rq rr r_ �modnamer
ZpathitemrT Z importersZexpected_importer�finderrK r3 r r r �test_iter_importers sJ
z#ExtendPathTests.test_iter_importersc C sF d } | j | � } | j | � } | j | | d d � t j t j j | | d � � | j | | d d � d d l } d d l } | j | j j
d � | j | j j
d � | j t | j
� t t j j | | � t j j | | � g � � t j | � t j | � t j d =t j d =t j d =t j d =t j d =d S) NrY rg r z __init__.pyrh r; zfoo.barzfoo.baz)ra rf r �unlinkr r ri rj r! rg re rh rk rl r r
r r" )r r_ rm rn rY r r r �test_mixed_namespace? s(
z$ExtendPathTests.test_mixed_namespaceN)rG rH rI ra rf ro r� r� r r r r r^ � s
0r^ c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�NestedNamespacePackageTestc C s) t j � | _ t j d d � | _ d S)N)r r �basedirr r �old_path)r r r r r ` s z NestedNamespacePackageTest.setUpc C s* | j t j d d � <t j | j � d S)N)r� r r r r
r� )r r r r r d s z#NestedNamespacePackageTest.tearDownc C s� | j d d � \ } } t j j | j | j d t j j � � } t j | d d �t t j j | | d � d � � } | j | � Wd QRXd S)N�.r; �exist_okTz.pyr) )
�rsplitr r r r� �replace�sep�makedirsr r )r r4 �contents�base�final� base_pathr% r r r rL h s
*%z(NestedNamespacePackageTest.create_modulec C sa d } | j d | � | j d | � | j d | � | j d | � | j d d � | j d d � t j j d
t j j | j d � � t j j d
t j j | j d � � d
d l } | j t d
� | j
t | j � d � d
d l
} | j t d � | j
t | j j � d � d
d l m } d
d l m } | j
| d � | j
| d � d S)NzBimport pkgutil; __path__ = pkgutil.extend_path(__path__, __name__)za.pkg.__init__zb.pkg.__init__za.pkg.subpkg.__init__zb.pkg.subpkg.__init__za.pkg.subpkg.czc = 1zb.pkg.subpkg.dzd = 2r �a�br# � z
pkg.subpkg)�c)r7 r; )rL r r r r r r� r# r r r! �lenrl Z
pkg.subpkgZsubpkgZ pkg.subpkg.cr� Z pkg.subpkg.dr7 )r Zpkgutil_boilerplater# r� r7 r r r � test_nestedo s& %% z&NestedNamespacePackageTest.test_nestedN)rG rH rI r r rL r� r r r r r� ^ s r� c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d S)�ImportlibMigrationTestsc C s t d t f � S)Nz5This emulation is deprecated, use 'importlib' instead)r �DeprecationWarning)r r r r �check_deprecated� s z(ImportlibMigrationTests.check_deprecatedc C s% | j � � t j d � Wd QRXd S)Nr* )r� r Z ImpImporter)r r r r �test_importer_deprecated� s
z0ImportlibMigrationTests.test_importer_deprecatedc C s. | j � � t j d d d d � Wd QRXd S)Nr* )r� r Z ImpLoader)r r r r �test_loader_deprecated� s
z.ImportlibMigrationTests.test_loader_deprecatedc
C sr t � �b } | j t j d � � | j t j d � � | j t j d � � | j t | j � d � Wd QRXd S)Nr r z test.supportr )r �assertIsNotNoner �
get_loaderr! r� �warnings)r r) r r r � test_get_loader_avoids_emulation� s
z8ImportlibMigrationTests.test_get_loader_avoids_emulationc C s] t } b zF t � �6 } | j t j t � � | j t | j � d � Wd QRXWd | a Xd S)Nr ) �
__loader__r r� r r� rG r! r� r� )r Z this_loaderr) r r r �0test_get_loader_handles_missing_loader_attribute� s $zHImportlibMigrationTests.test_get_loader_handles_missing_loader_attributec
C s_ d } t t � | � } | ` t | � �! | t j | <t j | � } Wd QRX| j | � d S)Nrp )�typer �__spec__r r" r r� r| )r r4 r? r3 r r r �.test_get_loader_handles_missing_spec_attribute� s
zFImportlibMigrationTests.test_get_loader_handles_missing_spec_attributec
C sb d } t t � | � } d | _ t | � �! | t j | <t j | � } Wd QRX| j | � d S)Nrp )r� r r� r r" r r� r| )r r4 r? r3 r r r �+test_get_loader_handles_spec_attribute_none� s
zCImportlibMigrationTests.test_get_loader_handles_spec_attribute_nonec C sE d } d t j | <z t j | � } Wd t j | =X| j | � d S)Nz
totally bogus)r r" r r� r| )r r4 r3 r r r �#test_get_loader_None_in_sys_modules� s
z;ImportlibMigrationTests.test_get_loader_None_in_sys_modulesc C s& d } t j | � } | j | � d S)Nz
totally bogus)r � find_loaderr| )r r4 r3 r r r � test_find_loader_missing_module� s z7ImportlibMigrationTests.test_find_loader_missing_modulec
C sr t � �b } | j t j d � � | j t j d � � | j t j d � � | j t | j � d � Wd QRXd S)Nr r z test.supportr )r r� r r� r! r� r� )r r) r r r �!test_find_loader_avoids_emulation� s
z9ImportlibMigrationTests.test_find_loader_avoids_emulationc
C sF t � �6 } | j t j d � � | j t | j � d � Wd QRXd S)Nz*??r )r r| r rr r! r� r� )r r) r r r �"test_get_importer_avoids_emulation� s z:ImportlibMigrationTests.test_get_importer_avoids_emulationc
C sG t � �7 } x t j � D] } q W| j t | j � d � Wd QRXd S)Nr )r r rq r! r� r� )r r) Zimporterr r r �$test_iter_importers_avoids_emulation� s z<ImportlibMigrationTests.test_iter_importers_avoids_emulationN)rG rH rI r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c C sI t t t t t t � d d l } d d l } | j j � | j
� d S)Nr ) r r rJ r^ r� r� � zipimportrt �_zip_directory_cache�clear�invalidate_caches)r� rt r r r � test_main� s
r� �__main__)Z test.supportr r r r Zunittestr rt �importlib.utilr r r Zos.pathr r r+ ZTestCaser rJ r^ r� r� r� rG r r r r �<module> s$ " �-})W