Your IP : 216.73.216.213
���]�� � @ s d d l 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 d d l Z d d l Z d d l Z d d l
m Z d d l
Z
d d l m
Z
d d l m Z e e d � r� d d l Z d Z n d Z e j e j k Z e e d � Z e j j d � rd Z n d
Z Gd d � d e
j � Z Gd
d � d e
j � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z! e j" d d � � Z# d d � Z$ Gd d � d � Z% Gd d � d e% e � Z& Gd! d" � d" e � Z' Gd# d$ � d$ e � Z( Gd% d& � d& e � Z) Gd' d( � d( e% e � Z* Gd) d* � d* e � Z+ Gd+ d, � d, e � Z, Gd- d. � d. e � Z- e j. e j/ k r�Gd/ d0 � d0 e � Z0 Gd1 d2 � d2 � Z1 Gd3 d4 � d4 e � Z2 e3 d5 k re
j4 � d S)6� N)�mock)�support)�
script_helper�stat� �spawnlZopenbsd�0 �d c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�TestLowLevelInternalsc C sO | j t t j d � � | j t t j d � � | j t t j d � � d S)N� � )�assertIs�str�tempfile�_infer_return_type�bytes)�self� r �2/opt/alt/python35/lib64/python3.5/test_tempfile.py� test_infer_return_type_singles( s z4TestLowLevelInternals.test_infer_return_type_singlesc C s� | j t t j d d � � | j t t j d d � � | j t � � t j d d � Wd QRX| j t � � t j d d � Wd QRXd S)Nr r )r
r r r r � assertRaises� TypeError)r r r r � test_infer_return_type_multiples- s z6TestLowLevelInternals.test_infer_return_type_multiplesc C s� | j t t j d d � � | j t t j d d � � | j t t j d d � � | j t t j d d � � | j t t j d d � � | j t � � t j d d d � Wd QRX| j t � � t j d d d � Wd QRXd S)Nr r )r
r r r r r r )r r r r �)test_infer_return_type_multiples_and_none5 s z?TestLowLevelInternals.test_infer_return_type_multiples_and_noneN)�__name__�
__module__� __qualname__r r r r r r r r
' s r
c @ sR e Z d Z e j d � Z e j d � Z d d � Z d d � Z d d � Z d S)
� BaseTestCasez^[a-z0-9_-]{8}$s ^[a-z0-9_-]{8}$c C s? t j � | _ | j j � t j d d t d d d t �d S)N�ignore�category�message�mktemp�module)r �check_warnings�_warnings_manager� __enter__�warnings�filterwarnings�RuntimeWarningr )r r r r �setUpH s
zBaseTestCase.setUpc C s | j j d d d � d S)N)r$ �__exit__)r r r r �tearDownN s zBaseTestCase.tearDownc
C s t j j | � \ } } | d t | � � } | t | � t | � d � } | d k r� | j t | � t | � t k r� t n t d � | d k r� | j t | � t | � t k r� t n t d � | d k r| j t | � t | � t k r� t n t d � | | | f d k r3| j t | � t d � | j t j j | � t j j | � d | | f � | j | | d | | f � | j | | d | | f � | t | � t | � t | � � } t
| t � r�| j n | j } | j
| j | � d | | j f � d S)Nzunexpected return typez default return type must be strzfile %r not in directory %rz file %r does not begin with %rz file %r does not end with %rz$random characters %r do not match %r)NNN)�os�path�split�lenr
�typer r � assertEqual�abspath�
isinstance� str_check�b_check�
assertTrue�match�pattern)
r �name�dir�pre�suf�ndir�nbaseZnpreZnsuf�checkr r r � nameCheckR s2 * * *$ &!zBaseTestCase.nameCheckN)
r r r �re�compiler4 r5 r) r+ r@ r r r r r C s
r c @ s e Z d Z d d � Z d S)� TestExportsc C s� t j } d d d d d d d d d d d d d d d d d
d d d d d d
d d d d d i } g } x7 | D]/ } | d d k rp | | k rp | j | � qp W| j t | � d k d | � d S)N�NamedTemporaryFiler �
TemporaryFile�mkstemp�mkdtempr! �TMP_MAX�
gettempprefix�gettempprefixb�
gettempdir� gettempdirb�tempdir�template�SpooledTemporaryFile�TemporaryDirectoryr �_zunexpected keys: %s)r �__dict__�appendr6 r/ )r �dict�expectedZunexp�keyr r r � test_exportss s*
zTestExports.test_exportsN)r r r rW r r r r rC r s rC c sm e Z d Z � f d d � Z d d � Z d d � Z d d � Z e j e e
d � d
� d d � � Z � S)
�TestRandomNameSequencec s t j � | _ t � j � d S)N)r �_RandomNameSequence�r�superr) )r )� __class__r r r) � s z TestRandomNameSequence.setUpc C s) t | j � } | j | d d d � d S)Nr )�nextrZ r@ )r �sr r r �test_get_six_char_str� s z,TestRandomNameSequence.test_get_six_char_strc C sf i } | j } xP t t � D]B } t | � } | j | d d d � | j | | � d | | <q Wd S)Nr r )rZ �range�
TEST_FILESr] r@ � assertNotIn)r rT rZ �ir^ r r r � test_many� s z TestRandomNameSequence.test_manyc C s; d } | j } x% | D] } | d 7} | d k r Pq Wd S)Nr r � )rZ )r rc rZ r^ r r r �
supports_iter� s
z$TestRandomNameSequence.supports_iter�forkz!os.fork is required for this testc C s t j � \ } } d } z� t j � } | sv t j | � t j | t | j � j d � � t j | � t j d � t | j � } t j | t
| � � j d � } Wd | r� y t j | t
j � Wn t k
r� Yn Xt j | � t j | � X| j | | � d S)N�asciir )r, �piperg �close�writer] rZ �encode�_exit�readr/ �decode�kill�signal�SIGKILL�OSError�assertNotEqual)r Zread_fdZwrite_fd�pidZ parent_valueZ child_valuer r r �test_process_awareness� s&
"
%
z-TestRandomNameSequence.test_process_awareness) r r r r) r_ rd rf �unittest�
skipUnless�hasattrr, rv r r )r\ r rX � s
rX c @ s( e Z d Z d d � Z d d � Z d S)�TestCandidateTempdirListc C sJ t j � } | j t | � d k � x | D] } | j | t � q, Wd S)Nr )r �_candidate_tempdir_list� assertFalser/ �assertIsInstancer )r �cand�cr r r �test_nonempty_list� s
z+TestCandidateTempdirList.test_nonempty_listc C s� t j � �� } x9 d D]1 } t j | � } | s t j j | � | | <q Wt j � } x9 d D]1 } t j | � } | s t � | j | | � q^ Wy t j
� } Wn! t t f k
r� t j
} Yn X| j | | � Wd QRXd S)N�TMPDIR�TEMP�TMP)r� r� r� )r� r� r� )r �EnvironmentVarGuardr, �getenvr- r2 r r{ �
ValueError�assertIn�getcwd�AttributeErrorrs �curdir)r �env�envname�dirnamer~ r r r �test_wanted_dirs� s
z)TestCandidateTempdirList.test_wanted_dirsN)r r r r� r� r r r r rz � s rz c @ s e Z d Z d d � Z d S)�TestGetDefaultTempdirc - s= t j � �*� � f d d � } t j t d | � �� t j � | j t j � � g � d d � � t j t d � � �? | j t
� � t j � Wd QRX| j t j � � g � Wd QRX� � f d d � } t j t d | � �A � | j t
� � t j � Wd QRX| j t j � � g � Wd QRXWd QRXWd QRXd S) Nc s � g S)Nr r )�our_temp_directoryr r �our_candidate_list s zKTestGetDefaultTempdir.test_no_files_left_behind.<locals>.our_candidate_listr{ c _ s
t � � d S)N)rs )�args�kwargsr r r �
raise_OSError s zFTestGetDefaultTempdir.test_no_files_left_behind.<locals>.raise_OSError�openc s � | | � } � | _ | S)N)rk )r� r� �fp)� orig_openr� r r �
bad_writer s zCTestGetDefaultTempdir.test_no_files_left_behind.<locals>.bad_writer) r rP r � swap_attr�_get_default_tempdirr1 r, �listdir�ior �FileNotFoundError)r r� r� r )r� r� r� r �test_no_files_left_behind� s
z/TestGetDefaultTempdir.test_no_files_left_behindN)r r r r� r r r r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�TestGetCandidateNamesc C s# t j � } | j | t j � d S)N)r �_get_candidate_namesr} rY )r �objr r r � test_retval# s z!TestGetCandidateNames.test_retvalc C s/ t j � } t j � } | j | | k � d S)N)r r� r6 )r �a�br r r �test_same_thing( s z%TestGetCandidateNames.test_same_thingN)r r r r� r� r r r r r� s r� c c sG t j � } z& t j t d | � �
d VWd QRXWd t j | � Xd S)NrM )r rG r r� �rmtree)r: r r r �_inside_empty_temp_dir0 s
r� c s t j t d � f d d � � S)Nr� c s
t � � S)N)�iterr )�namesr r �<lambda>= s z'_mock_candidate_names.<locals>.<lambda>)r r� r )r� r )r� r �_mock_candidate_names: s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�TestBadTempdirc C s� t � �� t j t j � j } } | t j t j Bt j BM} t j t j | � zf t j
t j t j � rz | j d � | j
t � � | j � Wd QRX| j t j t j � g � Wd t j t j | � XWd QRXd S)Nz!can't set the directory read-only)r� r, r r rM �st_mode�S_IWUSR�S_IWGRP�S_IWOTH�chmod�access�W_OKZskipTestr �PermissionError� make_tempr1 r� )r Zoldmode�moder r r �test_read_only_directoryB s
z'TestBadTempdir.test_read_only_directoryc C sk t � �[ t j j t j d � } t j t d | � �& | j t � � | j
� Wd QRXWd QRXWd QRXd S)N� nonexistentrM ) r� r, r- �joinr rM r r� r r� r� )r rM r r r �test_nonexisting_directoryP s
z)TestBadTempdir.test_nonexisting_directoryc C s� t � �t t j j t j d � } t | d � j � t j t d | � �, | j
t t f � � | j
� Wd QRXWd QRXWd QRXd S)N�file�wbrM )r� r, r- r� r rM r� rj r r� r �NotADirectoryErrorr� r� )r rM r r r �test_non_directoryW s
z!TestBadTempdir.test_non_directoryN)r r r r� r� r� r r r r r� @ s r� c @ s� e Z d Z Gd d � d � Z d d d d d d � Z d d � Z d d
� Z d d � Z d
d � Z e j
e d � d d � � Z e j
e
d � d d � � Z e j
e d � d d � � Z d d � Z d d � Z d d � Z d S) �TestMkstempInnerc @ sX e Z d Z e j Z e j Z e j Z
e j Z d d � Z
d d � Z d d � Z d S)zTestMkstempInner.mkstempedc C s[ | r | j } n | j } t j | | | � } t j | | | | | � \ | _ | _ d S)N)�_bflags�_tflagsr r �_mkstemp_inner�fdr9 )r r: r; r< �bin�flags� output_typer r r �__init__i s
z#TestMkstempInner.mkstemped.__init__c C s t j | j | � d S)N)r, rk r� )r r r r r rk p s z TestMkstempInner.mkstemped.writec C s$ | j | j � | j | j � d S)N)�_closer� �_unlinkr9 )r r r r �__del__s s z"TestMkstempInner.mkstemped.__del__N)r r r r �_bin_openflagsr� �_text_openflagsr� r, rj r� �unlinkr� r� rk r� r r r r � mkstempedc s r� Nr c C s� t j | | | � } | d k rH | t k r< t j � } n t j � } | d k r] | � } | d k rr | � } | j | | | | � } | j | j | | | � | S)N)r r r rK rL r� r@ r9 )r r: r; r< r� r� r� r r r � do_createw s zTestMkstempInner.do_createc C s� | j � j d � | j d d � j d � | j d d � j d � | j d d d d � j d � | j d d d d � j d � d S)Ns blatr; r� r< r� �aaz.txt)r� rk )r r r r �
test_basic� s
zTestMkstempInner.test_basicc C s_ t j � } | j d | d d � j d � | j d | d d � j d � | j d | d d � j d � | j d | d d d d � j d � | j d | d d d d � j d � | j t � �$ | j d d
d d � j d � Wd QRX| j t � �$ | j d | d d
� j d � Wd QRX| j t � �* | j d | d d d d
� j d � Wd QRXd S) Nr: r< r s blatr; � a� bs aas .txtr )r rL r� rk r r )r Zdir_br r r �test_basic_with_bytes_names� s %%&&z,TestMkstempInner.test_basic_with_bytes_namesc C s= t t t � � } x$ | D] } | j d d � | | <q Wd S)Nr; r� )�listr` ra r� )r �extantrc r r r �test_basic_many� s
z TestMkstempInner.test_basic_manyc C s> t j � } z | j d | � j d � Wd t j | � Xd S)Nr: s blat)r rG r� rk r, �rmdir)r r: r r r �test_choose_directory� s z&TestMkstempInner.test_choose_directoryzos.stat not availablec C sg | j � } t j t j | j � j � } d } t j d k rS | d ?} | d } | j | | � d S) Ni� �win32� r � �@ � �I ) r� r �S_IMODEr, r9 r� �sys�platformr1 )r r� r� rU �userr r r �test_file_mode� s
z TestMkstempInner.test_file_modezos.spawnl not availablec C s5 t j r d } n d } | j � } | j t j | j � d � d | j } y
t } Wn t k
rx t j
d } Yn Xt j j t j j
t j j | � � d � } t j d k r� d t j } d | } n t j } t j t j t j | | | | � } | j | d k d | � | j | d k d
| � d S) N�v�qFz%dr ztf_inherit_check.pyr� z"%s"z$child process caught fatal signal %dz child process reports failure %d)r �verboser� r1 r, �get_inheritabler� �__file__� NameErrorr� �argvr- r� r� r2 r� �
executabler �P_WAITr| )r r� r� r� �meZtesterZ decoratedZretvalr r r �test_noinherit� s(
$
$ z TestMkstempInner.test_noinheritztext mode not availablec C sh | j d d � } | j d � | j d � t j | j d t j � | j t j | j d � d � d S)Nr� r s blats extra
re s blat)r� rk r, �lseekr� �SEEK_SETr1 rn )r �fr r r �
test_textmode� s
z TestMkstempInner.test_textmodec C s( t j t j � t j � d t j t � S)Nr )r r� rK rI r� r )r r r r r� � s
zTestMkstempInner.make_tempc C s� t � �� t d d d � �o | j � \ } } t j | � | j | j d � � | j � \ } } t j | � | j | j d � � Wd QRXWd QRXd S)N�aaa�bbb)r� r� r� r, rj r6 �endswith)r �fd1Zname1�fd2Zname2r r r �!test_collision_with_existing_file� s
z2TestMkstempInner.test_collision_with_existing_filec C s� t � �v t d d d � �\ t j � } | j | j d � � | j � \ } } t j | � | j | j d � � Wd QRXWd QRXd S)Nr� r� ) r� r� r rG r6 r� r� r, rj )r r: r� r9 r r r �&test_collision_with_existing_directory s
z7TestMkstempInner.test_collision_with_existing_directory)r r r r� r� r� r� r� r� rw rx �has_statr� �
has_spawnlr� � has_textmoder� r� r� r� r r r r r� ` s &
r� c @ s( e Z d Z d d � Z d d � Z d S)�TestGetTempPrefixc C sh t j � } | j | t � | j t | � d � t j � } | j | t � | j t | � d � d S)Nr )r rI r} r Z
assertGreaterr/ rJ r )r �pZpbr r r �test_sane_template s z$TestGetTempPrefix.test_sane_templatec C s� t j � d } t j d d � } zO t j j | | � } t j | t j t j B� } t j | � t j
| � Wd t j | � Xd S)Nz
xxxxxx.xxx�prefixr ) r rI rG r, r- r� r� �O_RDWR�O_CREATrj r� r� )r r� �dr� r r r �test_usable_template s
z&TestGetTempPrefix.test_usable_templateN)r r r r 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 S)
�TestGetTempDirc C sv xo t j � t j � f D]U } | j t j j | � pC | t j k d | � | j t j j | � d | � q Wd S)Nz%r is not an absolute pathz%r is not a directory) r rK rL r6 r, r- �isabsr� �isdir)r r r r r �test_directory_exists1 s
$ z$TestGetTempDir.test_directory_existsc C s' t j � } | j d � | j � d S)Ns blat)r rD rk rj )r r� r r r �test_directory_writable: s
z&TestGetTempDir.test_directory_writablec C sp t j � } t j � } t j � } | j | | k � | j t | � t | � � | j | t j | � � d S)N) r rK rL r6 rt r0 r1 r, �fsdecode)r r� r� r r r r r� D s z TestGetTempDir.test_same_thingc C sz t j d � } t j d } t _ z: t j � �' } | | d <| j t j � | � Wd QRXWd | t _ t j | � Xd S)Nz-Tempr� )r rG rM r r� r1 rK r� )r Zcase_sensitive_tempdirZ_tempdirr� r r r �test_case_sensitiveN s
! z"TestGetTempDir.test_case_sensitiveN)r r r r r
r� r r r r r r . s
r c @ sI e Z d Z d d d d d � Z d d � Z d d � Z d d � Z d S)
� TestMkstempNc
C s t j | | | � } | d k rH | t k r< t j � } n t j � } | d k r] | � } | d k rr | � } t j d | d | d | � \ } } t j j | � \ } } t j j | � } | j
| | d | | f � z | j | | | | � Wd t j | � t j
| � Xd S)Nr: r �suffixz+Directory '%s' incorrectly returned as '%s')r r r rK rL rF r, r- r. r2 r1 r@ rj r� )
r r: r; r< r� r� r9 r= r> Zadirr r r r� ` s$ $
zTestMkstemp.do_createc C sj | j � | j d d � | j d d � | j d d d d � | j d d d d � | j d d � d S) Nr; r� r< r� r� z.txtr: �.)r� )r r r r r� w s
zTestMkstemp.test_basicc C s3 t j � } | j d | d d � | j d | d d � | j d | d d � | j d | d d d d � | j d | d d d d � | j d d � | j t � �! | j d d
d d d d � Wd QRX| j t � �! | j d d d d d d � Wd QRX| j t � �! | j d d d d d d � Wd QRXd S)
Nr: r< r r; r� r� s aas .txt� .r r� z.txt)r rL r� r r )r r r r r r� � s ##z'TestMkstemp.test_basic_with_bytes_namesc C s5 t j � } z | j d | � Wd t j | � Xd S)Nr: )r rG r� r, r� )r r: r r r r� � s z!TestMkstemp.test_choose_directory)r r r r� r� r� r� r r r r r
] s r
c @ s� e Z d Z d d � Z d d d d d � Z d d � Z d d � Z d
d � Z d d
� Z e j
e d � d d � � Z d d � Z
d d � Z d S)� TestMkdtempc C s
t j � S)N)r rG )r r r r r� � s zTestMkdtemp.make_tempNc C s� t j | | | � } | d k rH | t k r< t j � } n t j � } | d k r] | � } | d k rr | � } t j d | d | d | � } y | j | | | | � | SWn t j | � � Yn Xd S)Nr: r r ) r r r rK rL rG r@ r, r� )r r: r; r< r� r9 r r r r� � s
zTestMkdtemp.do_createc C s� t j | j � � t j | j d d � � t j | j d d � � t j | j d d d d � � t j | j d d d d � � d S)Nr; r� r< r� r� z.txt)r, r� r� )r r r r r� � s
zTestMkdtemp.test_basicc C se t j � } t j | j d | � � t j | j d | d d � � t j | j d | d d � � t j | j d | d d d d � � t j | j d | d d d d � � | j t � �* t j | j d | d d d d � � Wd QRX| j t � �* t j | j d | d d d d � � Wd QRX| j t � �* t j | j d d
d d d d � � Wd QRXd S) Nr: r; r� r< r� s aas .txtr� z.txtr )r rL r, r� r� r r )r r r r r r� � s %%,,z'TestMkdtemp.test_basic_with_bytes_namesc
C sr t t t � � } z+ x$ | D] } | j d d � | | <q WWd x* | D]" } t | t � rG t j | � qG WXd S)Nr; r� )r� r` ra r� r3 r r, r� )r r� rc r r r r� � s
zTestMkdtemp.test_basic_manyc C s> t j � } z t j | j d | � � Wd t j | � Xd S)Nr: )r rG r, r� r� )r r: r r r r� � s z!TestMkdtemp.test_choose_directoryzos.stat not availablec C s� | j � } zb t j t j | � j � } | d M} d } t j d k r] | d ?} | d } | j | | � Wd t j | � Xd S)
Ni� i� r� r� r r� r� r� r� ) r� r r� r, r� r� r� r1 r� )r r: r� rU r� r r r � test_mode� s
zTestMkdtemp.test_modec C s� t � �v t d d d � �\ t j d d � } | j � | j | j j d � � t j � } | j | j d � � Wd QRXWd QRXd S)Nr� r� �deleteF) r� r� r rD rj r6 r9 r� rG )r r� r: r r r r� � s
z-TestMkdtemp.test_collision_with_existing_filec C ss t � �c t d d d � �I t j � } | j | j d � � t j � } | j | j d � � Wd QRXWd QRXd S)Nr� r� )r� r� r rG r6 r� )r Zdir1Zdir2r r r r� � s
z2TestMkdtemp.test_collision_with_existing_directory)r r r r� r� r� r� r� r� rw rx r� r r� r� r r r r r � s r c sq e Z d Z � f d d � Z � f d d � Z Gd d � d � Z d d d d � Z d
d � Z d d
� Z � S)�
TestMktempc s t j � | _ t � j � d S)N)r rG r: r[ r) )r )r\ r r r) s zTestMktemp.setUpc s3 | j r" t j | j � d | _ t � j � d S)N)r: r, r� r[ r+ )r )r\ r r r+ s zTestMktemp.tearDownc @ s: e Z d Z e j Z e j Z d d � Z d d � Z
d S)zTestMktemp.mktempedc C sG t j d | d | d | � | _ t j t j | j | j d � � d S)Nr: r r i� )r r! r9 r, rj r� r� )r r: r; r< r r r r� s !z TestMktemp.mktemped.__init__c C s | j | j � d S)N)r� r9 )r r r r r� s zTestMktemp.mktemped.__del__N) r r r r, r� r� r r� r� r� r� r r r r �mktemped s r r c C s8 | j | j | | � } | j | j | j | | � | S)N)r r: r@ r9 )r r; r< r� r r r r� # s zTestMktemp.do_createc C sZ | j � | j d d � | j d d � | j d d d d � | j d d d d � d S)Nr; r� r< r� r� z.txt)r� )r r r r r� ) s
zTestMktemp.test_basicc C s= t t t � � } x$ | D] } | j d d � | | <q Wd S)Nr; r� )r� r` ra r� )r r� rc r r r rd 1 s
zTestMktemp.test_many) r r r r) r+ r r� r� rd r r )r\ r r s
r c @ s� e Z d Z d d d d 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)�TestNamedTemporaryFileNr Tc C sY | d k r t j � } t j d | d | d | d | � } | j | j | | | � | S)Nr: r r r )r rK rD r@ r9 )r r: r; r< r r� r r r r� F s z TestNamedTemporaryFile.do_createc C sZ | j � | j d d � | j d d � | j d d d d � | j d d d d � d S)Nr; r� r< r� r� z.txt)r� )r r r r r� P s
z!TestNamedTemporaryFile.test_basicc C sp | j � } t j | � } | j } | j } ~ | d � ~ | d � ~ t j d d � rl | j | � � d S)Ns foos barZcpythonT)r� �weakref�refrk r Zcheck_impl_detail� assertIsNone)r r� �wrrk Zwrite2r r r �test_method_lookupX s
z)TestNamedTemporaryFile.test_method_lookupc ss d d d g � � f d d � } x1 t | � � D] \ } } | j | � | � q1 W| j | t � � d � d S)Ns spam
s eggs
s beans
c s9 t j d d � } | j d j � � � | j d � | S)Nr� zw+br r )r rD rk r� �seek)r� )�linesr r � make_filel s
z3TestNamedTemporaryFile.test_iter.<locals>.make_filer )� enumerater1 r/ )r r rc �lr )r r � test_iterh s
z TestNamedTemporaryFile.test_iterc C s6 t j � } | j t j j | j � d | j � d S)Nz$NamedTemporaryFile %s does not exist)r rD r6 r, r- �existsr9 )r r� r r r �test_creates_namedu s z)TestNamedTemporaryFile.test_creates_namedc C st t j � } zS t j d | � } | j d � | j � | j t j j | j � d | j � Wd t j
| � Xd S)Nr: s blatz(NamedTemporaryFile %s exists after close) r rG rD rk rj r| r, r- r" r9 r� )r r: r� r r r �test_del_on_close{ s
z(TestNamedTemporaryFile.test_del_on_closec C s� t j � } d } zb t j d | d d � } | j } | j d � | j � | j t j j | j � d | j � Wd | d k r� t j
| � t j | � Xd S)Nr: r Fs blatz)NamedTemporaryFile %s missing after close) r rG rD r9 rk rj r6 r, r- r" r� r� )r r: �tmpr� r r r �test_dis_del_on_close� s
z,TestNamedTemporaryFile.test_dis_del_on_closec C s; t j � } | j d � | j � | j � | j � d S)Ns abc
)r rD rk rj )r r� r r r �test_multiple_close� s
z*TestNamedTemporaryFile.test_multiple_closec
st t j � �# � | j t j j � j � � Wd QRX| j t j j � j � � � f d d � } | j t | � d S)Nc s � � Wd QRXd S)Nr r )r� r r �
use_closed� s z?TestNamedTemporaryFile.test_context_manager.<locals>.use_closed)
r rD r6 r, r- r" r9 r| r r� )r r( r )r� r �test_context_manager� s
# z+TestNamedTemporaryFile.test_context_managerc s� g � t j � � � f d d � } t j d d | ��K t j d d t ��. | j t t j � | j t � � d � Wd QRXWd QRXd S)Nc s � j | � � | � d S)N)rS )r� )�closed�os_closer r rj � s
z5TestNamedTemporaryFile.test_no_leak_fd.<locals>.closezos.close� side_effectzio.openr )
r, rj r �patchr� r r rD r1 r/ )r rj r )r* r+ r �test_no_leak_fd� s z&TestNamedTemporaryFile.test_no_leak_fdc C s� t j � } | j t j | � | j t � � t j d d d | � Wd QRX| j t � � t j d d d | � Wd QRX| j t
j | � g � d S)Nr� r r: � ) r rG Z
addCleanupr r� r r� rD r r1 r, r� )r r: r r r �
test_bad_mode� s z$TestNamedTemporaryFile.test_bad_mode)r r r r� r� r r! r# r$ r& r' r) r. r0 r r r r r C s
r c @ s e Z d Z d d d d 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 d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d S)*�TestSpooledTemporaryFiler Nr c C s@ | d k r t j � } t j d | d | d | d | � } | S)N�max_sizer: r r )r rK rO )r r2 r: r; r< r� r r r r� � s $z"TestSpooledTemporaryFile.do_createc C sN | j � } | j | j � | j d d d d d d � } | j | j � d S)Nr2 r r; r� r< z.txt)r� r| �_rolled)r r� r r r r� � s z#TestSpooledTemporaryFile.test_basicc C s� t j � } z� t j d d d | � } | j | j � | j d d � | j | j � | j } | j � | j t | t
� o� t j j
| � d | � Wd t j | � Xd S)Nr2 �
r: s blat � z*SpooledTemporaryFile %s exists after close)r rG rO r| r3 rk r6 r9 rj r3 r r, r- r" r� )r r: r� �filenamer r r r$ � s
$z*TestSpooledTemporaryFile.test_del_on_closec C sj | j d d � } | j | j � x1 t d � D]# } | j d d � | j d � q/ W| j | j � d S)Nr2 � r5 r � xre s xxxxxxxxxxxxxxxxxxxx)r� r| r3 r` r rk )r r� rc r r r �test_rewrite_small� s z+TestSpooledTemporaryFile.test_rewrite_smallc C s} | j d d � } | j | j � | j d � | j | j � | j d � | j | j � | j d � | j | j � d S)Nr2 r7 r8 re r4 s xxxxxxxxxxxxxxxxxxxxs
xxxxxxxxxx)r� r| r3 rk r6 )r r� r r r �test_write_sequential� s
z.TestSpooledTemporaryFile.test_write_sequentialc C sF | j � } | j d � | j d � | j � } | j | d � d S)Nr8 � y� zr s xyz)r8 r; r< )r� �
writelinesr rn r1 )r r� �bufr r r �test_writelines� s
z(TestSpooledTemporaryFile.test_writelinesc C sP | j d d � } | j d
� | j | j � | j d � | j | j � d S) Nr2 �# r8 re r4 r5 � xxxxxxxxxxxxxxxxxxxx�
xxxxxxxxxx� xxxxx)rA rB rC )r� r= r| r3 rk r6 )r r� r r r �test_writelines_sequential� s
z3TestSpooledTemporaryFile.test_writelines_sequentialc C sc | j d d � } | j | j � | j d d � | j | j � | j d � | j | j � d S)Nr2 r7 r r r8 )r� r| r3 r rk r6 )r r� r r r � test_sparse s
z$TestSpooledTemporaryFile.test_sparsec C sO | j d d � } | j | j � | j | j � d k � | j | j � d S)Nr2 r7 r )r� r| r3 r6 �fileno)r r� r r r � test_fileno s z$TestSpooledTemporaryFile.test_filenoc C sK t j � } | j d � | j | j � | j � | j � | j � d S)Ns abc
)r rO rk r| r3 rj )r r� r r r �#test_multiple_close_before_rollover s
z<TestSpooledTemporaryFile.test_multiple_close_before_rolloverc C sQ t j d d � } | j d � | j | j � | j � | j � | j � d S)Nr2 r s abc
)r rO rk r6 r3 rj )r r� r r r �"test_multiple_close_after_rollover" s
z;TestSpooledTemporaryFile.test_multiple_close_after_rolloverc C s| | j d d � } | j } | j } | j } | d d � | d d � | d d � | j | d � d d d d � d S)Nr2 r7 r� r@ r� r �F )r� rn rk r r1 )r r� rn rk r r r r �test_bound_methods+ s
z+TestSpooledTemporaryFile.test_bound_methodsc C s t j d d � } | j d � | j | j � | j | j d � | j | j � | j t
� � | j Wd QRX| j t
� � | j Wd QRX| j d � | j
| j � | j | j d � | j | j � | j t
� � | j Wd QRX| j t
� � | j Wd QRXd S)Nr2 r4 r8 zw+bzrb+s
xxxxxxxxxx)r rO rk r| r3 r1 r� r r9 r r� �newlines�encodingr6 �assertIsNotNone)r r� r r r �test_properties9 s"
z(TestSpooledTemporaryFile.test_propertiesc C s� t j d d d d � } | j d � | j d � | j | j � d � | j d � | j d � | j | j � d � | j | j � | j | j d � | j | j
� | j | j � | j | j � | j d � | j d � | j | j � d
� | j d � | j d � | j | j � d � | j
| j � | j | j d � | j | j
� | j | j t j � | j | j � d S)
Nr� zw+r2 r4 zabc
r zdef
zabc
def
zxyzzy
zabc
def
xyzzy
zfoobar
zabc
def
xyzzy
foobar
)r rO rk r r1 rn r| r3 r� r r9 rL rM r6 rN r, �linesep)r r� r r r �test_text_modeM s.
z'TestSpooledTemporaryFile.test_text_modec C s= t j d d d d d d d d � } | j d � | j d
� | j | j � d � | j | j � | j | j d � | j | j
� | j | j � | j | j � | j d d
� | j d
� | j | j � d d d
� | j
| j � | j | j d � | j | j
� | j | j � | j | j d � d S)Nr� zw+r2 r4 �newliner rM zutf-8u Λ
r u Λre z
u( ΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛu( ΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛ)r rO rk r r1 rn r| r3 r� r r9 rL rM r6 rN )r r� r r r � test_text_newline_and_encodingj s$
z7TestSpooledTemporaryFile.test_text_newline_and_encodingc
sr t j d d � �' � | j � j � | j � j � Wd QRX| j � j � � f d d � } | j t | � d S)Nr2 r c s � � Wd QRXd S)Nr r )r� r r r( � s zQTestSpooledTemporaryFile.test_context_manager_before_rollover.<locals>.use_closed)r rO r| r3 r* r6 r r� )r r( r )r� r �$test_context_manager_before_rollover s z=TestSpooledTemporaryFile.test_context_manager_before_rolloverc
s� t j d d � �N � | j � j � � j d � � j � | j � j � | j � j � Wd QRX| j � j � � f d d � } | j t | � d S)Nr2 r s abc
c s � � Wd QRXd S)Nr r )r� r r r( � s zQTestSpooledTemporaryFile.test_context_manager_during_rollover.<locals>.use_closed)
r rO r| r3 rk �flushr6 r* r r� )r r( r )r� r �$test_context_manager_during_rollover� s
z=TestSpooledTemporaryFile.test_context_manager_during_rolloverc
s� t j d d � � � j d � � j � | j � j � � � | j � j � Wd QRX| j � j � � f d d � } | j t | � d S)Nr2 r s abc
c s � � Wd QRXd S)Nr r )r� r r r( � s zPTestSpooledTemporaryFile.test_context_manager_after_rollover.<locals>.use_closed)
r rO rk rU r6 r3 r| r* r r� )r r( r )r� r �#test_context_manager_after_rollover� s
z<TestSpooledTemporaryFile.test_context_manager_after_rolloverc C s t j d d � } | j d � | j d � | j � | j | j � | j | j j � d � t j d d � } | j d � | j d � | j | j � | j | j j � d � t j d d � } | j d � | j d � | j
| j � t r| j t j
| j � � j d � d S) Nr2 r4 s abcdefg
r r � s abcdre )r rO rk r �truncater| r3 r1 �_file�getvaluer6 r� r, �fstatrF �st_size)r r� r r r �!test_truncate_with_size_parameter� s"
z:TestSpooledTemporaryFile.test_truncate_with_size_parameter)r r r r� r� r$ r9 r: r? rD rE rG rH rI rK rO rQ rS rT rV rW r^ r r r r r1 � s&
r1 c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �TestTemporaryFilec C s t j � d S)N)r rE )r r r r r� � s z TestTemporaryFile.test_basicc C se t j � } t j d | � } | j d � y t j | � Wn" | j � t j | � � Yn Xd S)Nr: s blat)r rG rE rk r, r� rj )r r: r� r r r �test_has_no_name� s
z"TestTemporaryFile.test_has_no_namec C s; t j � } | j d � | j � | j � | j � d S)Ns abc
)r rE rk rj )r r� r r r r' � s
z%TestTemporaryFile.test_multiple_closec sV � f d d � } | d d � | d d � | d d d d �| d
d d d �d S)
Nc
sP t j | | � �7 } | j | � | j d � � j | | j � � Wd QRXd S)Nr )r rE rk r r1 rn )�inputr� r� �fileobj)r r r � roundtrip� s
z;TestTemporaryFile.test_mode_and_encoding.<locals>.roundtrips 1234zw+bzabdc
zw+u ΛrM zutf-16zfoo
rR r r )r rc r )r r �test_mode_and_encoding� s
z(TestTemporaryFile.test_mode_and_encodingc s� g � t j � � � f d d � } t j d d | ��K t j d d t ��. | j t t j � | j t � � d � Wd QRXWd QRXd S)Nc s � j | � � | � d S)N)rS )r� )r* r+ r r rj � s
z0TestTemporaryFile.test_no_leak_fd.<locals>.closezos.closer, zio.openr )
r, rj r r- r� r r rE r1 r/ )r rj r )r* r+ r r. � s z!TestTemporaryFile.test_no_leak_fdN)r r r r� r` r' rd r. r r r r r_ � s
r_ c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
NulledModulesc G s3 d d � | D� | _ d d � | j D� | _ d S)Nc S s g | ] } | j � q Sr )rR )�.0�modr r r �
<listcomp>� s z*NulledModules.__init__.<locals>.<listcomp>c S s g | ] } | j � � q Sr )�copy)rf r r r r rh s )�refs�contents)r �modulesr r r r� � s zNulledModules.__init__c C s3 x, | j D]! } x | D] } d | | <q Wq
Wd S)N)rj )r r rV r r r r% s
zNulledModules.__enter__c G sA x: t | j | j � D]# \ } } | j � | j | � q Wd S)N)�ziprj rk �clear�update)r �exc_infor r r r r r* s "
zNulledModules.__exit__N)r r r r� r% r* r r r r re � s re c @ s� e Z d Z d d d d d d � Z d d � Z d d � Z e j d
d � � Z e j d d
� � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d S)�TestTemporaryDirectoryNr r c C s� | d k r t j � } t j d | d | d | � } | j | j | | | � | r} | j | j | | | d � } d | _ t t j j | j d � d � � } | j
d � Wd QRX| S)Nr: r r r ztest.txtr� s Hello world!) r rK rP r@ r9 r� r� r, r- r� rk )r r: r; r< �recurser% �d1r� r r r r� s $z TestTemporaryDirectory.do_createc C s\ t j � � } Wd QRX| j t � � } t j d | � Wd QRX| j | j j t j � d S)Nr: )r rP r r� r1 � exception�errno�ENOENT)r r� �cmr r r �test_mkdtemp_failure s
z+TestTemporaryDirectory.test_mkdtemp_failurec C s� t j � } zl | j d | � } | j t j j | j � d | j � | j � | j t j j | j � d | j � Wd t j
| � Xd S)Nr: z$TemporaryDirectory %s does not existz*TemporaryDirectory %s exists after cleanup) r rG r� r6 r, r- r" r9 �cleanupr| r� )r r: r r r r �test_explicit_cleanup& s
z,TestTemporaryDirectory.test_explicit_cleanupc C s� | j � } | j d d � } t j | j t j j | j d � � | j � | j t j j | j � d | j � | j t j j | j � d � | j
t j | j � d g d � | j � d S)Nrr r Zfooz*TemporaryDirectory %s exists after cleanupz-Directory pointed to by a symlink was deletedztest.txtz>Contents of the directory pointed to by a symlink were deleted) r� r, �symlinkr9 r- r� ry r| r" r6 r1 r� )r rs Zd2r r r �(test_cleanup_with_symlink_to_a_directory3 s %
z?TestTemporaryDirectory.test_cleanup_with_symlink_to_a_directoryc C sc t j � } zB | j d | � } | j } ~ | j t j j | � d | � Wd t j | � Xd S)Nr: z*TemporaryDirectory %s exists after __del__) r rG r� r9 r| r, r- r" r� )r r: r r9 r r r �test_del_on_collectionH s z-TestTemporaryDirectory.test_del_on_collectionc
C s� | j � �� } x� d D]� } d j d | d | � } t j d
| � \ } } } | j � j � } | j t j j | � d | � | j d d
� } | j
d | � | j d | � q WWd QRXd S)N�builtinsr, �shutilr� r r& a� if True:
import builtins
import os
import shutil
import sys
import tempfile
import warnings
tmp = tempfile.TemporaryDirectory(dir={dir!r})
sys.stdout.buffer.write(tmp.name.encode())
tmp2 = os.path.join(tmp.name, 'test_dir')
os.mkdir(tmp2)
with open(os.path.join(tmp2, "test.txt"), "w") as f:
f.write("Hello world!")
{mod}.tmp = tmp
warnings.filterwarnings("always", category=ResourceWarning)
r: rg z-cz*TemporaryDirectory %s exists after cleanupzutf-8�backslashreplacez
Exception z'ResourceWarning: Implicitly cleaning up)r~ r, r r� r r& ) r� �formatr �assert_python_okro �stripr| r, r- r" rb r� )r r: rg �code�rc�out�err�tmp_namer r r �test_del_on_shutdownU s
z+TestTemporaryDirectory.test_del_on_shutdownc C s� | j � �� } d j d | � } t j d | � \ } } } | j � j � } | j t j j | � d | � | j d d � } | j
d | � | j d | � Wd QRXd S) Na� if True:
import sys
import tempfile
import warnings
def generator():
with tempfile.TemporaryDirectory(dir={dir!r}) as tmp:
yield tmp
g = generator()
sys.stdout.buffer.write(next(g).encode())
warnings.filterwarnings("always", category=ResourceWarning)
r: z-cz*TemporaryDirectory %s exists after cleanupzutf-8r� z
Exception z'ResourceWarning: Implicitly cleaning up) r� r� r r� ro r� r| r, r- r" rb r� )r r: r� r� r� r� r� r r r �test_exit_on_shutdownu s
z,TestTemporaryDirectory.test_exit_on_shutdownc C s� | j � �� } | j d | d d � } | j } t j d t f d d ��% t j d d t �~ t j � Wd QRX| j t j
j | � d | � Wd QRXd S)
Nr: rr � Z
Implicitly�quietF�alwaysr z*TemporaryDirectory %s exists after __del__) r� r9 r r# �ResourceWarningr& r' Z
gc_collectr| r, r- r" )r r: r r9 r r r �test_warnings_on_cleanup� s z/TestTemporaryDirectory.test_warnings_on_cleanupc C s. | j � } | j � | j � | j � d S)N)r� ry )r r r r r r' � s
z*TestTemporaryDirectory.test_multiple_closec
C se | j � } | �3 } | j t j j | � � | j | | j � Wd QRX| j t j j | � � d S)N)r� r6 r, r- r" r1 r9 r| )r r r9 r r r r) � s
z+TestTemporaryDirectory.test_context_manager)r r r r� rx rz r Zskip_unless_symlinkr| Z cpython_onlyr} r� r� r� r' r) r r r r rq s
rq �__main__)5r ru r� r, rq r� rA r& �
contextlibr rw r Ztestr Z test.supportr ry r r� r� r� r� r� r� �
startswithra ZTestCaser
r rC rX rz r� r� �contextmanagerr� r� r� r� r� r r
r r r r1 rD rE r_ re rq r �mainr r r r �<module> s\ / A*%
� />l<~�>