1h" d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ G d d
e Z G d d e Z G d
de Z G d de Z G d de Z G d de Z G d de Z G d de Z G d de Z G d d e Z G d d e Z G d d e Z G d! d"e Z G d# d$e Z G d% d&e Z g d'Z! G d( d)e Z" G d* d+e Z#y),a8 Tests 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 @ e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z y
) TestFindz&Basic tests of the find() method.
c f | j d }|j d j dk( sJ y )Nz 1234b2)soupfindstringselfr s Q/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/bs4/tests/test_tree.py
test_find_tagzTestFind.test_find_tag" s- yy;<yy~$$+++ c T | j d }|j d dk( sJ y )Nu Räksmörgås
Räksmörgåsr )r r r s r test_unicode_text_findz TestFind.test_unicode_text_find& s* yy12yyy0OCCCr c ~ | j d }t | d|j d j k( sJ y )Nu& here it is
z
here it isr id)r strr textr s r test_unicode_attribute_findz$TestFind.test_unicode_attribute_find* s7 yyAB D tyyOy < A AAAAr c b | j d }dt |j k( sJ y)z)Test an optimization that finds all tags.foobar Nr lenfind_allr s r test_find_everythingz TestFind.test_find_everything/ s* yy/0C
((((r c d | j d }dt |j d k( sJ y)z;Test an optimization that finds all tags with a given name.z foobarbazr& aNr' r s r test_find_everything_with_namez'TestFind.test_find_everything_with_name4 s- yy9:C
c*++++r c | j d }d|j j k( sJ d|j d j j k( sJ d|j d j j k( sJ y )Nzdivpr r namefind_previous_siblingfind_next_siblingr s r test_find_with_no_argumentsz$TestFind.test_find_with_no_arguments9 sv yy-. ((((( #<<>CCCCCdii&88:?????r c | j d }d|j j k( sJ d|j d j j k( sJ d|j d j j k( sJ y )N"texttext
texttext
r/ r0 r1 r s r +test_find_with_no_arguments_only_finds_tagsz4TestFind.test_find_with_no_arguments_only_finds_tags? sv yy=> ((((( #<<>CCCCCdii&88:?????r c | j d }d|j d j k( sJ d |j d k( sJ y )Nr7 r0 c | j dk( S )Nr0 r2 ts r zETestFind.test_find_with_function_can_only_find_tags..G s !&&#+ r c | dk( S )Nr" r< s r r> zETestFind.test_find_with_function_can_only_find_tags..H s
1f9 r )r r r2 r s r *test_find_with_function_can_only_find_tagsz3TestFind.test_find_with_function_can_only_find_tagsE sG yy=>dii 56;;;;;tyy!45555r N) __name__
__module__ __qualname____doc__r r r# r* r- r5 r8 rA r@ r r r r s3 ,DB
)
,
@ @ 6r r c 4 e Zd ZdZd Zd Zd Zd Zd Zd Z y) TestFindAllz%Basic tests of the find_all() method.c * | j d }dt |j j k( sJ dt |j d j k( sJ dt |j d j
k( sJ y )Nz/texttext
texttext
r& r r0 r/ )r r( bodyr) r find_previous_siblingsfind_next_siblingsr s r /test_find_all_with_no_arguments_only_finds_tagsz;TestFindAll.test_find_all_with_no_arguments_only_finds_tagsM sz yyJKC **,----C #==?@@@@C %(;;=>>>>r c | j d }|j d dgk( sJ |j ddg ddgk( sJ |j t j d g dk( sJ |j d g dk( sJ y) z'You can search the tree for text nodes.u Foobar»barr Fooz.*)rO rN »TN)r r) recompiler s r test_find_all_text_nodesz$TestFindAll.test_find_all_text_nodesS s yy9:}}E}*ug555 }}UEN}3u~EEE}}BJJt$4}59OOOO}}D})-CCCCr c L | j d }| j |j dd g d | j |j dd dg | j |j dd g d | j |j dd
g d y ) z7You can limit the number of items returned by find_all.z(12345r, limit)1r 3r rX
)rX r rY 45r Nr assert_selectsr) r s r test_find_all_limitz TestFindAll.test_find_all_limit` s yyCD DMM#QM 7 I DMM#QM 7#? DMM#RM 8:ST
DMM#QM 79RSr c | j d }| j |dd dg | j |j d dg y ) Nz!123r, r rV rX foor rY )r r^ r r s r %test_calling_a_tag_is_calling_findallz1TestFindAll.test_calling_a_tag_is_calling_findallj sG yy<= DA .6 DFFeF ,se4r c : | j d }g }|j | t j d 5 }g |j | k( sJ |\ }|j
t k( sJ t |j }|dk( sJ d d d y # 1 sw Y y xY w)NTrecordzLIgnoring nested list [[...]] to avoid the possibility of infinite recursion.) r appendwarningscatch_warningsr) filename__file__r! message)r r selfrefwwarningmsgs r Ttest_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursionz`TestFindAll.test_find_all_with_self_referential_data_structure_does_not_cause_infinite_recursiono s yy#w
$
$D
1 Q w// //IW##x/ //goo&Cab
b
s A BBc | j d }|j d }t |d sJ |j d }t |d sJ |j d }t |d sJ y)z%All find_all calls return a ResultSetrd r, sourceTra r N)r r) hasattrr r results r test_find_all_resultsetz#TestFindAll.test_find_all_resultset sl yy# s#vx((( t$vx((( e ,vx(((r N)
rB rC rD rE rL rS r_ rb rq rw r@ r r rG rG J s% /? DT5
(
)r rG c e Zd Zd Zy)TestFindAllBasicNamespacesc | j d }d|j d j k( sJ d|j ddi j k( sJ y )Nz04r[ z mathml:msqrtr, zsvg:fillredattrs)r r r r2 r s r test_find_by_namespaced_namez7TestFindAllBasicNamespaces.test_find_by_namespaced_name sR yyKLdii/66666diiz5&9i:?????r N)rB rC rD r~ r@ r r ry ry s @r ry c T e Zd ZdZddZd Zd Zd Zd Zd Z d Z
d
Z d Z d Z
d
Zy)TestFindAllByNamez&Test ways of finding tags by tag name.Nc 0 | j d | _ y )NzFirst tag.
Second tag.
Third Nested tag. tag.r treer s r setup_methodz TestFindAllByName.setup_method s II J K r c ^ | j | j j d ddg y Nr,
First tag. Nested tag.r^ r r) r s r test_find_all_by_tag_namez+TestFindAllByName.test_find_all_by_tag_name s& DII . .s 3lM5RSr c B | j | j j dd dg | j | j j dd ddg | j | j j dt j d ddg y )Nr, r r Tr tagr^ r r) rQ rR r s r test_find_all_by_name_and_textz0TestFindAllByName.test_find_all_by_name_and_text s II s< 8<.
II s4 0< 2O
II s2::e+< =
= )
r c p | j | j j j d dg y )Nr, r )r^ r cr) r s r !test_find_all_on_non_root_elementz3TestFindAllByName.test_find_all_on_non_root_element s' DIIKK 0 0 5
Gr c J | j | j d ddg y r )r^ r r s r %test_calling_element_invokes_find_allz7TestFindAllByName.test_calling_element_invokes_find_all s DIIcN\=,IJr c p | j | j j t d ddg y r )r^ r r) r
r s r test_find_all_by_tag_strainerz/TestFindAllByName.test_find_all_by_tag_strainer s. II |C 0 1L-3P
r c b | j | j j ddg g d y )Nr, r r z Second tag.r r r s r test_find_all_by_tag_namesz,TestFindAllByName.test_find_all_by_tag_names s) II Sz *,X
r c d | j | j j ddd g d y )NT)r, r r r r s r test_find_all_by_tag_dictz+TestFindAllByName.test_find_all_by_tag_dict s* II T 5 6 8
r c | j | j j t j d g d y )Nz^[ab]$r r r s r test_find_all_by_tag_rez)TestFindAllByName.test_find_all_by_tag_re s. II rzz( 3 4 8
r c r d }| j d }| j |j | ddg y )Nc > | j | j d k( S )Nr )r2 getr s r id_matches_namezRTestFindAllByName.test_find_all_with_tags_matching_method..id_matches_name s 88swwt}, ,r zMatch 1.
Does not match.
Match 2.zMatch 1.zMatch 2.r] )r r r s r 'test_find_all_with_tags_matching_methodz9TestFindAllByName.test_find_all_with_tags_matching_method s< - yy 6 7
DMM/ :Z12
3
r/ za dza brY rX )r r rQ rR r) r )r r r1r2r3r4s r %test_find_with_multi_valued_attributez7TestFindAllByName.test_find_with_multi_valued_attribute s yy V
YYue
$
YYubjj 0
1 uuen5Bbii bii bii bii r returnN)rB rC rD rE r r r r r r r r r r r r@ r r r r sA 0K
T
HK
V
r r c l e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zy)TestFindAllByAttributec n | j d }| j |j d ddg y )Nz
Matching a.
Non-matching Matching b.a.
firstr z Matching a.z Matching b.r] r r s r test_find_all_by_attribute_namez6TestFindAllByAttribute.test_find_all_by_attribute_name s9 yy ! "
DMMWM 5
}7UVr c \ dj d }dj d }| j | }|j g|j | k( sJ |j g|j |j d k( sJ |j gsJ |j |dg y )Nu םולשutf8u )titlezsomething else)encoder r, r) decode)r peacedatar s r %test_find_all_by_utf8_attribute_valuezName match.
Class match.
Non-match.
A tag called 'name1'.
name1r; zA tag called 'name1'.r2 r| z Name match.classclass2z Class match.r] r s r test_find_all_by_attribute_dictz6TestFindAllByAttribute.test_find_all_by_attribute_dict s} yy
DMMwM 7:Q9RS DMM0AM B]OT DMM(0CM D~FVWr c | j d }| j |j dd dg | j |j dd dg | j |j dd dg | j |j dd dg | j |j d
ddg | j |j dd dg | j |j dd dg y ) Nz
Class 1.
Class 2.
Class 1.
Class 3 and 4.
r, rX class_zClass 1.r rY zClass 3 and 4.r[ r| r] r s r test_find_all_by_classz-TestFindAllByAttribute.test_find_all_by_class s yy
DMM#cM :ZLI DMM#cM :=MFound itgaror Found itr, zo b)r r) rQ rR r^ )r r fs r 0test_find_by_class_when_multiple_classes_presentzGTestFindAllByAttribute.test_find_by_class_when_multiple_classes_present s yy>? MM%
3M 8 A
|, MM%
3M 8 A
|,
MM%
5(9M : A
|,r c ( | j d }| j |j dt j d dg d }| j |j d| g d }| j |j d| dg y )NzFound itr, bar c t | dkD S NrU r( values r big_attribute_valueznTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..big_attribute_value3 s u:> !r c t | dk S r r r s r small_attribute_valuezpTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class..small_attribute_value8 s u:? "r r r^ r) rQ rR )r r r r s r :test_find_all_with_non_dictionary_for_attrs_finds_by_classzQTestFindAllByAttribute.test_find_all_with_non_dictionary_for_attrs_finds_by_class. s yy67 DMM#rzz$/? @:,O "
DMM#/B CRH #
DMM#/D E
|Tr c H | j d }|j d \ }}||gsJ |j dd |gsJ |j dd |g|j dd k( sJ |g|j dd k( sJ g |j dd k( sJ y )Nz*r, ra rN zfoo barr zbar foor r) )r r r, a2s r :test_find_all_with_string_for_attrs_finds_multiple_classeszQTestFindAllByAttribute.test_find_all_with_string_for_attrs_finds_multiple_classes= s yyEF
c"22w1
c511ws-DMM#u--s sdmmC m::::sdmmC3333T]]3 2222r c | j d }t ddi }| j |j | dg y )Nzi
Match.
Non-match.r r r| zMatch.)r r
r^ r) )r r strainers r 'test_find_all_by_attribute_soupstrainerz>TestFindAllByAttribute.test_find_all_by_attribute_soupstrainerI s@ yy : ; tWo6 DMM( 3hZ@r c n | j d }| j |j dd dg y )NID present.
No ID present.
ID is empty.r, r zNo ID present.r] r s r $test_find_all_with_missing_attributez;TestFindAllByAttribute.test_find_all_with_missing_attributeQ s: yy 9 :
DMM#$M 7:J9KLr c n | j d }| j |j d ddg y )Nr Tr z ID present.z ID is empty.r] r s r $test_find_all_with_defined_attributez;TestFindAllByAttribute.test_find_all_with_defined_attributeY s9 yy 9 :
DMMTM 2]N4STr c | j d }ddg}| j |j d | | j |j d | y )Nz[Unquoted attribute.
Quoted attribute.zUnquoted attribute.zQuoted attribute.r r rX r] )r r expecteds r $test_find_all_with_numeric_attributez;TestFindAllByAttribute.test_find_all_with_numeric_attributea sY yy ? @ *+>? DMMQM /: DMMSM 181
2
3
No ID.)rX rY r[ r rX rY r] r s r (test_find_all_with_list_attribute_valuesz?TestFindAllByAttribute.test_find_all_with_list_attribute_valuesj sU yy - .
DMM_M =SzJ
DMMRM 0"5r c | j d }| j |j t j d ddg y )NzOne a.
Two as.
Mixed as and bs.
One b.
No ID.z^a+$r zOne a.zTwo as.r r s r 5test_find_all_with_regular_expression_attribute_valuezLTestFindAllByAttribute.test_find_all_with_regular_expression_attribute_valuev sB yy - .
DMMRZZ-?M @8YBWXr c | j d }|j }|g|j dd k( sJ g |j dd k( sJ y )Nz foobarfoor, ra r rN r r, r) r r r, s r 'test_find_by_name_and_containing_stringz>TestFindAllByAttribute.test_find_by_name_and_containing_string sR yy9: FFsdmmCm6666T]]3u]5555r c t | j d }|j d |j dd k( sJ y )Nz"foofoor, ra r r r s r =test_find_by_name_and_containing_string_when_string_is_buriedzTTestFindAllByAttribute.test_find_by_name_and_containing_string_when_string_is_buried s5 yy=>}}S!T]]3u]%EEEEr c | j d }|j }|g|j dd k( sJ g |j dd k( sJ y )Nz"foofoor& ra )r r r rN r r s r ,test_find_by_attribute_and_containing_stringzCTestFindAllByAttribute.test_find_by_attribute_and_containing_string sR yy=> FFsdmmqm7777T]]a]6666r N)rB rC rD r r r r r r r r r r r r r r r r r@ r r r r s\ WHX$I( -
U
3AMU=
6
Y6F7r r c e Zd ZdZd Zy)
TestSmoothzTest Tag.smooth.c | j d }|j }|j d |j d |j t d |j t d |j d | j }t ||d }|j d |j d |j | d |j j u sJ d
t |j k( sJ |j d t |j k( sJ d |j d
k( sJ d|j j k( sJ d|j d k( sJ d|j d k( sJ y )Nz a
r r z Comment 1z Comment 2dspanrX r abcr 12r r& ) r r/ rg r default_builderr r r r( contentssmooth)r r r/ builderr s r test_smoothzTestSmooth.test_smooth sI yy (hh
3
3
7;'(
7;'(
3&&(4&) C C
4 sxx&&&C %%%%
C %%%%
Q''' sxx&&&
cll1o---cll1o---r N)rB rC rD rE r r@ r r r r s
&.r r c e Zd ZdZd Zy) TestIndexzTest Tag.indexc " | j d }|j }t |j D ] \ }}||j | k( rJ t j t 5 |j d d d d y # 1 sw Y y xY w)Nah
Identical
Not identical
Identical
Identical with child
Also not identical
Identical with child
r )r r/ enumerater indexpytestraises
ValueError)r r r/ ielements r
test_indexzTestIndex.test_index s yy & ' hh#CLL1 +JAw '** ** +
]]:
& JJqM s *BBN)rB rC rD rE r r@ r r r r s
r r c N e Zd ZdZd
dZd Zd Zd Zd Zd Z d Z
d
Z d Z d Z
y)TestParentOperationsz;Test navigation and searching through an element's parents.Nc f | j d | _ | j j | _ y )Na<
)r r r startr s r r z!TestParentOperations.setup_method s( II + , YY[[
r c | j j d dk( sJ | j j j d dk( sJ | j j j j d dk( sJ y )Nr bottommiddletop)r parentr s r test_parentz TestParentOperations.test_parent sk zz &(222zz ''-999zz ''..t4===r c l | j j d }|j | j k( sJ y )Nr )r r r )r top_tags r %test_parent_of_top_tag_is_soup_objectz:TestParentOperations.test_parent_of_top_tag_is_soup_object s, ))$$Q'~~***r c 8 d | j j u sJ y N)r r r s r test_soup_object_has_no_parentz3TestParentOperations.test_soup_object_has_no_parent s tyy'''''r c | j | j j d g d | j | j j dd dg | j j d g k( sJ y )Nulr r r r r r )assert_selects_idsr find_parentsr s r test_find_parentsz&TestParentOperations.test_find_parents sm JJ # #D )+F
7 7 7 JXJWzz&&'&2b888r c | j j dd d dk( sJ | j j d J y )Nr r r r r )r find_parentr s r test_find_parentz%TestParentOperations.test_find_parent sG zz%%du%5d;uDDDzz%%%1999r c r | j j d }|j j dk( sJ y )N
Start herer r )r r r r2 r r" s r test_parent_of_text_elementz0TestParentOperations.test_parent_of_text_element s. yy~~\~2{{ 3&&&r c n | j j d }|j d d dk( sJ y )Nr# r r r r )r r r r$ s r test_text_element_find_parentz2TestParentOperations.test_text_element_find_parent s4 yy~~\~2 %d+x777r c | j j D cg c] }|d|j v r|d }}|g dk( sJ y c c}w )Nr r )r parentsr} )r r r) s r test_parent_generatorz*TestParentOperations.test_parent_generator sV **,,
!dfll&:
4L
5555
Ac | j j D cg c] }|d|j v r|d }}|g dk( sJ y c c}w )Nr )r r r r )r self_and_parentsr} )r r resultss r test_self_and_parent_generatorz3TestParentOperations.test_self_and_parent_generator sV **55
!dfll&:
4L
>>>>
r+ r )rB rC rD rE r r r r r r! r% r' r* r/ r@ r r r
r
s7 E!>
+(9: '86?r r
c e Zd ZddZy)
ProximityTestNc 0 | j d | _ y )NzOneTwoThreer r s r r zProximityTest.setup_method s II
P
r r rB rC rD r r@ r r r1 r1 s
r r1 c J e Zd Zd fd Zd Zd Zd Zd Zd Zd Z d Z
xZ S )
TestNextOperationsc ` t t | | j j | _ y r )superr5 r r r r r __class__s r r z TestNextOperations.setup_method s
$46YY[[
r c | j j dk( sJ | j j j d dk( sJ y )NOner r )r next_elementr s r test_nextz TestNextOperations.test_next s= zz&&%///zz&&33D9S@@@r c X | j j d }|j J y NThreer )r r r< )r lasts r test_next_of_last_item_is_nonez1TestNextOperations.test_next_of_last_item_is_none$ s( yy~~W~- (((r c 4 | j j J y r )r r< r s r test_next_of_root_is_nonez,TestNextOperations.test_next_of_root_is_none( s yy%%---r c | j | j j d ddg | j j d | j | j j d dg y )Nr Twor@ rU r )r^ r
find_all_nextr s r test_find_all_nextz%TestNextOperations.test_find_all_next, s_ DJJ 4 4S 9E7;KL
A & DJJ 4 4 4 :WIFr c | j j d d dk( sJ | j j d dk( sJ y )Nr r r r@ r )r find_nextr s r test_find_nextz!TestNextOperations.test_find_next1 sD zz##C(.#555zz##7#3w>>>r c | j j d }|j d j dk( sJ | j |j d ddg y )Nr; r r rF r@ )r r rJ r r^ rG r$ s r test_find_next_for_text_elementz2TestNextOperations.test_find_next_for_text_element5 sS yy~~U~+~~c"))U222 D . .s 3eW5EFr c | j j d }|j D cg c] }| }}|\ }}|d dk( sJ |dk( sJ |j D cg c] }| }}|dd |k( sJ |d |k( sJ y c c}w c c}w )NrF r r rY r@ r r )r r
next_elementsself_and_next_elements)r r node
successorsr r successors2s r test_next_generatorsz'TestNextOperations.test_next_generators: s e,',':':;td;
;"
X4yC 7"""(-(D(DEtE E12*,,,1~&&& < Fs A: A?r ) rB rC rD r r= rB rD rH rK rM rT
__classcell__r9 s @r r5 r5 s. !A).G
?G
'r r5 c J e Zd Zd fd Zd Zd Zd Zd Zd Zd Z d Z
xZ S )
TestPreviousOperationsc l t t | | j j d | _ y r? )r7 rX r r r endr8 s r r z#TestPreviousOperations.setup_methodH s'
$d8:99>>>1r c | j j d dk( sJ | j j j dk( sJ y )Nr rY rF )rZ previous_elementr s r
test_previousz$TestPreviousOperations.test_previousL s= xx((.#555xx((99UBBBr c V | j j d }|j J y )Nhtml)r r r\ )r r s r #test_previous_of_first_item_is_nonez:TestPreviousOperations.test_previous_of_first_item_is_noneP s& v&%%---r c 4 | j j J y r )r r\ r s r test_previous_of_root_is_nonez4TestPreviousOperations.test_previous_of_root_is_noneT s yy))111r c | j | j j d g d | j | j j d dg y )Nr r@ rF r; r r r; )r^ rZ find_all_previousr s r test_find_all_previousz-TestPreviousOperations.test_find_all_previousX sH
DHH 6 6s ;=TU DHH 6 6! 6