Your IP : 216.73.216.85
��Yf�� � @ 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 �7/opt/alt/python35/lib64/python3.5/test/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�nbaseZnpreZnsufZcheckr 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 rV r r r r rB r s rB c ss e Z d 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)�TestRandomNameSequencez6Test the internal iterator object _RandomNameSequence.c 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 )�nextrY 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 )rY �range�
TEST_FILESr\ r? � assertNotIn)r rS rY �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 � )rY )r rb rY 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, �piperf �close�writer\ rY �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 �__doc__r) r^ rc re �unittest�
skipUnless�hasattrr, ru r r )r[ r rW � s
rW c @ s. e Z d Z d Z d d � Z d d � Z d S)�TestCandidateTempdirListz3Test the internal function _candidate_tempdir_list.c 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�AttributeErrorrr �curdir)r �envZenvname�dirnamer~ r r r �test_wanted_dirs� s
z)TestCandidateTempdirList.test_wanted_dirsN)r r r rv r� r� r r r r rz � s rz c @ s" e Z d Z d Z d d � Z d S)�TestGetDefaultTempdirz Test _get_default_tempdir().c - 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)rr )�args�kwargsr r r �
raise_OSError s zFTestGetDefaultTempdir.test_no_files_left_behind.<locals>.raise_OSError�openc s � | | � } � | _ | S)N)rj )r� r� �fp)� orig_openr� r r �
bad_writer s zCTestGetDefaultTempdir.test_no_files_left_behind.<locals>.bad_writer) r rO r � swap_attrZ_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 rv r� r r r r r� � s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�TestGetCandidateNamesz0Test the internal function _get_candidate_names.c C s# t j � } | j | t j � d S)N)r �_get_candidate_namesr} rX )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 rv 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)NrL )r rF 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 rL �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� nonexistentrL ) r� r, r- �joinr rL r r� r r� r� )r rL 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�wbrL )r� r, r- r� r rL r� ri r r� r �NotADirectoryErrorr� r� )r rL 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 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) �TestMkstempInnerz*Test the internal function _mkstemp_inner.c @ 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, rj r� )r r r r r rj 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, ri r� �unlinkr� r� rj 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 rJ rK 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� rj )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 rK r� rj 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_ r` r� )r �extantrb 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 rF r� rj 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� rT �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
rd s blat)r� rj r, �lseekr� �SEEK_SETr1 rm )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� rJ rH 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, ri r6 �endswith)r Zfd1Zname1Zfd2Zname2r 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 rF r6 r� r� r, ri )r r: r� r9 r r r �&test_collision_with_existing_directory s
z7TestMkstempInner.test_collision_with_existing_directory)r r r rv 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 Z d d � Z d d � Z d S)�TestGetTempPrefixzTest gettempprefix().c C sh t j � } | j | t � | j t | � d � t j � } | j | t � | j t | � d � d S)Nr )r rH r} r Z
assertGreaterr/ rI 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 rH rF r, r- r� r� �O_RDWR�O_CREATri r� r� )r r� �dr� r r r �test_usable_template s
z&TestGetTempPrefix.test_usable_templateN)r r r rv r� r r r r r r� s r� c @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) �TestGetTempDirzTest gettempdir().c 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 rJ rK 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 rC rj ri )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 rJ rK r6 rs 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 rF rL r r� r1 rJ r� )r Zcase_sensitive_tempdirZ_tempdirr� r r r �test_case_sensitiveN s
! z"TestGetTempDir.test_case_sensitiveN)r r r rv r r r� r r r r r r . s
r c @ sO e Z d Z d Z d d d d d � Z d d � Z d d � Z d d
� Z d S) � TestMkstempzTest mkstemp().Nc
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 rJ rK rE r, r- r. r2 r1 r? ri 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 rK 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 rF r� r, r� )r r: r r r r� � s z!TestMkstemp.test_choose_directory)r r r rv r� r� r� r� r r r r r ] s
r c @ s� e Z d 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)� TestMkdtempzTest mkdtemp().c C s
t j � S)N)r rF )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 rJ rK rF 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 rK 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_ r` r� r3 r r, r� )r r� rb 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 rF 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� rT 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 rC ri r6 r9 r� rF )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 rF r6 r� )r Zdir1Zdir2r r r r� � s
z2TestMkdtemp.test_collision_with_existing_directory)r r r rv r� r� r� r� r� r� rw rx r� r r� r� r r r r r
� s r
c sw e Z d 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)�
TestMktempzTest mktemp().c s t j � | _ t � j � d S)N)r rF r: rZ r) )r )r[ r r r) s zTestMktemp.setUpc s3 | j r" t j | j � d | _ t � j � d S)N)r: r, r� rZ 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, ri 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_ r` r� )r r� rb r r r rc 1 s
zTestMktemp.test_many)
r r r rv r) r+ r r� r� rc r r )r[ r r s
r c @ s� e Z d 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)�TestNamedTemporaryFilezTest NamedTemporaryFile().Nr Tc C sY | d k r t j � } t j d | d | d | d | � } | j | j | | | � | S)Nr: r� r
r )r rJ rC 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�refrj r Zcheck_impl_detail� assertIsNone)r r� �wrrj 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 rC rj r� �seek)r� )�linesr r � make_filel s
z3TestNamedTemporaryFile.test_iter.<locals>.make_filer )� enumerater1 r/ )r r rb �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 rC 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 rF rC rj ri 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 rF rC r9 rj ri 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 rC rj ri )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 rC 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)rR )r� )�closed�os_closer r ri � s
z5TestNamedTemporaryFile.test_no_leak_fd.<locals>.closezos.close� side_effectzio.openr )
r, ri r �patchr� r r rC r1 r/ )r ri 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 rF Z
addCleanupr r� r r� rC r r1 r, r� )r r: r r r �
test_bad_mode� s z$TestNamedTemporaryFile.test_bad_mode)r r r rv r� r� r r r r r" r# r% r* r, r r r r r C s
r c @ s e Z d 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)+�TestSpooledTemporaryFilez Test SpooledTemporaryFile().r Nr c C s@ | d k r t j � } t j d | d | d | d | � } | S)N�max_sizer: r� r
)r rJ rN )r r. 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)Nr. 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)Nr. �
r: s blat � z*SpooledTemporaryFile %s exists after close)r rF rN r| r/ rj r6 r9 ri 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)Nr. � r1 r � xrd s xxxxxxxxxxxxxxxxxxxx)r� r| r/ r_ r rj )r r� rb 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)Nr. r3 r4 rd r0 s xxxxxxxxxxxxxxxxxxxxs
xxxxxxxxxx)r� r| r/ rj 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)Nr4 � y� zr s xyz)r4 r7 r8 )r� �
writelinesr rm r1 )r r� Zbufr r r �test_writelines� s
z(TestSpooledTemporaryFile.test_writelinesc C sP | j d d � } | j d
� | j | j � | j d � | j | j � d S) Nr. �# r4 rd r0 r1 � xxxxxxxxxxxxxxxxxxxx�
xxxxxxxxxx� xxxxx)r< r= r> )r� r9 r| r/ rj 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)Nr. r3 r r r4 )r� r| r/ r rj 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)Nr. r3 r )r� r| r/ 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 rN rj r| r/ ri )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)Nr. r s abc
)r rN rj r6 r/ ri )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)Nr. r3 r� r; r� r �F )r� rm rj r r1 )r r� rm rj 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)Nr. r0 r4 zw+bzrb+s
xxxxxxxxxx)r rN rj r| r/ 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+r. r0 zabc
r zdef
zabc
def
zxyzzy
zabc
def
xyzzy
zfoobar
zabc
def
xyzzy
foobar
)r rN rj r r1 rm r| r/ r� r r9 rG rH r6 rI 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+r. r0 �newliner rH zutf-8u Λ
r u Λrd z
u( ΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛu( ΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛΛ)r rN rj r r1 rm r| r/ r� r r9 rG rH r6 rI )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)Nr. r c s � � Wd QRXd S)Nr r )r� r r r$ � s zQTestSpooledTemporaryFile.test_context_manager_before_rollover.<locals>.use_closed)r rN r| r/ 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)Nr. 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 rN r| r/ rj �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)Nr. 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 rN rj rP r6 r/ 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) Nr. r0 s abcdefg
r r � s abcdrd )r rN rj r �truncater| r/ r1 Z_file�getvaluer6 r� r, �fstatrA �st_size)r r� r r r �!test_truncate_with_size_parameter� s"
z:TestSpooledTemporaryFile.test_truncate_with_size_parameter)r r r rv r� r� r r5 r6 r: r? r@ rB rC rD rF rJ rL rN rO rQ rR rX r r r r r- � s(
r- c @ sR e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
�TestTemporaryFilezTest TemporaryFile().c C s t j � d S)N)r rD )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 rF rD rj r, r� ri )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 rD rj ri )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 rD rj r r1 rm )�inputr� r� Zfileobj)r r r � roundtrip� s
z;TestTemporaryFile.test_mode_and_encoding.<locals>.roundtrips 1234zw+bzabdc
zw+u ΛrH zutf-16zfoo
rM r r )r r\ 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)rR )r� )r& r' r r ri � s
z0TestTemporaryFile.test_no_leak_fd.<locals>.closezos.closer( zio.openr )
r, ri r r) r� r r rD r1 r/ )r ri r )r& r' r r* � s z!TestTemporaryFile.test_no_leak_fdN) r r r rv r� rZ r# r] r* r r r r rY � s
rY 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 )rQ )�.0�modr r r �
<listcomp>� s z*NulledModules.__init__.<locals>.<listcomp>c S s g | ] } | j � � q Sr )�copy)r_ r r r r ra 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)rc )r r rU r r r r% s
zNulledModules.__enter__c G sA x: t | j | j � D]# \ } } | j � | j | � q Wd S)N)�ziprc rd �clear�update)r �exc_infor r r r r r* s "
zNulledModules.__exit__N)r r r r� r% r* r r r r r^ � s r^ c @ s� e Z d 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)�TestTemporaryDirectoryzTest TemporaryDirectory().Nr 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 rJ rO r? r9 r� r� r, r- r� rj )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 rO r r� r1 Z 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 rF 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)Nrk 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� rq r| r r6 r1 r� )r rl 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 rF 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: r` z-cz*TemporaryDirectory %s exists after cleanupzutf-8�backslashreplacez
Exception z'ResourceWarning: Implicitly cleaning up)rv r, rw r� r r& ) r� �formatr �assert_python_okrn �stripr| r, r- r ra r� )r r: r` �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-8rx z
Exception z'ResourceWarning: Implicitly cleaning up) r� ry r rz rn r{ r| r, r- r ra 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: rk � 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� rq )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 rv r� rp rr r Zskip_unless_symlinkrt Z cpython_onlyru r� r� r� r# r% r r r r rj s
rj �__main__)5r rm r� r, rp r� r@ r& �
contextlibr rw r Ztestr Z test.supportr ry r r� r� r� r� r� r� �
startswithr` ZTestCaser
r rB rW rz r� r� �contextmanagerr� r� r� r� r� r r r
r r r- rC rD rY r^ rj r �mainr r r r �<module> s\ / A*%
� />l<~�>