JfZdZddlmZddlmZddlmZddlmZddl m Z m Z m Z  ddZ Gdd eZGd d eZGd d eZy)z( Tests for L{twisted.web.http_headers}. ) annotations)Sequence)TestCase)Headers)bytesLinearWhitespaceComponentssanitizedBytestextLinearWhitespaceComponentsc|D]}g}|jt||git}|j|||j|t}|j||g|j||D]Q}|j t |j ||gfg|j |j||gSy)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)testCase componentsexpected componentheadersadded setHeaderheaders ]/var/www/cs2snipe.com/venv/lib/python3.12/site-packages/twisted/web/test/test_http_headers.pyassertSanitizedrs$ M w I;789  9i0uI  I;7y! MF  V,,./8hZ2H1I   !5!5h!?( L  MMceZdZdZddZddZddZddZddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZddZddZddZddZddZddZddZddZy)BytesHeadersTestszE Tests for L{Headers}, using L{bytes} arguments for methods. c.t|ttyzf Linear whitespace in header names or values is replaced with a single space. N)rrrselfs rtest_sanitizeLinearWhitespacez/BytesHeadersTests.test_sanitizeLinearWhitespace>s =~Nrcdtddgi}|j|jddgy)zx The header values passed to L{Headers.__init__} can be retrieved via L{Headers.getRawHeaders}. sFoobarfooNrrrr"hs rtest_initializerz"BytesHeadersTests.test_initializerEs0 VfX& ' 06(;rcddg}t}|jd||j|jd|j|jd|j |j d|y)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_setRawHeadersMsf y) I *  G,-  G,- 18 w) 1Hh3GHrcT|jtjdy)z L{Headers.getRawHeaders} returns L{None} if the header is not found and no default is specified. r.N assertIsNonerrr!s rtest_getRawHeadersNoDefaultz-BytesHeadersTests.test_getRawHeadersNoDefaultcs ')11':;rcpt}t}|j|jd||y)o L{Headers.getRawHeaders} returns the specified default value when no header is found. r.N)robjectassertIdenticalrr"r)defaults rtest_getRawHeadersDefaultValuez0BytesHeadersTests.test_getRawHeadersDefaultValuejs- I( Q__Wg>Hrct}dg}|jd||j|jd|dt|j |jd|dgy)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)rr assertIsInstancerbytesrrDs r*test_getRawHeadersWithDefaultMatchingValuezqA5I 9H:Frct}|jddg|j|jddg|j|jddgy)m L{Headers.getRawHeaders} returns the values which have been set for a given header. r.r7r/Nrr rrr(s rtest_getRawHeadersz$BytesHeadersTests.test_getRawHeaderssQ I (, 1H:> 1H:>rct}|jddg|j|jd|j|jdy)i Check that L{Headers.hasHeader} returns C{True} when the given header is found. r.r7r/Nrr r1r2r(s rtest_hasHeaderTruez$BytesHeadersTests.test_hasHeaderTruesE I (,  G,-  G,-rcT|jtjdy)c L{Headers.hasHeader} returns C{False} when the given header is not found. r.N assertFalserr2r!s rtest_hasHeaderFalsez%BytesHeadersTests.test_hasHeaderFalses ,,W56rct}|jddg|j|jd|j d|j |jd|jddg|j|jd|j d|j |jdy)N Check that L{Headers.removeHeader} removes the given header. r&r7r%r8sBarNrr r1r2 removeHeaderrYr(s rtest_removeHeaderz#BytesHeadersTests.test_removeHeaders I  +  F+, v V,-  +  F+, v V,-rct}|jd|jt|j gy)k L{Headers.removeHeader} is a no-operation when the specified header is not found. r.Nrr^rrrr(s rtest_removeHeaderDoesntExistz.BytesHeadersTests.test_removeHeaderDoesntExists5 I w a0023R8rct}|j|jdd|j|jdd|j|jdd|j|jdd|j|jd d |j|jd d |j|jd d|j|jdd|j|jdd|j|jdd|j|jddy)zk L{Headers._encodeName} returns the canonical capitalization for the given header. r.r/s test-stuffs Test-Stuffs content-md5 Content-MD5sdntsDNTsetagsETagsp3psP3PstesTEwww-authenticateWWW-AuthenticatesWWW-authenticatesWww-Authenticatesx-xss-protectionsX-XSS-ProtectionN)rr _encodeNamer(s rtest_encodeNamez!BytesHeadersTests.test_encodeNames& I w/9 }5}E ~6G v.7 w/9 v.7 u-u5 ':;=PQ ':;=PQ ':;=PQ ':;=PQrct}|jddg|jddg|jDchc]\}}|t|f}}}|j |ddhycc}}w) 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. r.lemursrfbasic aksljdlk=rg)rm)r/rlNrr rtuplerr"r)kv allHeaderss rtest_getAllRawHeadersz'BytesHeadersTests.test_getAllRawHeaderssv I )- +.@-AB010B0B0DE1q%(mE E   9;R S FsA3c t}|jddgt}|jddgt}|jdddg|j|||j|||j||y) A L{Headers} instance compares equal to itself and to another L{Headers} instance with the same values. r&r8r7Nrr rassertNotEqual)r"firstsecondthirds rtest_headersComparisonz(BytesHeadersTests.test_headersComparisons   FXJ/VhZ0  FXx$89 & ' E5)rct}|j|d|j|t|j|dy)f An instance of L{Headers} does not compare equal to other unrelated objects. r&NrrzrBr(s rtest_otherComparisonz&BytesHeadersTests.test_otherComparisons= I Ar" Avx( Av&rc d}d}d}|jtt|||gid|jd|d|dy) 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_reprsW   #Sz*+ ,),Cwbt D rc d}d}d}|jtt|||gid|jd|d|dy) 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_reprWithRawBytessY  #Sz*+ ,),Cwbt D rc d}d}d}Gddt}|jt||||gid|jd|d|d y ) 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%rc eZdZy)9BytesHeadersTests.test_subclassRepr..FunnyHeadersN__name__ __module__ __qualname__rrr FunnyHeadersr rrFunnyHeaders({rrrN)rrrr)r"rrrrs rtest_subclassReprz#BytesHeadersTests.test_subclassRepr se  7   sS#J/0 1cnn.1SG2cWD I rctt}|jddg|j}|j|j ddg|j dd|j|j ddg|j dd|j|j dddgy) L{Headers.copy} creates a new independent copy of an existing L{Headers} instance, allowing future modifications without impacts between the copies. r.r&r%rNrr copyrrr r"r)is r test_copyzBytesHeadersTests.test_copys I &* FFH 1F8< w' 1F8< w' 1FF3CDrct}ttjdzD]}|jd|d|j t tj tjy)zH Only a limited number of HTTP header names get cached. zhello-rHN)rrange_MAX_CACHED_HEADERSr rlen_canonicalHeaderCache)r"rrs rtest_max_cached_headersz)BytesHeadersTests.test_max_cached_headers-sd)w22S89 8A  6!w 7 8  -- .0K0K rNreturnNone)rrr__doc__r#r*r4r:r?rFrMrQrUrZr_rcrirvr~rrrrrrrrrrr9svO< =I<I G?.7. 9R$ " *'   " " E  rrceZdZdZddZddZddZddZddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZddZddZddZddZddZddZddZddZy)UnicodeHeadersTestszC Tests for L{Headers}, using L{str} arguments for methods. c.t|ttyr )rr rr!s rr#z1UnicodeHeadersTests.test_sanitizeLinearWhitespace>s UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValues] I* ( aooeW=a@#F 87)DrcNt}|jddg|j|jddg|j|jddg|j|jddg|j|jddgy)rOtestárTestátestr7TestNrPr(s rrQz&UnicodeHeadersTests.test_getRawHeaderss I  wi0 6 B 6 B 5zB 5zBrc>t}|jddg|j|jd|j|jd|j|jd|j|jdy)rSrrrrrNrTr(s rrUz&UnicodeHeadersTests.test_hasHeaderTrueso I  wi0  L12  L12  K01  K01rcT|jtjdy)rWrNrXr!s rrZz'UnicodeHeadersTests.test_hasHeaderFalses ,,\:;rc(t}|jddg|j|jd|j d|j |jd|j |jd|jddg|j|jd|j d|j |jd|j |jdy) r\rrr&rrBarr%Nr]r(s rr_z%UnicodeHeadersTests.test_removeHeaders I y)  E*+ u U+, V,- y)  E*+ u U+, V,-rct}|jd|jt|j gy)rarNrbr(s rrcz0UnicodeHeadersTests.test_removeHeaderDoesntExists5 I v a0023R8rct}|jddg|jddg|jddg|jDchc]\}}|t|f}}}|j |hdycc}}w) rkrlemurszwww-authenticatezbasic aksljdlk=z content-md5 kjdfdfgdfgnsd>rrore)s kjdfdfgdfgnsdrnNrprrs rrvz)UnicodeHeadersTests.test_getAllRawHeaderss I  xj1 *->,?@  '89010B0B0DE1q%(mE E    FsBc<t}|jddgt}|jddgt}|jdddg|j|||j|||j||t}|jddgt}|jddgt}|jdddg|j|||j|||j||y)rxfooárrsfoor8r7Nry)r"r{r|r} firstBytes secondBytes thirdBytess rr~z*UnicodeHeadersTests.test_headersComparisons   K'3[7)4  K'7);< & ' E5)Y   hZ8i !!*xj9Y   h-@A  + -  +rct}|j|d|j|t|j|dy)rrrNrr(s rrz(UnicodeHeadersTests.test_otherComparison$s= I Ar" Avx( Au%rc d}d}d}d}d}d|z}d|z}|jtt|||gidj|||j dy ) 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.ss ! + :% :%   #Sz*+ , ) 0 0J 6(:  rc d}d}d}d}d}Gddt}|jt||||gid|d |d |jd d y )rrrrz b'Foo\xe1'zb'bar\xe2\x98\x83'c eZdZy);UnicodeHeadersTests.test_subclassRepr..FunnyHeadersNrrrrrrMrrrrrrrrN)rrrr)r"rrrrrrs rrz%UnicodeHeadersTests.test_subclassReprBsd " ,  7   sS#J/0 1:szz&'9 ; rcBt}|jddg|j}|j|j ddg|j|j ddg|j dd|j|j ddg|j|j ddg|j dd|j|j dddg|j|j dddgy) rrufoo☃rsfoo☃rrr%Nrrs rrzUnicodeHeadersTests.test_copyVs I  {m4 FFH 6 F 58J7KL |U+ 6 F 58J7KL |V, 6e8LM 58JF7STrNr)rrrrr#r*r4rrrr:r?rFrMrQrUrZr_rcrvr~rrrrrrrrr9swN :D "/"N I;   E C 2<.$9 ,,8& ( (Urrc eZdZdZddZddZy)MixedHeadersTestszm Tests for L{Headers}, mixing L{bytes} and L{str} arguments for methods where that is permitted. ct}|jdd|jdd|j|jddg|j|jddgy)zL L{Headers.addRawHeader} accepts mixed L{str} and L{bytes}. bytesrBytesstrStrrLNr9r(s rr:z#MixedHeadersTests.test_addRawHeaderys] I x' uh' 2VH= /';rct}|jddg|jddg|jdddg|jdddg|j|jddg|j|jddg|j|jdddg|j|jd d dgy ) zM L{Headers.setRawHeaders} accepts mixed L{str} and L{bytes}. rrz mixed-strs mixed-bytesrrz Mixed-StrrLs Mixed-BytesrNrPr(s rr4z$MixedHeadersTests.test_setRawHeaderss I 8*- w'  h%67 (9: 2XJ? /%9 57GH 868:LMrNr)rrrrr:r4rrrrrss < NrrN)rrrzSequence[bytes] | Sequence[str]rrLrr)r __future__rtypingrtwisted.trial.unittestrtwisted.web.http_headersrtwisted.web.test.requesthelperrrr rrrrrrrrs{#+,"M"M$C"MOT"M "MJ} } @wU(wUt NNr