a #.cŹ@sdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!GdddeZ"Gddde"Z#Gddde"Z$Gdd d eZ%Gd!d"d"e%Z&Gd#d$d$e%Z'Gd%d&d&eZ(Gd'd(d(eZ)dS))a8Tests for Beautiful Soup's tree traversal methods. The tree traversal methods are the main advantage of using Beautiful Soup over just using a parser. Different parsers will build different Beautiful Soup trees given the same markup, but all Beautiful Soup trees can be traversed with the methods tested here. ) set_traceN) BeautifulSoup)builder_registryHTMLParserTreeBuilder) CDataComment DeclarationDoctype FormatterNavigableStringScript SoupStrainer StylesheetTagTemplateString)SoupTestskipIfc@s8eZdZdZddZddZddZdd Zd d Zd S) TestFindzBasic tests of the find() method. find() just calls find_all() with limit=1, so it's not tested all that thouroughly here. cCs"|d}|djdksJdS)Nz 1234b2)soupfindstringselfrrM/var/www/brookimports/venv/lib/python3.9/site-packages/bs4/tests/test_tree.py test_find_tag.s zTestFind.test_find_tagcCs"|d}|jdddksJdS)Nu

Räksmörgås

Räksmörgåsr)rrrrrrtest_unicode_text_find2s zTestFind.test_unicode_text_findcCs,|d}t|d|jddjks(JdS)Nu&

here it is

