a C g7@sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdd d eZGd!d"d"eZGd#d$d$eZGd%d&d&eZ gd'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. N) BeautifulSoup)builder_registry)!AttributeResemblesVariableWarningCDataCommentNavigableStringTag) SoupStrainer)SoupTestc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)TestFindz&Basic tests of the find() method. cCs"|d}|djdksJdS)Nz 1234b2)soupfindstringselfrrR/var/www/lab.imftr.de/x/nb_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_find&s zTestFind.test_unicode_text_findcCs,|d}t|d|jddjks(JdS)Nu&

here it is

z here it isrid)rstrrtextrrrrtest_unicode_attribute_find*s 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_name4s z'TestFind.test_find_everything_with_namecCsP|d}d|jksJd|djks4Jd|djksLJdS)Nz

divprrnamefind_previous_siblingfind_next_siblingrrrrtest_find_with_no_arguments9s z$TestFind.test_find_with_no_argumentscCsP|d}d|jksJd|djks4Jd|djksLJdS)Nz"text
text
text

text

r'r(r)rrrr+test_find_with_no_arguments_only_finds_tags?s z4TestFind.test_find_with_no_arguments_only_finds_tagsN) __name__ __module__ __qualname____doc__rrrr$r&r-r.rrrrr sr c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) TestFindAllz%Basic tests of the find_all() method.cCsX|d}dt|jks Jdt|dks:Jdt|dksTJdS)Nz/text
text
text

text

