File: //home/arjun/projects/env/lib/python3.10/site-packages/_pytest/__pycache__/junitxml.cpython-310.pyc
o
%wemd � @ s� d Z ddlZddlZddlZddlZddlm mZ ddl m Z ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd l
mZ dd
l
mZ ddlZdd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl m!Z! ddl"m#Z# ddl$m%Z% e#d � Z&de'de(fdd�Z)dCdd �Z*i Z+d d d gie+d < d g d!�ie+d"< e+d �,� e+d#< e*e+d# e+d" � e+d e+d$< G d%d&� d&�Z-d'e d(e(ddfd)d*�Z.ej/d'e de e(e'gdf fd+d,��Z0ej/d'e de e(e'gdf fd-d.��Z1d/e(d0e(ddfd1d2�Z2ej/d3d4�d'e de e(e'gdf fd5d6��Z3d7e ddfd8d9�Z4d:eddfd;d<�Z5d:eddfd=d>�Z6d?e(de
e( fd@dA�Z7G dBd� d�Z8dS )Da Report test results in JUnit-XML format, for use with Jenkins and build
integration servers.
Based on initial code from Ross Lawley.
Output conforms to
https://github.com/jenkinsci/xunit-plugin/blob/master/src/main/resources/org/jenkinsci/plugins/xunit/types/model/xsd/junit-10.xsd
� N)�datetime)�Callable)�Dict)�List)�Match)�Optional)�Tuple)�Union)�nodes)�timing)�
ExceptionRepr)�ReprFileLocation)�Config)� filename_arg)�Parser)�FixtureRequest)�
TestReport)�StashKey)�TerminalReporter�LogXML�arg�returnc C s, dt t dtfdd�}d}t�||t| ��S )a! Visually escape invalid XML characters.
For example, transforms
'hello\aworld\b'
into
'hello#x07world#x08'
Note that the #xABs are *not* XML escapes - missing the ampersand «.
The idea is to escape visually for the user rather than for XML itself.
�matchobjr c S s$ t | �� �}|dkrd| S d| S )N� z#x%02Xz#x%04X)�ord�group)r �i� r �I/home/arjun/projects/env/lib/python3.10/site-packages/_pytest/junitxml.py�repl3 s z bin_xml_escape.<locals>.replu [^
-~--�က0-ჿFF])r �str�re�sub)r r �illegal_xml_rer r r �bin_xml_escape( s �r$ c C sX i }| � � D ] \}}|� � D ]\}}t|t�s tt|���|| ||< qq| �|� d S �N)�items�
isinstance�list� TypeError�type�update)�left�right�result�kl�vl�kr�vrr r r � merge_familyC s
�r3 �testcase� classname�name�_base)�file�line�url� _base_legacy�xunit1�xunit2c @ sx e Zd Zdeeef ddddfdd�Zdejddfd d
�Z d ed e
ddfd
d�Z d ed e
ddfdd�Z de
ej fdd�Zdeddfdd�Zdejfdd�Zd5dedede
e ddfdd �Zd eddfd d �Zd ed!edefd"d#�Zd ed ed$eddfd%d&�Zd eddfd'd(�Zd eddfd)d*�Zd eddfd+d,�Zd eddfd-d.�Zd eddfd/d0�Zd eddfd1d2�Zd6d3d4�ZdS )7�
_NodeReporter�nodeid�xmlr r Nc C s< || _ || _| jj| _| jj| _d| _g | _g | _i | _d S )N� )�idr@ � add_stats�family�duration�
propertiesr
�attrs)�selfr? r@ r r r �__init__Z s
z_NodeReporter.__init__�nodec C s | j �|j� | j�|� d S r% )r@ rC �tagr
�append)rH rJ r r r rL d s z_NodeReporter.appendr6 �valuec C s | j �t|�t|�f� d S r% )rF rL r r$ �rH r6 rM r r r � add_propertyh s z_NodeReporter.add_propertyc C s t |�| jt|�< d S r% )r$ rG r rN r r r �
add_attributek � z_NodeReporter.add_attributec C �> | j r t�d�}| j D ]\}}|�tjd||d�� q |S dS �z9Return a Junit node containing custom properties, if any.rF �property�r6 rM N)rF �ET�ElementrL �rH rF r6 rM r r r �make_properties_noden �
z"_NodeReporter.make_properties_node�
testreportc C s� t |j�}| j}|d d� }| jjr|�d| jj� d�|�t|d �|jd d�}|jd d ur:t |jd �|d< t
|d�rD|j |d< || _| j� |� | j
dkrTd S i }| j�� D ]}|t| j
d v rm| j| ||< q[|| _d S )
N���r �.)r5 r6 r8 � r9 r: r< r4 )�mangle_test_addressr? rG r@ �prefix�insert�joinr$ �locationr �hasattrr: r+ rD �keys�families)rH r[ �names�existing_attrs�
classnamesrG �
temp_attrs�keyr r r �record_testreportw s.
�
�
z _NodeReporter.record_testreportc C sB t jd| jd| j d�}| �� }|d ur|�|� |�| j� |S )Nr4 �%.3f)�time)rV rW rG rE rY rL �extendr
)rH r4 rF r r r �to_xml� s
z_NodeReporter.to_xmlrK �message�datac C s& t j||d�}t|�|_| �|� d S )N)rq �rV rW r$ �textrL )rH rK rq rr rJ r r r � _add_simple� s
z_NodeReporter._add_simple�reportc C s� | j js |jr d S |j}|j}|j}| j jdkrd S d}| j jdv r(| �|d�}| j jdv r?|| �|d�7 }| �||d� d}| j jdv rV|| �|d �7 }| �||d
� d}|ra| �||d� d S d S ) N�no� )�log�allz Captured Log )�
system-out�out-errrz z Captured Out r{ )�
system-errr| rz z Captured Err r} ) r@ �log_passing_tests�passed� capstdout�caplog� capstderr�logging�_prepare_content�_write_content)rH rv � content_out� content_log� content_err� content_allr r r �write_captured_output� s* �z#_NodeReporter.write_captured_output�content�headerc C s d� |�dd�|dg�S )N�
�P �-rx )rb �center)rH r� r� r r r r� � � z _NodeReporter._prepare_content�jheaderc C s" t �|�}t|�|_| �|� d S r% rs )rH rv r� r� rK r r r r� � s
z _NodeReporter._write_contentc C s | � d� d S )Nr )rC �rH rv r r r � append_pass� � z_NodeReporter.append_passc C sp t |d�r
| �dd� d S |jd usJ �t|jdd �}|d ur#|j}nt|j�}t|�}| �d|t|j�� d S )N�wasxfail�skippedz%xfail-marked test passes unexpectedly� reprcrash�failure)rd ru �longrepr�getattrrq r r$ )rH rv r� rq r r r �append_failure� s
�
z _NodeReporter.append_failurec C s& |j d usJ �| �ddt|j �� d S )N�errorzcollection failure)r� ru r r� r r r �append_collect_error� s z"_NodeReporter.append_collect_errorc C s | � ddt|j�� d S )Nr� zcollection skipped)ru r r� r� r r r �append_collect_skipped� r� z$_NodeReporter.append_collect_skippedc C sv |j d usJ �t|j dd �}|d ur|j}nt|j �}|jdkr'd|� d�}nd|� d�}| �dt|�t|j �� d S )Nr� �teardownzfailed on teardown with "�"zfailed on setup with "r� )r� r� rq r �whenru r$ )rH rv r� �reason�msgr r r � append_error� s �
z_NodeReporter.append_errorc C s� t |d�r&|j}|�d�r|dd � }t|�}tjdd|d�}| �|� d S t|jt �s.J �|j\}}}|�d�r?|dd � }|� d |� d
|� �}tjdd |d�}t|�|_
| �|� | � |� d S ) Nr� zreason: � r� z pytest.xfail)r* rq z Skipped: � �:z: z pytest.skip) rd r� �
startswithr$ rV rW rL r' r� �tuplert r� )rH rv � xfailreasonr� �filename�lineno�
skipreason�detailsr r r �append_skipped� s
z _NodeReporter.append_skippedc s$ | � � � | j�� � fdd�| _ d S )Nc s � S r% r r �rr r r �<lambda> s z(_NodeReporter.finalize.<locals>.<lambda>)rp �__dict__�clear�rH r r� r �finalize s
z_NodeReporter.finalizer% �r N) �__name__�
__module__� __qualname__r r r rI rV rW rL �objectrO rP r rY rl rp ru r� r� r� r� r� r� r� r� r� r� r r r r r> Y s&
r> �request� fixture_namec C sV ddl m} | jj�td�}|dur'|jdvr)| j�|dj ||jd��� dS dS dS )z[Emit a PytestWarning about the given fixture being incompatible with newer xunit revisions.r )�
PytestWarningN)r< �legacyzV{fixture_name} is incompatible with junit_family '{family}' (use 'legacy' or 'xunit1'))r� rD )
�_pytest.warning_typesr� �config�stash�get�xml_keyrD rJ �warn�format)r� r� r� r@ r r r �!_warn_incompatibility_with_xunit2 s ����r� c s( t � d� dtdtddf� fdd� }|S )an Add extra properties to the calling test.
User properties become part of the test report and are available to the
configured reporters, like JUnit XML.
The fixture is callable with ``name, value``. The value is automatically
XML-encoded.
Example::
def test_function(record_property):
record_property("example_key", 1)
�record_propertyr6 rM r Nc s � j j�| |f� d S r% )rJ �user_propertiesrL rU �r� r r �append_property* rQ z(record_property.<locals>.append_property)r� r r� )r� r� r r� r r� s
r� c C sp ddl m} | j�|d�� t| d� dtdtddfd d
�}|}| jj� t
d�}|dur6|� | jj �}|j
}|S ) z�Add extra xml attributes to the tag for the calling test.
The fixture is callable with ``name, value``. The value is
automatically XML-encoded.
r )� PytestExperimentalApiWarningz/record_xml_attribute is an experimental feature�record_xml_attributer6 rM r Nc S s d S r% r rU r r r �
add_attr_noop@ s z+record_xml_attribute.<locals>.add_attr_noop)r� r� rJ r� r� r r� r� r� r� r� �
node_reporterr? rP )r� r� r� � attr_funcr@ r� r r r r� 0 s �
r� �param�vc C s. d}t |t�sd}t|j| t|�jd���dS )zcUsed by record_testsuite_property to check that the given parameter name is of the proper
type.Tz5{param} parameter needs to be a string, but {g} given)r� �gN)r' r r) r� r* r� )r� r� �__tracebackhide__r� r r r �_check_record_param_typeM s
�r� �session)�scopec C s< d}dt dtddfdd�}| jj�td�}|dur |j}|S )a+ Record a new ``<property>`` tag as child of the root ``<testsuite>``.
This is suitable to writing global information regarding the entire test
suite, and is compatible with ``xunit2`` JUnit family.
This is a ``session``-scoped fixture which is called with ``(name, value)``. Example:
.. code-block:: python
def test_foo(record_testsuite_property):
record_testsuite_property("ARCH", "PPC")
record_testsuite_property("STORAGE_TYPE", "CEPH")
:param name:
The property name.
:param value:
The property value. Will be converted to a string.
.. warning::
Currently this fixture **does not work** with the
`pytest-xdist <https://github.com/pytest-dev/pytest-xdist>`__ plugin. See
:issue:`7767` for details.
Tr6 rM r Nc S s d}t d| � dS )zENo-op function in case --junitxml was not passed in the command-line.Tr6 N)r� )r6 rM r� r r r � record_funcs s z.record_testsuite_property.<locals>.record_func)r r� r� r� r� r� �add_global_property)r� r� r� r@ r r r �record_testsuite_propertyV s r� �parserc
C s� | � d�}|jdddddtjtdd�d dd � |jd
d dd d d
d� | jdddd� | jdddd� | jddddd� | jdd d d� | jd d d d� d S )!Nzterminal reportingz
--junitxmlz --junit-xml�store�xmlpath�path)�optnamez0Create junit-xml style report file at given path)�action�dest�metavarr* �default�helpz
--junitprefixz--junit-prefixr z0Prepend prefix to classnames in junit-xml output)r� r� r� r� �junit_suite_namez Test suite name for JUnit report�pytest)r� �
junit_loggingz\Write captured log messages to JUnit report: one of no|log|system-out|system-err|out-err|allrw �junit_log_passing_testsz;Capture log information for passing tests to JUnit report: �boolT)r* r� �junit_duration_reportz*Duration time to report: one of total|call�total� junit_familyz0Emit XML for schema: one of legacy|xunit1|xunit2r= )�getgroup� addoption� functools�partialr �addini)r� r r r r �pytest_addoption~ sV
�
�����
�r� r� c
C st | j j}|r6t| d�s8| �d�}t|| j j| �d�| �d�| �d�|| �d��| jt< | j� | jt � d S d S d S )N� workerinputr� r� r� r� r� )
�optionr� rd �getinir � junitprefixr� r� �
pluginmanager�register)r� r� r� r r r �pytest_configure� s
� �r� c C s. | j �td �}|r| j t= | j�|� d S d S r% )r� r� r� r� �
unregister)r� r@ r r r �pytest_unconfigure� s
�r� �addressc C s^ | � d�\}}}|�d�}|d �tjd�|d< t�dd|d �|d< |d || 7 < |S )N�[z::r r] z\.py$rx r\ )� partition�split�replacer
�SEPr! r" )r� r� �possible_open_bracket�paramsrg r r r r_ � s
r_ c
@ s& e Zd Z d/dee deded ed
ed d f d
d�Zded d fdd�Zde eef d e
fdd�Z ded d fdd�Z ded e
fdd�Z
ded d fdd�Zded d fdd �Zded d fd d �Zd ed d fd d!�Zd0d"d#�Zd0d$d%�Zd&ed d fd'd(�Zd)ed*ed d fd+d,�Zd eej fd-d.�Zd S )1r r� rw r� r< Tr` �
suite_namer� �report_durationr~ r Nc C s� t j�t j�|��}t j�t j�|��| _|| _|| _|| _ || _
|| _ || _ t
�g d�d�| _i | _g | _g | _g | _d| _| j dkrId| _ d S d S )N)r� r r� r� r r� r< )�osr� �
expanduser�
expandvars�normpath�abspath�logfiler` r r� r~ r rD �dict�fromkeys�stats�node_reporters�node_reporters_ordered�global_properties� open_reports�cnt_double_fail_tests)rH r r` r r� r rD r~ r r r rI � s(
��
�zLogXML.__init__rv c C s` t |d|�}t |dd �}| j�||f�}|jD ] \}}|�|t|�� q|d ur.|�� d S d S �Nr? rJ )r� r �popr� rO r r� )rH rv r? �
workernode�reporter�propname� propvaluer r r r� � s �zLogXML.finalizec C sX t |d|�}t |dd �}||f}|| jv r| j| S t|| �}|| j|< | j�|� |S r )r� r r> r rL )rH rv r? r rk r r r r r� s
zLogXML.node_reporterrk c C s$ || j v r| j | d7 < d S d S )Nr^ )r )rH rk r r r rC s
�zLogXML.add_statsc C s | � |�}|�|� |S r% )r� rl �rH rv r r r r �
_opentestcase s
zLogXML._opentestcasec sr d}� j r� jdkr| �� �}|�� � nb� jrj� jdkrFt� dd��t� dd��t� ��fdd�| jD �d�}|rF| �|� | j d7 _ | �� �}� jdkrd|�
� � | j� � � | j sc|�
� � n|�� � n
� jrw| �� �}|�� � | �� � � jdkr�| �� �}|�
� � | �� � t� dd��t� dd��t� ��fd d�| jD �d�}|r�| j�|� dS dS dS )
a Handle a setup/call/teardown report, generating the appropriate
XML tags as necessary.
Note: due to plugins like xdist, this hook may be called in interlaced
order with reports from other nodes. For example:
Usual call order:
-> setup node1
-> call node1
-> teardown node1
-> setup node2
-> call node2
-> teardown node2
Possible call order in xdist:
-> setup node1
-> call node1
-> setup node2
-> call node2
-> teardown node2
-> teardown node1
N�callr� � worker_id�
item_indexc 3 �@ � | ]}|j � j kr t|d d��kr t|dd��kr|V qdS �r Nr �r? r� ��.0�rep�rv � report_ii�
report_widr r � <genexpr>= � � ��z2LogXML.pytest_runtest_logreport.<locals>.<genexpr>r^ c 3 r r r! r" r% r r r( b r) )r r� r r� �failedr� �nextr r� r r� rL r~ r� r� r� r� �update_testcase_duration�remove)rH rv � close_reportr r r% r �pytest_runtest_logreport s^
�
� �
�
� � �z LogXML.pytest_runtest_logreportc C s> | j dks |j| j kr | �|�}| jt|dd�7 _dS dS )z�Accumulate total duration for nodeid from given report and update
the Junit.testcase with the new total if already created.r� rE rA N)r r� r� rE r� r r r r r, p s
�z LogXML.update_testcase_durationc C s6 |j s| �|�}|jr|�|� d S |�|� d S d S r% )r r r* r� r� r r r r �pytest_collectreportw s
�zLogXML.pytest_collectreport�excreprc C s0 | � d�}|jjddd� |�ddt|�� d S )N�internalr� )r5 r6 r� zinternal error)r� rG r+ ru r )rH r1 r r r r �pytest_internalerror s
zLogXML.pytest_internalerrorc C s t �� | _d S r% )r rn �suite_start_timer� r r r �pytest_sessionstart� r� zLogXML.pytest_sessionstartc
C sT t j�t j�| j��}t j|dd� t| jddd���}t�� }|| j }| j
d | j
d | j
d | j
d | j }|� d
� t
jd | jt| j
d �t| j
d �t| j
d �t|�d | t�| j ��� t�� d
� }| �� }|d urw|�|� | jD ] }|�|�� � qzt
�d�} | �|� |� t
j| dd�� W d � d S 1 s�w Y d S )NT)�exist_ok�wzutf-8)�encodingr r� r� r� z&<?xml version="1.0" encoding="utf-8"?>� testsuiterm )r6 �errors�failuresr� �testsrn � timestamp�hostname�
testsuites�unicode)r r� �dirnamer
r �makedirs�openr rn r4 r r �writerV rW r r r �
fromtimestamp� isoformat�platformrJ �_get_global_properties_noderL r rp �tostring)
rH rA r �suite_stop_time�suite_time_delta�numtests�
suite_noder r� r? r r r �pytest_sessionfinish� sH
�����
�
"�zLogXML.pytest_sessionfinish�terminalreporterc C s |� dd| j� �� d S )Nr� zgenerated xml file: )� write_sepr )rH rO r r r �pytest_terminal_summary� r� z LogXML.pytest_terminal_summaryr6 rM c C s& d}t d|� | j�|t|�f� d S )NTr6 )r� r rL r$ )rH r6 rM r� r r r r� � s
zLogXML.add_global_propertyc C rR rS )r rV rW rL rX r r r rH � rZ z"LogXML._get_global_properties_node)r� rw r� r< Tr� ) r� r� r� r r r� rI r r� r r> r� rC r r/ r, r0 r r3 r5 rN r rQ r� r� rV rW rH r r r r r � s@ ������
�# U
&r� )9�__doc__r� r rG r! �xml.etree.ElementTree�etree� ElementTreerV r �typingr r r r r r r r� �_pytestr
r �_pytest._code.coder r
�_pytest.configr r �_pytest.config.argparsingr �_pytest.fixturesr �_pytest.reportsr �
_pytest.stashr �_pytest.terminalr r� r� r r$ r3 rf �copyr> r� �fixturer� r� r� r� r� r� r� r_ r r r r r �<module> sp
0��
�
'/