o Óó·iMaã@s¢dZddlmZddlmZddlmZmZddlm Z m Z m Z ddl m Z mZmZGdd„deƒZddd„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.eZdZdZd dd„Zd dd„Zd dd „Zd S) ÚNameEncoderTestsz7 Test L{twisted.web.http_headers._NameEncoder} ÚreturnÚNonecCóÊ| 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 úhC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/web/test/test_http_headers.pyÚtest_encodeNameóz NameEncoderTests.test_encodeNamecCr)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+r#z#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©r r)Ú__name__Ú __module__Ú __qualname__Ú__doc__r"r)r1r r r r!r s   r ÚtestCaserÚ componentsúSequence[bytes] | Sequence[str]ÚexpectedÚbytesr rc Csžd}|D]H}g}| t||giƒ¡tƒ}| ||¡| |¡tƒ}| ||g¡| |¡|D]}| t| ¡ƒ||gfg¡| | |¡|g¡q1qdS)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) r7r8r:ÚnameÚ componentr/ÚaddedZ setHeaderÚheaderr r r!ÚassertSanitizedHs   þôrEc@sÎeZdZdZd+dd„Zd+dd„Zd+dd „Zd+d d „Zd+d d „Zd+dd„Z d+dd„Z d+dd„Z d+dd„Z d+dd„Z d+dd„Zd+dd„Zd+dd„Zd+dd„Zd+d d!„Zd+d"d#„Zd+d$d%„Zd+d&d'„Zd+d(d)„Zd*S),ÚBytesHeadersTestszE Tests for L{Headers}, using L{bytes} arguments for methods. r rcCót|ttƒdS©zf Linear whitespace in header names or values is replaced with a single space. N)rEr r rr r r!Útest_sanitizeLinearWhitespacepóz/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ÚrawValuerOr 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@rNr r r!Útest_addRawHeader‹s   z#BytesHeadersTests.test_addRawHeadercCó| 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•óz-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@©rrOÚ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@r;rrer r r!Ú*test_getRawHeadersWithDefaultMatchingValue¥ó  zr?rNr r r!Útest_removeHeaderDoesntExistÜó z.BytesHeadersTests.test_removeHeaderDoesntExistcCsHtƒ}| ddg¡| ddg¡dd„| ¡Dƒ}| |ddh¡d S) úÇ 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ólemursróbasic aksljdlk=cSóh|] \}}|t|ƒf’qSr ©Útuple©Ú.0ÚkÚvr r r!Ú ïóz:BytesHeadersTests.test_getAllRawHeaders..©r)r‚©r)rN©rr=r?r©rrOZ 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. rLrZrYN©rr=rÚassertNotEqual)rÚfirstÚsecondÚthirdr r r!Útest_headersComparisonös  z(BytesHeadersTests.test_headersComparisoncCó0tƒ}| |d¡| |tƒ¡| |d¡dS)úf An instance of L{Headers} does not compare equal to other unrelated objects. r rLN©rr“rcrNr r r!Útest_otherComparisonó z&BytesHeadersTests.test_otherComparisonc CóDd}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. rLrKóbazú Headers({ú: [ú, ú]})N©rÚreprrÚ capitalize©rÚfooÚbarÚbazr r r!Ú test_reprsþzBytesHeadersTests.test_reprc Cr) 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. rLsbarásbazárŸr 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"}. rLrKržc@ó eZdZdS)z9BytesHeadersTests.test_subclassRepr..FunnyHeadersN©r3r4r5r r r r!Ú FunnyHeaders6ór¯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. rrLrKržN©rr=Úcopyrr@r-©rrOr0r r r!Ú test_copy>s  zBytesHeadersTests.test_copyNr2)r3r4r5r6rIrPrWr\r`rgrkrorrrvrzr~rr—r›rªr«r±r¶r r r r!rFks*         rFc@sìeZdZdZd1dd„Zd1dd„Zd1dd „Zd1d d „Zd1d d „Zd1dd„Z d1dd„Z d1dd„Z d1dd„Z d1dd„Z d1dd„Zd1dd„Zd1dd„Zd1dd„Zd1d d!„Zd1d"d#„Zd1d$d%„Zd1d&d'„Zd1d(d)„Zd1d*d+„Zd1d,d-„Zd1d.d/„Zd0S)2ÚUnicodeHeadersTestszC Tests for L{Headers}, using L{str} arguments for methods. r rcCrGrH)rEr r rr r r!rISrJz1UnicodeHeadersTests.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¨rLrKr§NrMrNr r r!rPZsz$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. Zvalue1Zvalue2rQrRr$rrÚTestNrS)rrVZrawEncodedValuerOr r r!rWfs z&UnicodeHeadersTests.test_setRawHeaderscCsxtƒ}| t¡| ddg¡Wdƒn1swY| t¡| d¡WdƒdS1s5wYdS)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=rUrNr 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=rNr 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¨rLrKNrSrNr 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¹rLóxs☃NrSrNr r r!Útest_rawHeadersValueEncoding¦sz0UnicodeHeadersTests.test_rawHeadersValueEncodingcCsbtƒ}| dd¡| | d¡dg¡| dd¡| | d¡ddg¡| | d¡ddg¡dS)rXr$ÚlemurÚpandarrYrZNr[rNr r r!r\°s   z%UnicodeHeadersTests.test_addRawHeadercCr])r^r$Nr_rr r r!r`»raz/UnicodeHeadersTests.test_getRawHeadersNoDefaultcCshtƒ}tƒ}| | d|¡|¡| | dd¡d¡| | ddg¡dg¡| | ddg¡dg¡dS)rbr$Nr¹)rrcrdr@rrer r r!rgÂ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. rirhÚkeyrr+N)rr=rjr@Ústrrrer r r!rkÑrlz>UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValuecCshtƒ}| ddg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡dS)rmr$rÅr¸rrYrNrnrNr r r!roÝs z&UnicodeHeadersTests.test_getRawHeaderscCsXtƒ}| ddg¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)rpr$rÅr¸rrNrqrNr r r!rrés z&UnicodeHeadersTests.test_hasHeaderTruecCr])rsr$Nrtrr r r!rvõraz'UnicodeHeadersTests.test_hasHeaderFalsecCsštƒ}| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡dS) rwr§rÅrLr¨rÆÚBarrKNrxrNr r r!rzüs  z%UnicodeHeadersTests.test_removeHeadercCr{)r|r$Nr}rNr r r!r~rz0UnicodeHeadersTests.test_removeHeaderDoesntExistcCsVtƒ}| ddg¡| ddg¡| ddg¡dd„| ¡Dƒ}| |hd £¡d S) r€r$Zlemursr(zbasic aksljdlk=r%Z kjdfdfgdfgnsdcSrƒr r„r†r r r!rŠ"r‹z.>rŒr)r)s kjdfdfgdfgnsdNrŽrr r r!rsþz)UnicodeHeadersTests.test_getAllRawHeaderscCsÈtƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡tƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡dS)r‘r§rÆrÅrLrZrYNr’)rr”r•r–Z firstBytesZ secondBytesZ thirdBytesr r r!r—-s$     z*UnicodeHeadersTests.test_headersComparisoncCr˜)r™r r§NršrNr r r!r›Irœ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) r¬r§rÊr©zb'Foo'zb'bar\xe2\x98\x83'c@r­)z;UnicodeHeadersTests.test_subclassRepr..FunnyHeadersNr®r r r r!r¯rr°r¯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¸ržrKNr³rµr r r!r¶{s  zUnicodeHeadersTests.test_copyNr2)r3r4r5r6rIrPrWr½rÀrÁrÄr\r`rgrkrorrrvrzr~rr—r›rªr±r¶r r r r!r·Ns0             r·c@s$eZdZdZd dd„Zd 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ÚStrr;Nr[rNr r r!r\ž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-Strr;s Mixed-BytesrÓNrnrNr r r!rW©sz$MixedHeadersTests.test_setRawHeadersNr2)r3r4r5r6r\rWr r r r!rИs  rÐN)r7rr8r9r:r;r r)r6Ú __future__rÚtypingrZtwisted.trial.unittestrrZtwisted.web.http_headersrrrZtwisted.web.test.requesthelperr r r r rErFr·rÐr r r r!Ús   3#dL