o Эj<@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZGdddeZGdddejZGdddeZGd d d ejed Zd d ZejddZd*ddZGdddZGdddeZ GdddeZ!ddZ"e"Z"GdddZ#Gdddej$Z%Gdd d e%Z&Gd!d"d"e%Z'd#d$Z(d+d&d'Z)ejd(d)Z*dS),z8Test utilities. Don't use outside of the uvloop project.Nc@eZdZddZdS) MockPatterncCsttt||tjSN)boolresearchstrS)selfotherr M/var/www/html/CropPilot/venv/lib/python3.10/site-packages/uvloop/_testbase.py__eq__szMockPattern.__eq__N)__name__ __module__ __qualname__rr r r r r rcs(eZdZfddZfddZZS) TestCaseDictcst||_dSr)super__init__namer r __class__r r rs  zTestCaseDict.__init__cs.||jvrtd|j|t||dS)Nzduplicate test {}.{})data RuntimeErrorformatrr __setitem__)r keyvaluerr r r#s zTestCaseDict.__setitem__)rrrrr __classcell__r r rr rs rcs(eZdZeddZfddZZS)BaseTestCaseMetacCt|Sr)r)mclsrbasesr r r __prepare__,szBaseTestCaseMeta.__prepare__csT|D]}|ds q|D]}t||rtd|||jq qt|||t|S)Ntest_z6duplicate test {}.{} (also defined in {} parent class)) startswithhasattrrrrr__new__dict)r#rr$dct test_namebaserr r r)0s  zBaseTestCaseMeta.__new__)rrr classmethodr%r)r r r rr r!*s r!c@seZdZddZddZddZddZd d Zd d d dZddZ ddZ ddZ ddZ e jdddddddZe jdfddZd d!Zd"d#Zejd$d%Zd&d'ZdS)( BaseTestCasecCtrNotImplementedErrorr r r r new_loop@zBaseTestCase.new_loopcCr0rr1r3r r r new_policyCr5zBaseTestCase.new_policycCr"r)r)r rr r r mock_patternFzBaseTestCase.mock_patternc s@t|tjs dSz |IdHWdSttfyYdSwr) isinstanceasyncio StreamWriter wait_closedBrokenPipeErrorConnectionError)r objr r r r<Is zBaseTestCase.wait_closedcCst|jjdS)Nzasyncio.)typelooprr'r3r r r is_asyncio_loopQzBaseTestCase.is_asyncio_loopg{Gz?)delaycCs|jt|dSr)rArun_until_completer:sleep)r rDr r r run_loop_brieflyTzBaseTestCase.run_loop_brieflycCs|j||j|dSr)#_BaseTestCase__unhandled_exceptionsappendrAdefault_exception_handler)r rAcontextr r r loop_exception_handlerWs z#BaseTestCase.loop_exception_handlercCsB||_t|t|jd|_|j|jg|_ dS)NT) r4rAr:set_event_loop_policyr6set_event_loop"_check_unclosed_resources_in_debugset_exception_handlerrMrIr3r r r setUp[s   zBaseTestCase.setUpc Cs|j|jrtdt|j|ddS|jsdStttt |jddrttt| |jj |jj d| |jj dd| |jjdd| |jjdd |jjD]%\}}|jd |d | |dd |Wdn1swYqi|jjD])\}}|jd |d | ||jj|d|Wdn1swYqtdtdd|_dS)Nz2Unexpected calls to loop.call_exception_handler():z1unexpected calls to loop.call_exception_handler() _debug_ccFz¬ all uv_handle_t handles were freedrz¬ all callbacks (call_soon) are GCedz-not all timer callbacks (call_later) are GCedz¬ all stream write contexts are GCedzAlive handle after test) handle_namezalive {} after testzTotal/closed handlesztotal != closed for {})rAcloserIprintpprintfailrPgccollectgetattr assertEqual_debug_uv_handles_total_debug_uv_handles_freed_debug_cb_handles_count_debug_cb_timer_handles_count_debug_stream_write_ctx_cnt_debug_handles_currentitemssubTestr_debug_handles_total_debug_handles_closedr:rOrN)r h_nameh_cntr r r tearDowndsn       zBaseTestCase.tearDowncCs d|_dSNF)rPr3r r r skip_unclosed_handles_check z(BaseTestCase.skip_unclosed_handles_checkN )familyaddrtimeoutbacklog max_clientsc Cs|dur#|tjkr!t }|j}Wdn1swYnd}t|tj}|dur2td|dkr:td||z ||| |Wnt y]} z| | d} ~ wwt |||||S)N)z 127.0.0.1rtimeout is requiredr#only blocking sockets are supported) socketAF_UNIXtempfileNamedTemporaryFiler SOCK_STREAMr settimeoutbindlistenOSErrorrUTestThreadedServer) r server_progrprqrrrsrttmpsockexr r r tcp_servers0     zBaseTestCase.tcp_servercCsFt|tj}|durtd|dkrtd||t||||S)Nrurrv)rwr{rr|TestThreadedClient)r client_progrprrrr r r tcp_clients zBaseTestCase.tcp_clientcO|j|dtji|SNrp)rrwrxr argskwargsr r r unix_serverrHzBaseTestCase.unix_servercOrr)rrwrxrr r r unix_clientrHzBaseTestCase.unix_clientccst?}tj|d}z|VWzt|Wnty"Ynwzt|Wwty3YwwWddSWddS1sGwYdS)Nr)ryTemporaryDirectoryospathjoinunlinkr)r tdfnr r r unix_sock_names&   "zBaseTestCase.unix_sock_namec Cs(z |jW||dS||wr)rAstoprXr rr r r _abort_socket_tests zBaseTestCase._abort_socket_test)rrrr4r6r7r<rBrGrMrRrirkrwAF_INETrrrr contextlibcontextmanagerrrr r r r r/>s2 9 !   r/) metaclasscCs4tjtjtj|d|}tj|sJ|S)Ncerts)rrabspathrdirnameisfile)test_file_namecert_file_namefullnamer r r _cert_fullnames rc csRGdddtj}td}|}||z dVW||dS||w)Nc@r)z)silence_long_exec_warning..FiltercSs|jdo |jd S)N Executingseconds)msgr'endswith)r recordr r r filters  z0silence_long_exec_warning..Filter.filterN)rrrrr r r r Filterrrr:)loggingr getLogger addFilter removeFilter)rloggerrr r r silence_long_exec_warnings  rPc Cst||dD]6}t}|&z |d|fWntjy)YWdqw|WdS1s8wYqtd)Nizcould not find a free port)rangerwr}errorr) start_fromportrr r r find_free_ports&rc@s2eZdZd ddZddddZejdd ZdS) SSLTestCaseNcCs\ttdr ttj}nttdrttj}nttj}|jtjO_||||S)NPROTOCOL_TLS_SERVER PROTOCOL_TLS) r(ssl SSLContextrrPROTOCOL_SSLv23options OP_NO_SSLv2load_cert_chain)r certfilekeyfile sslcontextr r r _create_server_ssl_contexts    z&SSLTestCase._create_server_ssl_contextT)disable_verifycCst}d|_|r tj|_|Srj)rcreate_default_contextcheck_hostname CERT_NONE verify_mode)r rrr r r _create_client_ssl_contexts z&SSLTestCase._create_client_ssl_contextc csDtd}td}||z dVW||dS||w)Nr:zhas no effect when using ssl)rrrrr)r rrr r r _silence_eof_received_warning!s   z)SSLTestCase._silence_eof_received_warningr)rrrrrrrrr r r r r s   rc@s eZdZdZddZddZdS) UVTestCaseuvloopcCtSr)rnew_event_loopr3r r r r41r8zUVTestCase.new_loopcCrr)rEventLoopPolicyr3r r r r64r8zUVTestCase.new_policyN)rrrimplementationr4r6r r r r r-s rcs<eZdZdZfddZfddZddZdd ZZS) AIOTestCaser:cs:ttjdkrt}||jt|dSdSN) ) rrRsys version_infor:SafeChildWatcher attach_looprAset_child_watcher)r watcherrr r rR<s   zAIOTestCase.setUpcs"tjdkr tdtdSr)rrr:rrrir3rr r riDs  zAIOTestCase.tearDowncCrr)r:rr3r r r r4Ir8zAIOTestCase.new_loopcCrr)r:DefaultEventLoopPolicyr3r r r r6Lr8zAIOTestCase.new_policy) rrrrrRrir4r6r r r rr r8s   rc Cshttj}|"z|dWnty YWddSw WddS1s-wYdS)N)z::1rFT)rwAF_INET6r}r) server_sockr r r has_IPv6Ps  $rc@s>eZdZddZddZddddd d Zd d Zd dZdS)TestSocketWrappercCs ||_dSr)_TestSocketWrapper__sockr rr r r rerlzTestSocketWrapper.__init__cCsFd}t||kr!||t|}|dkrt||7}t||ks|S)N)lenrecvConnectionAbortedError)r nbufrr r r recv_allhs  zTestSocketWrapper.recv_allFNT server_sideserver_hostnamedo_handshake_on_connectcCsDt|tjsJ|j|j|||d}|r||j||_dS)Nr)r9rr wrap_socketr do_handshakerU)r ssl_contextrrrssl_sockr r r starttlsqs  zTestSocketWrapper.starttlscCs t|j|Sr)r[rrr r r __getattr__ zTestSocketWrapper.__getattr__cCsdt|j|jS)Nz <{} {!r}>)rr@rrr3r r r __repr__szTestSocketWrapper.__repr__)rrrrrrrrr r r r rcs   rc@s$eZdZddZddZddZdS) SocketThreadcCsd|_|dSrj)_activerr3r r r rs zSocketThread.stopcCs ||Sr)startr3r r r __enter__szSocketThread.__enter__cGs |dSr)r)r excr r r __exit__rzSocketThread.__exit__N)rrrrrrr r r r rs rc@seZdZddZddZdS)rcCs:tj|dddd|_||_||_d|_||_||_dS)Nz test-clientT) threadingThreadrdaemon_timeout_sockr_prog_test)r testrprogrrr r r rs zTestThreadedClient.__init__c Cs\z |t|jWdSttfyty-}z |j|WYd}~dSd}~wwr)rrrKeyboardInterrupt SystemExit BaseExceptionrrrr r r runszTestThreadedClient.runN)rrrrr r r r r rs rcsHeZdZddZfddZddZddZd d Zed d Z Z S) rcCshtj|dddd|_d|_d|_||_||_||_d|_ ||_ t \|_ |_|j d||_dS)Nz test-serverTrF)rrrr_clients_finished_clients _max_clientsrrrrrw socketpair_s1_s2 setblockingr)r rrrrrrtr r r rs  zTestThreadedServer.__init__csz<|jr|jdkr%z|jdWntyYnwWtdSWtdSWtdSWtdStw)Nsstop)rfilenosendrrrr3rr r rs zTestThreadedServer.stopcCsz7|j|jd|Wdn1swYW|j|jdSW|j|jdS|j|jw)Nr)rr_runrrUrr3r r r r s     zTestThreadedServer.runc Cs.|jr|j|jkr dSt|j|jggg|j\}}}|j|vr"dS|j|vrz |j\}}Wnty9Yqt j yG|jsFYdSw|jd7_| |jz| | |Wdn1shwYWn!t tfyxty}z d|_z|j|wd}~ww|jsdSdS)NrnF)rr rselectrrracceptBlockingIOErrorrwrrr|_handle_clientrr r rr)r rwxconnrqrr r r rsF      zTestThreadedServer._runcCs|t|dSr)rrrr r r rrCz!TestThreadedServer._handle_clientcCs |jSr)r getsocknamer3r r r rqs zTestThreadedServer.addr) rrrrrr rrpropertyrqr r r rr rs   $rcCsBdd}|}||}d|_z ||W|dS|w)NcsdSrr r r r r onceszrun_briefly..onceF) create_task_log_destroy_pendingrErU)rAr!gentr r r run_brieflys  r&cCsXt|}|s*|dur|t}|dkrtj|tjd|r dSdS)NrgMbP?)timer:futures TimeoutErrorrEtasksrF)rApredrrdeadliner r r run_untils   r.c csLtjjj}ztjjtjddVWtjj|dStjj|w)zrContext manager to disable asyncio logger. For example, it can be used to ignore warnings in debug mode. rnN)r:logrlevelsetLevelrCRITICAL) old_levelr r r disable_loggers  "r4)r)r')+__doc__r:asyncio.events collectionsrrYrrrWrrrwrrryrr(unittestrrrUserDictrr@r!TestCaser/rrrrrrrrrrrrrr&r.r4r r r r sR +      'X