U ‰ª/e¥ã @sâUdZddlZddlZddlZddlmZddlmZddlm Z ddl Z ddl Z ddl m Z ddlmZddlmZd d lmZd d d d ddddddœ Zeed<Gdd„deƒZdd„Zdd„Zdd„Zedd„ƒZdd „ZdS)!z%Tests for kernel connection utilitieséN)ÚTemporaryDirectory)Ú no_type_check)Úpatch)ÚConfig)Úconnect)Ú IPKernelAppé)ÚTemporaryWorkingDirectoryz1.2.3.4Zipcéééésabc123zhmac-md5) ÚipÚ transportZ shell_portZhb_portZ iopub_portZ stdin_portZ control_portÚkeyZsignature_schemeÚ sample_infoc@seZdZdd„Zddd„ZdS)ÚDummyKernelAppcCsdS)Nr©)Úselfrrú@/tmp/pip-unpacked-wheel-g43y689f/ipykernel/tests/test_connect.pyÚ_default_shell_port$sz"DummyKernelApp._default_shell_portNcCs| ¡| ¡dS)N)Zinit_profile_dirZinit_connection_file)rÚargvrrrÚ initialize'szDummyKernelApp.initialize)N)Ú__name__Ú __module__Ú __qualname__rrrrrrr#src Cs´tƒ}tƒž}||j_d}t||d}| ¡tj |j |¡}||j ksLt ‚t |dƒ}|  d¡W5QRXtj |¡s|t ‚t |¡|ksŽt ‚||_t |¡|ks¦t ‚W5QRXdS)Nú kernel.json©ÚconfigÚconnection_fileÚwz{})rr Ú ProfileDirÚlocationrrÚosÚpathÚjoinZconnection_dirZabs_connection_fileÚAssertionErrorÚopenÚwriteÚexistsrZget_connection_filer)ÚcfgÚdÚcfÚappZ profile_cfÚfrrrÚtest_get_connection_file,s  r/c Cs°tƒ:}tj |d¡}tj|ftŽt |¡}tj|dd}W5QRXt|t ƒsTt ‚dd„|  ¡Dƒ}|tksrt ‚t   |¡}|d d¡|d<dd„|  ¡Dƒ}|tks¬t ‚dS) NrT)ÚunpackcSsi|]\}}|tkr||“qSr©r©Ú.0ÚkÚvrrrÚ Gsz,test_get_connection_info..rzutf-8cSsi|]\}}|tkr||“qSrr1r2rrrr6Ls)rr#r$r%rZwrite_connection_filerZget_connection_infoÚ isinstanceÚstrr&ÚitemsÚjsonÚloadsÚencode)r+r,Z json_infoÚinfoZsub_infoZinfo2Z sub_info2rrrÚtest_get_connection_info?s   r>c Csœtƒ}tƒ†}||j_d}t||d}| |j¡| ¡t  |d¡B}t   dd¡|_ t  t j ¡| ¡W5QRX|jdks„t‚W5QRXW5QRXdS)NrrÚ_try_bind_socketiœÿÿÿzfails for unknown error typesr)rr r!r"rÚ addfinalizerÚcloserrÚobjectÚzmqÚZMQErrorÚ side_effectÚpytestÚraisesÚ init_socketsÚ call_countr&©Úrequestr*r+r,r-Ú mock_try_bindrrrÚtest_port_bind_failure_raisesPs  rMc CsÖz tjWn:tk rDtjtdddd}| ¡| |j¡YnXtƒ}t ƒz}||j _ d}t ||d}| |j ¡| ¡t |d¡6}t tjd¡t tjd ¡gd gd |_| ¡W5QRXW5QRXdS) NÚ WSAEADDRINUSEi90T)Úcreaterrr?zfails for non-bind unixzfails for non-bind windowsréd)ÚerrnorNÚAttributeErrorrrBÚstartr@Ústoprr r!r"rrArrCrDÚ EADDRINUSErErH)rKÚpr*r+r,r-rLrrrÚtest_port_bind_failure_recovery_s(     þýrWc Csžtƒ}tƒˆ}||j_d}t||d}| |j¡| ¡t  |d¡D}t   t j d¡|_t t j ¡| ¡W5QRX|jdks†t‚W5QRXW5QRXdS)Nrrr?zfails for non-bindrP)rr r!r"rr@rArrrBrCrDrQrUrErFrGrHrIr&rJrrrÚ'test_port_bind_failure_gives_up_retriesys  rX)Ú__doc__rQr:r#ÚtempfilerÚtypingrZ unittest.mockrrFrCZtraitlets.config.loaderrZ ipykernelrZipykernel.kernelapprÚutilsr rÚdictÚ__annotations__rr/r>rMrWrXrrrrÚs:       ÷