a
    C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   R/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/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$J d S )Nz
<!DOCTYPE>r    )soupcontentsstrip)r   r   doctyper   r   r   test_empty_doctype.   s    

z&TestLXMLTreeBuilder.test_empty_doctypec                 C   sp   t jdd}td}W d    n1 s*0    Y  dt|jksFJ |\}|jtksZJ dt|jv slJ d S )NT)recordz<b />z<b/>z+The BeautifulStoneSoup class was deprecated)warningscatch_warningsr   strb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u sJ d |jju s.J d S )Nz=
   <p>

<sourceline>
<b>text</b></sourceline><sourcepos></p>T)Zstore_line_numbers)r   pZ
sourcelineZ	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 J |jjddiks4J |jjdddksJJ |jjdddks`J |jjdddksvJ 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prefix2r7   )r8   r7   )r9   r7   )r   _namespacestagtag2subtagZ	subsubtagr,   r   r   r   test_namespace_indexing\   s.    
z.TestLXMLXMLTreeBuilder.test_namespace_indexingzSoup Sieve not installedr   c                 C   s   |  d}|djdksJ |djdks2J |dd u sDJ |jd|jjdjdks`J |jdjdksvJ |djdksJ |djdksJ |d	jdksJ |jd	jdksJ 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:   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J t|}t|}d|jjks@J |j|jksPJ t|j| jsbJ d S )Nz<a>some markup</a>zsome markup)	r   
isinstanceZbuilderr   pickledumpsloadsastring)r   r   ZpickledZ	unpickledr   r   r   test_pickle_restores_builder   s    


z3TestLXMLXMLTreeBuilder.test_pickle_restores_builderN)r.   r/   r0   r1   r2   r   r>   r3   r4   r5   r
   rB   rI   r   r   r   r   r6   Q   s   
4
(r6   )r1   rD   r3   r!   r   r   r   Zbs4.builder._lxmlr   r   Zbs4r   r   r	   r
   r4   r5   r   r6   r   r   r   r   <module>   s$   8