o ¯büZã@s\dZddlmZddlmZddlmZmZmZdd„Z Gdd„deƒZ Gd d „d eƒZ d S) z( Tests for L{twisted.web.http_headers}. é)ÚTestCase)ÚHeaders)ÚbytesLinearWhitespaceComponentsÚsanitizedBytesÚtextLinearWhitespaceComponentscCsš|D]H}g}| t||giƒ¡tƒ}| ||¡| |¡tƒ}| ||g¡| |¡|D]}| t| ¡ƒ||gfg¡| | |¡|g¡q/qdS)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Ú assertEqualÚlistÚgetAllRawHeadersÚ getRawHeaders)ÚtestCaseÚ componentsÚexpectedÚ componentÚheadersÚaddedÚ setHeaderÚheader©rúD/usr/lib/python3/dist-packages/twisted/web/test/test_http_headers.pyÚassertSanitizeds    ÿüôrc@sØeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4S)5ÚBytesHeadersTestszE Tests for L{Headers}, using L{bytes} arguments for methods. cCót|ttƒdS©zf Linear whitespace in header names or values is replaced with a single space. N)rrr©ÚselfrrrÚtest_sanitizeLinearWhitespace:óz/BytesHeadersTests.test_sanitizeLinearWhitespacecCs&tddgiƒ}| | d¡dg¡dS)zx The header values passed to L{Headers.__init__} can be retrieved via L{Headers.getRawHeaders}. óFooóbarófooN©rr r ©rÚhrrrÚtest_initializerAsz"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óvalue2ótestóTestN©rr Ú assertTrueÚ hasHeaderr r )rÚrawValuer%rrrÚtest_setRawHeadersIs  z$BytesHeadersTests.test_setRawHeaderscCó tƒ}| t|jdddi¡dS)zN L{Headers.setRawHeaders} requires values to be of type list. ókeyr r!N©rÚ assertRaisesÚ TypeErrorr r$rrrÚ)test_rawHeadersTypeCheckingValuesIterableUóz;BytesHeadersTests.test_rawHeadersTypeCheckingValuesIterablecCs0tƒ}| t|jddg¡}| |jdd¡dS)zf L{Headers.setRawHeaders} requires C{name} to be a L{bytes} or L{str} string. Nr"rúBHeader name is an instance of , not bytes or str©rr3r4r r Úargs©rr%ÚerrrÚtest_rawHeadersTypeCheckingName\s þz1BytesHeadersTests.test_rawHeadersTypeCheckingNamecCs2tƒ}| t|jdddg¡}| |jdd¡dS)zn L{Headers.setRawHeaders} requires values to a L{list} of L{bytes} or L{str} strings. r1r!NrzQHeader value at position 1 is an instance of , not bytes or strr8r:rrrÚ*test_rawHeadersTypeCheckingValuesAreStringhs þz, not bytes or strrCr:rrrÚtest_addRawHeaderTypeCheckValue‹rEz1BytesHeadersTests.test_addRawHeaderTypeCheckValuecCó| tƒ d¡¡dS)úz L{Headers.getRawHeaders} returns L{None} if the header is not found and no default is specified. r)N©Ú assertIsNonerr rrrrÚtest_getRawHeadersNoDefault—óz-BytesHeadersTests.test_getRawHeadersNoDefaultcCs$tƒ}tƒ}| | d|¡|¡dS)úo L{Headers.getRawHeaders} returns the specified default value when no header is found. r)N)rÚobjectÚassertIdenticalr ©rr%ÚdefaultrrrÚ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. Úvaluer1róvalueN)rr ÚassertIsInstancer Úbytesr rPrrrÚ*test_getRawHeadersWithDefaultMatchingValue§ó  zóz:BytesHeadersTests.test_getAllRawHeaders..©rn)rs)r*©rrN©rr r r ©rr%Ú allHeadersrrrÚ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. r"r@r?N©rr r ÚassertNotEqual)rÚfirstÚsecondÚthirdrrrÚtest_headersComparisons  z(BytesHeadersTests.test_headersComparisoncCó0tƒ}| |d¡| |tƒ¡| |d¡dS)úf An instance of L{Headers} does not compare equal to other unrelated objects. rr"N©rr…rNr$rrrÚtest_otherComparisonó z&BytesHeadersTests.test_otherComparisonc Có@d}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. r"r!óbazú Headers({ú: [ú, ú]})N©r Úreprr©rÚfooÚbarÚbazrrrÚ test_repr!sþ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. r"sbarásbazár‘r’r“r”Nr•r—rrrÚtest_reprWithRawBytes.sþz'BytesHeadersTests.test_reprWithRawBytesc CsPd}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"}. r"r!rc@ó eZdZdS)z9BytesHeadersTests.test_subclassRepr..FunnyHeadersN©Ú__name__Ú __module__Ú __qualname__rrrrÚ FunnyHeadersHór£zFunnyHeaders({r’r“r”N)rr r–)rr˜r™ršr£rrrÚ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. r)r"r!rN©rr Úcopyr r r©rr%ÚirrrÚ test_copyPs  zBytesHeadersTests.test_copyN)r r¡r¢Ú__doc__rr&r/r5r<r=rBrDrFrKrRrWr[r^rbrfrjrpr‚r‰rr›rœr¥r«rrrrr5s6              rc@sÀeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.S)/ÚUnicodeHeadersTestszC Tests for L{Headers}, using L{str} arguments for methods. cCrr)rrrrrrrrerz1UnicodeHeadersTests.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™r"r!r˜Nr#r$rrrr&lsz$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Úvalue2r'r(Útestr)r*ÚTestNr+)rr.ÚrawEncodedValuer%rrrr/xs 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)rr3ÚUnicodeEncodeErrorr r-r$rrrÚtest_nameNotEncodableˆs ÿ  "ÿz)UnicodeHeadersTests.test_nameNotEncodablecCsLtƒ}| 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"óáNr+r$rrrÚ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. r¸r´r"r¹s☃Nr+r$rrrÚtest_rawHeadersValueEncoding¨sz0UnicodeHeadersTests.test_rawHeadersValueEncodingcCr0)zQ L{Headers.setRawHeaders} requires values to be of type sequence Úkeyr®r™Nr2r$rrrÚtest_rawHeadersTypeChecking²r6z/UnicodeHeadersTests.test_rawHeadersTypeCheckingcCsbtƒ}| dd¡| | d¡dg¡| dd¡| | d¡ddg¡| | d¡ddg¡dS)r>r±ÚlemurÚpandar)r?r@NrAr$rrrrB¹s   z%UnicodeHeadersTests.test_addRawHeadercCrG)rHr±NrIrrrrrKÄrLz/UnicodeHeadersTests.test_getRawHeadersNoDefaultcCshtƒ}tƒ}| | d|¡|¡| | dd¡d¡| | ddg¡dg¡| | ddg¡dg¡dS)rMr±Nr´)rrNrOr r rPrrrrRË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. rTr1r¼rrSN)rr rUr Ústrr rPrrrrWÚrXz>UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValuecCshtƒ}| ddg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡| | d¡dg¡dS)rYõtestár¾õTestáótestár?óTestáNrZr$rrrr[æs z&UnicodeHeadersTests.test_getRawHeaderscCsXtƒ}| ddg¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)r\rÁr¾rÂrÃrÄNr]r$rrrr^òs z&UnicodeHeadersTests.test_hasHeaderTruecCrG)r_rÁNr`rrrrrbþrLz'UnicodeHeadersTests.test_hasHeaderFalsecCsštƒ}| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡| ddg¡| | d¡¡| d¡| | d¡¡| | d¡¡dS) rcr˜r¾r"r™r¿ÚBarr!Nrdr$rrrrfs  z%UnicodeHeadersTests.test_removeHeadercCrg)rhr±Nrir$rrrrjrkz0UnicodeHeadersTests.test_removeHeaderDoesntExistcCsVtƒ}| ddg¡| ddg¡| ddg¡dd„| ¡Dƒ}| |hd £¡d S) rqrÁÚlemurszwww-authenticatezbasic aksljdlk=z content-md5Ú kjdfdfgdfgnsdcSrtrrurwrrrr{+r|z.>©rÄr~©rl)s kjdfdfgdfgnsdr}Nrr€rrrr‚ sþz)UnicodeHeadersTests.test_getAllRawHeaderscCsÈtƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡tƒ}| ddg¡tƒ}| ddg¡tƒ}| dddg¡| ||¡| ||¡| ||¡dS)rƒõfooár¿r¾sfooár@r?Nr„)rr†r‡rˆÚ firstBytesÚ secondBytesÚ thirdBytesrrrr‰6s$     z*UnicodeHeadersTests.test_headersComparisoncCrŠ)r‹rr˜NrŒr$rrrrRrŽ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\xe1'z'bar\xe2\x98\x83'ÚbzHeaders({{{}: [{}, {!r}]}})Úutf8N)r r–rÚformatÚencode)rr˜r™ršÚ fooEncodedÚ barEncodedrrrr›\s ÿþzUnicodeHeadersTests.test_reprc CsTd}d}d}d}d}Gdd„dtƒ}| t||||giƒƒd||| d ¡f¡d S) rrÊrÎršz b'foo\xe1'zb'bar\xe2\x98\x83'c@rž)z;UnicodeHeadersTests.test_subclassRepr..FunnyHeadersNrŸrrrrr£{r¤r£zFunnyHeaders({%s: [%s, %r]})rÐN)rr r–rÒ)rr˜r™ršrÓrÔr£rrrr¥psÿþz%UnicodeHeadersTests.test_subclassReprcCs´tƒ}| ddg¡| ¡}| | d¡dg¡| | d¡dg¡| dd¡| | d¡dg¡| | d¡dg¡| dd¡| | d¡ddg¡| | d¡ddg¡dS) r¦rÁufoo☃rÃsfoo☃r™rr!Nr§r©rrrr«„s  zUnicodeHeadersTests.test_copyN)r r¡r¢r¬rr&r/r·rºr»r½rBrKrRrWr[r^rbrfrjr‚r‰rr›r¥r«rrrrr­`s0         r­N) r¬Útwisted.trial.unittestrÚtwisted.web.http_headersrÚtwisted.web.test.requesthelperrrrrrr­rrrrÚs  #-