U n5{iMaã@s¬dZddlmZddlmZddlmZmZddlm Z m Z m Z ddl m Z mZmZGdd„deƒZd d d d d œdd„ZGdd„deƒZGdd„deƒZGdd„deƒZdS)z( Tests for L{twisted.web.http_headers}. é)Ú annotations)ÚSequence)ÚSynchronousTestCaseÚTestCase)ÚHeadersÚInvalidHeaderNameÚ _nameEncoder)ÚbytesLinearWhitespaceComponentsÚsanitizedBytesÚtextLinearWhitespaceComponentsc@s:eZdZdZddœdd„Zddœdd„Zddœdd „Zd S) ÚNameEncoderTestsz7 Test L{twisted.web.http_headers._NameEncoder} ÚNone©ÚreturncCsÊ| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d ¡d ¡| t d ¡d ¡| t d ¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡dS)zk L{_NameEncoder.encode} returns the canonical capitalization for the given header. ótestóTests test-stuffó Test-Stuffs content-md5ó Content-MD5sdntóDNTsetagóETagsp3póP3PsteóTEówww-authenticateóWWW-AuthenticatesWWW-authenticatesWww-Authenticatesx-xss-protectionóX-XSS-ProtectionN©Ú assertEqualrÚencode©Úself©r úF/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/web/test/test_http_headers.pyÚtest_encodeNamesz NameEncoderTests.test_encodeNamecCsÊ| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d ¡d ¡| t d ¡d ¡| t d ¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡dS)zz L{_NameEncoder.encode} returns the canonical capitalization for a header name given as a L{str}. Útestrz test-stuffrú content-md5rZdntrÚetagrZp3prÚterúwww-authenticaterzWWW-authenticatezWww-Authenticatezx-xss-protectionrNrrr r r!Útest_encodeNameStr+sz#NameEncoderTests.test_encodeNameStrcCsDtƒ}ttjdƒD]}| d|›d¡q| ttjƒtj¡dS)zH Only a limited number of HTTP header names get cached. éÈzhello-ÚvalueN)rÚrangerZ_MAX_CACHED_HEADERSÚ addRawHeaderrÚlenZ_canonicalHeaderCache)rÚheadersÚir r r!Útest_maxCachedHeaders<sÿz&NameEncoderTests.test_maxCachedHeadersN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r"r(r0r r r r!r sr rzSequence[bytes] | Sequence[str]Úbytesr )ÚtestCaseÚ componentsÚexpectedrc Csžd}|D]}g}| t||giƒ¡tƒ}| ||¡| |¡tƒ}| ||g¡| |¡|D]4}| t| ¡ƒ||gfg¡| | |¡|g¡qbqdS)aÞ Assert that the components are sanitized to the expected value as both a header 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 values; see C{textLinearWhitespaceComponents} and C{bytesLinearWhitespaceComponents} @param expected: The expected sanitized form of the component as a header value. sNameN)Úappendrr,Ú setRawHeadersrÚlistÚgetAllRawHeadersÚ getRawHeaders) r6r7r8ÚnameÚ componentr.ÚaddedZ setHeaderÚheaderr r r!ÚassertSanitizedHs   rBc@seZdZdZddœdd„Zddœdd„Zddœdd „Zddœd d „Zddœd d „Zddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Zddœdd„Zddœdd„Zddœdd„Zddœd d!„Zddœd"d#„Zddœd$d%„Zddœd&d'„Zddœd(d)„Zd*S)+ÚBytesHeadersTestszE Tests for L{Headers}, using L{bytes} arguments for methods. r rcCst|ttƒdS©zf Linear whitespace in header names or values is replaced with a single space. N)rBr r rr r r!Útest_sanitizeLinearWhitespacepsz/BytesHeadersTests.test_sanitizeLinearWhitespacecCs&tddgiƒ}| | d¡dg¡dS)zx The header values passed to L{Headers.__init__} can be retrieved via L{Headers.getRawHeaders}. sFooóbarófooN©rrr=©rÚhr r r!Útest_initializerwsz"BytesHeadersTests.test_initializercCsPddg}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. óvalue1óvalue2rrN©rr:Ú assertTrueÚ hasHeaderrr=)rÚrawValuerJr r r!Útest_setRawHeaderss  z$BytesHeadersTests.test_setRawHeaderscCsLtƒ}| dd¡| | d¡dg¡| dd¡| | d¡ddg¡dS)úN L{Headers.addRawHeader} adds a new value for a given header. rólemurópandaN©rr,rr=rIr r r!Útest_addRawHeader‹s   z#BytesHeadersTests.test_addRawHeadercCs| tƒ d¡¡dS)úz L{Headers.getRawHeaders} returns L{None} if the header is not found and no default is specified. rN©Z assertIsNonerr=rr r r!Útest_getRawHeadersNoDefault•sz-BytesHeadersTests.test_getRawHeadersNoDefaultcCs$tƒ}tƒ}| | d|¡|¡dS)úo L{Headers.getRawHeaders} returns the specified default value when no header is found. rN)rÚobjectÚassertIdenticalr=©rrJÚdefaultr r r!Útest_getRawHeadersDefaultValueœsz0BytesHeadersTests.test_getRawHeadersDefaultValuecCsJtƒ}dg}| d|¡| | d|¡dt¡| | d|¡dg¡dS)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. r*ókeyróvalueN)rr:ÚassertIsInstancer=r5rr^r r r!Ú*test_getRawHeadersWithDefaultMatchingValue¥s  zïsz:BytesHeadersTests.test_getAllRawHeaders..©r)rx©r)rwN©rr:r<r©rrJZ allHeadersr r r!Útest_getAllRawHeadersåsþz'BytesHeadersTests.test_getAllRawHeaderscCsftƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡dS)ú A L{Headers} instance compares equal to itself and to another L{Headers} instance with the same values. rGrUrTN©rr:rÚassertNotEqual)rÚfirstÚsecondÚthirdr r r!Útest_headersComparisonös  z(BytesHeadersTests.test_headersComparisoncCs0tƒ}| |d¡| |tƒ¡| |d¡dS)úf An instance of L{Headers} does not compare equal to other unrelated objects. r rGN©rr‡r\rIr r r!Útest_otherComparisons z&BytesHeadersTests.test_otherComparisonc CsDd}d}d}| tt|||giƒƒd| ¡›d|›d|›d¡dS) zy The L{repr} of a L{Headers} instance shows the names and values of all the headers it contains. rGrFóbazú Headers({ú: [ú, ú]})N©rÚreprrÚ capitalize©rÚfooÚbarÚbazr r r!Ú test_reprsþzBytesHeadersTests.test_reprc CsDd}d}d}| tt|||giƒƒd| ¡›d|›d|›d¡dS) 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. rGsbarásbazárr‘r’r“Nr”r—r r r!Útest_reprWithRawBytessþz'BytesHeadersTests.test_reprWithRawBytesc CsTd}d}d}Gdd„dtƒ}| t||||giƒƒd| ¡›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"}. rGrFrc@s eZdZdS)z9BytesHeadersTests.test_subclassRepr..FunnyHeadersN©r1r2r3r r r r!Ú FunnyHeaders6srŸzFunnyHeaders({r‘r’r“N)rrr•r–)rr˜r™ršrŸr r r!Útest_subclassRepr-sþz#BytesHeadersTests.test_subclassReprcCsvtƒ}| ddg¡| ¡}| | d¡dg¡| dd¡| | d¡dg¡| dd¡| | d¡ddg¡dS)ú¶ L{Headers.copy} creates a new independent copy of an existing L{Headers} instance, allowing future modifications without impacts between the copies. rrGrFrN©rr:Úcopyrr=r,©rrJr/r r r!Ú test_copy>s  zBytesHeadersTests.test_copyN)r1r2r3r4rErKrRrWrZr`rdrgrjrnrrrur„r‹rŽr›rœr r¥r r r r!rCks(         rCc@sDeZdZdZddœdd„Zddœdd„Zddœdd „Zddœd d „Zddœd d „Zddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Zddœdd„Zddœdd„Zddœdd„Zddœd d!„Zddœd"d#„Zddœd$d%„Zddœd&d'„Zddœd(d)„Zddœd*d+„Zddœd,d-„Zddœd.d/„Zd0S)1ÚUnicodeHeadersTestszC Tests for L{Headers}, using L{str} arguments for methods. r rcCst|ttƒdSrD)rBr r rr r r!rESsz1UnicodeHeadersTests.test_sanitizeLinearWhitespacecCs:tddgiƒ}| | d¡dg¡| | d¡dg¡dS)aC The header values passed to L{Headers.__init__} can be retrieved via L{Headers.getRawHeaders}. If a L{bytes} argument is given, it returns L{bytes} values, and if a L{str} argument is given, it returns L{str} values. Both are the same header value, just encoded or decoded. ZFoor™rGrFr˜NrHrIr r r!rKZsz$UnicodeHeadersTests.test_initializercCsŠddg}ddg}tƒ}| d|¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡| | d¡|¡| | d¡|¡d S) zˆ L{Headers.setRawHeaders} sets the header values for the given header name to the sequence of strings, encoded. Zvalue1Zvalue2rLrMr#rrÚTestNrN)rrQZrawEncodedValuerJr r r!rRfs z&UnicodeHeadersTests.test_setRawHeadersc CsNtƒ}| t¡| ddg¡W5QRX| t¡| d¡W5QRXdS)zÉ Passing L{str} to any function that takes a header name will encode said header name as ISO-8859-1, and if it cannot be encoded, it will raise a L{UnicodeDecodeError}. õ☃ÚvalN)rÚ assertRaisesÚUnicodeEncodeErrorr:rPrIr r r!Útest_nameNotEncodablevs   z)UnicodeHeadersTests.test_nameNotEncodablecCsntƒ}| t|jddg¡| t|jddg¡| t|jddg¡| t|jddg¡| t|jddg¡dS)z„ HTTP header names must be tokens, so any names containing non-token characters raises L{InvalidHeaderName} óásvalõása bzc dzc dN)rrªrr:rIr r r!Útest_nameNotToken…s z%UnicodeHeadersTests.test_nameNotTokencCsLtƒ}| ddg¡| | d¡¡| | d¡dg¡| | d¡¡dS)z} Passing L{str} to any function that takes a header name will encode said header name as ISO-8859-1. r™rGrFNrNrIr r r!Útest_nameEncoding•s z%UnicodeHeadersTests.test_nameEncodingcCs@tƒ}| dddg¡| | d¡¡| | d¡ddg¡dS)z| Passing L{str} to L{Headers.setRawHeaders} will encode the name as ISO-8859-1 and values as UTF-8. Úxr¨rGóxs☃NrNrIr r r!Útest_rawHeadersValueEncoding¦sz0UnicodeHeadersTests.test_rawHeadersValueEncodingcCsbtƒ}| dd¡| | d¡dg¡| dd¡| | d¡ddg¡| | d¡ddg¡dS)rSr#ÚlemurÚpandarrTrUNrVrIr r r!rW°s   z%UnicodeHeadersTests.test_addRawHeadercCs| tƒ d¡¡dS)rXr#NrYrr r r!rZ»sz/UnicodeHeadersTests.test_getRawHeadersNoDefaultcCshtƒ}tƒ}| | d|¡|¡| | dd¡d¡| | ddg¡dg¡| | ddg¡dg¡dS)r[r#Nr¨)rr\r]r=rr^r r r!r`Âs þz2UnicodeHeadersTests.test_getRawHeadersDefaultValuecCsJtƒ}dg}| d|¡| | d|¡dt¡| | d|¡dg¡dS)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. rbraÚkeyrr*N)rr:rcr=Ústrrr^r r r!rdÑs  z>UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValuecCshtƒ}| ddg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡dS)rer#r´r§rrTrNrfrIr r r!rgÝs z&UnicodeHeadersTests.test_getRawHeaderscCsXtƒ}| ddg¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)rhr#r´r§rrNrirIr r r!rjés z&UnicodeHeadersTests.test_hasHeaderTruecCs| tƒ d¡¡dS)rkr#Nrlrr r r!rnõsz'UnicodeHeadersTests.test_hasHeaderFalsecCsštƒ}| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡dS) ror˜r´rGr™rµÚBarrFNrprIr r r!rrüs  z%UnicodeHeadersTests.test_removeHeadercCs(tƒ}| d¡| t| ¡ƒg¡dS)rsr#NrtrIr r r!rus z0UnicodeHeadersTests.test_removeHeaderDoesntExistcCsXtƒ}| ddg¡| ddg¡| ddg¡dd„| ¡Dƒ}| |d d d h¡d S) rvr#Zlemursr'zbasic aksljdlk=r$Z kjdfdfgdfgnsdcSsh|]\}}|t|ƒf’qSr ryr{r r r!r"sz.r€)r)s kjdfdfgdfgnsdrNr‚rƒr r r!r„sýþz)UnicodeHeadersTests.test_getAllRawHeaderscCsÈtƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡tƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡dS)r…r˜rµr´rGrUrTNr†)rrˆr‰rŠZ firstBytesZ secondBytesZ thirdBytesr r r!r‹-s$     z*UnicodeHeadersTests.test_headersComparisoncCs0tƒ}| |d¡| |tƒ¡| |d¡dS)rŒr r˜NrrIr r r!rŽIs z(UnicodeHeadersTests.test_otherComparisonc CsTd}d}d}d}d}d|}d|}| tt|||giƒƒd ||| d¡¡¡d 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. r˜õbar☃ršz'Foo'z'bar\xe2\x98\x83'ÚbzHeaders({{{}: [{}, {!r}]}})Úutf8N)rr•rÚformatr)rr˜r™ršÚ fooEncodedÚ barEncodedr r r!r›SsÿþzUnicodeHeadersTests.test_reprc CsTd}d}d}d}d}Gdd„dtƒ}| t||||giƒƒd||| d ¡f¡d S) rr˜r¹ršzb'Foo'zb'bar\xe2\x98\x83'c@s eZdZdS)z;UnicodeHeadersTests.test_subclassRepr..FunnyHeadersNržr r r r!rŸrsrŸzFunnyHeaders({%s: [%s, %r]})r»N)rrr•r)rr˜r™ršr½r¾rŸr r r!r gsÿþz%UnicodeHeadersTests.test_subclassReprcCs´tƒ}| ddg¡| ¡}| | d¡dg¡| | d¡dg¡| dd¡| | d¡dg¡| | d¡dg¡| dd¡| | d¡ddg¡| | d¡ddg¡d S) r¡r#ufoo☃rsfoo☃r™r§rrFNr¢r¤r r r!r¥{s  zUnicodeHeadersTests.test_copyN)r1r2r3r4rErKrRr¬r¯r°r³rWrZr`rdrgrjrnrrrur„r‹rŽr›r r¥r r r r!r¦Ns.        r¦c@s,eZdZdZddœdd„Zddœdd„ZdS) ÚMixedHeadersTestszm Tests for L{Headers}, mixing L{bytes} and L{str} arguments for methods where that is permitted. r rcCsJtƒ}| dd¡| dd¡| | d¡dg¡| | d¡dg¡dS)zL L{Headers.addRawHeader} accepts mixed L{str} and L{bytes}. óbytesr·óBytesóstrÚStrr5NrVrIr r r!rWžs   z#MixedHeadersTests.test_addRawHeadercCsštƒ}| ddg¡| ddg¡| dddg¡| dddg¡| | d¡dg¡| | d¡dg¡| | d¡ddg¡| | d ¡d dg¡d S) zM L{Headers.setRawHeaders} accepts mixed L{str} and L{bytes}. rÀr·z mixed-strs mixed-bytesrÁrÃz Mixed-Strr5s Mixed-BytesrÂNrfrIr r r!rR©sz$MixedHeadersTests.test_setRawHeadersN)r1r2r3r4rWrRr r r r!r¿˜s r¿N)r4Ú __future__rÚtypingrZtwisted.trial.unittestrrZtwisted.web.http_headersrrrZtwisted.web.test.requesthelperr r r r rBrCr¦r¿r r r r!Ús  3#dL