ϪfadZddlmZddlmZddlmZddlmZddl 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 D/usr/lib/python3/dist-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 ZddZddZddZddZddZddZddZddZddZddZddZddZddZy)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}. FoobarfooNrrrr"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,- 18NOrct}|jt|jddg}|j |j ddy)zf L{Headers.setRawHeaders} requires C{name} to be a L{bytes} or L{str} string. Nr'rBHeader name is an instance of , not bytes or strrr9r:r rargsr"r*es rtest_rawHeadersTypeCheckingNamez1BytesHeadersTests.test_rawHeadersTypeCheckingName`sC I   i$ I  FF1I S rct}|jt|jdddg}|j |j ddy)zn L{Headers.setRawHeaders} requires values to a L{list} of L{bytes} or L{str} strings. r7r&NrzQHeader value at position 1 is an instance of , not bytes or strr>r@s r*test_rawHeadersTypeCheckingValuesAreStringz w) 1Hh3GHrct}|jt|jdd}|j |j ddy)ze L{Headers.addRawHeader} requires C{name} to be a L{bytes} or L{str} string. Nr'rr=rr9r:r rr?r@s rtest_addRawHeaderTypeCheckNamez0BytesHeadersTests.test_addRawHeaderTypeCheckNamesA I   iv F  FF1I S rct}|jt|jdd}|j |j ddy)zc L{Headers.addRawHeader} requires value to be a L{bytes} or L{str} string. r7NrzCHeader value is an instance of , not bytes or strrLr@s rtest_addRawHeaderTypeCheckValuez1BytesHeadersTests.test_addRawHeaderTypeCheckValuesA I   i F  FF1I T rcT|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_getRawHeadersNoDefaults ')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_getRawHeadersDefaultValues- 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. valuer7rvalueN)rr assertIsInstancerbytesrrYs 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/rGr0Nrr 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/rGr0Nrr r2r3r)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 assertFalserr3r!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'rGr&rHsBarNrr r2r3 removeHeaderrmr)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/Nrrrrrrr)s rtest_removeHeaderDoesntExistz.BytesHeadersTests.test_removeHeaderDoesntExists5 I w a0023R8rcjt}|j|jdd|j|jdd|j|jdd|j|jdd|j|jd d |j|jd d |j|jd d|j|jdd|j|jddy)zr L{Headers._canonicalNameCaps} returns the canonical capitalization for the given header. r/r0s test-stuffs Test-Stuffs content-md5 Content-MD5sdntsDNTsetagsETagsp3psP3PstesTEwww-authenticateWWW-Authenticatesx-xss-protectionsX-XSS-ProtectionN)rr_canonicalNameCapsr)s rtest_canonicalNameCapsz(BytesHeadersTests.test_canonicalNameCapss I --g6@ --m --g6@ --f5v> --e4e< --.ABDWX --.ABDWXrct}|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/lemursrzbasic aksljdlk=r{)r)r0rNrr 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'rHrGNrr rassertNotEqual)r"firstsecondthirds rtest_headersComparisonz(BytesHeadersTests.test_headersComparison s   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'NrrrWr)s rtest_otherComparisonz&BytesHeadersTests.test_otherComparisons= I Ar" Avx( Av&rc td}d}d}|jtt|||gid|d|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({: [, ]})Nrreprrr"foobarbazs r test_reprzBytesHeadersTests.test_repr%sO   #Sz*+ ,s3'C7$ 7 rc td}d}d}|jtt|||gid|d|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_reprWithRawBytes2sQ  #Sz*+ ,s3'C7$ 7 rc d}d}d}Gddt}|jt||||gid|d|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 FunnyHeadersrL rrFunnyHeaders({rrrN)rrr)r"rrrrs rtest_subclassReprz#BytesHeadersTests.test_subclassReprCs]  7   sS#J/0 1cWCwbt < 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_copyTs I &* FFH 1F8< w' 1F8< w' 1FF3CDrNreturnNone)rrr__doc__r#r+r5r;rBrDrJrMrOrTr[rarerirnrsrwr}rrrrrrrrrrrr9sO< =P    I    <I G?.7. 9Y  " *'   " " ErrceZdZdZddZddZddZddZddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZddZddZddZddZddZddZddZddZddZy)UnicodeHeadersTestszC Tests for L{Headers}, using L{str} arguments for methods. c.t|ttyr )rr rr!s rr#z1UnicodeHeadersTests.test_sanitizeLinearWhitespaceis UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValues] I* ( aooeW=a@#F 87)DrcNt}|jddg|j|jddg|j|jddg|j|jddg|j|jddgy)rctestárTestátestrGTestNrdr)s rrez&UnicodeHeadersTests.test_getRawHeaderss I  wi0 6 B 6 B 5zB 5zBrc>t}|jddg|j|jd|j|jd|j|jd|j|jdy)rgrrrrrNrhr)s rriz&UnicodeHeadersTests.test_hasHeaderTrueso I  wi0  L12  L12  K01  K01rcT|jtjdy)rkrNrlr!s rrnz'UnicodeHeadersTests.test_hasHeaderFalses ,,\:;rc(t}|jddg|j|jd|j d|j |jd|j |jd|jddg|j|jd|j d|j |jd|j |jdy) rprrr'rrBarr&Nrqr)s rrsz%UnicodeHeadersTests.test_removeHeader s I y)  E*+ u U+, V,- y)  E*+ u U+, V,-rct}|jd|jt|j gy)rurNrvr)s rrwz0UnicodeHeadersTests.test_removeHeaderDoesntExists5 I v a0023R8rct}|jddg|jddg|jddg|jDchc]\}}|t|f}}}|j |hdycc}}w) rrlemurszwww-authenticatezbasic aksljdlk=z content-md5 kjdfdfgdfgnsd>rrry)s kjdfdfgdfgnsdrNrrs rrz)UnicodeHeadersTests.test_getAllRawHeaders$s I  xj1 *->,?@  '89010B0B0DE1q%(mE E    FsBc<t}|jddgt}|jddgt}|jdddg|j|||j|||j||t}|jddgt}|jddgt}|jdddg|j|||j|||j||y)rfooárrsfoorHrGNr)r"rrr firstBytes secondBytes thirdBytess rrz*UnicodeHeadersTests.test_headersComparison:s   K'3[7)4  K'7);< & ' E5)Y   hZ8i !!*xj9Y   h-@A  + -  +rct}|j|d|j|t|j|dy)rrrNrr)s rrz(UnicodeHeadersTests.test_otherComparisonVs= 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..FunnyHeadersNrrrrrrrrrrrrrrN)rrrr)r"rrrrrrs rrz%UnicodeHeadersTests.test_subclassReprtsd " ,  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_copys I  {m4 FFH 6 F 58J7KL |U+ 6 F 58J7KL |V, 6e8LM 58JF7STrNr)rrrrr#r+r5rrrrrJrTr[rarerirnrsrwrrrrrrrrrrrds}N :D "/"NM 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}. bytesrBytesstrStrr`NrIr)s rrJz#MixedHeadersTests.test_addRawHeaders] 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-Strr`s Mixed-BytesrNrdr)s rr5z$MixedHeadersTests.test_setRawHeaderss I 8*- w'  h%67 (9: 2XJ? /%9 57GH 868:LMrNr)rrrrrJr5rrrrrs < NrrN)rrrzSequence[bytes] | Sequence[str]rr`rr)r __future__rtypingrtwisted.trial.unittestrtwisted.web.http_headersrtwisted.web.test.requesthelperrrr rrrrrrrr s}#+,"M"M$C"MOT"M "MJhEhEV ~U(~UB NNr