U
n^p, @ s
d dl mZ z d dlZW n ek
r4 edY nX d dlmZ d dlZd dlZd dlZd dl Z ddl
m Z m Z d dl
mZ d dlmZ d dlZdd lmZ d dlZd d
lmZmZ G d d d e ZG d
d deZG dd deZG dd deZG dd deZ dS ) )absolute_importNz"lxml is required to run the tests.)BytesIO )HelperTestCaseskipIf)xmlfile)LxmlSyntaxError) compare_xml)Elementparsec @ s" e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zejjd d Zejjd d Zd d! Zd"d# Zejjd$d% Zejjd&d' Zejjd(d) Zd*d+ Zd,d- Z d.d/ Z ejjd0d1 Z d2d3 Z d4d5 Z d6d7 Z!d;d9d:Z"dS )<_XmlFileTestCaseBaseNc C s t | _d S Nr _fileself r /private/var/folders/sd/whlwsn6x1_qgglc0mjv25_695qk2gl/T/pip-install-4zq3fp6i/et-xmlfile/et_xmlfile/tests/test_incremental_xmlfile.pysetUp! s z_XmlFileTestCaseBase.setUpc
C s: t | j }|d W 5 Q R X W 5 Q R X | d d S )Ntestz
r r element assertXmlr xfr r r test_element$ s z!_XmlFileTestCaseBase.test_elementc
C sD t | j&}|d |d W 5 Q R X W 5 Q R X | d d S )Nr toasttoastr r r writer r r r r test_element_write_text* s z,_XmlFileTestCaseBase.test_element_write_textc C sp t | jR}|d< |d& |d |d W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X | d d S )Nr r tasteZconTentz2conTentr r r r r test_element_nested0 s 2z(_XmlFileTestCaseBase.test_element_nestedc C s t | jz}|dd |d |d: |d |d |d W 5 Q R X |d W 5 Q R X |d W 5 Q R X W 5 Q R X | d d S )
Nr conr Ztentr! ZinsideZtnetZnocz?contentinsidetnetnocr r r r r test_element_nested_with_text8 s
z2_XmlFileTestCaseBase.test_element_nested_with_textc C s2 t | j}|td W 5 Q R X | d d S )Nr z)r r r r
r r r r r test_write_ElementE s z'_XmlFileTestCaseBase.test_write_Elementc
C s t d}t| j4}|d tdD ]}|| q(W 5 Q R X W 5 Q R X | }| |d k | dt |
| t dgt dd |
D d S )Nr d c s s | ] }|j V qd S r
)tag).0elr r r T s zE_XmlFileTestCaseBase.test_write_Element_repeatedly..) r
r r r ranger _parse_file
assertTrueZ assertEquallengetrootset)r r r itreer r r test_write_Element_repeatedlyJ s z2_XmlFileTestCaseBase.test_write_Element_repeatedlyc
C sB t | j$}|jdddid W 5 Q R X W 5 Q R X | d d S )N {nsURI}testxnsURIZnsmapz!r r r r r test_namespace_nsmapV s z)_XmlFileTestCaseBase.test_namespace_nsmapc C sX t | j:}|jdddid |d W 5 Q R X W 5 Q R X W 5 Q R X | d d S )Nr r5 r6 r7 {nsURI}toastz0r r r r r test_namespace_nested_nsmap\ s
z0_XmlFileTestCaseBase.test_namespace_nested_nsmapc
C s: t | j }|d W 5 Q R X W 5 Q R X | d d S )Nr4 z'r r r r r test_anonymous_namespacec s z-_XmlFileTestCaseBase.test_anonymous_namespacec C sP t | j2}|d |d W 5 Q R X W 5 Q R X W 5 Q R X | d d S )Nr r9 z6r r r r r test_namespace_nested_anonymousi s
z4_XmlFileTestCaseBase.test_namespace_nested_anonymousc
C sB t | j$}|jdd did W 5 Q R X W 5 Q R X | d d S )Nr4 r6 r7 zr r r r r test_default_namespacep s z+_XmlFileTestCaseBase.test_default_namespacec C sX t | j:}|jdd did |d W 5 Q R X W 5 Q R X W 5 Q R X | d d S )Nr4 r6 r7 r9 z*r r r r r test_nested_default_namespacev s
z2_XmlFileTestCaseBase.test_nested_default_namespacec
C sT ddl m} t| j*}||d |d W 5 Q R X W 5 Q R X | d d S )Nr )ProcessingInstructionpypir z)et_xmlfile.xmlfiler? r r r r r )r r? r r r r test_pi} s z _XmlFileTestCaseBase.test_pic
C sJ t | j,}|td |d W 5 Q R X W 5 Q R X | d d S )Nz a commentr z )r r r etreeCommentr r r r r r test_comment s
z!_XmlFileTestCaseBase.test_commentc
C sB t | j$}|jdddid W 5 Q R X W 5 Q R X | d d S )Nr kv)attribzr r r r r test_attribute s z#_XmlFileTestCaseBase.test_attributec
C sN t | j0}|d |d |d W 5 Q R X W 5 Q R X | d d S )Nr zComments:
zEntities: &z>Comments: <!-- text -->
Entities: &r r r r r
test_escaping s
z"_XmlFileTestCaseBase.test_escapingc
C sL t | jdd&}|d |d W 5 Q R X W 5 Q R X | jddd d S )Nutf16)encodingr r r r r r r r
test_encoding s z"_XmlFileTestCaseBase.test_encodingc C s t | jdd}|d | d |d | d |d. | d |d td
d | d W 5 Q R X | d |d
| d W 5 Q R X | d W 5 Q R X | d d S )NFZbufferedr zr z toastr! toastZsomeZmorez"toastsometoastz*toastsometoastendz-toastsometoastendz4toastsometoastend)r r r r r rC r
r r r r test_buffering s
z#_XmlFileTestCaseBase.test_bufferingc C s t | jdd}|d` | d |d | d |d" | d | | d W 5 Q R X | d W 5 Q R X | d W 5 Q R X | d d S ) NTrN r r r! rO z!toast)r r r r r flushr r r r
test_flush s
z _XmlFileTestCaseBase.test_flushc C sR z$t | j}|d W 5 Q R X W n tk
rB | d Y n X | d d S )Nr TF)r r r r r- r r r r test_failure_preceding_text s z0_XmlFileTestCaseBase.test_failure_preceding_textc
C sh t | jT}|d W 5 Q R X z|d W n tk
rN | d Y n X | d W 5 Q R X d S )Nr r TF)r r r r r r- r r r r test_failure_trailing_text s
z/_XmlFileTestCaseBase.test_failure_trailing_textc
C sl t | jX}|d W 5 Q R X z|td W n tk
rR | d Y n X | d W 5 Q R X d S )Nr TF)r r r r r
r r- r r r r test_failure_trailing_Element s
z2_XmlFileTestCaseBase.test_failure_trailing_Elementc C s d }z:t | j&}|d}|j}| tdW 5 Q R X W n\ tk
r | | z|ttdd W n tk
r | d Y n X | d Y n X | d d S )Nr Z123ZhuhuTF)r r r __exit__ __enter__
ValueErrorr- r )r cm_exitr r5 r r r 'test_closing_out_of_order_in_error_case s
z<_XmlFileTestCaseBase.test_closing_out_of_order_in_error_casec C s8 | j }| j d z| j W S | j | X d S Nr )r tellseekreadr posr r r
_read_file s
z _XmlFileTestCaseBase._read_filec C s8 | j }| j d zt| j W S | j | X d S r] )r r^ r_ r ra r r r r, s
z _XmlFileTestCaseBase._parse_filec C s | j d k r| j d S r
)r closer r r r tearDown s
z _XmlFileTestCaseBase.tearDownutf8c C s( t | ||}|d ks$t|d S r
)r rc decodeAssertionError)r expectedrL Zdiffr r r r
s z _XmlFileTestCaseBase.assertXml)rf )#__name__
__module__ __qualname__r r r r r" r$ r% r3 r8 r: r; r< r= r> pytestmarkZxfailrB rE rI rJ rM rQ rT rU rV rW r\ rc r, re r r r r r r sF
r c @ s e Zd Zdd Zdd ZdS )BytesIOXmlFileTestCasec C s t | _d S r
r r r r r r s z BytesIOXmlFileTestCase.setUpc
C sD t | jdd }|d W 5 Q R X W 5 Q R X | t| jj d S NT)rd r )r r r Z assertRaisesrZ getvaluer r r r test_filelike_close s z*BytesIOXmlFileTestCase.test_filelike_closeN)rj rk rl r rr r r r r ro s ro c @ s e Zd Zdd ZdS )TempXmlFileTestCasec C s t | _d S r
)tempfile
TemporaryFiler r r r r r s zTempXmlFileTestCase.setUpN)rj rk rl r r r r r rs s rs c @ sP e Zd Zdd Zdd Zdd Zdd Zed d
d d Zed d
d
d Z dS )TempPathXmlFileTestCasec C s t jdd| _| jj| _d S )NF)delete)rt NamedTemporaryFile_tmpfilenamer r r r r r s z TempPathXmlFileTestCase.setUpc C s4 z| j W 5 t j| jjr.t | jj X d S r
)ospathexistsry rz unlinkrd r r r r re $ s z TempPathXmlFileTestCase.tearDownc C s | j d | j S r] )ry r_ r` r r r r rc + s z"TempPathXmlFileTestCase._read_filec C s | j d t| j S r] )ry r_ r r r r r r, / s z#TempPathXmlFileTestCase._parse_fileTz1temp file behaviour is too platform specific herec C s d S r
r r r r r rQ 3 s z&TempPathXmlFileTestCase.test_bufferingc C s d S r
r r r r r rT 7 s z"TempPathXmlFileTestCase.test_flushN)
rj rk rl r re rc r, r rQ rT r r r r rv s
rv c @ sD e Zd ZG dd deZdd Zdd Zdd Zd d
Zd d Z d
S ) SimpleFileLikeXmlFileTestCasec @ s e Zd Zdd Zdd ZdS )z,SimpleFileLikeXmlFileTestCase.SimpleFileLikec C s( || _ |j| _|j| _|j| _d| _d S )NF)_targetr r^ r_ closed)r targetr r r __init__> s
z5SimpleFileLikeXmlFileTestCase.SimpleFileLike.__init__c C s | j r
td| _ | j d S )NT)r rh r rd r r r r rd E s
z2SimpleFileLikeXmlFileTestCase.SimpleFileLike.closeN)rj rk rl r rd r r r r SimpleFileLike= s r c C s t | _| | j| _d S r
)r r r r r r r r r J s z#SimpleFileLikeXmlFileTestCase.setUpc C s
| j S r
)r rq r r r r rc N s z(SimpleFileLikeXmlFileTestCase._read_filec C s8 | j }| jd zt| jW S | j| X d S r] )r r^ r r_ r ra r r r r, Q s
z)SimpleFileLikeXmlFileTestCase._parse_filec
C s> t | j }|d W 5 Q R X W 5 Q R X | | jj d S )Nr )r r r Z assertFalser r r r r test_filelike_not_closingY s z7SimpleFileLikeXmlFileTestCase.test_filelike_not_closingc
C sH t | jdd }|d W 5 Q R X W 5 Q R X | | jj d | _d S rp )r r r r- r r r r r rr _ s
z1SimpleFileLikeXmlFileTestCase.test_filelike_closeN)
rj rk rl objectr r rc r, r rr r r r r r < s
r )
__future__r lxml ImportErrorior Zunittestrt r{ sysZcommon_importsr r Z
et_xmlfiler rA r rm helperr xml.etree.ElementTreexmlr
r r ro rs rv r r r r r s( r