a xd3@s~ddlmZddlZddlZddlZddlZddlZddlZddlZ ddl Z ddl Z ddl m Z ejdvZGddde ZdS))absolute_importN) TestFormat)win32cygwinc@sDeZdZgfddZddZddZddZd d Zed d Z d S) GoogleTestcCsDt|_t|d|_t|}tr,|d7}||dh|_||_dS)N;z.exe.py)setseen_executablesstrsplit test_sub_dirs kIsWindows test_suffixes run_under)selfrZ test_suffixrZ exe_suffixr_/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lit/formats/googletest.py__init__szGoogleTest.__init__c Cs||ddg}ztj||jd}WnFtjyj}z,|d|td|jfWYd}~dSd}~00t t dd| dS) Nz--gtest_list_testsz--gtest_filter=-*DISABLED_*)envz=unable to discover google-tests in %r: %s. Process output: %srcSstj|dS)Nz )litutilZ to_string startswith)linerrr*z*GoogleTest.get_num_tests..F) prepareCmd subprocess check_output environmentCalledProcessErrorwarningsysexc_infooutputsummap splitlines)rpath litConfig localConfigZ list_test_cmdoutexcrrr get_num_testss& zGoogleTest.get_num_testsc csd}tj}||}|jD]\}tj||} tj| s@qtjj | |j dD]&} tj||| } | |j vr| d| qRn |j | || ||} | durV|} tt| | }||kr| dkr| d} tt| | }qt|D]`}||| t|t|f}d| |jjttt|t|gd}tjj|||| |dVqqR||| d f}tjj|||| d VqRqdS) Ni)suffixesz7Skip adding %r since it has been added to the test poolr-z.json) file_pathgtest_json_fileZ#failed_to_discover_tests_from_gtest)r2)rrZusable_core_count getSourcePathrosr)joinisdirZ listdir_filesrr r"addr.intmathceilranger confignamegetpidTest)rZ testSuite path_in_suiter*r+Zinit_shard_sizeZ core_count source_pathsubdirZdir_pathfnZexecpathZ num_testsZ shard_sizeZnshardidxtestPathZ json_filerrrgetTestsInDirectory-sh            zGoogleTest.getTestsInDirectoryc s`|jdurtjjdfStj|\}}tj|sTtj|\}}|d|}q*tj|\}}tj|\}}ddl m }||j |j k} d|j| rdndtj d|tj d |d } |jj| |g||jr|j|jrtjjdfSfd d } | | } z$tjj|jj|jd d\} }Wn^tjjy}z@d|jd|jd}tjj| |d|jdfWYd}~Sd}~00tj|jsd|j}dd|d}tjj| ||fS|dkrtjjdfSfdd}d}t|jdd}t |}| r2t!|d| d<| | d}|dD]}|dD]}|d}|dksR|d krvqR|d!d"|d!}d#d$"|f}d%|vrd }||7}||}|r||d&7}|d%D]}||d'd7}q|d7}n|d(krR||7}|d)7}qRqFWdn1s.0Y|sT|dd|d7}tjj|fS)*N/r) TestOrderzjson:10GTEST_TOTAL_SHARDSGTEST_SHARD_INDEX)Z GTEST_OUTPUTZ GTEST_SHUFFLErMrNcs*ddd|D}d|dfS)N cSsg|]\}}|d|qS)=r).0kvrrr rz@GoogleTest.execute..get_shard_header..zScript(shard): -- %s %s -- )r6items) shard_envZ shard_envs)cmdrrget_shard_headersz,GoogleTest.execute..get_shard_headerT)rtimeoutredirect_stderr z -- exit: z -- zReached timeout of z secondsz$shard JSON output does not exist: %scsg}d|}d|}d}D]J}||r4d}q ||r\t|dkrNnd|S|r ||q ds|Jd|dS)Nz [ RUN ] z [ FAILED ] FTrr[z$gtest did not report the result for )r(rlenr6append)Z test_nameresheaderfooterZin_rangel)r,rrget_test_stdouts    z+GoogleTest.execute..get_test_stdoutFutf-8encodingZ random_seedZGTEST_RANDOM_SEED testsuites testsuiteresult SUPPRESSEDSKIPPEDr>.#Script: -- %s --gtest_filter=%s -- rOfailuresz failure COMPLETEDunresolved test result )#r3rr@FAILr5r)r r4existsZlit.cl_argumentsrJorderZRANDOMenvirongetr=r updaterZ useValgrindZ valgrindArgsZ noExecutePASSrZexecuteCommandZmaxIndividualTestTimeZExecuteCommandTimeoutExceptionr,exitCodeTIMEOUTopenjsonloadr r6)rtestr*rFZtestNameZ namePrefixZ total_shardsZ shard_idxrJZ use_shufflerVrXZ shard_header_rxeZ stream_msgerrmsgrbZfound_failed_testfZjfr%testcasetestinforhtestnamer_Ztest_outfailr)rWr,rexecutehs                         0zGoogleTest.executecCsL|ddrtjg|}|jrHt|jtr8|j|}nt|j|}|S)a?Insert interpreter if needed. It inserts the python exe into the command if cmd[0] ends in .py or caller specified run_under. We cannot rely on the system to interpret shebang lines for us on Windows, so add the python executable to the command if this is a .py script. rr )endswithr# executabler isinstancelistshlexr )rrWrrrrs    zGoogleTest.prepareCmdc Csdd}||}dd|D}||}|D]J}tj|jsT||||q*|jjp^d}d}t|jdd}zt |d }Wn>tj y} z$t d |jd | j WYd} ~ n d} ~ 00|D]Z} | d D]J} | d dkrq|j dd| d| df} tj|j| |j|j} | dd| d}d|j|f}d}| d dkrbtjj}ndd| vrd}tjj}|}| dD]}||dd 7}qn(| d dkrtjj}ntjj}|d}t| ddd}tj|||}|jjpd|_||_||}| ||| || qqWdn1s@0Yt|j|s*|r*||||q*||fS)NcSsdd|DS)NcSsg|]}|jdur|qS)Nr3rQtrrrrTrzOGoogleTest.post_process_shard_results..remove_gtest..r)testsrrr remove_gtestsz;GoogleTest.post_process_shard_results..remove_gtestcSsg|]}|jr|qSrrrrrrrTrz9GoogleTest.post_process_shard_results..gFrcrdrfzFailed to parse json file: r[rgrhrir>rkrlrHrjrmTrnrorptimer)r5r)rrr3r]rhstartrzr{r|JSONDecodeError RuntimeErrordocrArr@Zsuiter=r2rjrqrwZ UNRESOLVEDfloatResultpidZ setResultremoveZ isFailure)Zselected_testsZdiscovered_testsrZgtestsr} start_timeZhas_failure_in_shardrrfrrrrFZsubtestrr_r%Z returnCoderZ elapsed_timer^rrrpost_process_shard_resultss|             .    z%GoogleTest.post_process_shard_resultsN) __name__ __module__ __qualname__rr.rGrr staticmethodrrrrrrs ;jr) __future__rr{r:r5rrr#Zlit.TestrZlit.TestRunnerZlit.utilbaserplatformrrrrrrs