o 2hS@s,dZddlZddlZddlZddlZddlZddlZddlmZm Z ddl Z ddl m Z ddl mZeeddhdZdd eDZd d d d dddZeeedZddZddZddZd'ddZejZdedefddZGdd d ZGd!d"d"Z Gd#d$d$ej!Z"Gd%d&d&e j#Z$dS)(zGA Python test reporter that generates test reports in JUnit XML format.N)AnyDict)saxutils)_pretty_print_reporter > cCsi|] }t|d|dqS)z\x02x)chr).0irT/var/www/html/chatgem/venv/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)contentrrr_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)scharescapedrrr _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) timestamprrr_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 attributerrr_print_xml_element_header^s  r4objreturncCs6zt|WStydt|jt|jfYSw)z-Returns a string representation of an object.z)str Exceptiontype __module____name__)r5rrr _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_reasonerrorstestidr7&_CLASS_OR_MODULE_LEVEL_TEST_DESC_REGEXmatchgroupunittestutilstrclass __class__ isinstancecase_SubTest test_case startswithr.rsplitrnamefull_class_name)selfrD test_descrGrSrT class_namepartsrrr__init__s*     z_TestCaseResult.__init__cC ||_dSN)r@rU time_in_secsrrr set_run_time z_TestCaseResult.set_run_timecCrZr[rAr\rrrset_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 suppressedrS%sstatusresulttime%.3f classnamer(testcasez z ) rBrSr@rTr)rAr4_print_testcase_detailsr-)rUr1rgrhtest_case_attributesrrrprint_xml_summarys       z!_TestCaseResult.print_xml_summaryc CsT|jD]$}|\}}}}tt|}tt|}t|}|d|||||fqdS)Nz1 <%s message="%s" type="%s"> )rCrr<r7r!r-)rUr1erroroutcomeexception_typemessage error_msgrrrrms     z'_TestCaseResult._print_testcase_detailsN) r;r: __qualname____doc__rYr^rarormrrrrr=~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_propertiesrUrrrrYs  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,rp)r9rDr;rTrRrMrIrNrOrP_setup_test_suiterxappendrCryrz)rUtest_case_result suite_namerprrradd_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|VqdSr[)r.r xrrr sz5_TestSuiteResult.print_xml_summary..)rSr*testsz%dfailuresrCrirjr( testsuitesz z1 z css|] }|j|jVqdSr[)rAr@rrrrrscss|]}|jVqdSr[r`rrrrrsrSrf testsuitecSs|jSr[)rS)trrr.sz4_TestSuiteResult.print_xml_summary..)keyz z )sumrxvaluesryrzr|r{r)r4r}r-sortedrrr7maxminr.ro)rUr1overall_test_countoverall_failuresoverall_errorsoverall_attributesrSvaluersuitesuite_end_timesuite_start_timerrCsuite_attributesrrrrrosH                  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)rxryrz)rUrrrrr3s   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|rUtimestamp_in_secsrrr 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)r{rrrrraGrz_TestSuiteResult.set_start_timeN) r;r:rurvrYrrorrrarrrrrws+ rwcseZdZdZeZeZedffdd 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 |_ dSr[) superrY xml_streampending_test_case_results_TEST_SUITE_RESULT_CLASSrr} time_getter threadingRLock_pending_test_case_results_lock)rUrr1 descriptions verbosityrtestsuites_propertiesrLrrrYYs z_TextAndXMLTestResult.__init__cs||_t|dSr[)rrAr startTestrUrDrrrrfs z_TextAndXMLTestResult.startTestcs|j\t|||}|s*|pt|}tjd| WddSt |dddur7| |_ t|}| |j }| || |j |j||j|=WddS1sbwYdS)NzNo pending test case: %s rA)rrstopTestget_pending_test_case_resultrEr7sysstderrr-getattrrrAr^rarrr)rUrDrh test_nametest_idr@rrrrjs"       "z_TextAndXMLTestResult.stopTestcs|j|tdSr[)rrarr startTestRunr~rrrr}sz"_TextAndXMLTestResult.startTestRuncCs|j||j9|jD]'}|j|}t|dddur0|jj|j}||| |j|j |q|j WddS1sGwYdS)NrA) rrrrrrr|rAr^rarclear)rUrrhr@rrr 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)rUerrrDrrrrs 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)rrEr_TEST_CASE_RESULT_CLASSrCrrB)rUrD error_summaryrBrrrradd_pending_test_case_results  "z2_TextAndXMLTestResult.add_pending_test_case_resultcCs<|jt|}|j|=WddS1swYdSr[)rrErrUrDrrrrdelete_pending_test_case_results "z5_TextAndXMLTestResult.delete_pending_test_case_resultcCst|}|j|dSr[)rErgetrrrrrsz2_TextAndXMLTestResult.get_pending_test_case_resultcst|||dSr[)r addSuccessrrrrrrs z _TextAndXMLTestResult.addSuccessc>t||d|d|d|j||df}|j||ddS)Nrprr,rDr)raddErrorrrrUrDrrrrrr  z_TextAndXMLTestResult.addErrorcr)Nrrr,rr)r addFailurerrrrrrrrz _TextAndXMLTestResult.addFailurecs t|||j||ddS)N)rB)raddSkipr)rUrDreasonrrrrsz_TextAndXMLTestResult.addSkipcsBt||tt|ddr|d|j||d||dS)NrecordPropertyEXPECTED_FAILUREr)raddExpectedFailurecallablerrrr)rUrDrrrrrs  z(_TextAndXMLTestResult.addExpectedFailurecs>t||p t|}dddd|f}|j||ddS)Nrpr*z,Test case %s should have failed, but passed.r)raddUnexpectedSuccessrEr7r)rUrDrrrrrrs z*_TextAndXMLTestResult.addUnexpectedSuccesscs~t||||dur4t|d|jr$d|d|d|j||df}nd|d|d|j||df}nd}|j||ddS)Nrrr,rrpr)r addSubTest issubclassfailureExceptionrr)rUrDsubtestrrrrrrs  z _TextAndXMLTestResult.addSubTestcs(t|jd|j|jdS)Nz )r printErrorsrr-rror~rrrrs  z!_TextAndXMLTestResult.printErrorsr[)NN)r;r:rurvrwrr=r _time_copyrYrrrrrrrrrrrrrrrr __classcell__rrrrrPs0            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)rrY _xml_stream)rUrargskwargsrrrrYs  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)clsrrrrset_default_xml_stream#s z+TextAndXMLTestRunner.set_default_xml_streamcs2|jdur tS|j|j|j|j|j|jdS)N)r)rr _makeResult_TEST_RESULT_CLASSr1rrr}r~rrrr0s  z TextAndXMLTestRunner._makeResultcCs||j|<dSr[)r})rrrrrrset_testsuites_property8sz,TextAndXMLTestRunner.set_testsuites_propertyr[)r;r:rurvrrrr}rr__annotations__rY classmethodrrrrrrrrrs   r)r*)%rvr#rerrrirtypingrrrIxml.saxr absl.testingrsetrange_bad_control_character_codesrrupdatecompilerFrr!r)r4robjectr7r<r=rwTextTestResultrTextTestRunnerrrrrrsF      hj7