U UcnS@s&dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z e e dddddhZd d eDZd d d ddddZeeedZddZddZddZd%ddZejZeedrejZneZGdddeZGdd d eZGd!d"d"e j Z!Gd#d$d$ej"Z#dS)&zGA Python test reporter that generates test reports in JUnit XML format.N)saxutils)_pretty_print_reporter cCsi|]}t|d|qS)z\x{:02x})chrformat).0ir =/tmp/pip-unpacked-wheel-yg88_g5m/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_attr7srcCs*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 rcCs,|dks|dkrdStjj|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_timestampMs r'cCsh|d||f|D]B}t|dkr|ddk r|ddk r|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  r2 _some_strc@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|_||_|p,t|}t|}|rR| d}| d}nt j |j }t|t jjr~t j |jj }||dr|t|dd}|}n,|dd}|d}t|dkr|dnd}t||_t||_dS)Nr*r).rr()run_time start_time skip_reasonerrorstestidstr&_CLASS_OR_MODULE_LEVEL_TEST_DESC_REGEXmatchgroupunittestutilZstrclass __class__ isinstancecase_SubTest test_case startswithr,rsplitrnamefull_class_name)selfr;Z test_descr?rJrK class_namepartsr r r __init__s*     z_TestCaseResult.__init__cCs ||_dSN)r7rLZ time_in_secsr r r set_run_timesz_TestCaseResult.set_run_timecCs ||_dSrPr8rQr r r set_start_timesz_TestCaseResult.set_start_timecCs|jdkrd}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 NrunZ completedZnotrunZ suppressedrJ%sstatusresulttime%.3f classnamer&Ztestcasez z ) r9rJr7rKr'r8r2_print_testcase_detailsr+)rLr/rWrXZtest_case_attributesr r r print_xml_summarys       z!_TestCaseResult.print_xml_summaryc CsT|jD]H}|\}}}}tt|}tt|}t|}|d|||||fqdS)Nz1 <%s message="%s" type="%s"> )r:r _safe_strr=rr+)rLr/errorZoutcomeZexception_typemessage error_msgr r r r\s     z'_TestCaseResult._print_testcase_detailsN) __name__ __module__ __qualname____doc__rOrRrTr]r\r r r r r4ws %r4c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)_TestSuiteResultz)Private helper for _TextAndXMLTestResult.cCs(i|_i|_i|_d|_d|_i|_dS)Nr5)suitesfailure_counts error_countsoverall_start_timeoverall_end_time_testsuites_propertiesrLr r r rOs z_TestSuiteResult.__init__cCst|jj}|dkr$|jdd}t|jtjjrBt|jj j}| ||j | ||j D]J}|ddkr|j|d7<qqb|ddkrb|j|d7<qqbdS)NZ _ErrorHolderr6r5rfailurer*r_)typer;rbrKrIrDrArErFrG_setup_test_suitergappendr:rhri)rLtest_case_result suite_namer_r r r add_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 r| d t |j D]&\}}| d t |t t|fq| d|jD]}|j|} tdd| D} tdd| D} |j|} |j|} dd|fddt| fdd| fdd| fdd | | fd t| fg}td||| D]}||qv| dq| ddS)Ncss|]}t|VqdSrP)r,r xr r r sz5_TestSuiteResult.print_xml_summary..)rJr(testsz%dfailuresr:rYrZr&Z testsuitesz z1 z css|]}|j|jVqdSrP)r8r7rur r r rwscss|] }|jVqdSrPrSrur r r rwsrJrVZ testsuitez z )sumrgvaluesrhrirkrjr'r2rlr+sortedrrr=maxminr,r])rLr/Zoverall_test_countZoverall_failuresZoverall_errorsZoverall_attributesrJvaluerssuiteZsuite_end_timeZsuite_start_timeryr:Zsuite_attributesrrr r r r]sH                 z"_TestSuiteResult.print_xml_summarycCs0||jkrdSg|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)rgrhri)rLrsr r r rp*s    z"_TestSuiteResult._setup_test_suitecCs ||_dS)zvSets the start timestamp of this test suite. Args: timestamp_in_secs: timestamp in seconds since epoch N)rkrLZtimestamp_in_secsr r r set_end_time6sz_TestSuiteResult.set_end_timecCs ||_dS)ztSets the end timestamp of this test suite. Args: timestamp_in_secs: timestamp in seconds since epoch N)rjrr r r rT>sz_TestSuiteResult.set_start_timeN) rbrcrdrerOrtr]rprrTr r r r rfs) rfcseZdZdZeZeZedffdd Z fddZ fddZ fd d Z d d Z d%fd d Zd&ddZddZddZfddZfddZfddZfddZfddZfdd Zfd!d"Zfd#d$ZZS)'_TextAndXMLTestResultzoPrivate TestResult class that produces both formatted text results and XML. Used by TextAndXMLTestRunner. NcsJtt||||||_i|_||_|r6||j_||_t |_ dSrP) superrrO xml_streampending_test_case_results_TEST_SUITE_RESULT_CLASSrrl time_getter threadingRLock_pending_test_case_results_lock)rLrr/ descriptions verbosityrtestsuites_propertiesrCr r rOPs z_TextAndXMLTestResult.__init__cs||_tt||dSrP)rr8rr startTestrLr;rr r r]s z_TextAndXMLTestResult.startTestc s|jtt||||}|sT|p4t|}tj d|W5QRdSt|}| |j }| || |j |j||j|=W5QRXdS)NzNo pending test case: %s )rrrstopTestget_pending_test_case_resultr<r=sysstderrr+rr8rRrTrrtr)rLr;rX test_nametest_idr7rr r ras    z_TextAndXMLTestResult.stopTestcs"|j|tt|dSrP)rrTrrr startTestRunrmrr r rqsz"_TextAndXMLTestResult.startTestRunc Cs|j||j`|jD]H}|j|}t|drZ|jj|j}||| |j|j |q|j W5QRXdS)Nr8) rrrrrhasattrrkr8rRrTrtclear)rLrrXr7r r r stopTestRunus     z!_TextAndXMLTestResult.stopTestRuncs&|rtt|||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()rr_exc_info_to_stringjoin tracebackformat_exception)rLerrr;rr r rs z)_TextAndXMLTestResult._exc_info_to_stringc Cs^|jNt|}||jkr*|||j|<|r@|j|j||rP||j|_W5QRXdS)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)rr<r_TEST_CASE_RESULT_CLASSr:rqr9)rLr; error_summaryr9rr r r add_pending_test_case_results  z2_TextAndXMLTestResult.add_pending_test_case_resultc Cs&|jt|}|j|=W5QRXdSrP)rr<rrLr;rr r r delete_pending_test_case_resultsz5_TextAndXMLTestResult.delete_pending_test_case_resultcCst|}|j|dSrP)r<rgetrr r r rsz2_TextAndXMLTestResult.get_pending_test_case_resultcstt||||dSrP)rr addSuccessrrrr r rsz _TextAndXMLTestResult.addSuccesscsBtt|||d|d|d|j||df}|j||ddS)Nr_rr*r;r)rraddErrorrrrLr;rrrr r rs  z_TextAndXMLTestResult.addErrorcsBtt|||d|d|d|j||df}|j||ddS)Nrnrr*rr)rr addFailurerrrrr r rs  z _TextAndXMLTestResult.addFailurecs$tt||||j||ddS)N)r9)rraddSkipr)rLr;reasonrr r rsz_TextAndXMLTestResult.addSkipcsFtt|||tt|ddr8|d|j||d||dS)NrecordPropertyZEXPECTED_FAILUREr)rraddExpectedFailurecallablegetattrrrr)rLr;rrr r rs  z(_TextAndXMLTestResult.addExpectedFailurecsBtt|||pt|}dddd|f}|j||ddS)Nr_r(z,Test case %s should have failed, but passed.r)rraddUnexpectedSuccessr<r=r)rLr;rrrr r rsz*_TextAndXMLTestResult.addUnexpectedSuccesscstt|||||dk rlt|d|jrLd|d|d|j||df}qpd|d|d|j||df}nd}|j||ddS)Nrrnr*rr_r)rr addSubTest issubclassZfailureExceptionrr)rLr;Zsubtestrrrr r rs  z _TextAndXMLTestResult.addSubTestcs,tt||jd|j|jdS)Nz )rr printErrorsrr+rr]rmrr r rs z!_TextAndXMLTestResult.printErrors)N)NN)rbrcrdrerfrr4r _time_copyrOrrrrrrrrrrrrrrrr __classcell__r r rr rGs0           rcsReZdZdZeZdZiZd fdd Ze ddZ fddZ e d d Z Z S) 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. Ncs$tt|j|||dk r ||_dS)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)rrrO _xml_stream)rLrargskwargsrr r rOs zTextAndXMLTestRunner.__init__cCs ||_dS)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 r set_default_xml_streams z+TextAndXMLTestRunner.set_default_xml_streamcs:|jdkrtt|S|j|j|j|j|j|jdSdS)N)r) rrr _makeResult_TEST_RESULT_CLASSr/rrrlrmrr r r$s z TextAndXMLTestRunner._makeResultcCs||j|<dSrP)rl)rkeyrr r r set_testsuites_property,sz,TextAndXMLTestRunner.set_testsuites_property)N)rbrcrdrerrrrlrO classmethodrrrrr r rr rs r)r()$rer!rerrrYrrAZxml.saxrZ absl.testingrsetrangeZ_bad_control_character_codesrrupdatecompiler>rrr'r2rrr3r^r=objectr4rfZTextTestResultrZTextTestRunnerrr r r r sF      hh4