o ¼>hMaã@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 ú‚/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/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-md5rÚdntrÚetagrÚp3prÚ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ÚrangerÚ_MAX_CACHED_HEADERSÚ addRawHeaderrÚlenÚ_canonicalHeaderCache)rÚheadersÚir r r!Útest_maxCachedHeaders<s  ÿz&NameEncoderTests.test_maxCachedHeadersN©r r)Ú__name__Ú __module__Ú __qualname__Ú__doc__r"r+r5r 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)Úappendrr0Ú setRawHeadersrÚlistÚgetAllRawHeadersÚ getRawHeaders) r;r<r>ÚnameÚ componentr3ÚaddedÚ setHeaderÚheaderr r r!ÚassertSanitizedHs   þôrJc@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)rJr 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©rrrD©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©rrAÚ assertTrueÚ hasHeaderrrD)rÚrawValuerTr 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©rr0rrDrSr 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©Ú assertIsNonerrDrr 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ÚassertIdenticalrD©rrTÚ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)rrAÚassertIsInstancerDr?rrkr r r!Ú*test_getRawHeadersWithDefaultMatchingValue¥ó  zïóz:BytesHeadersTests.test_getAllRawHeaders..©r)rˆ©r)r‡N©rrArCr©rrTÚ 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. rQr_r^N©rrArÚ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 rQN©rršrirSr 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. rQrPó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. rQsbará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"}. rQrPr¥c@ó eZdZdS)z9BytesHeadersTests.test_subclassRepr..FunnyHeadersN©r7r8r9r 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. rrQrPr¥N©rrAÚcopyrrDr0©rrTr4r r r!Ú test_copy>s  zBytesHeadersTests.test_copyNr6)r7r8r9r:rNrUr\rarfrmrqrurxr|r€r„r—ržr¢r±r²r¸r½r r r r!rKks*         rKc@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 rcCrLrM)rJr r rr r r!rNSrOz1UnicodeHeadersTests.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. ÚFoor¯rQrPr®NrRrSr r r!rUZsz$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. Úvalue1Úvalue2rVrWr$rrÚTestNrX)rr[ÚrawEncodedValuerTr r r!r\fs 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ÚUnicodeEncodeErrorrArZrSr 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ÆrrArSr 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¯rQrPNrXrSr 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ÄrQóxs☃NrXrSr r r!Útest_rawHeadersValueEncoding¦sz0UnicodeHeadersTests.test_rawHeadersValueEncodingcCsbtƒ}| dd¡| | d¡dg¡| dd¡| | d¡ddg¡| | d¡ddg¡dS)r]r$ÚlemurÚpandarr^r_Nr`rSr r r!ra°s   z%UnicodeHeadersTests.test_addRawHeadercCrb)rcr$Nrdrr r r!rf»rgz/UnicodeHeadersTests.test_getRawHeadersNoDefaultcCshtƒ}tƒ}| | d|¡|¡| | dd¡d¡| | ddg¡dg¡| | ddg¡dg¡dS)rhr$NrÄ)rrirjrDrrkr r r!rmÂ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. rornÚkeyrr-N)rrArprDÚstrrrkr r r!rqÑrrz>UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValuecCshtƒ}| ddg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡dS)rsr$rÐrÂrr^rNrtrSr r r!ruÝs z&UnicodeHeadersTests.test_getRawHeaderscCsXtƒ}| ddg¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)rvr$rÐrÂrrNrwrSr r r!rxés z&UnicodeHeadersTests.test_hasHeaderTruecCrb)ryr$Nrzrr r r!r|õrgz'UnicodeHeadersTests.test_hasHeaderFalsecCsštƒ}| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡dS) r}r®rÐrQr¯rÑÚBarrPNr~rSr r r!r€üs  z%UnicodeHeadersTests.test_removeHeadercCr)r‚r$NrƒrSr r r!r„r…z0UnicodeHeadersTests.test_removeHeaderDoesntExistcCsVtƒ}| ddg¡| ddg¡| ddg¡dd„| ¡Dƒ}| |hd £¡d S) r†r$Úlemursr*zbasic aksljdlk=r%Ú kjdfdfgdfgnsdcSr‰r rŠrŒr r r!r"r‘z.>©r)s kjdfdfgdfgnsdr“r’Nr”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ÐrQr_r^Nr™)rr›rœrÚ firstBytesÚ secondBytesÚ thirdBytesr r r!rž-s$     z*UnicodeHeadersTests.test_headersComparisoncCrŸ)r r r®Nr¡rSr 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¥rPNrºr¼r r r!r½{s  zUnicodeHeadersTests.test_copyNr6)r7r8r9r:rNrUr\rÈrËrÌrÏrarfrmrqrurxr|r€r„r—rž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`rSr r r!raž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äNrtrSr r r!r\©sz$MixedHeadersTests.test_setRawHeadersNr6)r7r8r9r:rar\r r r r!rá˜s  ráN)r;rr<r=r>r?r r)r:Ú __future__rÚtypingrÚtwisted.trial.unittestrrÚtwisted.web.http_headersrrrÚtwisted.web.test.requesthelperr r r r rJrKr¾rár r r r!Ús   3#dL