← Back
Editing: test_http_headers.cpython-38.pyc
U W[�O � @ s` d Z ddlmZmZ ddlmZ ddlmZmZ ddl m Z G dd� de�Z G dd � d e�Z d S ) z( Tests for L{twisted.web.http_headers}. � )�division�absolute_import)�TestCase)�_PY3�unicode)�Headersc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+�BytesHeadersTestszE Tests for L{Headers}, using L{bytes} arguments for methods. c C s& t ddgi�}| �|�d�dg� dS )zx The header values passed to L{Headers.__init__} can be retrieved via L{Headers.getRawHeaders}. � Foo� bar� fooN�r � assertEqual� getRawHeaders��self�h� r �D/usr/lib/python3/dist-packages/twisted/web/test/test_http_headers.py�test_initializer s z"BytesHeadersTests.test_initializerc C sP ddg}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�r � setRawHeaders� assertTrue� hasHeaderr r )r �rawValuer r r r �test_setRawHeaders s z$BytesHeadersTests.test_setRawHeadersc C s t � }| �t|jdddi� dS )�N L{Headers.setRawHeaders} requires values to be of type list. � keyr r N�r � assertRaises� TypeErrorr r r r r �test_rawHeadersTypeChecking( s z-BytesHeadersTests.test_rawHeadersTypeCheckingc C sL t � }|�dd� | �|�d�dg� |�dd� | �|�d�ddg� dS )�N L{Headers.addRawHeader} adds a new value for a given header. r � lemur� pandaN�r � addRawHeaderr r r r r r �test_addRawHeader0 s z#BytesHeadersTests.test_addRawHeaderc C s | � t� �d�� dS )�z L{Headers.getRawHeaders} returns L{None} if the header is not found and no default is specified. r N�Z assertIsNoner r �r r r r �test_getRawHeadersNoDefault; s z-BytesHeadersTests.test_getRawHeadersNoDefaultc C s$ t � }t� }| �|�d|�|� dS )�o L{Headers.getRawHeaders} returns the specified default value when no header is found. r N)r �object�assertIdenticalr �r r �defaultr r r � test_getRawHeadersDefaultValueC s z0BytesHeadersTests.test_getRawHeadersDefaultValuec C sJ t � }dg}|�d|� | �|�d|�d t� | �|�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. �valuer r � valueN)r r �assertIsInstancer �bytesr r2 r r r �*test_getRawHeadersWithDefaultMatchingValueM s z<BytesHeadersTests.test_getRawHeadersWithDefaultMatchingValuec C s@ t � }|�ddg� | �|�d�dg� | �|�d�dg� dS )�m L{Headers.getRawHeaders} returns the values which have been set for a given header. r r& r N�r r r r r r r r �test_getRawHeadersZ s z$BytesHeadersTests.test_getRawHeadersc C s8 t � }|�ddg� | �|�d�� | �|�d�� dS )�i Check that L{Headers.hasHeader} returns C{True} when the given header is found. r r& r N�r r r r r r r r �test_hasHeaderTruee s z$BytesHeadersTests.test_hasHeaderTruec C s | � t� �d�� dS )�c L{Headers.hasHeader} returns C{False} when the given header is not found. r N�� assertFalser r r- r r r �test_hasHeaderFalsep s z%BytesHeadersTests.test_hasHeaderFalsec C sz t � }|�ddg� | �|�d�� |�d� | �|�d�� |�ddg� | �|�d�� |�d� | �|�d�� dS )�N Check that L{Headers.removeHeader} removes the given header. r r& r r' s BarN�r r r r � removeHeaderrB r r r r �test_removeHeaderx s z#BytesHeadersTests.test_removeHeaderc C s( t � }|�d� | �t|�� �g � dS )�k L{Headers.removeHeader} is a no-operation when the specified header is not found. r N�r rF r �list�getAllRawHeadersr r r r � test_removeHeaderDoesntExist� s z.BytesHeadersTests.test_removeHeaderDoesntExistc C s� t � }| �|�d�d� | �|�d�d� | �|�d�d� | �|�d�d� | �|�d �d � | �|�d �d � | �|�d �d� | �|�d�d� | �|�d�d� dS )zr L{Headers._canonicalNameCaps} returns the canonical capitalization for the given header. r r s test-stuffs Test-Stuffs content-md5� Content-MD5s dnts DNTs etags ETags p3ps P3Ps tes TE� www-authenticate� WWW-Authenticates x-xss-protections X-XSS-ProtectionN)r r Z_canonicalNameCapsr r r r �test_canonicalNameCaps� s � �z(BytesHeadersTests.test_canonicalNameCapsc C sP t � }|�ddg� |�ddg� tdd� |�� D ��}| �|tddg�� 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 � lemursrN � basic aksljdlk=c S s g | ]\}}|t |�f�qS r ��tuple��.0�k�vr r r � <listcomp>� s z;BytesHeadersTests.test_getAllRawHeaders.<locals>.<listcomp>�rO )rS )r �rR N�r r �setrK r �r r Z allHeadersr r r �test_getAllRawHeaders� s ��z'BytesHeadersTests.test_getAllRawHeadersc C sf t � }|�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�r r r �assertNotEqual)r �first�second�thirdr r r �test_headersComparison� s z(BytesHeadersTests.test_headersComparisonc C s0 t � }| �|d� | �|t� � | �|d� dS )�f An instance of L{Headers} does not compare equal to other unrelated objects. r r N�r rc r0 r r r r �test_otherComparison� s z&BytesHeadersTests.test_otherComparisonc C s6 d}d}d}| � tt|||gi��d|||f � 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({%r: [%r, %r]})N�r �reprr �r �foo�bar�bazr r r � test_repr� s �zBytesHeadersTests.test_reprc C s6 d}d}d}| � tt|||gi��d|||f � 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. r s bar�s baz�rl Nrm ro r r r �test_reprWithRawBytes� s �z'BytesHeadersTests.test_reprWithRawBytesc C sF d}d}d}G dd� dt �}| �t||||gi��d|||f � dS )�� 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 rk c @ s e Zd ZdS )z9BytesHeadersTests.test_subclassRepr.<locals>.FunnyHeadersN��__name__� __module__� __qualname__r r r r � FunnyHeaders� s rz z FunnyHeaders({%r: [%r, %r]})N)r r rn )r rp rq rr rz r r r �test_subclassRepr� s �z#BytesHeadersTests.test_subclassReprc C sv t � }|�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 rk N�r r �copyr r r) �r r �ir r r � test_copy� s zBytesHeadersTests.test_copyN)rw rx ry �__doc__r r r$ r* r. r4 r9 r< r? rC rG rL rP r` rg rj rs rt r{ r� r r r r r s* r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,S )-�UnicodeHeadersTestszG Tests for L{Headers}, using L{unicode} arguments for methods. c C s: t ddgi�}| �|�d�dg� | �|�d�dg� dS )aK 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{unicode} argument is given, it returns L{unicode} values. Both are the same header value, just encoded or decoded. �Foorq r r rp Nr r r r r r s z$UnicodeHeadersTests.test_initializerc C s� 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. Zvalue1Zvalue2r r �testr r ZTestNr )r r ZrawEncodedValuer r r r r ! s z&UnicodeHeadersTests.test_setRawHeadersc C sN t � }| �t�� |�ddg� W 5 Q R X | �t�� |�d� W 5 Q R X dS )z� Passing L{unicode} 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 r" �UnicodeEncodeErrorr r r r r r �test_nameNotEncodable2 s z)UnicodeHeadersTests.test_nameNotEncodablec C sL t � }|�ddg� | �|�d�� | �|�d�dg� | �|�d�� dS )z� Passing L{unicode} to any function that takes a header name will encode said header name as ISO-8859-1. � ár � �Nr r r r r �test_nameEncodingB s z%UnicodeHeadersTests.test_nameEncodingc C s@ t � }|�dddg� | �|�d�� | �|�d�ddg� dS )z� Passing L{unicode} to L{Headers.setRawHeaders} will encode the name as ISO-8859-1 and values as UTF-8. r� r� r r� s ☃Nr r r r r � test_rawHeadersValueEncodingT s z0UnicodeHeadersTests.test_rawHeadersValueEncodingc C s t � }| �t|jdddi� dS )r �keyr� rq Nr! r r r r r$ _ s z/UnicodeHeadersTests.test_rawHeadersTypeCheckingc C sb t � }|�dd� | �|�d�dg� |�dd� | �|�d�ddg� | �|�d�ddg� dS )r% r� �lemur�pandar r& r' Nr( r r r r r* g s z%UnicodeHeadersTests.test_addRawHeaderc C s | � t� �d�� dS )r+ r� Nr, r- r r r r. s s z/UnicodeHeadersTests.test_getRawHeadersNoDefaultc C sh t � }t� }| �|�d|�|� | �|�dd�d� | �|�ddg�dg� | �|�ddg�dg� dS )r/ r� Nr� )r r0 r1 r r r2 r r r r4 { s �z2UnicodeHeadersTests.test_getRawHeadersDefaultValuec C sJ t � }dg}|�d|� | �|�d|�d t� | �|�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. r6 r r� r r5 N)r r r7 r r r r2 r r r r9 � s z>UnicodeHeadersTests.test_getRawHeadersWithDefaultMatchingValuec C sh t � }|�ddg� | �|�d�dg� | �|�d�dg� | �|�d�dg� | �|�d�dg� dS )r: � testár� � Testá� test�r& � Test�Nr; r r r r r< � s z&UnicodeHeadersTests.test_getRawHeadersc C sX t � }|�ddg� | �|�d�� | �|�d�� | �|�d�� | �|�d�� dS )r= r� r� r� r� r� Nr> r r r r r? � s z&UnicodeHeadersTests.test_hasHeaderTruec C s | � t� �d�� dS )r@ r� NrA r- r r r rC � s z'UnicodeHeadersTests.test_hasHeaderFalsec C s� t � }|�ddg� | �|�d�� |�d� | �|�d�� | �|�d�� |�ddg� | �|�d�� |�d� | �|�d�� | �|�d�� dS ) rD rp r� r rq r� ZBarr NrE r r r r rG � s z%UnicodeHeadersTests.test_removeHeaderc C s( t � }|�d� | �t|�� �g � dS )rH r� NrI r r r r rL � s z0UnicodeHeadersTests.test_removeHeaderDoesntExistc C s` t � }|�ddg� |�ddg� |�ddg� tdd� |�� D ��}| �|td d d g�� d S ) rQ r� Zlemurszwww-authenticatezbasic aksljdlk=z content-md5Z kjdfdfgdfgnsdc S s g | ]\}}|t |�f�qS r rT rV r r r rZ � s z=UnicodeHeadersTests.test_getAllRawHeaders.<locals>.<listcomp>r[ )rM )s kjdfdfgdfgnsd)r� r\ Nr] r_ r r r r` � s ��z)UnicodeHeadersTests.test_getAllRawHeadersc C s� t � }|�ddg� t � }|�ddg� t � }|�dddg� | �||� | �||� | �||� t � }|�ddg� t � }|�ddg� t � }|�dddg� | �||� | �||� | �||� dS )ra � fooár� r� s foo�r' r& Nrb )r rd re rf Z firstBytesZ secondBytesZ thirdBytesr r r rg � s$ z*UnicodeHeadersTests.test_headersComparisonc C s0 t � }| �|d� | �|t� � | �|d� dS )rh r rp Nri r r r r rj s z(UnicodeHeadersTests.test_otherComparisonc C sX d}d}d}d}d}t r(d| }d| }| �tt|||gi��d|||�d�f � 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☃rr � 'foo\xe1'�'bar\xe2\x98\x83'�bzHeaders({%s: [%s, %r]})�utf8N)r r rn r �encode)r rp rq rr � fooEncoded� barEncodedr r r rs s ��z UnicodeHeadersTests.test_reprc C sh d}d}d}d}d}t r(d| }d| }G dd� dt�}| �t||||gi��d |||�d �f � d S ) ru r� r� rr r� r� r� c @ s e Zd ZdS )z;UnicodeHeadersTests.test_subclassRepr.<locals>.FunnyHeadersNrv r r r r rz 4 s rz z FunnyHeaders({%s: [%s, %r]})r� N)r r r rn r� )r rp rq rr r� r� rz r r r r{ ' s ��z%UnicodeHeadersTests.test_subclassReprc C s� t � }|�ddg� |�� }| �|�d�dg� | �|�d�dg� |�dd� | �|�d�dg� | �|�d�dg� |�dd� | �|�d�ddg� | �|�d�ddg� dS ) r| r� u foo☃r� s foo☃rq rk r Nr} r r r r r� = s" � �z UnicodeHeadersTests.test_copyN)rw rx ry r� r r r� r� r� r$ r* r. r4 r9 r< r? rC rG rL r` rg rj rs r{ r� r r r r r� s, r� N) r� Z __future__r r Ztwisted.trial.unittestr Ztwisted.python.compatr r Ztwisted.web.http_headersr r r� r r r r �<module> s
Save File
Cancel