Your IP : 216.73.216.52
o
0�i=� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZ d dlmZ d dlmZ G dd� de
j�Zd ZG d
d � d e
j�ZG d d
� d
e
j�ZG dd� de
j�ZG dd� de
j�ZG dd� de
j�ZG dd� de
j�ZG dd� de
j�Z G dd� de
j�Z G dd� de
j�Z G d d � d e
j�Z G d d � d e
j�Z G d d!� d!e
j�Z!G d"d#� d#e
j�Z"G d$d%� d%ej#�Z$e
�%e
j&d&k d'�G d(d)� d)e$e
j��Z'e(d*kr�e
�)� dS dS )+� )�BytesION)�expat)�errors)�
import_helper)�sortdictc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�SetAttributeTestc C s t jdd�| _d S �N�!�Znamespace_separator)r � ParserCreate�parser��self� r �8/opt/alt/python310/lib64/python3.10/test/test_pyexpat.py�setUp � zSetAttributeTest.setUpc C �: | � | jjd� dD ]}|| j_| � | jjt|�� q
d S �NF)r � � r )�assertIsr � buffer_text�bool�r �xr r r �test_buffer_text �
�z!SetAttributeTest.test_buffer_textc C r r )r r Znamespace_prefixesr r r r r �test_namespace_prefixes! r z(SetAttributeTest.test_namespace_prefixesc C r r )r r Zordered_attributesr r r r r �test_ordered_attributes' r z(SetAttributeTest.test_ordered_attributesc C r r )r r Zspecified_attributesr r r r r �test_specified_attributes- r z*SetAttributeTest.test_specified_attributesc C s� | � t�� d| j_W d � n1 sw Y | � t�� | jj W d � n1 s-w Y | � tt| jtd�d� | � t| jjtd�d� | � tt| jtd�� d S )Nr � r ) � assertRaises�AttributeErrorr Zreturns_unicode� TypeError�setattr�range� __setattr__�getattrr
r r r �test_invalid_attributes3 s
�
�z(SetAttributeTest.test_invalid_attributesN) �__name__�
__module__� __qualname__r r r r r r) r r r r r s r s� <?xml version="1.0" encoding="iso-8859-1" standalone="no"?>
<?xml-stylesheet href="stylesheet.css"?>
<!-- comment data -->
<!DOCTYPE quotations SYSTEM "quotations.dtd" [
<!ELEMENT root ANY>
<!ATTLIST root attr1 CDATA #REQUIRED attr2 CDATA #IMPLIED>
<!NOTATION notation SYSTEM "notation.jpeg">
<!ENTITY acirc "â">
<!ENTITY external_entity SYSTEM "entity.file">
<!ENTITY unparsed_entity SYSTEM "entity.file" NDATA notation>
%unparsed_entity;
]>
<root attr1="value1" attr2="value2ὀ">
<myns:subelement xmlns:myns="http://www.python.org/namespace">
Contents of subelements
</myns:subelement>
<sub2><![CDATA[contents of CDATA section]]></sub2>
&external_entity;
&skipped_entity;
�
</root>
c @ sR e Zd ZG dd� d�Zg d�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� ParseTestc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/S )0zParseTest.Outputterc C s
g | _ d S �N)�outr
r r r �__init__\ �
z ParseTest.Outputter.__init__c C s$ | j �dt|� d t|� � d S )NzStart element: � )r/ �append�reprr �r �name�attrsr r r �StartElementHandler_ s
�z'ParseTest.Outputter.StartElementHandlerc C � | j �dt|� � d S )Nz
End element: �r/ r3 r4 �r r6 r r r �EndElementHandlerc � z%ParseTest.Outputter.EndElementHandlerc C s( |� � }|r| j�dt|� � d S d S )NzCharacter data: )�stripr/ r3 r4 �r �datar r r �CharacterDataHandlerf s �z(ParseTest.Outputter.CharacterDataHandlerc C �$ | j �dt|� d t|� � d S )NzPI: r2 r: )r �targetr@ r r r � ProcessingInstructionHandlerk � $z0ParseTest.Outputter.ProcessingInstructionHandlerc C rB )Nz NS decl: r2 r: )r �prefixZurir r r �StartNamespaceDeclHandlern rE z-ParseTest.Outputter.StartNamespaceDeclHandlerc C r9 )NzEnd of NS decl: r: )r rF r r r �EndNamespaceDeclHandlerq r= z+ParseTest.Outputter.EndNamespaceDeclHandlerc C � | j �d� d S )N�Start of CDATA section�r/ r3 r
r r r �StartCdataSectionHandlert � z,ParseTest.Outputter.StartCdataSectionHandlerc C rI )N�End of CDATA sectionrK r
r r r �EndCdataSectionHandlerw rM z*ParseTest.Outputter.EndCdataSectionHandlerc C r9 )Nz Comment: r: �r �textr r r �CommentHandlerz r= z"ParseTest.Outputter.CommentHandlerc G s" |\}}}}| j �d|f � d S )NzNotation declared: %srK )r �argsr6 �baseZsysidZpubidr r r �NotationDeclHandler} s z'ParseTest.Outputter.NotationDeclHandlerc G s$ |\}}}}}| j �d|f � d S )NzUnparsed entity decl: %srK )r rS Z
entityNamerT ZsystemIdZpublicIdZ notationNamer r r �UnparsedEntityDeclHandler� s z-ParseTest.Outputter.UnparsedEntityDeclHandlerc C � | j �d� dS )N�Not standaloner rK r
r r r �NotStandaloneHandler� � z(ParseTest.Outputter.NotStandaloneHandlerc G s* |\}}}}| j �d|dd � f � dS )NzExternal entity ref: %sr rK )r rS �contextrT �sysId�pubIdr r r �ExternalEntityRefHandler� s z,ParseTest.Outputter.ExternalEntityRefHandlerc G � | j �d|f� dS )N�
Start doctyper rK �r rS r r r �StartDoctypeDeclHandler� � z+ParseTest.Outputter.StartDoctypeDeclHandlerc C rW )N� End doctyper rK r
r r r �EndDoctypeDeclHandler� rZ z)ParseTest.Outputter.EndDoctypeDeclHandlerc G r_ )N�Entity declarationr rK ra r r r �EntityDeclHandler� rc z%ParseTest.Outputter.EntityDeclHandlerc G r_ )N�XML declarationr rK ra r r r �XmlDeclHandler� rc z"ParseTest.Outputter.XmlDeclHandlerc G r_ )N�Element declarationr rK ra r r r �ElementDeclHandler� rc z&ParseTest.Outputter.ElementDeclHandlerc G r_ )N�Attribute list declarationr rK ra r r r �AttlistDeclHandler� rc z&ParseTest.Outputter.AttlistDeclHandlerc G r_ )N�Skipped entityr rK ra r r r �SkippedEntityHandler� rc z(ParseTest.Outputter.SkippedEntityHandlerc C � d S r. r �r ZuserDatar r r �DefaultHandler� � z"ParseTest.Outputter.DefaultHandlerc C rp r. r rq r r r �DefaultHandlerExpand� rs z(ParseTest.Outputter.DefaultHandlerExpandN)r* r+ r, r0 r8 r< rA rD rG rH rL rO rR rU rV rY r^ rb re rg ri rk rm ro rr rt r r r r � Outputter[ s0 ru )r8 r< rA rD rV rU rG rH rR rL rO rr rt rY r^ rb re rg ri rk rm ro c C s" | j D ] }t||t||�� qdS )z{
Set each of the callbacks defined on handler and named in
self.handler_names on the given parser.
N)�
handler_namesr% r( )r r �handlerr6 r r r �_hookup_callbacks� s
�zParseTest._hookup_callbacksc C s, g d�}t ||�D ]
\}}| �||� q d S )N) )rh )z1.0�
iso-8859-1r z,PI: 'xml-stylesheet' 'href="stylesheet.css"'zComment: ' comment data 'rX )r` )Z
quotationszquotations.dtdNr )rj )�root)r r Nr )rl )rz Zattr1�CDATANr )rl )rz Zattr2r{ Nr z<Notation declared: ('notation', None, 'notation.jpeg', None))rf )Zacircr � âNNNN)rf )Zexternal_entityr NNz entity.fileNNzPUnparsed entity decl: ('unparsed_entity', None, 'entity.file', None, 'notation')rX rd u? Start element: 'root' {'attr1': 'value1', 'attr2': 'value2ὀ'}z1NS decl: 'myns' 'http://www.python.org/namespace'z>Start element: 'http://www.python.org/namespace!subelement' {}z)Character data: 'Contents of subelements'z9End element: 'http://www.python.org/namespace!subelement'zEnd of NS decl: 'myns'zStart element: 'sub2' {}rJ z+Character data: 'contents of CDATA section'rN zEnd element: 'sub2'z0External entity ref: (None, 'entity.file', None))rn )Zskipped_entityr u Character data: 'µ'zEnd element: 'root')�zip� assertEqual)r �
operationsZexpected_operationsZ operationZexpected_operationr r r �_verify_parse_output� s "�z ParseTest._verify_parse_outputc C sP | � � }tjdd�}| �||� |�td� |j}| �|� | �t t
|d� d S )Nr r
Tu �) ru r r rx �Parser@ r/ r� r" r# r( �r r/ r r r r r �test_parse_bytes� s
zParseTest.test_parse_bytesc C sF | � � }tjdd�}| �||� |�t�d�d� |j}| �|� d S )Nr r
ry T) ru r r rx r� r@ �decoder/ r� r� r r r �test_parse_str� s zParseTest.test_parse_strc C sF | � � }tjdd�}| �||� tt�}|�|� |j}| �|� d S r ) ru r r rx r r@ � ParseFiler/ r� )r r/ r �filer r r r �test_parse_file s
zParseTest.test_parse_filec C sn t �� }tt�}|�|� | �t j��
}|�|� W d � n1 s#w Y | �t �|j j
�t j j � d S r. )
r r r r@ r� r" �errorr~ Z ErrorStringZ exception�coder ZXML_ERROR_FINISHED)r r r� �cmr r r �test_parse_again s
��zParseTest.test_parse_againN) r* r+ r, ru rv rx r� r� r� r� r� r r r r r- Z s U &
r- c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�NamespaceSeparatorTestc C s$ t �� t jd d� t jdd� d S )Nr
r2 �r r r
r r r �
test_legal s z!NamespaceSeparatorTest.test_legalc
C s� z t jdd� | �� W n ty& } z| �t|�d� W Y d }~nd }~ww z
t jdd� | �� W d S tyO } z| �t|�d� W Y d }~d S d }~ww )N�* r
zJParserCreate() argument 'namespace_separator' must be str or None, not intztoo longzCnamespace_separator must be at most one character, omitted, or None)r r �failr$ r~ �str�
ValueError)r �er r r � test_illegal s$
���
���z#NamespaceSeparatorTest.test_illegalc C s t jdd� d S )N� r
r� r
r r r �test_zero_length/ s z'NamespaceSeparatorTest.test_zero_lengthN)r* r+ r, r� r� r� r r r r r� s r� c @ � e Zd Zdd� Zdd� ZdS )�
InterningTestc sd t �� }g � � fdd�}||_||_|�dd� � d }| �t� �d� � D ] }| �||u � q&d S )Nc � � � | � d S r. �r3 )r6 rS ��Lr r � collector@ � z%InterningTest.test.<locals>.collectors <e> <e/> <e></e> </e>Tr � )r r r8 r< r� r~ �len�
assertTrue)r �pr� �tag�entryr r� r �test< s �zInterningTest.testc C sN G dd� d�}t jdd�}d|_||�}|j|_|�td� | �|jd� d S )Nc @ r� )z7InterningTest.test_issue9402.<locals>.ExternalOutputterc S s || _ d | _d S r. )r �
parser_result�r r r r r r0 N s
z@InterningTest.test_issue9402.<locals>.ExternalOutputter.__init__c S s | j �d�}|�dd�| _dS )Nr� � Tr )r �ExternalEntityParserCreater� r� )r r[ rT r\ r] Zexternal_parserr r r r^ R s zPInterningTest.test_issue9402.<locals>.ExternalOutputter.ExternalEntityRefHandlerN)r* r+ r, r0 r^ r r r r �ExternalOutputterM s r� r r
r T)r r r r^ r� r@ r~ r� )r r� r r/ r r r �test_issue9402K s
z InterningTest.test_issue9402N)r* r+ r, r� r� r r r r r� ; s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zg fd
d�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�BufferTextTestc C s& g | _ t�� | _d| j_| j| j_d S �Nr )�stuffr r r r rA r
r r r r ` s
zBufferTextTest.setUpc
C s& | � | j|d|| jtt|�f � d S )Nz %s
stuff = %r
expected = %r)r~ r� �mapr� )r �expectedZlabelr r r �checkf s
��zBufferTextTest.checkc C s | j �|� d S r. �r� r3 rP r r r rA k rM z#BufferTextTest.CharacterDataHandlerc C sF | j �d| � |�d�}|dkrd| j_d S |dkr!d| j_d S d S )Nz<%s>z buffer-textZyesr Znor )r� r3 �getr r )r r6 r7 Zbtr r r r8 n s
�z"BufferTextTest.StartElementHandlerc C � | j �d| � d S )Nz</%s>r� r; r r r r< v � z BufferTextTest.EndElementHandlerc C r� )Nz <!--%s-->r� r? r r r rR y r� z BufferTextTest.CommentHandlerc C s" |D ] }t | j|t| |�� qd S r. )r% r r( )r �handlersr6 r r r � setHandlers| s �zBufferTextTest.setHandlersc C s t �� }| �|j� d S r. )r r � assertFalser r� r r r �test_default_to_disabled� s z'BufferTextTest.test_default_to_disabledc C s2 | � | jj� | j�dd� | �| jdgd� d S )N� <a>1<b/>2<c/>3</a>TZ123�$buffered text not properly collapsed)r� r r r� r~ r� r
r r r �test_buffering_enabled� s
�z%BufferTextTest.test_buffering_enabledc C �2 | � dg� | j�dd� | �| jg d�d� d S )Nr8 s9 <a>1<b buffer-text='no'/>2
3<c buffer-text='yes'/>4
5</a>T)�<a>�1�<b>�2�
�3�<c>z4
5z*buffering control not reacting as expected�r� r r� r~ r� r
r r r �test1� s �zBufferTextTest.test1c C s$ | j �dd� | �| jdgd� d S )Ns& <a>1<b/><2><c/> 
 3</a>Tz1<2>
