U
    }g                     @   s   d Z ddlZddlZddlZddlmZmZ er@ddlmZm	Z	 ddl
mZ ddlmZmZmZ ejje dd	G d
d deZejje dd	G dd deZdS )z@Tests to ensure that the lxml tree builder generates good trees.    N   )LXML_PRESENTLXML_VERSION)LXMLTreeBuilderLXMLTreeBuilderForXML)BeautifulStoneSoup)HTMLTreeBuilderSmokeTestXMLTreeBuilderSmokeTestSOUP_SIEVE_PRESENTz;lxml seems not to be present, not testing its tree builder.reasonc                   @   s^   e Zd ZdZedd Zdd Zdd Zej	j
e p:edk d	d
dd Zdd Zdd ZdS )TestLXMLTreeBuilder!See ``HTMLTreeBuilderSmokeTest``.c                 C   s   t S N)r   self r   7/tmp/pip-unpacked-wheel-kgiupv3k/bs4/tests/test_lxml.pydefault_builder   s    z#TestLXMLTreeBuilder.default_builderc                 C   s(   |  dd |  dd |  dd d S )Nz<p>foo&#10000000000000;bar</p>z<p>foobar</p>z<p>foo&#x10000000000000;bar</p>z<p>foo&#1000000000;bar</p>)Zassert_soupr   r   r   r   test_out_of_range_entity    s    z,TestLXMLTreeBuilder.test_out_of_range_entityc                 C   s   d S r   r   r   r   r   r   *test_entities_in_foreign_document_encoding%   s    z>TestLXMLTreeBuilder.test_entities_in_foreign_document_encoding)         r   z@Skipping doctype test for old version of lxml to avoid segfault.r   c                 C   s(   |  d}|jd }d| ks$td S )Nz
<!DOCTYPE>r    )soupcontentsstripAssertionError)r   r   doctyper   r   r   test_empty_doctype.   s    

z&TestLXMLTreeBuilder.test_empty_doctypec              	   C   s\   t jdd}td}W 5 Q R X dt|jks2t|\}|jtksFtdt|jksXtd S )NT)recordz<b />z<b/>z+The BeautifulStoneSoup class was deprecated)	warningscatch_warningsr   strbr   filename__file__message)r   wr   warningr   r   r   %test_beautifulstonesoup_is_xml_parser7   s    z9TestLXMLTreeBuilder.test_beautifulstonesoup_is_xml_parserc                 C   s2   | j ddd}d |jjkstd |jjks.td S )Nz=
   <p>

<sourceline>
<b>text</b></sourceline><sourcepos></p>T)Zstore_line_numbers)r   pZ
sourceliner   Z	sourceposr   r   r   r   r   test_tracking_line_numbersA   s    z.TestLXMLTreeBuilder.test_tracking_line_numbersN)__name__
__module____qualname____doc__propertyr   r   r   pytestmarkskipifr   r   r    r+   r.   r   r   r   r   r      s   
	

r   z?lxml seems not to be present, not testing its XML tree builder.c                   @   sF   e Zd ZdZedd Zdd Zejj	e
 dddd	 Zd
d ZdS )TestLXMLXMLTreeBuilderr   c                 C   s   t S r   )r   r   r   r   r   r   X   s    z&TestLXMLXMLTreeBuilder.default_builderc                 C   sz   |  d}|jddddks t|jjddiks4t|jjdddksJt|jjdddks`t|jjdddksvtd S )	Nam  <?xml version="1.1"?>
<root><tag xmlns="http://unprefixed-namespace.com">content</tag><prefix:tag2 xmlns:prefix="http://prefixed-namespace.com">content</prefix:tag2><prefix2:tag3 xmlns:prefix2="http://another-namespace.com"><subtag xmlns="http://another-unprefixed-namespace.com"><subsubtag xmlns="http://yet-another-unprefixed-namespace.com"></prefix2:tag3></root>z$http://www.w3.org/XML/1998/namespacezhttp://prefixed-namespace.comzhttp://another-namespace.com)xmlprefixprefix2r8   )r9   r8   )r:   r8   )r   _namespacesr   tagtag2subtagZ	subsubtagr-   r   r   r   test_namespace_indexing\   s0     
z.TestLXMLXMLTreeBuilder.test_namespace_indexingzSoup Sieve not installedr   c                 C   s   |  d}|djdkst|djdks2t|dd ksDt|jd|jjdjdks`t|jdjdksvt|djdkst|djdkst|d	jdkst|jd	jdkstd S )
Nz<?xml version="1.1"?>
<root><tag xmlns="http://unprefixed-namespace.com">content</tag><prefix:tag2 xmlns:prefix="http://prefixed-namespace.com">content</tag><subtag xmlns:prefix="http://another-namespace-same-prefix.com"><prefix:tag3></subtag></root>r<   zprefix|tag2r=   zprefix|tag3)
namespacesZtag3zprefix:tag2zprefix:tag3)r   Z
select_onenamer   r>   r;   findr-   r   r   r   /test_namespace_interaction_with_select_and_find   s    zFTestLXMLXMLTreeBuilder.test_namespace_interaction_with_select_and_findc                 C   sf   |  d}t|j| jstt|}t|}d|jj	ks@t|j|jksPtt|j| jsbtd S )Nz<a>some markup</a>zsome markup)
r   
isinstanceZbuilderr   r   pickledumpsloadsastring)r   r   ZpickledZ	unpickledr   r   r   test_pickle_restores_builder   s    


z3TestLXMLXMLTreeBuilder.test_pickle_restores_builderN)r/   r0   r1   r2   r3   r   r?   r4   r5   r6   r
   rC   rJ   r   r   r   r   r7   Q   s   
4
(r7   )r2   rE   r4   r"   r   r   r   Zbs4.builder._lxmlr   r   Zbs4r   r   r	   r
   r5   r6   r   r7   r   r   r   r   <module>   s$   8