U
|
Z ±d ã @ sê d dl Z d dlZd dlmZ d dlZe jdkZejje ddZ ddl
|
m Z m Z m Z dd lmZmZmZ dd
|
lmZ erd d lmZmZmZmZmZ ej d ¡dd Zdd Z edd Z e dd Z e dd Z dd Z dd Z!dS ) é N)ÚcontextmanagerÚntz windows only)Úreasoné )ÚslowÚgc_collect_harderÚrestore_unraisablehooké )Ú_coreÚsleepÚ move_on_after)Úwait_all_tasks_blockedé )ÚffiÚkernel32ÚINVALID_HANDLE_VALUEÚraise_winerrorÚ FileFlagsz3ignore:.*UnboundedQueue:trio.TrioDeprecationWarningc à s¾ dd } t ¡ ¤\}}t ¡ 4 I d H ~}| | |¡ d}td |2 zP3 d H W }td| |D ](}|jdkspt|j|ks~t|d7 }q^|dkrD qqD6 td W 5 Q I d H R X W 5 Q R X d S ) Nc Ó s` t dt ¡ ¡}tdD ]B}td| |d dkr@t ¡ I d H t ||| t j ¡}|st
|
qd S )NÚHANDLEé
|
Úpostr r ) r Úcastr
|
Z current_iocpÚrangeÚprintÚ
|
checkpointr ZPostQueuedCompletionStatusÚNULLÚAssertionError)ÚkeyZiocpÚiÚsuccess© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_windows.pyr s
|
z(test_completion_key_listen.<locals>.postr Zloopzgot somer r zend loop)r
|
Zmonitor_completion_keyÚ open_nurseryÚ
|
start_soonr Z lpOverlappedr ZdwNumberOfBytesTransferred)r r ÚqueueÚnurseryr ÚbatchÚinfor r r! Útest_completion_key_listen s
|
|
r( c
|
sj d} t t| }t ¡ F}tj |d¡}t|d}| | ¡ | ¡ W 5 Q R X |
|
d¡d }t |¡}t t d|¡tjtjt jtjtjt j¡tkr¢t z¬t| fdd}t ¡ t ¡ 4 I d H *}t d d
|
d D ]} | || | d ¡ qäW 5 Q I d H R X || kst t !t"¡ t #d ¡I d H W 5 Q R X W 5 Q R X W 5 t ¡ X W 5 Q R X d S ) Ns 1111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222233333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333334444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444444z numbers.txtÚwbzutf-16les ÚLPCWSTRc s t | |
| ¡I d H d S )N)r
|
Úreadinto_overlapped)ÚstartÚend©Z buffer_viewÚhandler r! Ú read_regionR s
|
|
ÿz-test_readinto_overlapped.<locals>.read_regionr i i s immutable)$Ú bytearrayÚlenÚtempfileÚTemporaryDirectoryÚosÚpathÚjoinÚopenÚwriteÚflushÚencoder Ú from_bufferr Z CreateFileWr r Ú GENERIC_READZFILE_SHARE_READr Ú OPEN_EXISTINGÚFILE_FLAG_OVERLAPPEDr r Ú CloseHandleÚ
|
memoryviewr
|
Úregister_with_iocpr" r r# r ÚpytestÚraisesÚ BufferErrorr+ )
|
ÚdataÚbufferZtdirZtfileÚfpZrawnameZ rawname_bufr0 r% r, r r. r! Útest_readinto_overlapped7 s>
|
|
|
ù
|
|
$ *rI c c st ddl m} dd l}| dd\}}z&| |d¡}tj |ddd|fV W 5 t t d|¡¡ t t d|¡¡ X d S ) Nr )Úpipe)TF)Z
|
overlappedr r) F)Úclosefd)
|
Zasyncio.windows_utilsrJ Úmsvcrtr r@ r r Úopen_osfhandler5 Úfdopen)rJ rL Ú read_handleZ write_handleZwrite_fdr r r! Úpipe_with_overlapped_readd s rP c
|
s t \} | | d¡ W 5 Q R X d fdd}t tj¡}t |¡ W 5 Q R X d dt|jkspt dt|jkst ~t
|
W 5 Q R X d S )Ns test
|
Fc
|
sh t d} zPt ¡ 4 I d H 0}|jtj| dd t I d H |j ¡ W 5 Q I d H R X W 5 sbtX d S )Nr Zxyz)Úname) r1 r r
|
r" r# r+ r Ú cancel_scopeÚcancel)Útargetr% ©Z left_run_yetrO r r! Úmainz s ÿ z/test_forgot_to_register_with_iocp.<locals>.mainTz'Failed to cancel overlapped I/O in xyz z$forget to call register_with_iocp()?) rP r9 rC rD r
|
ZTrioInternalErrorÚrunÚstrÚvaluer r )Úwrite_fprV Úexc_infor rU r! Ú!test_forgot_to_register_with_iocpr s r\ c à sâ dd l } t Ê\}}t |¡ td}t ¡ 4 I d H P}| tj||¡ t I d H | | d¡ W 5 Q R X |
|
d¡ |j ¡ W 5 Q I d H R X |d d
dks¦t t ||¡I d H dksÀt |d d
dksÔt W 5 Q R X d S )Nr é s test1
|
test2
|
r s test1
|
s test2
|
)ÚtimerP r
|
rB r1 r" r# r+ r r9 r rR rS r )r^ rZ rO rT r% r r r! Útest_too_late_to_cancel s
|
|
r_ c sj ddl m m ddlm} jd fdd
|
}| |d|¡ tjt dd t
|
t d
|
¡ W 5 Q R X d S ) Nr ©Ú WSAIoctlsÚ_handle©Ú _io_windows©Úwhichc s4 t | dr| ¡ } | jkr(| d S | S d S )NÚfilenor )Úhasattrrg ZSIO_BSP_HANDLE_SELECT©Úsockrf r` r r! Úpatched_get_underlyingµ s
|
|
|
zKtest_lsp_that_hooks_select_gives_good_error.<locals>.patched_get_underlyingÚ_get_underlying_socketz0SIO_BASE_HANDLE and SIO_BSP_HANDLE_SELECT differ©Úmatchr © Ú _windows_cffira rb Ú rd ÚSIO_BASE_HANDLEÚsetattrrC rD Ú RuntimeErrorr
|
rW r ©Z monkeypatchrd rk r r` r! Ú+test_lsp_that_hooks_select_gives_good_error± s ÿrv c sj ddl m m ddlm} jd fdd
|
}| |d|¡ tjt dd t
|
t d
|
¡ W 5 Q R X d S ) Nr r` rc re c s2 t | dr| ¡ } | jkr&tdn| S d S )Nrg Znope)rh rg rr ÚOSErrorri r` r r! rk Í s
|
|
|
|
z[test_lsp_that_completely_hides_base_socket_gives_good_error.<locals>.patched_get_underlyingrl zCSIO_BASE_HANDLE failed and SIO_BSP_HANDLE_POLL didn't return a diffrm r ro ru r r` r! Ú;test_lsp_that_completely_hides_base_socket_gives_good_errorÄ s þrx )"r5 r3 Ú
|
contextlibr rC rQ Ú
|
on_windowsÚmarkZskipifZ
|
pytestmarkZtutilr r r rq r
|
r r Ztestingr rp r r r r r Úfilterwarningsr( rI rP r\ r_ rv rx r r r r! Ú<module> s*
|
|
|
-
|
|
!
|
|