r r r(r')rr"bodyr#rfind_previous_siblingsfind_next_siblingsrrrr/test_find_all_with_no_arguments_only_finds_tagsIs z;TestFindAll.test_find_all_with_no_arguments_only_finds_tagscCsv|d}|jdddgks J|jddgdddgksFoobar»barrFooz.*)r9r8»TN)rr#recompilerrrrtest_find_all_text_nodesOs  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)1r3r rA )rArrB45rNrassert_selectsr#rrrrtest_find_all_limit\s  zTestFindAll.test_find_all_limitcCs:|d}||ddddg||jdddgdS) Nz!123r%r r?rAfoorrB)rrGr rrrr%test_calling_a_tag_is_calling_findallfs z1TestFindAll.test_calling_a_tag_is_calling_findallcCs|d}g}||tjddL}g||ks8J|\}|jtksLJt|j}|dksbJWdn1sv0YdS)NTrecordzLIgnoring nested list [[...]] to avoid the possibility of infinite recursion.) rappendwarningscatch_warningsr#filename__file__rmessage)rrselfrefwwarningmsgrrrTtest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursionks   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 ResultSetrKr%sourceTrIrN)rr#hasattrrrresultrrrtest_find_all_resultsets    z#TestFindAll.test_find_all_resultsetN) r/r0r1r2r7r=rHrJrXr]rrrrr3Fs  r3c@seZdZddZdS)TestFindAllBasicNamespacescCs<|d}d|djksJd|jddidjks8JdS)Nz04rDz mathml:msqrtr%zsvg:fillZredattrs)rrrr*rrrrtest_find_by_namespaced_names z7TestFindAllBasicNamespaces.test_find_by_namespaced_nameN)r/r0r1rarrrrr^sr^c@sneZdZdZddddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS)TestFindAllByNamez&Test ways of finding tags by tag name.NreturncCs|d|_dS)NzFirst tag. Second tag. Third Nested tag. tag.rtreerrrr setup_methodszTestFindAllByName.setup_methodcCs||jdddgdSNr% First tag. Nested tag.rGrfr#rgrrrtest_find_all_by_tag_namesz+TestFindAllByName.test_find_all_by_tag_namecCs\||jjddddg||jjdddddg||jjdtddddgdS)Nr%rjrTrktagrGrfr#r;r<rgrrrtest_find_all_by_name_and_textsz0TestFindAllByName.test_find_all_by_name_and_textcCs||jjddgdS)Nr%rk)rGrfcr#rgrrr!test_find_all_on_non_root_elementsz3TestFindAllByName.test_find_all_on_non_root_elementcCs||dddgdSri)rGrfrgrrr%test_calling_element_invokes_find_allsz7TestFindAllByName.test_calling_element_invokes_find_allcCs ||jtdddgdSri)rGrfr#r rgrrrtest_find_all_by_tag_strainersz/TestFindAllByName.test_find_all_by_tag_strainercCs ||jddggddS)Nr%r rjz Second tag.rkrlrgrrrtest_find_all_by_tag_namessz,TestFindAllByName.test_find_all_by_tag_namescCs"||jdddgddS)NT)r%r rurlrgrrrtest_find_all_by_tag_dictsz+TestFindAllByName.test_find_all_by_tag_dictcCs"||jtdgddS)Nz^[ab]$rurorgrrrtest_find_all_by_tag_resz)TestFindAllByName.test_find_all_by_tag_recCs,dd}|d}|||ddgdS)NcSs|j|dkSNr)r*getrnrrrid_matches_nameszRTestFindAllByName.test_find_all_with_tags_matching_method..id_matches_namezMatch 1. Does not match. Match 2.zMatch 1.zMatch 2.rF)rr|rfrrr'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
r'za dza brBrA)rrr;r<r#r)rrZr1Zr2Zr3Zr4rrr%test_find_with_multi_valued_attributes z7TestFindAllByName.test_find_with_multi_valued_attribute)r/r0r1r2rhrmrprrrsrtrvrwrxr}r~rrrrrbs rbc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestFindAllByAttributecCs&|d}||jddddgdS)Nz Matching a. Non-matching Matching b.a. firstrz Matching a.z Matching b.rFrrfrrrtest_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'. Zname1r*zA tag called 'name1'.r*r_z Name match.classZclass2z Class match.rFrrrrtest_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%rAclass_zClass 1.rqrBzClass 3 and 4.rDr_rFrrrrtest_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#r;r<rG)rrffrrr0test_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_value4szpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..small_attribute_valuerrGr#r;r<)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%rIr8zfoo barrzbar foorr#)rrr%Za2rrr:test_find_all_with_string_for_attrs_finds_multiple_classes9s zQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classescCs0|d}tddid}|||dgdS)Nzi Match. Non-match.rrr_zMatch.)rr rGr#)rrfZstrainerrrr'test_find_all_by_attribute_soupstrainerEs z>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainercCs&|d}||jddddgdS)NID present. No ID present. ID is empty.r%rzNo ID present.rFrrrr$test_find_all_with_missing_attributeMs z;TestFindAllByAttribute.test_find_all_with_missing_attributecCs&|d}||jddddgdS)NrTrz ID present.z ID is empty.rFrrrr$test_find_all_with_defined_attributeUs z;TestFindAllByAttribute.test_find_all_with_defined_attributecCs>|d}ddg}||jdd|||jdd|dS)Nz[Unquoted attribute. Quoted attribute.zUnquoted attribute.zQuoted attribute.r rrArF)rrfexpectedrrr$test_find_all_with_numeric_attribute]s z;TestFindAllByAttribute.test_find_all_with_numeric_attributecCs>|d}||jgddddg||jgdgdS)Nz1 2 3 No ID.)rArBrDrrArBrFrrrr(test_find_all_with_list_attribute_valuesfs 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+$rzOne a.zTwo as.rrrrr5test_find_all_with_regular_expression_attribute_valuers zLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuecCsB|d}|j}|g|jdddks(Jg|jdddks>JdS)Nzfoobarfoor%rIrr8rr%r#rrr%rrr'test_find_by_name_and_containing_string~s z>TestFindAllByAttribute.test_find_by_name_and_containing_stringcCs*|d}|d|jdddks&JdS)Nz"foofoor%rIrrrrrr=test_find_by_name_and_containing_string_when_string_is_burieds zTTestFindAllByAttribute.test_find_by_name_and_containing_string_when_string_is_buriedcCsB|d}|j}|g|jdddks(Jg|jdddks>JdS)Nz"foofoor rI)rrr r8rrrrr,test_find_by_attribute_and_containing_strings zCTestFindAllByAttribute.test_find_by_attribute_and_containing_stringN)r/r0r1rrrrrrrrrrrrrrrrrrrrrs      rc@seZdZdZddZdS) TestSmoothzTest Tag.smooth.cCs|d}|j}|d|d|td|td|d|}t||d}|d|d ||d|jjusJd t|j ksJ| d t|j ksJd |j d ksJd|jjksJd|j dksJd|j dksJdS)Nz
a
r rqz Comment 1z Comment 2dspanrArabcrZ12r r ) rr'rNrdefault_builderrrrr"contentsZsmooth)rrr'builderrrrr 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 )rr' enumeraterindexpytestraises ValueError)rrfr'ielementrrr test_indexs   zTestIndex.test_indexN)r/r0r1r2rrrrrrsrc@sfeZdZdZddddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ dS)TestParentOperationsz;Test navigation and searching through an element's parents.NrccCs|d|_|jj|_dS)Na<