z here it isrid)rstrrtextrrrrtest_unicode_attribute_find6s z$TestFind.test_unicode_attribute_findcCs"|d}dt|ksJdS)z)Test an optimization that finds all tags.foobarNrlenfind_allrrrrtest_find_everything<s zTestFind.test_find_everythingcCs$|d}dt|dks JdS)z;Test an optimization that finds all tags with a given name.zfoobarbazr(aNr)rrrrtest_find_everything_with_nameAs z'TestFind.test_find_everything_with_nameN) __name__ __module__ __qualname____doc__rr!r&r,r.rrrrr's rc@s8eZdZdZddZddZddZdd Zd d Zd S) TestFindAllz%Basic tests of the find_all() method.cCsv|d}|jdddgks J|jddgdddgksFoobar»barr Fooz.*)r5r4»TN)rr+recompilerrrrtest_find_all_text_nodesIs  z$TestFindAll.test_find_all_text_nodescCst|d}||jdddgd||jddddg||jdddgd ||jdd dgd d S) z7You can limit the number of items returned by find_all.z(12345r-limit)1r3rr= )r=rr>45rNrassert_selectsr+rrrrtest_find_all_limitWs zTestFindAll.test_find_all_limitcCs:|d}||ddddg||jdddgdS) Nz!123r-rr;r=foor"r>)rrCrrrrr%test_calling_a_tag_is_calling_findallcs z1TestFindAll.test_calling_a_tag_is_calling_findallcCs.|d}g}||g||ks*JdS)N)rappendr+)rrlrrrTtest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursionhs  z`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursioncCsX|d}|d}t|ds"J|d}t|ds:J|jdd}t|dsTJdS)z%All find_all calls return a ResultSetrGr-sourceTrEr N)rr+hasattr)rrresultrrrtest_find_all_resultsetrs    z#TestFindAll.test_find_all_resultsetN) r/r0r1r2r9rDrFrJrNrrrrr3Fs   r3c@seZdZddZdS)TestFindAllBasicNamespacescCs<|d}d|djksJd|jddidjks8JdS)Nz04r@z mathml:msqrtr-zsvg:fillredattrs)rrrnamerrrrtest_find_by_namespaced_names z7TestFindAllBasicNamespaces.test_find_by_namespaced_nameN)r/r0r1rTrrrrrOsrOc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)TestFindAllByNamez&Test ways of finding tags by tag name.cCs|d|_dS)NzFirst tag. Second tag. Third Nested tag. tag.rtreerrrr setup_methodszTestFindAllByName.setup_methodcCs||jdddgdSNr- First tag. Nested tag.rCrWr+rXrrrtest_find_all_by_tag_namesz+TestFindAllByName.test_find_all_by_tag_namecCs\||jjddddg||jjdddddg||jjdtddddgdS)Nr-r[r Tr\tagrCrWr+r7r8rXrrrtest_find_all_by_name_and_textsz0TestFindAllByName.test_find_all_by_name_and_textcCs||jjddgdS)Nr-r\)rCrWcr+rXrrr!test_find_all_on_non_root_elementsz3TestFindAllByName.test_find_all_on_non_root_elementcCs||dddgdSrZ)rCrWrXrrr%test_calling_element_invokes_find_allsz7TestFindAllByName.test_calling_element_invokes_find_allcCs ||jtdddgdSrZ)rCrWr+r rXrrrtest_find_all_by_tag_strainersz/TestFindAllByName.test_find_all_by_tag_strainercCs ||jddggddS)Nr-rr[z Second tag.r\r]rXrrrtest_find_all_by_tag_namessz,TestFindAllByName.test_find_all_by_tag_namescCs"||jdddgddS)NT)r-rrfr]rXrrrtest_find_all_by_tag_dictsz+TestFindAllByName.test_find_all_by_tag_dictcCs"||jtdgddS)Nz^[ab]$rfr`rXrrrtest_find_all_by_tag_resz)TestFindAllByName.test_find_all_by_tag_recCs,dd}|d}|||ddgdS)NcSs|j|dkSNr#)rSget)r_rrrid_matches_nameszRTestFindAllByName.test_find_all_with_tags_matching_method..id_matches_namezMatch 1. Does not match. Match 2.zMatch 1.zMatch 2.rB)rrlrWrrr'test_find_all_with_tags_matching_methods  z9TestFindAllByName.test_find_all_with_tags_matching_methodcCsx|d}|dd}|dtd}|dddg\}}d|jksJJd|jksXJd|jksfJd|jkstJdS)NzH
1
2
3
divza dza br>r=)rrr7r8r+r)rrr1r2Zr3Zr4rrr%test_find_with_multi_valued_attributes z7TestFindAllByName.test_find_with_multi_valued_attributeN)r/r0r1r2rYr^rarcrdrergrhrirmrqrrrrrUs  rUc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestFindAllByAttributecCs&|d}||jddddgdS)Nz Matching a. Non-matching Matching b.a. firstr"z Matching a.z Matching b.rBrrWrrrtest_find_all_by_attribute_names z6TestFindAllByAttribute.test_find_all_by_attribute_namecCstdd}dd}||}|jg|j|dks6J|jg|j|ddksTJ|jgspJ|j|dgddS)Nuםולשutf8u)titlezsomething else)encoderr-r+decode)rZpeacedatarrrr%test_find_all_by_utf8_attribute_values    zName match. Class match. Non-match. A tag called 'name1'. Zname1rSzA tag called 'name1'.rSrQz Name match.classZclass2z Class match.rBrtrrrtest_find_all_by_attribute_dicts z6TestFindAllByAttribute.test_find_all_by_attribute_dictcCs|d}||jddddg||jddddg||jdd ddg||dddg||jdd ddg||dddg||dd dgdS) Nz Class 1. Class 2. Class 1. Class 3 and 4. r-r=class_zClass 1.rbr>zClass 3 and 4.r@rQrBrtrrrtest_find_all_by_classs z-TestFindAllByAttribute.test_find_all_by_classcCst|d}|jdtdd}||dg|jdtdd}||dg|jdtdd}||dgdS)Nz#Found itZgarorFound itr-zo b)rr+r7r8rC)rrWfrrr0test_find_by_class_when_multiple_classes_presents zGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_presentcCsd|d}||dtddgdd}||d|gdd}||d|dgdS) NzFound itr-barcSs t|dkSNr:r*valuerrrbig_attribute_value&sznTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..big_attribute_valuecSs t|dkSrrrrrrsmall_attribute_value+szpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..small_attribute_valuerrCr+r7r8)rrrrrrr:test_find_all_with_non_dictionary_for_attrs_finds_by_class!s zQTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_classcCs|d}|d\}}||gs0J|dd|gsFJ|dd|g|jdddks^J|g|ddkstJg|ddksJdS)Nz*r-rEr4zfoo barrzbar foorr+)rrr-Za2rrr:test_find_all_with_string_for_attrs_finds_multiple_classes1s zQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classescCs0|d}tddid}|||dgdS)Nzi Match. Non-match.r#rsrQzMatch.)rr rCr+)rrWstrainerrrr'test_find_all_by_attribute_soupstrainer=s z>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainercCs&|d}||jddddgdS)NID present. No ID present. ID is empty.r-r"zNo ID present.rBrtrrr$test_find_all_with_missing_attributeEs z;TestFindAllByAttribute.test_find_all_with_missing_attributecCs&|d}||jddddgdS)NrTr"z ID present.z ID is empty.rBrtrrr$test_find_all_with_defined_attributeMs z;TestFindAllByAttribute.test_find_all_with_defined_attributecCs>|d}ddg}||jdd|||jdd|dS)Nz[Unquoted attribute. Quoted attribute.zUnquoted attribute.zQuoted attribute.rr"r=rB)rrWexpectedrrr$test_find_all_with_numeric_attributeVs z;TestFindAllByAttribute.test_find_all_with_numeric_attributecCs*|d}||jgddddgdS)Nz1 2 3 No ID.)r=r>r@r"r=r>rBrtrrr(test_find_all_with_list_attribute_values_s z?TestFindAllByAttribute.test_find_all_with_list_attribute_valuescCs,|d}||jtddddgdS)NzOne a. Two as. Mixed as and bs. One b. No ID.z^a+$r"zOne a.zTwo as.rrtrrr5test_find_all_with_regular_expression_attribute_valueis zLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuecCsB|d}|j}|g|jdddks(Jg|jdddks>JdS)Nzfoobarfoor-rEr r4rr-r+rrr-rrr'test_find_by_name_and_containing_stringvs z>TestFindAllByAttribute.test_find_by_name_and_containing_stringcCs*|d}|d|jdddks&JdS)Nz"foofoor-rEr rrrrr=test_find_by_name_and_containing_string_when_string_is_buried}s zTTestFindAllByAttribute.test_find_by_name_and_containing_string_when_string_is_buriedcCsB|d}|j}|g|jdddks(Jg|jdddks>JdS)Nz"foofoor(rE)r#rrr4rrrrr,test_find_by_attribute_and_containing_strings zCTestFindAllByAttribute.test_find_by_attribute_and_containing_stringN)r/r0r1rur{r~rrrrrrrrrrrrrrrrrrrs       rrc@seZdZdZddZdS) TestSmoothzTest Tag.smooth.cCs|d}|j}|d|d|td|td|d|}t||d}|d|d ||d|jjksJd t|j ksJ| d t|j ksJd |j d ksJd|jjksJd|j dksJd|j dksJdS)Nz
a
rrbz Comment 1z Comment 2dspanr=rabcrZ12rr() rrnrHrdefault_builderrrrr*contentsZsmooth)rrrnbuilderrrrr test_smooths(        zTestSmooth.test_smoothN)r/r0r1r2rrrrrrsrc@seZdZdZddZdS) TestIndexzTest Tag.indexcCsn|d}|j}t|jD]\}}|||ksJqtt|dWdn1s`0YdS)Nah
Identical Not identical Identical Identical with child Also not identical Identical with child
r)rrn enumeraterindexpytestraises ValueError)rrWrnielementrrr test_indexs   zTestIndex.test_indexN)r/r0r1r2rrrrrrsrc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)TestParentOperationsz;Test navigation and searching through an element's parents.cCs|d|_|jj|_dS)Na1