o SiS@s(dZddlZddlZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl m Z eeddhdZdd eDZd d d d dddZeeedZddZddZddZd'ddZejZdedefddZGdd d ZGd!d"d"ZGd#d$d$e j Z!Gd%d&d&e j"Z#dS)(zGA Python test reporter that generates test reports in JUnit XML format.N)Any)saxutils)_pretty_print_reporter > cCsi|] }t|d|dqS)z\x02x)chr).0ir M/home/ubuntu/.local/lib/python3.10/site-packages/absl/testing/xml_reporter.py !srz"z'z z z z )"'    z^(\w+) \((\S+)\)$cCs t|tS)zEscapes xml attributes.)rescape_escape_xml_attr_conversions)contentr r r_escape_xml_attr9s rcCs*tD] \}}|||}q|ddS)aEscapes a string to be used as XML CDATA. CDATA characters are treated strictly as character data, not as XML markup, but there are still certain restrictions on them. Args: s: the string to be escaped. Returns: An escaped version of the input string. z]]>z]] >)_control_character_conversionsitemsreplace)scharescapedr r r _escape_cdata?s  r cCs,|dus|dkr dStjj|tjjdS)zProduces an ISO8601 datetime. Args: timestamp: an Epoch based timestamp in seconds. Returns: A iso8601 format timestamp if the input is a valid timestamp, None otherwise Nr)tz)datetime fromtimestamptimezoneutc isoformat) timestampr r r_iso8601_timestampOs r(cCsh|d||f|D]!}t|dkr,|ddur,|ddur,|d|d|dfq |ddS)a+Prints an XML header of an arbitrary element. Args: element: element name (testsuites, testsuite, testcase) attributes: 2-tuple list with (attributes, values) already escaped stream: output stream to write test report XML to indentation: indentation added to the element header z%s<%srNz %s="%s"z> )writelen)element attributesstream indentation attributer r r_print_xml_element_header^s  r3objreturncCs6zt|WStydt|jt|jfYSw)z-Returns a string representation of an object.z)str Exceptiontype __module____name__)r4r r r _safe_strss   r;c@s8eZdZdZddZddZddZdd Zd d Zd S) _TestCaseResultajPrivate helper for _TextAndXMLTestResult that represents a test result. Attributes: test: A TestCase instance of an individual test method. name: The name of the individual test method. full_class_name: The full name of the test class. run_time: The duration (in seconds) it took to run the test. start_time: Epoch relative timestamp of when test started (in seconds) errors: A list of error 4-tuples. Error tuple entries are 1) a string identifier of either "failure" or "error" 2) an exception_type 3) an exception_message 4) a string version of a sys.exc_info()-style tuple of values ('error', err[0], err[1], self._exc_info_to_string(err)) If the length of errors is 0, then the test is either passed or skipped. skip_reason: A string explaining why the test was skipped. cCsd|_d|_d|_g|_||_|pt|}t|}|r)| d}| d}n@t j |j }t|t jjr?t j |jj }||drS|t|dd}|}n|dd}|d}t|dkrg|dnd}t||_t||_dS)Nr+r*.rr))run_time start_time skip_reasonerrorstestidr6&_CLASS_OR_MODULE_LEVEL_TEST_DESC_REGEXmatchgroupunittestutilstrclass __class__ isinstancecase_SubTest test_case startswithr-rsplitrnamefull_class_name)selfrC test_descrFrRrS class_namepartsr r r__init__s*     z_TestCaseResult.__init__cC ||_dSN)r?rT time_in_secsr r r set_run_time z_TestCaseResult.set_run_timecCrYrZr@r[r r rset_start_timer^z_TestCaseResult.set_start_timecCs|jdur d}d}nd}d}dd|jfdd|fd d|fd d |jfd |jfd t|jfg}td||d|||ddS)aQPrints an XML Summary of a TestCase. Status and result are populated as per JUnit XML test result reporter. A test that has been skipped will always have a skip reason, as every skip method in Python's unittest requires the reason arg to be passed. Args: stream: output stream to write test report XML to Nrun completednotrun suppressedrR%sstatusresulttime%.3f classnamer'testcasez z ) rArRr?rSr(r@r3_print_testcase_detailsr,)rTr0rfrgtest_case_attributesr r rprint_xml_summarys       z!_TestCaseResult.print_xml_summaryc CsT|jD]$}|\}}}}tt|}tt|}t|}|d|||||fqdS)Nz1 <%s message="%s" type="%s"> )rBrr;r6r r,)rTr0erroroutcomeexception_typemessage error_msgr r rrls     z'_TestCaseResult._print_testcase_detailsN) r:r9 __qualname____doc__rXr]r`rnrlr r r rr<~s% r<c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)_TestSuiteResultz)Private helper for _TextAndXMLTestResult.cCs(i|_i|_i|_d|_d|_i|_dS)Nr=)suitesfailure_counts error_countsoverall_start_timeoverall_end_time_testsuites_propertiesrTr r rrXs  z_TestSuiteResult.__init__cCst|jj}|dkr|jdd}t|jtjjr!t|jj j}| ||j | ||j D]&}|ddkrE|j|d7<dS|ddkrW|j|d7<dSq1dS)N _ErrorHolderr>r=rfailurer+ro)r8rCr:rSrQrLrHrMrNrO_setup_test_suiterwappendrBrxry)rTtest_case_result suite_nameror r radd_test_case_results     z%_TestSuiteResult.add_test_case_resultc Cstdd|jD}t|j}t|j}ddd|fdd|fdd|fdd |j|jfd t|jfg}td |||j ri| d t |j D]\}}| d t |t t|fqP| d|jD]b}|j|} tdd| D} tdd| D} |j|} |j|} dd|fddt| fdd| fdd| fdd | | fd t| fg}td||t | dddD]}||q| dql| ddS)Ncss|]}t|VqdSrZ)r-r xr r r sz5_TestSuiteResult.print_xml_summary..)rRr)testsz%dfailuresrBrhrir' testsuitesz z1 z css|] }|j|jVqdSrZ)r@r?rr r rrscss|]}|jVqdSrZr_rr r rrsrRre testsuitecSs|jSrZ)rR)tr r r.sz4_TestSuiteResult.print_xml_summary..)keyz z )sumrwvaluesrxryr{rzr(r3r|r,sortedrrr6maxminr-rn)rTr0overall_test_countoverall_failuresoverall_errorsoverall_attributesrRvaluersuitesuite_end_timesuite_start_timerrBsuite_attributesrr r rrnsH                  z"_TestSuiteResult.print_xml_summarycCs0||jvrdSg|j|<d|j|<d|j|<dS)zAdds a test suite to the set of suites tracked by this test run. Args: suite_name: string, The name of the test suite being initialized. Nr)rwrxry)rTrr r rr3s   z"_TestSuiteResult._setup_test_suitecC ||_dS)zvSets the start timestamp of this test suite. Args: timestamp_in_secs: timestamp in seconds since epoch N)r{rTtimestamp_in_secsr r r set_end_time? z_TestSuiteResult.set_end_timecCr)ztSets the end timestamp of this test suite. Args: timestamp_in_secs: timestamp in seconds since epoch N)rzrr r rr`Grz_TestSuiteResult.set_start_timeN) r:r9rtrurXrrnrrr`r r r rrvs+ rvcseZdZdZeZeZedffdd Z fddZ fddZ fd d Z d d Z d%fd d Z  d&ddZddZddZfddZfddZfddZfddZfddZfdd Zfd!d"Zfd#d$ZZS)'_TextAndXMLTestResultzoPrivate TestResult class that produces both formatted text results and XML. Used by TextAndXMLTestRunner. NcsFt|||||_i|_||_|r||j_||_t |_ dSrZ) superrX xml_streampending_test_case_results_TEST_SUITE_RESULT_CLASSrr| time_getter threadingRLock_pending_test_case_results_lock)rTrr0 descriptions verbosityrtestsuites_propertiesrKr rrXYs z_TextAndXMLTestResult.__init__cs||_t|dSrZ)rr@r startTestrTrCrr rrfs z_TextAndXMLTestResult.startTestcs|j\t|||}|s*|pt|}tjd| WddSt |dddur7| |_ t|}| |j }| || |j |j||j|=WddS1sbwYdS)NzNo pending test case: %s r@)rrstopTestget_pending_test_case_resultrDr6sysstderrr,getattrrr@r]r`rrr)rTrCrg test_nametest_idr?rr rrjs"       "z_TextAndXMLTestResult.stopTestcs|j|tdSrZ)rr`rr startTestRunr}rr rr}sz"_TextAndXMLTestResult.startTestRuncCs|j||j9|jD]'}|j|}t|dddur0|jj|j}||| |j|j |q|j WddS1sGwYdS)Nr@) rrrrrrr{r@r]r`rclear)rTrrgr?r r r stopTestRuns     "z!_TextAndXMLTestResult.stopTestRuncs"|r t||Sdtj|S)aWConverts a sys.exc_info()-style tuple of values into a string. This method must be overridden because the method signature in unittest.TestResult changed between Python 2.2 and 2.4. Args: err: A sys.exc_info() tuple of values for an error. test: The test method. Returns: A formatted exception string. r))r_exc_info_to_stringjoin tracebackformat_exception)rTerrrCrr rrs z)_TextAndXMLTestResult._exc_info_to_stringcCs|j5t|}||jvr|||j|<|r |j|j||r0||j|_WddSWddS1s;wYdS)aAdds result information to a test case result which may still be running. If a result entry for the test already exists, add_pending_test_case_result will add error summary tuples and/or overwrite skip_reason for the result. If it does not yet exist, a result entry will be created. Note that a test result is considered to have been run and passed only if there are no errors or skip_reason. Args: test: A test method as defined by unittest error_summary: A 4-tuple with the following entries: 1) a string identifier of either "failure" or "error" 2) an exception_type 3) an exception_message 4) a string version of a sys.exc_info()-style tuple of values ('error', err[0], err[1], self._exc_info_to_string(err)) If the length of errors is 0, then the test is either passed or skipped. skip_reason: a string explaining why the test was skipped N)rrDr_TEST_CASE_RESULT_CLASSrBrrA)rTrC error_summaryrArr r radd_pending_test_case_results  "z2_TextAndXMLTestResult.add_pending_test_case_resultcCs<|jt|}|j|=WddS1swYdSrZ)rrDrrTrCrr r rdelete_pending_test_case_results "z5_TextAndXMLTestResult.delete_pending_test_case_resultcCst|}|j|dSrZ)rDrgetrr r rrsz2_TextAndXMLTestResult.get_pending_test_case_resultcst|||dSrZ)r addSuccessrrrr rrs z _TextAndXMLTestResult.addSuccessc>t||d|d|d|j||df}|j||ddS)Nrorr+rCr)raddErrorrrrTrCrrrr rr  z_TextAndXMLTestResult.addErrorcr)Nrrr+rr)r addFailurerrrrr rrrz _TextAndXMLTestResult.addFailurecs t|||j||ddS)N)rA)raddSkipr)rTrCreasonrr rrsz_TextAndXMLTestResult.addSkipcsBt||tt|ddr|d|j||d||dS)NrecordPropertyEXPECTED_FAILUREr)raddExpectedFailurecallablerrrr)rTrCrrr rrs  z(_TextAndXMLTestResult.addExpectedFailurecs>t||p t|}dddd|f}|j||ddS)Nror)z,Test case %s should have failed, but passed.r)raddUnexpectedSuccessrDr6r)rTrCrrrr rrs z*_TextAndXMLTestResult.addUnexpectedSuccesscs~t||||dur4t|d|jr$d|d|d|j||df}nd|d|d|j||df}nd}|j||ddS)Nrrr+rror)r addSubTest issubclassfailureExceptionrr)rTrCsubtestrrrr rrs  z _TextAndXMLTestResult.addSubTestcs(t|jd|j|jdS)Nz )r printErrorsrr,rrnr}rr rrs  z!_TextAndXMLTestResult.printErrorsrZ)NN)r:r9rtrurvrr<r _time_copyrXrrrrrrrrrrrrrrrr __classcell__r r rrrPs0            rcsdeZdZUdZeZdZiZee e fe d<d fdd Z e ddZ fdd Ze d d ZZS) TextAndXMLTestRunnerzA test runner that produces both formatted text results and XML. It prints out the names of tests as they are run, errors as they occur, and a summary of the results at the end of the test run. Nr|cs(tj|i||dur||_dSdS)aInitialize a TextAndXMLTestRunner. Args: xml_stream: file-like or None; XML-formatted test results are output via this object's write() method. If None (the default), the new instance behaves as described in the set_default_xml_stream method documentation below. *args: passed unmodified to unittest.TextTestRunner.__init__. **kwargs: passed unmodified to unittest.TextTestRunner.__init__. N)rrX _xml_stream)rTrargskwargsrr rrXs  zTextAndXMLTestRunner.__init__cCr)aSets the default XML stream for the class. Args: xml_stream: file-like or None; used for instances when xml_stream is None or not passed to their constructors. If None is passed, instances created with xml_stream=None will act as ordinary TextTestRunner instances; this is the default state before any calls to this method have been made. N)r)clsrr r rset_default_xml_stream#s z+TextAndXMLTestRunner.set_default_xml_streamcs2|jdur tS|j|j|j|j|j|jdS)N)r)rr _makeResult_TEST_RESULT_CLASSr0rrr|r}rr rr0s  z TextAndXMLTestRunner._makeResultcCs||j|<dSrZ)r|)rrrr r rset_testsuites_property8sz,TextAndXMLTestRunner.set_testsuites_propertyrZ)r:r9rtrurrrr|dictr__annotations__rX classmethodrrrrr r rrrs   r)r))$rur"rerrrhrtypingrrHxml.saxr absl.testingrsetrange_bad_control_character_codesrrupdatecompilerErr r(r3robjectr6r;r<rvTextTestResultrTextTestRunnerrr r r rsF       hj7