tfZdZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddZ GddeZGddeZGddeZdS)z( Tests for L{twisted.web.http_headers}. ) annotations)Sequence)TestCase)Headers)bytesLinearWhitespaceComponentssanitizedBytestextLinearWhitespaceComponentstestCaser componentsSequence[bytes] | Sequence[str]expectedbytesreturnNonec|D]}g}|t||git}|||||t}|||g|||D]e}|t |||gfg||||gfdS)a  Assert that the components are sanitized to the expected value as both a header name and value, across all of L{Header}'s setters and getters. @param testCase: A test case. @param components: A sequence of values that contain linear whitespace to use as header names and values; see C{textLinearWhitespaceComponents} and C{bytesLinearWhitespaceComponents} @param expected: The expected sanitized form of the component for both headers names and their values. N)appendr addRawHeader setRawHeaders assertEquallistgetAllRawHeaders getRawHeaders)r r r componentheadersadded setHeaderheaders a/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/web/test/test_http_headers.pyassertSanitizedrs($ MM w I;788999  9i000uII  I;777y!!! M MF  V,,..//8hZ2H1I     !5!5h!?!?( L L L L  MMMceZdZdZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z ddZddZddZddZddZddZddZddZddZddZddZdS)BytesHeadersTestszE Tests for L{Headers}, using L{bytes} arguments for methods. rrc<t|ttdSzf Linear whitespace in header names or values is replaced with a single space. N)rrrselfs rtest_sanitizeLinearWhitespacez/BytesHeadersTests.test_sanitizeLinearWhitespace>s =~NNNNNr c~tddgi}||ddgdS)zx The header values passed to L{Headers.__init__} can be retrieved via L{Headers.getRawHeaders}. sFoobarfooNrrrr&hs rtest_initializerz"BytesHeadersTests.test_initializerEsB VfX& ' ' 006(;;;;;r cHddg}t}|d|||d||d||d|dS)z L{Headers.setRawHeaders} sets the header values for the given header name to the sequence of byte string values. value1value2testTestNrr assertTrue hasHeaderrr)r&rawValuer-s rtest_setRawHeadersz$BytesHeadersTests.test_setRawHeadersMs y) II ***  G,,---  G,,--- 118<<<<>> w))) 11Hh3GHHHHHr cn|tddS)z L{Headers.getRawHeaders} returns L{None} if the header is not found and no default is specified. r2N assertIsNonerrr%s rtest_getRawHeadersNoDefaultz-BytesHeadersTests.test_getRawHeadersNoDefaultcs0 '))11'::;;;;;r ct}t}||d||dS)o L{Headers.getRawHeaders} returns the specified default value when no header is found. r2N)robjectassertIdenticalrr&r-defaults rtest_getRawHeadersDefaultValuez0BytesHeadersTests.test_getRawHeadersDefaultValuejs@ II(( Q__Wg>>HHHHHr ct}dg}|d|||d|dt||d|dgdS)z If the object passed as the value list to L{Headers.setRawHeaders} is later passed as a default to L{Headers.getRawHeaders}, the result nevertheless contains encoded values. valuekeyrvalueN)rrassertIsInstancerrrrHs r*test_getRawHeadersWithDefaultMatchingValuez>qA5III 99H:FFFFFr ct}|ddg||ddg||ddgdS)m L{Headers.getRawHeaders} returns the values which have been set for a given header. r2r;r3Nrrrrr,s rtest_getRawHeadersz$BytesHeadersTests.test_getRawHeaderssq II (,,, 11H:>>> 11H:>>>>>r ct}|ddg||d||ddS)i Check that L{Headers.hasHeader} returns C{True} when the given header is found. r2r;r3Nrrr5r6r,s rtest_hasHeaderTruez$BytesHeadersTests.test_hasHeaderTruese II (,,,  G,,---  G,,-----r cn|tddS)c L{Headers.hasHeader} returns C{False} when the given header is not found. r2N assertFalserr6r%s rtest_hasHeaderFalsez%BytesHeadersTests.test_hasHeaderFalses0 ,,W5566666r ct}|ddg||d|d||d|ddg||d|d||ddS)N Check that L{Headers.removeHeader} removes the given header. r*r;r)r<sBarNrrr5r6 removeHeaderr\r,s rtest_removeHeaderz#BytesHeadersTests.test_removeHeaders II  +++  F++,,, v V,,---  +++  F++,,, v V,,-----r ct}|d|t|gdS)k L{Headers.removeHeader} is a no-operation when the specified header is not found. r2Nrrarrrr,s rtest_removeHeaderDoesntExistz.BytesHeadersTests.test_removeHeaderDoesntExistsO II w a002233R88888r ct}||dd||dd||dd||dd||d d ||d d ||d d||dd||dd||dd||dddS)zk L{Headers._encodeName} returns the canonical capitalization for the given header. r2r3s test-stuffs Test-Stuffs content-md5 Content-MD5sdntsDNTsetagsETagsp3psP3PstesTEwww-authenticateWWW-AuthenticatesWWW-authenticatesWww-Authenticatesx-xss-protectionsX-XSS-ProtectionN)rr _encodeNamer,s rtest_encodeNamez!BytesHeadersTests.test_encodeNames II w//999 }55}EEE ~66GGG v..777 w//999 v..777 u--u555 ':;;=PQQQ ':;;=PQQQ ':;;=PQQQ ':;;=PQQQQQr ct}|ddg|ddgd|D}||ddhdS) L{Headers.getAllRawHeaders} returns an iterable of (k, v) pairs, where C{k} is the canonicalized representation of the header name, and C{v} is a sequence of values. r2lemursribasic aksljdlk=c6h|]\}}|t|fStuple.0kvs r z:BytesHeadersTests.test_getAllRawHeaders..&EEE1q%((mEEEr rj)rp)r3roNrrrrr&r- allHeaderss rtest_getAllRawHeadersz'BytesHeadersTests.test_getAllRawHeaderss II )--- +.@-ABBBEE0B0B0D0DEEE    9;R S     r cjt}|ddgt}|ddgt}|dddg|||||||||dS) A L{Headers} instance compares equal to itself and to another L{Headers} instance with the same values. r*r<r;NrrrassertNotEqual)r&firstsecondthirds rtest_headersComparisonz(BytesHeadersTests.test_headersComparisons   FXJ///VhZ000  FXx$8999 &&& ''' E5)))))r ct}||d||t||ddS)f An instance of L{Headers} does not compare equal to other unrelated objects. rrr*NrrrFr,s rtest_otherComparisonz&BytesHeadersTests.test_otherComparisonsY II Ar""" Avxx((( Av&&&&&r c d}d}d}|tt|||gid|d|d|ddS) zy The L{repr} of a L{Headers} instance shows the names and values of all the headers it contains. r*r)baz Headers({: [, ]})Nrreprr capitalizer&foobarbazs r test_reprzBytesHeadersTests.test_reprsy   #Sz*++ , , D)) D D D D D D D     r c d}d}d}|tt|||gid|d|d|ddS) z The L{repr} of a L{Headers} instance shows the names and values of all the headers it contains, not attempting to decode any raw bytes. r*sbarsbazrrrrNrrs rtest_reprWithRawBytesz'BytesHeadersTests.test_reprWithRawBytess{  #Sz*++ , , D)) D D D D D D D     r c d}d}d}Gddt}|t||||gid|d|d|d d S) The L{repr} of an instance of a subclass of L{Headers} uses the name of the subclass instead of the string C{"Headers"}. r*r)rceZdZdS)9BytesHeadersTests.test_subclassRepr..FunnyHeadersN__name__ __module__ __qualname__rrr r FunnyHeadersr Dr rFunnyHeaders({rrrN)rrrr)r&rrrrs rtest_subclassReprz#BytesHeadersTests.test_subclassRepr s      7     sS#J/00 1 1 Icnn.. I IS I Ic I I I     r ct}|ddg|}||ddg|dd||ddg|dd||dddgdS) L{Headers.copy} creates a new independent copy of an existing L{Headers} instance, allowing future modifications without impacts between the copies. r2r*r)rNrrcopyrrrr&r-is r test_copyzBytesHeadersTests.test_copys II &*** FFHH 11F8<<< w''' 11F8<<< w''' 11FF3CDDDDDr ct}ttjdzD]}|d|d|t tjtjdS)zH Only a limited number of HTTP header names get cached. zhello-rLN)rrange_MAX_CACHED_HEADERSrrlen_canonicalHeaderCache)r&rrs rtest_max_cached_headersz)BytesHeadersTests.test_max_cached_headers-s))w2S899 8 8A  !w 7 7 7 7  - . .0K     r Nrr)rrr__doc__r'r.r8r>rCrJrPrTrXr]rbrfrlrrrrrrrrrrr rr"r"9sOOOO<<<< = = = =IIII<<<<IIII G G G G????....7777.... 9999RRRR$    " * * * *''''         "    " E E E E       r r"ceZdZdZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z ddZddZddZddZddZddZddZddZddZddZddZdS)UnicodeHeadersTestszC Tests for L{Headers}, using L{str} arguments for methods. rrc<t|ttdSr$)rr rr%s rr'z1UnicodeHeadersTests.test_sanitizeLinearWhitespace>s z%UnicodeHeadersTests.test_addRawHeaders II vw''' 007)<<< vw''' 007G2DEEE 11Hh3GHHHHHr cn|tddS)r@rNrAr%s rrCz/UnicodeHeadersTests.test_getRawHeadersNoDefaults0 '))11&99:::::r ct}t}||d||||ddd||ddgdg||ddgdgdS)rErNr)rrFrGrrrHs rrJz2UnicodeHeadersTests.test_getRawHeadersDefaultValues II(( Q__VW==wGGG Q__VT::DAAA $884&AAA  OOF]O 4 4 O     r ct}dg}|d|||d|dt||d|dgdS)z If the object passed as the value list to L{Headers.setRawHeaders} is later passed as a default to L{Headers.getRawHeaders}, the result nevertheless contains decoded values. rNrMkeyrrLN)rrrOrstrrrHs rrPz>UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValues} II* ((( aooeW==a@#FFF 887)DDDDDr ct}|ddg||ddg||ddg||ddg||ddgdS)rRtestárTestátestr;TestNrSr,s rrTz&UnicodeHeadersTests.test_getRawHeaderss II  wi000 66 BBB 66 BBB 55zBBB 55zBBBBBr ct}|ddg||d||d||d||ddS)rVrrrrrNrWr,s rrXz&UnicodeHeadersTests.test_hasHeaderTrues II  wi000  L11222  L11222  K00111  K0011111r cn|tddS)rZrNr[r%s rr]z'UnicodeHeadersTests.test_hasHeaderFalses0 ,,\::;;;;;r ct}|ddg||d|d||d||d|ddg||d|d||d||ddS) r_rrr*rrBarr)Nr`r,s rrbz%UnicodeHeadersTests.test_removeHeaders II y)))  E**+++ u U++,,, V,,--- y)))  E**+++ u U++,,, V,,-----r ct}|d|t|gdS)rdrNrer,s rrfz0UnicodeHeadersTests.test_removeHeaderDoesntExistsO II v a002233R88888r ct}|ddg|ddg|ddgd|D}||hdd S) rnrlemurszwww-authenticatezbasic aksljdlk=z content-md5 kjdfdfgdfgnsdc6h|]\}}|t|fSrrrsrus rryz.rzr >rr|rh)s kjdfdfgdfgnsdr{Nr}r~s rrz)UnicodeHeadersTests.test_getAllRawHeaderss II  xj111 *->,?@@@  '8999EE0B0B0D0DEEE           r ct}|ddgt}|ddgt}|dddg|||||||||t}|ddgt}|ddgt}|dddg|||||||||dS)rfooárrsfoor<r;Nr)r&rrr firstBytes secondBytes thirdBytess rrz*UnicodeHeadersTests.test_headersComparisons\   K'333[7)444  K'7);<<< &&& ''' E5)))YY   hZ888ii !!*xj999YY   h-@AAA  +++ ---  +++++r ct}||d||t||ddS)rrrrNrr,s rrz(UnicodeHeadersTests.test_otherComparison$sY II Ar""" Avxx((( Au%%%%%r c d}d}d}d}d}d|z}d|z}|tt|||gid|||dd S) z The L{repr} of a L{Headers} instance shows the names and values of all the headers it contains. This shows only reprs of bytes values, as undecodable headers may cause an exception. rbar☃rz 'Foo\xe1'z'bar\xe2\x98\x83'bzHeaders({{{}: [{}, {!r}]}})utf8N)rrrformatencode)r&rrr fooEncoded barEncodeds rrzUnicodeHeadersTests.test_repr.s ! + :% :%   #Sz*++ , , ) 0 0J 6(:(:       r c d}d}d}d}d}Gddt}|t||||gid|d |d |d d d S)rrrrz b'Foo\xe1'zb'bar\xe2\x98\x83'ceZdZdS);UnicodeHeadersTests.test_subclassRepr..FunnyHeadersNrrrr rrrMrr rrrrrrN)rrrr)r&rrrrrrs rrz%UnicodeHeadersTests.test_subclassReprBs " ,      7     sS#J/00 1 1 1zz:::szz&'9'9'9'9 ;     r ct}|ddg|}||ddg||ddg|dd||ddg||ddg|dd||dddg||dddgdS) rrufoo☃rsfoo☃rrr)Nrrs rrzUnicodeHeadersTests.test_copyVsK II  {m444 FFHH 66 FFF 558J7KLLL |U+++ 66 FFF 558J7KLLL |V,,, 66e8LMMM 558JF7STTTTTr Nr)rrrrr'r.r8rrrr>rCrJrPrTrXr]rbrfrrrrrrrrr rrr9sNNNN : : : :DDDD " " " "////"NNNN I I I I;;;;      E E E E C C C C 2 2 2 2<<<<....$9999    ,,,,,8&&&&    (    (UUUUUUr rc"eZdZdZddZddZdS)MixedHeadersTestszm Tests for L{Headers}, mixing L{bytes} and L{str} arguments for methods where that is permitted. rrc"t}|dd|dd||ddg||ddgdS)zL L{Headers.addRawHeader} accepts mixed L{str} and L{bytes}. bytesrBytesstrStrrNr=r,s rr>z#MixedHeadersTests.test_addRawHeaderys II x''' uh''' 22VH=== //';;;;;r c2t}|ddg|ddg|dddg|dddg||ddg||ddg||dddg||d d dgd S) zM L{Headers.setRawHeaders} accepts mixed L{str} and L{bytes}. rrz mixed-strs mixed-bytesrrz Mixed-Strrs Mixed-BytesrNrSr,s rr8z$MixedHeadersTests.test_setRawHeaderss II 8*--- w'''  h%6777 (9::: 22XJ??? //%999 557GHHH 8868:LMMMMMr Nr)rrrrr>r8rrr rrrssL < < < < N N N N N Nr rN)r rr r r rrr)r __future__rtypingrtwisted.trial.unittestrtwisted.web.http_headersrtwisted.web.test.requesthelperrrr rr"rrrrr rrsP#"""""++++++,,,,,,"M"M"M"MJ} } } } } } } } @wUwUwUwUwU(wUwUwUt NNNNNNNNNNr