File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/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