3 d@sdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZ 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)SoupTestc@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"|jd}|jdjdkstdS)Nz 1234b2)soupfindstringAssertionError)selfrr=/tmp/pip-build-8z3xcdsh/beautifulsoup4/bs4/tests/test_tree.py test_find_tag-s zTestFind.test_find_tagcCs"|jd}|jdddkstdS)Nu

Räksmörgås

u Räksmörgås)r)rrr)rrrrrtest_unicode_text_find1s zTestFind.test_unicode_text_findcCs,|jd}t|d|jddjks(tdS)Nu&

here it is

z here it isu Räksmörgås)id)rstrrtextr)rrrrrtest_unicode_attribute_find5s z$TestFind.test_unicode_attribute_findcCs"|jd}dt|jkstdS)z)Test an optimization that finds all tags.zfoobarN)rlenfind_allr)rrrrrtest_find_everything;s zTestFind.test_find_everythingcCs$|jd}dt|jdks tdS)z;Test an optimization that finds all tags with a given name.zfoobarbazr#aN)rr$r%r)rrrrrtest_find_everything_with_name@s z'TestFind.test_find_everything_with_nameN) __name__ __module__ __qualname____doc__rrr"r&r(rrrrr&s rc@s8eZdZdZddZddZddZdd Zd d Zd S) TestFindAllz%Basic tests of the find_all() method.cCsz|jd}|jdddgks t|jddgdddgksFoobar»bar)rFooz.*»TN)rr%rrecompile)rrrrrtest_find_all_text_nodesHs   z$TestFindAll.test_find_all_text_nodescCs|jd}|j|jddddddg|j|jddddg|j|jdd ddddd d g|j|jdd ddddd d gd S)z7You can limit the number of items returned by find_all.z(12345r')limit1r3r 45rN)rassert_selectsr%)rrrrrtest_find_all_limitVs zTestFindAll.test_find_all_limitcCs:|jd}|j|ddddg|j|jdddgdS) Nz!123r'r)r5r6foo)rr7)rr;r)rrrrr%test_calling_a_tag_is_calling_findallbs z1TestFindAll.test_calling_a_tag_is_calling_findallcCs.|jd}g}|j|g|j|ks*tdS)Nz)rappendr%r)rrlrrrTtest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursiongs  z`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursioncCsX|jd}|jd}t|ds"t|jd}t|ds:t|jdd}t|dsTtdS)z%All find_all calls return a ResultSetzr'sourceTr=)rN)rr%hasattrr)rrresultrrrtest_find_all_resultsetqs    z#TestFindAll.test_find_all_resultsetN) r)r*r+r,r3r<r>rArErrrrr-Es   r-c@seZdZddZdS)TestFindAllBasicNamespacescCs<|jd}d|jdjkstd|jddidjks8tdS)Nz04r9z mathml:msqrtr'zsvg:fillZred)attrs)rrrrname)rrrrrtest_find_by_namespaced_names z7TestFindAllBasicNamespaces.test_find_by_namespaced_nameN)r)r*r+rIrrrrrF~srFc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)TestFindAllByNamez&Test ways of finding tags by tag name.cCs|jd|_dS)NzFirst tag. Second tag. Third Nested tag. tag.)rtree)rrrr setup_methodszTestFindAllByName.setup_methodcCs|j|jjdddgdS)Nr'z First tag.z Nested tag.)r;rKr%)rrrrtest_find_all_by_tag_namesz+TestFindAllByName.test_find_all_by_tag_namecCs\|j|jjddddg|j|jjdddddg|j|jjdtjddddgdS)Nr'z First tag.)rTz Nested tag.tag)r;rKr%r1r2)rrrrtest_find_all_by_name_and_textsz0TestFindAllByName.test_find_all_by_name_and_textcCs|j|jjjddgdS)Nr'z Nested tag.)r;rKcr%)rrrr!test_find_all_on_non_root_elementsz3TestFindAllByName.test_find_all_on_non_root_elementcCs|j|jdddgdS)Nr'z First tag.z Nested tag.)r;rK)rrrr%test_calling_element_invokes_find_allsz7TestFindAllByName.test_calling_element_invokes_find_allcCs |j|jjtdddgdS)Nr'z First tag.z Nested tag.)r;rKr%r )rrrrtest_find_all_by_tag_strainersz/TestFindAllByName.test_find_all_by_tag_strainercCs"|j|jjddgdddgdS)Nr'rz First tag.z Second tag.z Nested tag.)r;rKr%)rrrrtest_find_all_by_tag_namessz,TestFindAllByName.test_find_all_by_tag_namescCs$|j|jjddddddgdS)NT)r'rz First tag.z Second tag.z Nested tag.)r;rKr%)rrrrtest_find_all_by_tag_dictsz+TestFindAllByName.test_find_all_by_tag_dictcCs$|j|jjtjddddgdS)Nz^[ab]$z First tag.z Second tag.z Nested tag.)r;rKr%r1r2)rrrrtest_find_all_by_tag_resz)TestFindAllByName.test_find_all_by_tag_recCs,dd}|jd}|j|j|ddgdS)NcSs|j|jdkS)Nr)rHget)rNrrrid_matches_nameszRTestFindAllByName.test_find_all_with_tags_matching_method..id_matches_namezMatch 1. Does not match. Match 2.zMatch 1.zMatch 2.)rr;r%)rrXrKrrr'test_find_all_with_tags_matching_methods z9TestFindAllByName.test_find_all_with_tags_matching_methodcCsx|jd}|jdd}|jdtjd}|jdddg\}}d|jksJtd|jksXtd|jksftd|jksttdS)NzH
1
2
3
divza dza br7r6)rrr1r2r%rr)rrZr1Zr2Zr3Zr4rrr%test_find_with_multi_valued_attributes z7TestFindAllByName.test_find_with_multi_valued_attributeN)r)r*r+r,rLrMrOrQrRrSrTrUrVrYr[rrrrrJs  rJc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestFindAllByAttributecCs&|jd}|j|jddddgdS)Nz Matching a. Non-matching Matching b.a. first)rz Matching a.z Matching b.)rr;r%)rrKrrrtest_find_all_by_attribute_namesz6TestFindAllByAttribute.test_find_all_by_attribute_namecCstdjd}djd}|j|}|jg|j|dks6t|jg|j|jddksTt|jgspt|j|dgddS)Nuםולשutf8u)titlezsomething else)encoderr'r%rdecode)rZpeacedatarrrr%test_find_all_by_utf8_attribute_values    zName match. Class match. Non-match. A tag called 'name1'. Zname1)rHzA tag called 'name1'.rH)rGz Name match.classZclass2z Class match.)rr;r%)rrKrrrtest_find_all_by_attribute_dictsz6TestFindAllByAttribute.test_find_all_by_attribute_dictcCs|jd}|j|jddddg|j|jddddg|j|jdd ddg|j|jdddg|j|jdd ddg|j|jdddg|j|jdd dgdS) Nz Class 1. Class 2. Class 1. Class 3 and 4. r'r6)class_zClass 1.rPr7zClass 3 and 4.r9)rG)rr;r%)rrKrrrtest_find_all_by_classsz-TestFindAllByAttribute.test_find_all_by_classcCst|jd}|jdtjdd}|j|dg|jdtjdd}|j|dg|jdtjdd}|j|dgdS)Nz#Found itZgaro)rgzFound itr'zo b)rr%r1r2r;)rrKfrrr0test_find_by_class_when_multiple_classes_presents zGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_presentcCsd|jd}|j|jdtjddgdd}|j|jd|gdd}|j|jd|dgdS) NzFound itr'bazFound itcSs t|dkS)Nr4)r$)valuerrrbig_attribute_value%sznTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..big_attribute_valuecSs t|dkS)Nr4)r$)rmrrrsmall_attribute_value*szpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..small_attribute_value)rr;r%r1r2)rrrnrorrr:test_find_all_with_non_dictionary_for_attrs_finds_by_class s zQTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_classcCs|jd}|jd\}}||gs0t|jdd|gsFt|jdd|g|jdddks^t|g|jddksttg|jddkstdS)Nz*r'r=r.zfoo bar)rgzbar foo)rr%r)rrr'Za2rrr:test_find_all_with_string_for_attrs_finds_multiple_classes0s zQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classescCs0|jd}tddid}|j|j|dgdS)Nzi Match. Non-match.rr])rGzMatch.)rr r;r%)rrKstrainerrrr'test_find_all_by_attribute_soupstrainer<sz>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainercCs&|jd}|j|jddddgdS)NzID present. No ID present. ID is empty.r')rzNo ID present.)rr;r%)rrKrrr$test_find_all_with_missing_attributeDsz;TestFindAllByAttribute.test_find_all_with_missing_attributecCs&|jd}|j|jddddgdS)NzID present. No ID present. ID is empty.T)rz ID present.z ID is empty.)rr;r%)rrKrrr$test_find_all_with_defined_attributeLsz;TestFindAllByAttribute.test_find_all_with_defined_attributecCs>|jd}ddg}|j|jdd||j|jdd|dS)Nz[Unquoted attribute. Quoted attribute.zUnquoted attribute.zQuoted attribute.r)rr6)rr;r%)rrKexpectedrrr$test_find_all_with_numeric_attributeUs z;TestFindAllByAttribute.test_find_all_with_numeric_attributecCs,|jd}|j|jdddgdddgdS)Nz1 2 3 No ID.r6r7r9)r)rr;r%)rrKrrr(test_find_all_with_list_attribute_values^sz?TestFindAllByAttribute.test_find_all_with_list_attribute_valuescCs,|jd}|j|jtjddddgdS)NzOne a. Two as. Mixed as and bs. One b. No ID.z^a+$)rzOne a.zTwo as.)rr;r%r1r2)rrKrrr5test_find_all_with_regular_expression_attribute_valuehszLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuecCsB|jd}|j}|g|jdddks(tg|jdddks>tdS)Nzfoobarfoor'r=)rr.)rr'r%r)rrr'rrr'test_find_by_name_and_containing_stringus z>TestFindAllByAttribute.test_find_by_name_and_containing_stringcCs*|jd}|jd|jdddks&tdS)Nz"foofoor'r=)r)rr%r)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|jd}|j}|g|jdddks(tg|jdddks>tdS)Nz"foofoor#r=)rrrr.)rr'r%r)rrr'rrr,test_find_by_attribute_and_containing_strings zCTestFindAllByAttribute.test_find_by_attribute_and_containing_stringN)r)r*r+r^rdrfrhrkrprqrsrtrurwrxryrzr{r|rrrrr\s       r\c@seZdZdZddZdS) TestSmoothzTest Tag.smooth.cCs|jd}|j}|jd|jd|jtd|jtd|jd|j}t||d}|jd|jd |j|d|jjkstd t |j kst|j d t |j kstd |j d kstd|jjkstd|j dkstd|j dkstdS)Nz
a
rrPz Comment 1z Comment 2dspanr6rabcrZ12rr#) rrZr?rdefault_builderrrrrr$contentsZsmooth)rrrZbuilderrrrr test_smooths(        zTestSmooth.test_smoothN)r)r*r+r,rrrrrr}sr}c@seZdZdZddZdS) TestIndexzTest Tag.indexc Cs^|jd}|j}x(t|jD]\}}||j|kstqWtjt|jdWdQRXdS)Nah
Identical Not identical Identical Identical with child Also not identical Identical with child
r) rrZ enumeraterindexrpytestraises ValueError)rrKrZielementrrr test_indexs zTestIndex.test_indexN)r)r*r+r,rrrrrrsrc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)TestParentOperationsz;Test navigation and searching through an element's parents.cCs|jd|_|jj|_dS)Na1