3r� )r r� r~ r� r
r r r �test2� s �zBufferTextTest.test2c C r� )Nr8 r� T)r� r� r� r� r� r� � buffered text not properly splitr� r
r r r �test3� s
�zBufferTextTest.test3c C s: | � ddg� d | j_| j�dd� | �| jg d�� d S )Nr8 r< r� T)r� r� �</b>r� �</c>�</a>)r� r rA r� r~ r� r
r r r �test4� s �zBufferTextTest.test4c C s2 | � ddg� | j�dd� | �| jg d�� d S )Nr8 r< s <a>1<b></b>2<c/>3</a>T) r� r� r� r� r� r� r� r� r� r� r
r r r �test5� s
�zBufferTextTest.test5c C �4 | � g d�� | j�dd� | �| jg d�d� d S )N�rR r< r8 s <a>1<b/>2<c></c>345</a> T) r� r� r� r� r� r� r� Z345r� r� r� r
r r r �test6� s �zBufferTextTest.test6c C r� )Nr� s, <a>1<b/>2<c></c>3<!--abc-->4<!--def-->5</a> T)
r� r� r� r� r� r� r� r� z
<!--abc-->�4z
<!--def-->�5r� r� r� r
r r r �test7� s �zBufferTextTest.test7N)r* r+ r, r r� rA r8 r< rR r� r� r� r� r� r� r� r� r� r� r r r r r� _ s" r� c @ r� )�HandlerExceptionTestc C s t |��r. )� RuntimeErrorr5 r r r r8 � s z(HandlerExceptionTest.StartElementHandlerc C s, | � tj�|d �|� | � |d |� d S )Nr r )r~ �os�path�basename)r r� �filename�funcnamer r r �check_traceback_entry� s z*HandlerExceptionTest.check_traceback_entryc C s t �� }| j|_z
|�dd� | �� W d S ty� } zh| �|jd ddd|jd � t� |j
�}| �t |�d� | � |d dd � | � |d
d d � | � |d
dd� t
�� rwtjdkrft�� dks~| �d|d
d � W Y d }~d S W Y d }~d S W Y d }~d S d }~ww )Ns <a><b><c/></b></a>Tr �az*Expected RuntimeError for element 'a', butz found %r� ztest_pyexpat.py�test_exceptionr z pyexpat.cZ StartElementr r8 Zwin32ZARMz call_with_frame("StartElement")r r r8 r� r� r� r~ rS � traceback�
extract_tb�
__traceback__r� r� � sysconfigZis_python_build�sys�platform�machineZassertIn)r r r� �entriesr r r r� � s6 ��
�
�
� " ���z#HandlerExceptionTest.test_exceptionN)r* r+ r, r8 r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� PositionTestc C � | � d� d S )N�s�� check_posr5 r r r r8 � r� z PositionTest.StartElementHandlerc C r� )Nr� r� r; r r r r< � r� z PositionTest.EndElementHandlerc C sd || j j| j j| j jf}| �| jt| j�k d� | j| j }| �||d||f � | jd7 _d S )Nztoo many parser eventsz%Expected position %s, got position %sr ) r ZCurrentByteIndexZCurrentLineNumberZCurrentColumnNumberr� �uptor� �
expected_listr~ )r Zevent�posr� r r r r� � s ��
�zPositionTest.check_posc C sD t �� | _| j| j_| j| j_d| _g d�| _d}| j�|d� d S )Nr ))r� r r r )r� � r r )r� � r� r )r� r! r� r� )r� � � r )r� � r� r s <a>
<b>
<c/>
</b>
</a>T)r r r r8 r< r� r� r� )r �xmlr r r r� � s
zPositionTest.testN)r* r+ r, r8 r< r� r� r r r r r� � s
r� c @ s e Zd Zdd� ZdS )�
sf1296433Testc sL dd }G dd� dt �� � fdd�}t�� }||_| �t |j|�d�� d S )Nz1<?xml version='1.0' encoding='iso8859'?><s>%s</s>A aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaac @ s e Zd ZdS )zAsf1296433Test.test_parse_only_xml_data.<locals>.SpecificExceptionN)r* r+ r, r r r r �SpecificException s r� c s � �r. r �rQ �r� r r rw rs z7sf1296433Test.test_parse_only_xml_data.<locals>.handler�iso8859)� Exceptionr r rA r" r� �encode)r r� rw r r r� r �test_parse_only_xml_data� s z&sf1296433Test.test_parse_only_xml_dataN)r* r+ r, r� r r r r r� � s r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�ChardataBufferTestz.
test setting of chardata buffer size
c C � | � | �d�d� d S )Ni r �r~ �small_buffer_testr
r r r �test_1025_bytes � z"ChardataBufferTest.test_1025_bytesc C r )Ni� r r r
r r r �test_1000_bytes r z"ChardataBufferTest.test_1000_bytesc C s� t �� }d|_| �t�� d|_W d � n1 sw Y | �t�� d|_W d � n1 s2w Y | �ttf�� tjd |_W d � n1 sOw Y | �t �� d|_W d � d S 1 shw Y d S )Nr ���r g �@)
r r r r" r� � buffer_size�
OverflowErrorr� �maxsizer$ r� r r r �test_wrong_size s � � � "�z"ChardataBufferTest.test_wrong_sizec C st d}d}t �� }| j|_d|_d|_d| _|�|� | �| jd� |j|_| �| jd� |�|� | �| jd� d S )Ns+ <?xml version='1.0' encoding='iso8859'?><s>aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaas aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa</s>i r r r ) r r �counting_handlerrA r r �nr� r~ �r �xml1�xml2r r r r �test_unchanged_size( s
z&ChardataBufferTest.test_unchanged_sizec C s� d}d}d}t �� }| j|_d|_d|_| �|jd� d| _|�|d� | �|jd� | �| jd� d|_| � |j� | �|jd� t
d�D ]}|�|d� qF| �| jd � d|_| � |j� | �|jd� |�|d
� | �| jd � d S ) Ns+ <?xml version='1.0' encoding='iso8859'?><a>aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaas bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbs cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc</a>r � r F�
r� T� ) r r r rA r r r~ r
r� r� r& r� )r r r Zxml3r �ir r r �test_disabling_buffer? s. z(ChardataBufferTest.test_disabling_bufferc C s | j d7 _ d S r� )r
rP r r r r ] r z#ChardataBufferTest.counting_handlerc C sB dd| d }t �� }| j|_d|_d|_d| _|�|� | jS )Ns+ <?xml version='1.0' encoding='iso8859'?><s>� as </s>r r r )r r r rA r r r
r� )r Z
buffer_lenr� r r r r r ` s
z$ChardataBufferTest.small_buffer_testc C s~ d}d}t �� }| j|_d|_d|_| �|jd� d| _|�|d� | jd9 _| �|jd� |�|d � | �| jd� d S )
Ns. <?xml version='1.0' encoding='iso8859'?><a><s>aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa� aaa</s><s>aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa</s></a>r r r Fr � T� r r r rA r r r~ r
r� r r r r �test_change_size_1k s z%ChardataBufferTest.test_change_size_1c C s| d}d}t �� }| j|_d|_d|_| �|jd� d| _|�|d� |jd |_| �|jd� |�|d � | �| jd
� d S ) Ns. <?xml version='1.0' encoding='iso8859'?><a>a<s>aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar r r r Fr r Tr� r r r r r �test_change_size_2{ s z%ChardataBufferTest.test_change_size_2N)
r* r+ r, �__doc__r r r r r r r r r r r r r r s r c @ r� )�MalformedInputTestc
C s` d}t �� }z
|�|d� | �� W d S t jy/ } z| �t|�d� W Y d }~d S d }~ww )Ns
Tz unclosed token: line 2, column 0)r r r� r� �
ExpatErrorr~ r� �r r� r r� r r r r� � s ��zMalformedInputTest.test1c C sP d}t �� }d}| �t j|�� |�|d� W d � d S 1 s!w Y d S )Ns <?xml version
='1.0'?>
z3XML declaration not well-formed: line 1, column \d+T)r r �assertRaisesRegexr r� )r r� r Z err_patternr r r r� � s "�zMalformedInputTest.test2N)r* r+ r, r� r� r r r r r � s r c @ r� )�ErrorMessageTestc C s | � tjtjtjtj � d S r. )r~ r ZXML_ERROR_SYNTAXZmessages�codesr
r r r �
test_codes� s �zErrorMessageTest.test_codesc
C sf d}t �� }z
|�|d� | �� W d S t jy2 } z| �|jtjtj � W Y d }~d S d }~ww )N� <T)
r r r� r� r r~ r� r r# ZXML_ERROR_UNCLOSED_TOKENr r r r �test_expaterror� s
���z ErrorMessageTest.test_expaterrorN)r* r+ r, r$ r& r r r r r" � s r" c @ s e Zd ZdZdd� Zdd� ZdS )�ForeignDTDTestszE
Tests for the UseForeignDTD method of expat parser objects.
c s� g � � fdd�}t �� }|�d� |�t j� ||_|�d� | �� dg� g � dd�< t �� }|�� |�t j� ||_|�d� | �� dg� dS )z�
If UseForeignDTD is passed True and a document without an external
entity reference is parsed, ExternalEntityRefHandler is first called
with None for the public and system ids.
c � � � ||f� dS r� r� �r[ rT Z system_idZ public_id�Zhandler_call_argsr r �resolve_entity� � z<ForeignDTDTests.test_use_foreign_dtd.<locals>.resolve_entityTs <?xml version='1.0'?><element/>)NNN�r r Z
UseForeignDTDZSetParamEntityParsingZ XML_PARAM_ENTITY_PARSING_ALWAYSr^ r� r~ �r r+ r r r* r �test_use_foreign_dtd� s
z$ForeignDTDTests.test_use_foreign_dtdc sP g � � fdd�}t �� }|�d� |�t j� ||_|�d� | �� dg� dS )z�
If UseForeignDTD is passed True and a document with an external
entity reference is parsed, ExternalEntityRefHandler is called with
the public and system ids from the document.
c r( r� r� r) r* r r r+ � r, zCForeignDTDTests.test_ignore_use_foreign_dtd.<locals>.resolve_entityTs@ <?xml version='1.0'?><!DOCTYPE foo PUBLIC 'bar' 'baz'><element/>)ZbarZbazNr- r. r r* r �test_ignore_use_foreign_dtd� s
�z+ForeignDTDTests.test_ignore_use_foreign_dtdN)r* r+ r, r r/ r0 r r r r r' � s r' c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ParentParserLifetimeTestz�
Subparsers make use of their parent XML_Parser inside of Expat.
As a result, parent parsers need to outlive subparsers.
See https://github.com/python/cpython/issues/139400.
c C s t �� }|�d �}~d S r. �r r r� )r r � subparserr r r �2test_parent_parser_outlives_its_subparsers__single� s
zKParentParserLifetimeTest.test_parent_parser_outlives_its_subparsers__singlec C s" t �� }|�d �}|�d �}~d S r. r2 )r r Z
subparser_oneZ
subparser_twor r r �4test_parent_parser_outlives_its_subparsers__multiple� s
zMParentParserLifetimeTest.test_parent_parser_outlives_its_subparsers__multiplec C s$ t �� }|�d �}|�d �}~~d S r. r2 )r r r3 Z subsubparserr r r �1test_parent_parser_outlives_its_subparsers__chain� s
zJParentParserLifetimeTest.test_parent_parser_outlives_its_subparsers__chainN)r* r+ r, r r4 r5 r6 r r r r r1 � s
r1 c @ r� )�ReparseDeferralTestc C sZ t �� }t jdk}| �|�� |� |�d� | �|�� d� |�d� | �|�� |� d S )N�r r� r FT)r r � version_infor �GetReparseDeferralEnabled�SetReparseDeferralEnabled)r r Zenabledr r r � test_getter_setter_round_trip s
z1ReparseDeferralTest.test_getter_setter_round_tripc s� t jdk r| �dt j� d�� g � � fdd�}t �� }||_| �|�� � dD ]}|�|d� q'| �� g � |�dd � | �� d
g� d S ) Nr8 zExpat z" does not support reparse deferralc r� r. r� �r6 �_�Zstartedr r �
start_element r� zHReparseDeferralTest.test_reparse_deferral_enabled.<locals>.start_element�s <docs />Fr� T�doc) r r9 ZskipTestr r8 r� r: r� r~ �r r@ r �chunkr r? r � test_reparse_deferral_enabled s
z1ReparseDeferralTest.test_reparse_deferral_enabledc sh g � � fdd�}t �� }||_t jdkr|�d� | �|�� � dD ]}|�|d� q"| �� dg� d S )Nc r� r. r� r= r? r r r@ , r� zIReparseDeferralTest.test_reparse_deferral_disabled.<locals>.start_elementr8 FrA rB ) r r r8 r9 r; r� r: r� r~ rC r r? r � test_reparse_deferral_disabled) s
z2ReparseDeferralTest.test_reparse_deferral_disabledN)r* r+ r, r< rE rF r r r r r7 s
r7 c @ s� e Zd ZdZedd�dd��Zdd� Zdd � Zej d
d � �Z
ej d d
� �Z ej dd� �Z ej dd� �Z
ej dd� �Zdd� Zdd� Zdd� Zej dd� �Zej d d � �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&S )'�AttackProtectionTestBasez�
Base class for testing protections against XML payloads with
disproportionate amplification.
The protections being tested should detect and prevent attacks
that leverage disproportionate amplification from small inputs.
�.r� c sP t �d|� d| � d���� }d�� fdd�t| �D ��}t �|d�}|j|d�S ) z�Create a billion laughs attack payload.
Be careful: the number of total items is pow(n, k), thereby
requiring at least pow(ncols, nrows) * sizeof(text) memory!
z] <?xml version="1.0"?>
<!DOCTYPE doc [
<!ENTITY row0 "zc">
<!ELEMENT doc (#PCDATA)>
{body}
]>
<doc>&rowz;</doc>
r� c 3 s2 � | ]}d |d � dd|� d�� � d�V qdS )z <!ENTITY rowr z "z&row�;z">Nr )�.0r ��ncolsr r � <genexpr>V s
� �
�zIAttackProtectionTestBase.exponential_expansion_payload.<locals>.<genexpr>z )�body)�textwrap�dedent�rstrip�joinr& �indent�format)�nrowsrL rQ �templaterN r rK r � exponential_expansion_payloadE s ���
� z6AttackProtectionTestBase.exponential_expansion_payloadc C s( | j ddd�}| �|t�d��� � d S )Nr r� )rU rL a' <?xml version="1.0"?>
<!DOCTYPE doc [
<!ENTITY row0 ".">
<!ELEMENT doc (#PCDATA)>
<!ENTITY row1 "&row0;&row0;&row0;">
<!ENTITY row2 "&row1;&row1;&row1;">
]>
<doc>&row2;</doc>
)rW r~ rO rP rQ )r �payloadr r r �test_payload_generation] s �z0AttackProtectionTestBase.test_payload_generationc O �& d}| j tj||g|�R i |�� dS )z=Check that func(*args, **kwargs) is invalid for a sub-parser.z parser must be a root parserN�r! r r �r �funcrS �kwargs�msgr r r �assert_root_parser_failurek � "z3AttackProtectionTestBase.assert_root_parser_failurec O � dS )a Assert that func(*args, **kwargs) triggers the attack protection.
Note: this method must ensure that the attack protection being tested
is the one that is actually triggered at runtime, e.g., by matching
the exact error message.
Nr )r r] rS r^ r r r �assert_rejectedp � z(AttackProtectionTestBase.assert_rejectedc C rb )z7Set the activation threshold for the tested protection.Nr �r r Z thresholdr r r �set_activation_thresholdy rd z1AttackProtectionTestBase.set_activation_thresholdc C rb )z?Set the maximum amplification factor for the tested protection.Nr �r r Z
max_factorr r r �set_maximum_amplification} rd z2AttackProtectionTestBase.set_maximum_amplificationc C rb )z/Test when the activation threshold is exceeded.Nr r
r r r �0test_set_activation_threshold__threshold_reached� rd zIAttackProtectionTestBase.test_set_activation_threshold__threshold_reachedc C rb )z3Test when the activation threshold is not exceeded.Nr r
r r r �4test_set_activation_threshold__threshold_not_reached� rd zMAttackProtectionTestBase.test_set_activation_threshold__threshold_not_reachedc C sD t �� }t�| j|�}| �t|d� | �t|d� | �t|d� d S )N� �?g �����)r r � functools�partialrf r" r$ r� �r r �setterr r r �5test_set_activation_threshold__invalid_threshold_type� s
zNAttackProtectionTestBase.test_set_activation_threshold__invalid_threshold_typec C s8 t �d�}t�� }t�| j|�}| �t||j d � d S )N� _testcapir )
r �
import_moduler r rm rn rf r" r Z
ULLONG_MAX)r rr r rp r r r �6test_set_activation_threshold__invalid_threshold_range� s
zOAttackProtectionTestBase.test_set_activation_threshold__invalid_threshold_rangec C �0 t �� }|�d �}t�| j|�}| �|d� d S )Ni90 )r r r� rm rn rf r` �r r r3 rp r r r �1test_set_activation_threshold__fail_for_subparser� �
zJAttackProtectionTestBase.test_set_activation_threshold__fail_for_subparserc C rb )z/Test when the amplification factor is exceeded.Nr r
r r r �6test_set_maximum_amplification__amplification_exceeded� rd zOAttackProtectionTestBase.test_set_maximum_amplification__amplification_exceededc C rb )z3Test when the amplification factor is not exceeded.Nr r
r r r �:test_set_maximum_amplification__amplification_not_exceeded� rd zSAttackProtectionTestBase.test_set_maximum_amplification__amplification_not_exceededc C s& t d�}t�� }| �| �||�� d S )N�inf)�floatr r � assertIsNonerh )r r{ r r r r �(test_set_maximum_amplification__infinity� s zAAttackProtectionTestBase.test_set_maximum_amplification__infinityc C s6 t �� }t�| j|�}| �t|d � | �t|d� d S )N�abc)r r rm rn rh r" r$ ro r r r �7test_set_maximum_amplification__invalid_max_factor_type� s zPAttackProtectionTestBase.test_set_maximum_amplification__invalid_max_factor_typec C sL t �� }t�| j|�}t�d�}| �t j||t d�� | �t j||d� d S )Nz!'max_factor' must be at least 1.0�nang�G�z��?)
r r rm rn rh �re�escaper! r r| )r r rp r_ r r r �8test_set_maximum_amplification__invalid_max_factor_range� s
zQAttackProtectionTestBase.test_set_maximum_amplification__invalid_max_factor_rangec C ru )Ng������^@)r r r� rm rn rh r` rv r r r �2test_set_maximum_amplification__fail_for_subparser� rx zKAttackProtectionTestBase.test_set_maximum_amplification__fail_for_subparserN)r* r+ r, r � staticmethodrW rY r` r �abstractmethodrc rf rh ri rj rq rt rw ry rz r~ r� r� r� r r r r rG < s6
rG )r � r zrequires Expat >= 2.7.2c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�MemoryProtectionTestc O rZ )z;Check that func(*args, **kwargs) hits the allocation limit.z#out of memory: line \d+, column \d+Nr[ r\ r r r rc � ra z$MemoryProtectionTest.assert_rejectedc C �
|� |�S r. )Z"SetAllocTrackerActivationThresholdre r r r rf � r1 z-MemoryProtectionTest.set_activation_thresholdc C r� r. )Z#SetAllocTrackerMaximumAmplificationrg r r r rh � r1 z.MemoryProtectionTest.set_maximum_amplificationc C �H t �� }| �|d� | �| �|d�� | jddd�}| �|j|d� d S )Nr� rk r r� �rL rU T�r r rf r} rh rW rc r� �r r rX r r r ri � s
zEMemoryProtectionTest.test_set_activation_threshold__threshold_reachedc C sP t �� }| �|tdd�� | �| �|d�� | jddd�}| �|�|d�� d S )Nr r� rk r� r� T) r r rf �powr} rh rW �assertIsNotNoner� r� r r r rj � s
zIMemoryProtectionTest.test_set_activation_threshold__threshold_not_reachedc C r� )Nr rk r r r� Tr� r� r r r ry � s
zKMemoryProtectionTest.test_set_maximum_amplification__amplification_exceededc C sJ t �� }| �|d� | �| �|d�� | jddd�}| �|�|d�� d S )Nr g ��@r r r� T)r r rf r} rh rW r� r� r� r r r rz � s
zOMemoryProtectionTest.test_set_maximum_amplification__amplification_not_exceededN)
r* r+ r, rc rf rh ri rj ry rz r r r r r� � s
r� �__main__)*�ior r rm r� r� r� r� r� rO Zunittestr� Z xml.parsersr Zxml.parsers.expatr Z test.supportr r ZTestCaser r@ r- r� r� r� r� r� r� r r r" r' r1 r7 �ABCrG ZskipIfr9 r� r* �mainr r r r �<module> sL ( ?#$` z4$6 < �