File: //lib/python3/dist-packages/twisted/web/test/__pycache__/test_static.cpython-38.pyc
U
W[ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z
ddl m Z ddl
mZmZ ddlmZ ddlmZ ddlmZmZ dd lmZmZ dd
lmZ dd l m Z m Z m Z m Z dd l!m"Z" dd
l#m$Z$ ddl%m&Z& ddl'm(Z( G dd� de�Z)G dd� de�Z*G dd� de�Z+G dd� de�Z,G dd� de�Z-G dd� de�Z.G d d � d e�Z/G d d � d e�Z0G d d!� d!e�Z1G d"d#� d#e�Z2G d$d%� d%e�Z3dS )&z"
Tests for L{twisted.web.static}.
� N)�BytesIO)� verifyObject)�abstract�
interfaces)�platform)�FilePath)�compat�log)�
intToBytes�
networkString)�TestCase)�static�http�script�resource)�UnsupportedMethod)� DummyRequest��_render)�FOUNDc @ s e Zd ZdZdd� Zdd� ZdS )�StaticDataTestsz
Tests for L{Data}.
c sB t �dd�}tdg�� d� _t|� �}� �fdd�}|�|� |S )zV
L{Data.render} returns an empty response body for a I{HEAD} request.
� fooZbar� � HEADc s �� d�� j�d� d S �N� �� assertEqual�join�written��ignored��request�self� �>/usr/lib/python3/dist-packages/twisted/web/test/test_static.py�
cbRendered- s z4StaticDataTests.test_headRequest.<locals>.cbRendered)r
�Datar �methodr � addCallback)r$ �data�dr' r% r" r& �test_headRequest% s
z StaticDataTests.test_headRequestc C s0 t �dd�}tdg�}d|_| �t|j|� dS )zv
L{Data.render} raises L{UnsupportedMethod} in response to a non-I{GET},
non-I{HEAD} request.
r s barr � POSTN)r
r( r r) � assertRaisesr �render)r$ r+ r# r% r% r& �test_invalidMethod3 s
z"StaticDataTests.test_invalidMethodN)�__name__�
__module__� __qualname__�__doc__r- r1 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
e�� rjde
_dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Ze�� �� d#kr�d$e�� f e_d%d&� Zd'd(� Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z!d5d6� Z"d7d8� Z#d9d:� Z$d;S )<�StaticFileTestsz2
Tests for the basic behavior of L{File}.
c C s
t ||�S �Nr )r$ r r# r% r% r& r C s zStaticFileTests._renderc C sN t jdd��(}tj| �� dd�}| �|jdg� W 5 Q R X | �t|�d� dS )z�
Passing C{1} as the value to L{File}'s C{ignoredExts} argument
issues a warning and sets the ignored extensions to the
wildcard C{"*"}.
T��record� �� ignoredExts�*N��warnings�catch_warningsr
�File�mktempr r< �len�r$ ZcaughtWarnings�filer% r% r& �test_ignoredExtTrueG s z#StaticFileTests.test_ignoredExtTruec C sL t jdd��&}tj| �� dd�}| �|jg � W 5 Q R X | �t|�d� dS )z�
Passing C{1} as the value to L{File}'s C{ignoredExts} argument
issues a warning and sets the ignored extensions to the empty
list.
Tr8 r r; r: Nr> rD r% r% r& �test_ignoredExtFalseT s z$StaticFileTests.test_ignoredExtFalsec C sN t jdd��(}tj| �� dd�}| �|jdg� W 5 Q R X | �t|�d� dS )z�
Passing C{1} as the value to L{File}'s C{allowExt} argument
issues a warning and sets the ignored extensions to the
wildcard C{*}.
Tr8 r; r= r: Nr> rD r% r% r& �
test_allowExta s z StaticFileTests.test_allowExtc C sF t dg�}d|_t| �� �}|�d� t�|j�}| �t |j
|� dS )zv
L{File.render} raises L{UnsupportedMethod} in response to a non-I{GET},
non-I{HEAD} request.
r r. r N) r r) r rB �
setContentr
rA �pathr/ r r0 )r$ r# rJ rE r% r% r& r1 n s
z"StaticFileTests.test_invalidMethodc s^ t ��� �}|�� t�|j�}tdg�� t�|� �}�� |� �}� �fdd�}|�
|� |S )z�
If a request is made which encounters a L{File} before a final segment
which does not correspond to any file in the path the L{File} was
created with, a not found response is sent.
� foobarc s �� � jd� d S �N� �r � responseCoder r" r% r& r' � s z1StaticFileTests.test_notFound.<locals>.cbRendered� r rB �makedirsr
rA rJ r r �getChildForRequestr r* �r$ �baserE �childr, r' r% r" r& �
test_notFound{ s
z StaticFileTests.test_notFoundc C sX t | �� �}|�� t�|j�}tdg�}t�||�}| � |tj
� | � |j|j� dS )z�
The C{''} child of a L{File} which corresponds to a directory in the
filesystem is a L{DirectoryLister}.
r N) r rB rQ r
rA rJ r r rR �assertIsInstance�DirectoryListerr �r$ rT rE r# rU r% r% r& �test_emptyChild� s
z StaticFileTests.test_emptyChildc C s� t | �� ��� }|�� |�d��d��� t�|j �}t
dg�}t � ||�}| �
|tj� t�|j �}| �|j |� |�|�}| �
|t� dS )z�
The C{u''} child of a L{File} which corresponds to a directory
whose path is text is a L{DirectoryLister} that renders to a
binary listing.
@see: U{https://twistedmatrix.com/trac/ticket/9438}
z text-file�wr N)r rB Z
asTextModerQ rU �open�closer
rA rJ r r rR rW rX r Z nativeStringr r0 �bytes)r$ ZtextBaseZtextFiler# rU Z
nativePathZresponser% r% r& � test_emptyChildUnicodeParent� s
z,StaticFileTests.test_emptyChildUnicodeParentc s^ t ��� �}|�� t�|j�}tdg�� t�|� �}�� |� �}� �fdd�}|�
|� |S )z�
If a request is made which encounters a L{File} before a final segment
which cannot be looked up in the filesystem due to security
considerations, a not found response is sent.
s ..c s �� � jd� d S rL rN r r" r% r& r' � s zBStaticFileTests.test_securityViolationNotFound.<locals>.cbRenderedrP rS r% r" r& � test_securityViolationNotFound� s
z.StaticFileTests.test_securityViolationNotFoundc sl t ��� �}|�d� ��|jd� |�d� t�|j�}tdg�� �� |� �}� �fdd�}|�
|� |S )z�
If the file in the filesystem which would satisfy a request cannot be
read, L{File.render} sets the HTTP response code to I{FORBIDDEN}.
r i� r c s �� � jd� d S )Ni� rN r r" r% r& r' � s z:StaticFileTests.test_forbiddenResource.<locals>.cbRendered) r rB rI Z
addCleanup�chmodr
rA rJ r r r* )r$ rT rE r, r' r% r" r& �test_forbiddenResource� s
z&StaticFileTests.test_forbiddenResourcez(Cannot remove read permission on Windowsc sv �� � }t|t�r |�d�}t|�}|�� t�|j�}t dg�� t
� |� �}�� |� �}� �fdd�}|�
|� |S )z�
A request whose path cannot be decoded as UTF-8 receives a not
found response, and the failure is logged.
�ascii� �c s( �� � jd� �� t��t��d� d S )NrM r: )r rO rC ZflushLoggedErrors�UnicodeDecodeErrorr r" r% r& r' � s �z8StaticFileTests.test_undecodablePath.<locals>.cbRendered)rB �
isinstancer^ �decoder rQ r
rA rJ r r rR r r* )r$ rJ rT rE rU r, r' r% r" r& �test_undecodablePath� s
z$StaticFileTests.test_undecodablePathc C s | � t�d�jtj� dS )zN
L{File.forbidden} defaults to L{resource.ForbiddenResource}.
� .N)rW r
rA � forbiddenr ZForbiddenResource�r$ r% r% r& � test_forbiddenResource_default� s
�z.StaticFileTests.test_forbiddenResource_defaultc s� t | �� �}|�d� d�dd� }G �fdd�dtj�� G � fdd�dtj�}||j�}||_t dg�}|�
|�}| � �|� d S )
z�
The resource rendered for forbidden requests is stored as a class
member so that users can customize it.
r s custom-forbidden-responsec S s t tjd��d S )Nr )�IOError�errnoZEACCESr% r% r% r& �failingOpenForReading s zOStaticFileTests.test_forbiddenResource_customize.<locals>.failingOpenForReadingc s e Zd Z� fdd�ZdS )zQStaticFileTests.test_forbiddenResource_customize.<locals>.CustomForbiddenResourcec s � S r7 r% �r$ r# ��markerResponser% r& r0 s zXStaticFileTests.test_forbiddenResource_customize.<locals>.CustomForbiddenResource.renderN�r2 r3 r4 r0 r% rq r% r& �CustomForbiddenResource s rt c s e Zd Z� � ZdS )zJStaticFileTests.test_forbiddenResource_customize.<locals>.CustomStaticFileN)r2 r3 r4 rj r% )rt r% r& �CustomStaticFile s ru N) r rB rI r �Resourcer
rA rJ �openForReadingr r0 r )r$ rT ro ru � fileResourcer# �resultr% )rt rr r& � test_forbiddenResource_customize s
z0StaticFileTests.test_forbiddenResource_customizec sv t ��� �}|�� |�d��d� t�|j�}dg|_t dg�� t
� |� �}�� |� �}� �fdd�}|�
|� |S )a
If a request is made which encounters a L{File} before a final empty
segment, a file in the L{File} instance's C{indexNames} list which
exists in the path the L{File} was created with is served as the
response to the request.
�foo.bar� bazr c s0 �� d�� j�d� �� � j�d�d d� d S �Nr r| � content-lengthr � 3�r r r �responseHeaders�
getRawHeadersr r" r% r& r' / s
�z3StaticFileTests.test_indexNames.<locals>.cbRendered)r rB rQ rU rI r
rA rJ Z
indexNamesr r rR r r* rS r% r" r& �test_indexNames s
z StaticFileTests.test_indexNamesc sn t ��� �}|�� |�d��d� t�|j�}tdg�� t �
|� �}�� |� �}� �fdd�}|� |� |S )z�
If a request is made which encounters a L{File} before a final segment
which names a file in the path the L{File} was created with, that file
is served as the response to the request.
r{ r| � foo.barc s0 �� d�� j�d� �� � j�d�d d� d S r} r� r r" r% r& r' G s
�z3StaticFileTests.test_staticFile.<locals>.cbRendered�
r rB rQ rU rI r
rA rJ r r rR r r* rS r% r" r& �test_staticFile8 s
z StaticFileTests.test_staticFilec s~ d}d� t ��� �}|�� |�|��� � t�|j�}t|� d�g��t
� |��}�� |��}� ��fdd�}|�
|� |S )zx
A request for a existing unicode file path encoded as UTF-8
returns the contents of that file.
u ῆs content�utf-8c s< �� d��j�� � �� �j�d�d ttt� ���� d S )Nr r~ r )r r r r� r� r �strrC r ��contentr# r$ r% r& r' a s
�zBStaticFileTests.test_staticFileUnicodeFileName.<locals>.cbRendered)r rB rQ rU rI r
rA rJ r �encoder rR r r* )r$ �namerT rE rU r, r' r% r� r& � test_staticFileUnicodeFileNameP s
z.StaticFileTests.test_staticFileUnicodeFileName)r� Zmcbsz>Cannot write unicode filenames with file system encoding of %sc C s6 t �| �� �}tdg�}|�d|�}| �||j� dS )z�
A L{static.File} created for a directory which does not exist should
return childNotFound from L{static.File.getChild}.
r� N)r
rA rB r �getChildr �
childNotFound)r$ �
staticFiler# rU r% r% r& � test_staticFileDeletedGetChildn s
z.StaticFileTests.test_staticFileDeletedGetChildc s\ t ���� �}tdg��tdg����|��� ��|j��}� ���fdd�}|�|� |S )z{
A L{static.File} created for a file which does not exist should render
its C{childNotFound} page.
r� c s ���fdd�}� � |� � S )Nc s �� d�� j�d��j�� d S r r r )r# �request2r$ r% r& r' � s
�zUStaticFileTests.test_staticFileDeletedRender.<locals>.cbRendered2.<locals>.cbRendered)r* )r! r' �r, r# r� r$ r% r& � cbRendered2� s
zAStaticFileTests.test_staticFileDeletedRender.<locals>.cbRendered2)r
rA rB r r r� r* )r$ r� Zd2r� r% r� r& � test_staticFileDeletedRendery s
z,StaticFileTests.test_staticFileDeletedRenderc s� t | �� �}|�d� d�G �fdd�dtj�� G � fdd�dtj�}||j�}tdg�}|� d|�}|�
|�}| � �|� dS ) zs
The resource rendered for child not found requests can be customize
using a class member.
r s custom-child-not-found-responsec s e Zd Z� fdd�ZdS )zYStaticFileTests.test_getChildChildNotFound_customize.<locals>.CustomChildNotFoundResourcec s � S r7 r% rp rq r% r& r0 � s z`StaticFileTests.test_getChildChildNotFound_customize.<locals>.CustomChildNotFoundResource.renderNrs r% rq r% r& �CustomChildNotFoundResource� s r� c s e Zd Z� � ZdS )zNStaticFileTests.test_getChildChildNotFound_customize.<locals>.CustomStaticFileN)r2 r3 r4 r� r% )r� r% r& ru � s ru s no-child.txtN) r rB rI r rv r
rA rJ r r� r0 r )r$ rT ru rx r# rU ry r% )r� rr r& �$test_getChildChildNotFound_customize� s
z4StaticFileTests.test_getChildChildNotFound_customizec sX t ��� �}|�d� t�|j�}tdg�� d� _t|� �}� �fdd�}|� |� |S )zd
L{static.File.render} returns an empty response body for I{HEAD}
requests.
r r r c s �� d�� j�d� d S r r r r" r% r& r' � s z4StaticFileTests.test_headRequest.<locals>.cbRendered)
r rB rI r
rA rJ r r) r r* )r$ rJ rE r, r' r% r" r& r- � s
z StaticFileTests.test_headRequestc sz t ��� �}|�� |�d��d� t�|j�}dtj i|_
t dg�� t �
|� �}��|� �}� �fdd�}|�|� |S )a!
If a request is made which encounters a L{File} before a final segment
which names a file with an extension which is in the L{File}'s
C{processors} mapping, the processor associated with that extension is
used to serve the response to the request.
r{ sT from twisted.web.static import Data
resource = Data(b'dynamic world', 'text/plain')
�.barr� c s0 �� d�� j�d� �� � j�d�d d� d S )Nr s
dynamic worldr~ r s 13r� r r" r% r& r' � s
�z3StaticFileTests.test_processors.<locals>.cbRendered)r rB rQ rU rI r
rA rJ r ZResourceScriptZ
processorsr r rR r r* rS r% r" r& �test_processors� s
�
z StaticFileTests.test_processorsc C sb t �d�}| �|jg � |�d� |�d� | �|jddg� t jddd�}| �|jddg� dS )z�
The list of ignored extensions can be set by passing a value to
L{File.__init__} or by calling L{File.ignoreExt} later.
ri z.foor� )r� �.bazr; r� N)r
rA r r< Z ignoreExt)r$ rE r% r% r& �test_ignoreExt� s
z StaticFileTests.test_ignoreExtc s� t ��� �}|�� |�d��d� |�d��d� tj|jdd�}tdg�� t �
|� �}�� |� �}� �fdd �}|� |� |S )
a
A request for the I{base} child of a L{File} succeeds with a resource
for the I{base<extension>} file in the path the L{File} was created
with if such a file exists and the L{File} has been configured to
ignore the I{<extension>} extension.
r{ r| zfoo.quuxrK )r� r; r c s �� d�� j�d� d S )Nr r| r r r" r% r& r' � s zAStaticFileTests.test_ignoredExtensionsIgnored.<locals>.cbRenderedr� rS r% r" r& � test_ignoredExtensionsIgnored� s
z-StaticFileTests.test_ignoredExtensionsIgnoredc C s� t | �� �}|�� |�d��� t�|j�}tdg�}d|_t �
||�}| � | � ||�� | �
|jt� | �
|j�d�dg� dS )z�
A request for a path which is a directory but does not have a trailing
slash will be redirected to a URL which does have a slash by L{File}.
Zfolders folders http://dummy/folder#baz?foo=bars locations http://dummy/folder/#baz?foo=barN)r rB rQ rU r
rA rJ r �urir rR �successResultOfr r rO r r� r� rY r% r% r& �+test_directoryWithoutTrailingSlashRedirects� s
�z;StaticFileTests.test_directoryWithoutTrailingSlashRedirectsc s< t � � t| �� �}|�� t�|j�}� fdd�|_� |fS )z�
Create a L{File} that when opened for reading, returns a L{StringIO}.
@return: 2-tuple of the opened "file" and the L{File}.
@rtype: L{tuple}
c s � S r7 r% r% ��fakeFiler% r& �<lambda> r z;StaticFileTests._makeFilePathWithStringIO.<locals>.<lambda>)�StringIOr rB Ztouchr
rA rJ r\ )r$ rJ rE r% r� r& �_makeFilePathWithStringIO
s z)StaticFileTests._makeFilePathWithStringIOc C sP | � � \}}tdg�}d|_| �t||�� | �d�|j�d� | �|j � dS )zm
A HEAD request opens the file, gets the size, and then closes it after
the request.
r r r N)
r� r r) r� r r r r �
assertTrue�closed�r$ r� rE r# r% r% r& �test_HEADClosesFile s
z#StaticFileTests.test_HEADClosesFilec C sZ | � � \}}tdg�}d|_dd� |_| �t||�� | �d�|j�d� | � |j
� dS )zQ
A GET request that is cached closes the file after the request.
r s GETc S s t jS r7 )r ZCACHED)�_r% r% r& r� 2 r z>StaticFileTests.test_cachedRequestClosesFile.<locals>.<lambda>r N) r� r r) ZsetLastModifiedr� r r r r r� r� r� r% r% r& � test_cachedRequestClosesFile* s
z,StaticFileTests.test_cachedRequestClosesFileN)%r2 r3 r4 r5 r rF rG rH r1 rV rZ r_ r` rb r Z isWindows�skiprh rl rz r� r� r� �sys�getfilesystemencoding�lowerr� r� r� r- r� r� r� r� r� r� r� r% r% r% r& r6 ? sF
��
r6 c @ s� e Zd ZdZd)dd�Zdd� Zdd� Zd d
� 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(� ZdS )*�StaticMakeProducerTestsz)
Tests for L{File.makeProducer}.
Nc C s4 t | �� �}|�|� t�|�� �}||_||_|S )z�
Make a L{static.File} resource that has C{content} for its content.
@param content: The L{bytes} to use as the contents of the resource.
@param type: Optional value for the content type of the resource.
)r rB rI r
rA Z _asBytesPath�encoding�type)r$ r� r� r� ZfileNamer r% r% r& �makeResourceWithContent? s
z/StaticMakeProducerTests.makeResourceWithContentc C s: i }|j �� D ]&\}}|�� �d�r|d ||�� < q|S )z�
Extract the content-* headers from the L{DummyRequest} C{request}.
This returns the subset of C{request.outgoingHeaders} of headers that
start with 'content-'.
s content-r )r� ZgetAllRawHeadersr� �
startswith)r$ r# �contentHeaders�k�vr% r% r& r� N s
z&StaticMakeProducerTests.contentHeadersc C sD | � d�}tg �}|�� � }|�||�}| �|tj� W 5 Q R X dS )zp
makeProducer when no Range header is set returns an instance of
NoRangeStaticProducer.
r N)r� r rw � makeProducerrW r
�NoRangeStaticProducer)r$ r r# rE �producerr% r% r& �,test_noRangeHeaderGivesNoRangeStaticProducer\ s
zDStaticMakeProducerTests.test_noRangeHeaderGivesNoRangeStaticProducerc C sF | � d�}tg �}|�� �"}|�||� | �tj|j� W 5 Q R X dS )zp
makeProducer when no Range header is set sets the responseCode on the
request to 'OK'.
r N)r� r rw r� r r �OKrO )r$ r r# rE r% r% r& �test_noRangeHeaderSets200OKh s
z3StaticMakeProducerTests.test_noRangeHeaderSets200OKc C sr d}d}d}| j d| ||d�}tg �}|�� �8}|�||� | �t|�t|�t|�d�| �|�� W 5 Q R X dS )zo
makeProducer when no Range header is set sets the Content-* headers
for the response.
�{ �
text/plain�gzip� a�r� r� )� content-typer~ � content-encodingN)r� r rw r� r r r
r� )r$ Zlength� contentType�contentEncodingr r# rE r% r% r& �$test_noRangeHeaderSetsContentHeaderst s" �
��z<StaticMakeProducerTests.test_noRangeHeaderSetsContentHeadersc C sR t g �}|j�dd� | �d�}|�� � }|�||�}| �|tj� W 5 Q R X dS )z�
makeProducer when the Range header requests a single byte range
returns an instance of SingleRangeStaticProducer.
� range� bytes=1-3� abcdefN� r �requestHeaders� addRawHeaderr� rw r� rW r
�SingleRangeStaticProducer�r$ r# r rE r� r% r% r& �.test_singleRangeGivesSingleRangeStaticProducer� s
zFStaticMakeProducerTests.test_singleRangeGivesSingleRangeStaticProducerc C sT t g �}|j�dd� | �d�}|�� �"}|�||� | �tj|j � W 5 Q R X dS )z�
makeProducer when the Range header requests a single, satisfiable byte
range sets the response code on the request to 'Partial Content'.
r� r� r� N�
r r� r� r� rw r� r r �PARTIAL_CONTENTrO �r$ r# r rE r% r% r& �%test_singleRangeSets206PartialContent� s
�z=StaticMakeProducerTests.test_singleRangeSets206PartialContentc C sv t g �}|j�dd� d}d}| jd||d�}|�� �6}|�||� | �t|�t|�ddd �| �|�� W 5 Q R X d
S ) ��
makeProducer when the Range header requests a single, satisfiable byte
range sets the Content-* headers appropriately.
r� r� r� r� r� r� s bytes 1-3/6r )r� r� �
content-ranger~ N) r r� r� r� rw r� r r r� )r$ r# r� r� r rE r% r% r& �"test_singleRangeSetsContentHeaders� s
��z:StaticMakeProducerTests.test_singleRangeSetsContentHeadersc C sR t g �}|j�dd� | �d�}|�� � }|�||�}| �|tj� W 5 Q R X dS )z�
makeProducer still returns an instance of L{SingleRangeStaticProducer}
when the Range header requests a single unsatisfiable byte range.
r� �
bytes=4-10� abcNr� r� r% r% r& �=test_singleUnsatisfiableRangeReturnsSingleRangeStaticProducer� s
zUStaticMakeProducerTests.test_singleUnsatisfiableRangeReturnsSingleRangeStaticProducerc C sT t g �}|j�dd� | �d�}|�� �"}|�||� | �tj|j � W 5 Q R X dS )z�
makeProducer sets the response code of the request to of 'Requested
Range Not Satisfiable' when the Range header requests a single
unsatisfiable byte range.
r� r� r� N�
r r� r� r� rw r� r r � REQUESTED_RANGE_NOT_SATISFIABLErO r� r% r% r& �?test_singleUnsatisfiableRangeSets416ReqestedRangeNotSatisfiable� s
�zWStaticMakeProducerTests.test_singleUnsatisfiableRangeSets416ReqestedRangeNotSatisfiablec C sf t g �}|j�dd� d}| jd|d�}|�� �,}|�||� | �dddd �| �|�� W 5 Q R X d
S ) z�
makeProducer when the Range header requests a single, unsatisfiable
byte range sets the Content-* headers appropriately.
r� r� r� r� �r� �
text/plain� 0� bytes */3�r� r~ r� N�r r� r� r� rw r� r r� �r$ r# r� r rE r% r% r& �/test_singleUnsatisfiableRangeSetsContentHeaders� s
��zGStaticMakeProducerTests.test_singleUnsatisfiableRangeSetsContentHeadersc C sf t g �}|j�dd� d}| jd|d�}|�� �,}|�||� | �dddd �| �|�� W 5 Q R X d
S ) z�
makeProducer when the Range header requests a single byte range that
partly overlaps the resource sets the Content-* headers appropriately.
r� s
bytes=2-10r� r� r� r� � 1s bytes 2-2/3r� Nr� r� r% r% r& �6test_singlePartiallyOverlappingRangeSetsContentHeaders� s
��zNStaticMakeProducerTests.test_singlePartiallyOverlappingRangeSetsContentHeadersc C sR t g �}|j�dd� | �d�}|�� � }|�||�}| �|tj� W 5 Q R X dS )z�
makeProducer when the Range header requests a single byte range
returns an instance of MultipleRangeStaticProducer.
r� �
bytes=1-3,5-6r� N� r r� r� r� rw r� rW r
�MultipleRangeStaticProducerr� r% r% r& �2test_multipleRangeGivesMultipleRangeStaticProducer� s
zJStaticMakeProducerTests.test_multipleRangeGivesMultipleRangeStaticProducerc C sT t g �}|j�dd� | �d�}|�� �"}|�||� | �tj|j � W 5 Q R X dS )z�
makeProducer when the Range header requests a multiple satisfiable
byte ranges sets the response code on the request to 'Partial
Content'.
r� r� r� Nr� r� r% r% r& �'test_multipleRangeSets206PartialContent s
�z?StaticMakeProducerTests.test_multipleRangeSets206PartialContentc C s� t g �}|j�dd� | jddd�}|�� ��}|�||�}| �|�}| �tddg�t|� � �� d}|j
D ]\}}} |t |�7 }qj| �t |�|d � | �
d|� |d }
| �d t�d
|
�� | �d |� W 5 Q R X d S ) r� r� r� s abcdefghijklr� )r� r~ r� � Ns( multipart/byteranges; boundary="[^"]*"\Zr� )r r� r� r� rw r� r� r �set�keysZ rangeInforC r
�assertInZassertNotIdentical�re�matchZ assertNotIn) r$ r# r rE r� r� ZexpectedLength�boundary�offset�sizer� r% r% r& �#test_mutipleRangeSetsContentHeaders s6 �
� ��z;StaticMakeProducerTests.test_mutipleRangeSetsContentHeadersc C sR t g �}|j�dd� | �d�}|�� � }|�||�}| �|tj� W 5 Q R X dS )z�
makeProducer still returns an instance of L{SingleRangeStaticProducer}
when the Range header requests multiple ranges, none of which are
satisfiable.
r� � bytes=10-12,15-20r� Nr� r� r% r% r& �Btest_multipleUnsatisfiableRangesReturnsMultipleRangeStaticProducer2 s
zZStaticMakeProducerTests.test_multipleUnsatisfiableRangesReturnsMultipleRangeStaticProducerc C sT t g �}|j�dd� | �d�}|�� �"}|�||� | �tj|j � W 5 Q R X dS )z�
makeProducer sets the response code of the request to of 'Requested
Range Not Satisfiable' when the Range header requests multiple ranges,
none of which are satisfiable.
r� r� r� Nr� r� r% r% r& �Btest_multipleUnsatisfiableRangesSets416ReqestedRangeNotSatisfiable@ s
�zZStaticMakeProducerTests.test_multipleUnsatisfiableRangesSets416ReqestedRangeNotSatisfiablec C st t g �}|j�dd� d}|j�dd� | jd|d�}|�� �,}|�||� | �ddd d
�| �|�� W 5 Q R X d S ) z�
makeProducer when the Range header requests multiple ranges, none of
which are satisfiable, sets the Content-* headers appropriately.
r� r� r� r� r� r� r� r� r� )r~ r� r� Nr� r� r% r% r& �1test_multipleUnsatisfiableRangeSetsContentHeadersO s
��zIStaticMakeProducerTests.test_multipleUnsatisfiableRangeSetsContentHeadersc C sT t g �}|j�dd� | �d�}|�� �"}|�||� | �tj|j � W 5 Q R X dS )z�
makeProducer when the Range header requests multiple ranges, at least
one of which matches, sets the response code to 'Partial Content'.
r� s bytes=1-3,100-200r� Nr� r� r% r% r& � test_oneSatisfiableRangeIsEnoughb s
�z8StaticMakeProducerTests.test_oneSatisfiableRangeIsEnough)NN)r2 r3 r4 r5 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r% r% r% r& r� 9 s(
"r� c @ s e Zd ZdZdd� Zdd� ZdS )�StaticProducerTestsz3
Tests for the abstract L{StaticProducer}.
c C s* t � }t�d|�}|�� | �|j� dS )zu
L{StaticProducer.stopProducing} closes the file object the producer is
producing data from.
N)r� r
�StaticProducer�
stopProducingr� r� �r$ Z
fileObjectr� r% r% r& � test_stopProducingClosesFilew s z0StaticProducerTests.test_stopProducingClosesFilec C s0 t � }t�tg �|�}|�� | �d|j� dS )z�
L{StaticProducer.stopProducing} sets the request instance variable to
None, which indicates to subclasses' resumeProducing methods that no
more data should be produced.
N)r� r
r� r r� �assertIdenticalr# r� r% r% r& �#test_stopProducingSetsRequestToNone� s z7StaticProducerTests.test_stopProducingSetsRequestToNoneN)r2 r3 r4 r5 r� r r% r% r% r& r� r s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �NoRangeStaticProducerTestsz-
Tests for L{NoRangeStaticProducer}.
c C s t tjt�dd�� dS )zG
L{NoRangeStaticProducer} implements L{IPullProducer}.
N)r r �
IPullProducerr
r� rk r% r% r& � test_implementsIPullProducer� s
�z7NoRangeStaticProducerTests.test_implementsIPullProducerc C s< t g �}d}t�|t|��}|�� | �|d�|j�� dS )zs
L{NoRangeStaticProducer.resumeProducing} writes content from the
resource to the request.
r� r N)r r
r� r� �startr r r �r$ r# r� r� r% r% r& �#test_resumeProducingProducesContent� s �z>NoRangeStaticProducerTests.test_resumeProducingProducesContentc C st t g �}tjj}dd| d }t�|t|��}|�� |d|� ||d| � |d| d� g}| �||j � dS )z�
L{NoRangeStaticProducer.start} writes at most
C{abstract.FileDescriptor.bufferSize} bytes of content from the
resource to the request at once.
r� � r: r N)
r r �FileDescriptor�
bufferSizer
r� r� r r r �r$ r# r
r� r� �expectedr% r% r& �!test_resumeProducingBuffersOutput� s �
�z<NoRangeStaticProducerTests.test_resumeProducingBuffersOutputc C sJ t g �}|�� }g }|�|j� t�|td��}|�� | �dg|� dS )z�
L{NoRangeStaticProducer.resumeProducing} calls finish() on the request
after it is done producing content.
r� N) r � notifyFinishr* �appendr
r� r� r r �r$ r# ZfinishDeferredZ callbackListr� r% r% r& �test_finishCalledWhenDone� s �z4NoRangeStaticProducerTests.test_finishCalledWhenDoneN�r2 r3 r4 r5 r r r
r r% r% r% r& r � s
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � SingleRangeStaticProducerTestsz1
Tests for L{SingleRangeStaticProducer}.
c C s t tjt�dddd�� dS )zK
L{SingleRangeStaticProducer} implements L{IPullProducer}.
N)r r r r
r� rk r% r% r& r � s �z;SingleRangeStaticProducerTests.test_implementsIPullProducerc C sH t g �}d}t�|t|�dd�}|�� | �|dd� d�|j�� dS )z�
L{SingleRangeStaticProducer.resumeProducing} writes the given amount
of content, starting at the given offset, from the resource to the
request.
r� r: � � r N)r r
r� r� r r r r r r% r% r& r � s �zBSingleRangeStaticProducerTests.test_resumeProducingProducesContentc C sn t g �}tjj}d| }t�|t|�d|d �}|�� |d|d � ||d |d � g}| �||j � dS )z�
L{SingleRangeStaticProducer.start} writes at most
C{abstract.FileDescriptor.bufferSize} bytes of content from the
resource to the request at once.
r� r: �
� N)
r r r r
r
r� r� r r r r r% r% r& r
� s ��z@SingleRangeStaticProducerTests.test_resumeProducingBuffersOutputc C sN t g �}|�� }g }|�|j� t�|td�dd�}|�� | �dg|� dS )z�
L{SingleRangeStaticProducer.resumeProducing} calls finish() on the
request after it is done producing content.
r� r: N) r r r* r r
r� r� r r r r% r% r& r s �z8SingleRangeStaticProducerTests.test_finishCalledWhenDoneNr r% r% r% r& r � s
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � MultipleRangeStaticProducerTestsz3
Tests for L{MultipleRangeStaticProducer}.
c C s t tjt�ddd�� dS )zM
L{MultipleRangeStaticProducer} implements L{IPullProducer}.
N)r r r r
r� rk r% r% r& r s �z=MultipleRangeStaticProducerTests.test_implementsIPullProducerc C sB t g �}d}t�|t|�ddg�}|�� | �dd�|j�� dS )z�
L{MultipleRangeStaticProducer.resumeProducing} writes the requested
chunks of content from the resource to the request, with the supplied
boundaries in between each chunk.
r� )r� r: r )� 2r� r: s 1bcd2fr N)r r
r� r� r r r r r r% r% r& r ' s �zDMultipleRangeStaticProducerTests.test_resumeProducingProducesContentc C st t g �}d}t�|t|�dddg�}d|_|�� d|dd� d |d
d � |d d � d
g}| �||j� dS )a�
L{MultipleRangeStaticProducer.start} writes about
C{abstract.FileDescriptor.bufferSize} bytes of content from the
resource to the request at once.
To be specific about the 'about' above: it can write slightly more,
for example in the case where the first boundary plus the first chunk
is less than C{bufferSize} but first boundary plus the first chunk
plus the second boundary is more, but this is unimportant as in
practice the boundaries are fairly small. On the other side, it is
important for performance to bundle up several small chunks into one
call to request.write.
s 01234567890123456789)r� r r )� br� r )� cr r r r� r r r r� r � r N)r r
r� r� r
r r r )r$ r# r� r� r r% r% r& r
7 s � �zBMultipleRangeStaticProducerTests.test_resumeProducingBuffersOutputc C sN t g �}|�� }g }|�|j� t�|td�dg�}|�� | �dg|� dS )z�
L{MultipleRangeStaticProducer.resumeProducing} calls finish() on the
request after it is done producing content.
r� )r r: r N) r r r* r r
r� r� r r r r% r% r& r U s �z:MultipleRangeStaticProducerTests.test_finishCalledWhenDoneNr 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 )-�
RangeTestsa�
Tests for I{Range-Header} support in L{twisted.web.static.File}.
@type file: L{file}
@ivar file: Temporary (binary) file containing the content to be served.
@type resource: L{static.File}
@ivar resource: A leaf web resource using C{file} as content.
@type request: L{DummyRequest}
@ivar request: A fake request, requesting C{resource}.
@type catcher: L{list}
@ivar catcher: List which gathers all log information.
c C sp t | �� �}d| _|�| j� |�� | _t�| jj�| _ d| j _
t dg�| _ | jj| j _
g | _t�| jj� dS )a
Create a temporary file with a fixed payload of 64 bytes. Create a
resource for that file and create a request which will be for that
resource. Each test can set a different range header to test different
aspects of the implementation.
s@ �u�E�7� ��a0y0S���M 6k{ � мkY�I ��oZ �b�k� &�%݂q/AY�r: r N)r rB �payloadrI r\ rE r
rA r� r ZisLeafr r# r� �catcherr Z addObserverr )r$ rJ r% r% r& �setUpw s
zRangeTests.setUpc C s | j �� t�| jj� dS )zB
Clean up the resource file and the log observer.
N)rE r] r ZremoveObserverr r rk r% r% r& �tearDown� s
zRangeTests.tearDownc C s8 | j �� }| �|d d |� | �| j g d|f � dS )zU
Asserts that a given log message occurred with an expected message.
�messager z An additional log occurred: %rN)r �popr )r$ r ZlogItemr% r% r& �
_assertLogged� s
�zRangeTests._assertLoggedc C sn | j j}| �t|d� | �t|d� | �t|d� | �t|d� | �t|d� | �t|d� | �t|d� dS ) zw
L{File._parseRangeHeader} raises L{ValueError} when passed
syntactically invalid byte ranges.
s bytess unknown=1-2s bytes=3s bytes=-s
bytes=foo-s
bytes=-foos bytes=5-4N)r �_parseRangeHeaderr/ �
ValueError)r$ �fr% r% r& �test_invalidRanges� s z RangeTests.test_invalidRangesc C s | � | j�d�dg� dS )z�
A single bytes range without an explicit stop position is parsed into a
two-tuple giving the start position and L{None}.
s bytes=0-)r NN�r r r% rk r% r% r& �test_rangeMissingStop� s
�z RangeTests.test_rangeMissingStopc C s | � | j�d�dg� dS )z�
A single bytes range without an explicit start position is parsed into
a two-tuple of L{None} and the end position.
s bytes=-3)Nr Nr) rk r% r% r& �test_rangeMissingStart� s
�z!RangeTests.test_rangeMissingStartc C s | � | j�d�dg� dS )z�
A single bytes range with explicit start and stop positions is parsed
into a two-tuple of those positions.
s bytes=2-5)r r� Nr) rk r% r% r& �
test_range� s
�zRangeTests.test_rangec C s� | � | j�d�dg� | � | j�d�dg� | � | j�d�dg� | � | j�d�dg� | � | j�d�dg� | � | j�d�dg� dS ) z�
A single bytes range with whitespace in allowed places is parsed in
the same way as it would be without the whitespace.
s bytes=1-2 �r: r s bytes =1-2 s
bytes= 1-2s
bytes=1 -2s
bytes=1- 2s
bytes=1-2 Nr) rk r% r% r& �test_rangeWithSpace� s0
�
�
�
�
�
�z RangeTests.test_rangeWithSpacec C s | � | j�d�dg� dS )z�
If there are multiple byte ranges but only one is non-null, the
non-null range is parsed and its start and stop returned.
s bytes=1-2,
, , r- Nr) rk r% r% r& �test_nullRangeElements� s
�z!RangeTests.test_nullRangeElementsc C s | � | j�d�ddg� dS )zd
If multiple byte ranges are specified their starts and stops are
returned.
s
bytes=1-2,3-4r- )r r Nr) rk r% r% r& �test_multipleRanges� s
�z RangeTests.test_multipleRangesc C s< | j j�dd� | j�| j � | �td�| j j��d� dS )zp
A correct response to a range request is as long as the length of the
requested range.
r� s
bytes=0-43r �, N) r# r� r� r r0 r rC r r rk r% r% r& �test_bodyLength s zRangeTests.test_bodyLengthc C s� d}| j j�d|� | j�| j � d|�� f }| �|� | �d�| j j �| j
� | �| j j t j
� | �| j j�d�d tt| j
��� dS )a&
An incorrect range request (RFC 2616 defines a correct range request as
a Bytes-Unit followed by a '=' character followed by a specific range.
Only 'bytes' is defined) results in the range header value being logged
and a normal 200 response being sent.
s foobar=0-43r� z"Ignoring malformed Range header %rr r~ r N)r# r� r� r r0 rg r$ r r r r rO r r� r� r� r
rC )r$ �ranger r% r% r& �test_invalidRangeRequest s
�z#RangeTests.test_invalidRangeRequestc C s� d| }|� |�}| �d|d � | �d|d � g }|dd� D ]�}|� dd�\}}} }
} |d | } | �d|� | �d|
� t�d
| tjtjB ��d�}
t�d | tjtjB ��� \}}}|�|
|||f| d �� qB|S )
z�
Parse C{body} as a multipart MIME response separated by C{boundary}.
Note that this with fail the calling test on certain syntactic
problems.
s
--r r s --
���r: s
r �
s ^content-type: (.*)$s1 ^content-range: bytes ([0-9]+)-([0-9]+)/([0-9]+)$)� contentType� contentRange� body) �splitr r� �search�I�M�group�groupsr )r$ �bodyr� �sep�partsZ parsed_parts�partZbeforeZheader1Zheader2ZblankZpartBodyZheadersZpartContentTypeValuer �endr� r% r% r& �parseMultipartBody s:
��
���z RangeTests.parseMultipartBodyc C s2 dddg}d� dd� |D ��}| jj�dd| � | j�| j� | �| jjtj � t
� d | jj �
d
�d ��d �}| �d
� | jj�|�}| �t|�t|�� t||�D ]�\}\}}| �t| jj�|d � |d \}} }
| �t|�|� | �t| �|� | �t|
�| j�� � | �| j||d � |d � q�dS )zo
The response to a request for multiple bytes ranges is a MIME-ish
multipart response.
�r r )� � )�( �2 � ,c S s g | ]\}}t d ||f ��qS �z%s-%s�r ��.0�s�er% r% r& �
<listcomp>B s z8RangeTests.test_multipleRangeRequest.<locals>.<listcomp>r� � bytes=�' ^multipart/byteranges; boundary="(.*)"$r� r r: r r7 r8 r9 N)r r# r� r� r r0 r rO r r� r� r� r� r� r> rE r rC �zipr r� �int� getFileSizer � r$ Z startEndsZrangeHeaderValuer� rB rC rP rQ r rD r� r% r% r& �test_multipleRangeRequest< s0
����z$RangeTests.test_multipleRangeRequestc C sN ddt | j�d fg}d�dd� |D ��}| jj�dd| � | j�| j� | �| jj t
j � t �
d | jj�d
�d ��d �}| �d
�| jj�|�}| �t |�t |�� t||�D ]�\}\}}| �t| jj�|d � |d \}} }
| �t|�|� | �t| �t|| j�� d �� | �t|
�| j�� � | �| j||d � |d � q�dS )z�
The response to a request for multiple bytes ranges is a MIME-ish
multipart response, even when one of the ranged falls off the end of
the resource.
rF rI r rK c S s g | ]\}}t d ||f ��qS rL rM rN r% r% r& rR ] s zORangeTests.test_multipleRangeRequestWithRangeOverlappingEnd.<locals>.<listcomp>r� rS rT r� r r: r r7 r8 r9 N)rC r r r# r� r� r r0 r rO r r� r� r� r� r� r> rE r rU r r� rV �minrW rX r% r% r& �0test_multipleRangeRequestWithRangeOverlappingEndV s0
���� z;RangeTests.test_multipleRangeRequestWithRangeOverlappingEndc C s� | j j�dd� | j�| j � | �d�| j j�| jdd� � | �t d�| j j��d� | �| j j
t j � | �| j j
�d�d d � | �| j j
�d
�d d � dS ) z�
If the end byte position is omitted, then it is treated as if the
length of the resource was specified by the end byte position.
r� s bytes=23-r � N�) r� r s bytes 23-63/64r~ s 41�r# r� r� r r0 r r r r rC rO r r� r� r� rk r% r% r& �test_implicitEndq s ��zRangeTests.test_implicitEndc C s� | j j�dd� | j�| j � | �d�| j j�| jdd� � | �t d�| j j��d� | �| j j
t j � | �| j j
�d�d d � | �| j j
�d
�d d � dS ) z�
If the start byte position is omitted but the end byte position is
supplied, then the range is treated as requesting the last -N bytes of
the resource, where N is the end byte position.
r� s bytes=-17r i��N� r� r s bytes 47-63/64r~ s 17r^ rk r% r% r& �test_implicitStart� s ��z RangeTests.test_implicitStartc C s� | j j�dd� | j�| j � d�| j j�}| �|| jdd� � | �| j j t
j � | �| j j �
d�d d� | �tt|��| j j �
d �d � d
S ) z�
A correct response to a bytes range header request from A to B starts
with the A'th byte and ends with (including) the B'th byte. The first
byte of a page is numbered with 0.
r� s
bytes=3-43r r r1 r� r s
bytes 3-43/64r~ N�r# r� r� r r0 r r r r rO r r� r� r� r
rC �r$ r r% r% r& �test_explicitRange� s �
�z RangeTests.test_explicitRangec C s� | j j�dd� | j�| j � d�| j j�}| �|| jdd� � | �| j j t
j � | �| j j �
d�d d� | �tt|��| j j �
d �d � dS )
a
A correct response to a bytes range header request from A to B when B
is past the end of the resource starts with the A'th byte and ends
with the last byte of the resource. The first byte of a page is
numbered with 0.
r� s bytes=40-100r rI Nr� r s bytes 40-63/64r~ rb rc r% r% r& � test_explicitRangeOverlappingEnd� s �
�z+RangeTests.test_explicitRangeOverlappingEndc C sp | j j�dd� | j�| j � | �| j jtj� | �d� | j j
�| j � | �| j j �
d�d tt| j ��� dS )z�
If a range is syntactically invalid due to the start being greater than
the end, the range header is ignored (the request is responded to as if
it were not present).
r� s bytes=20-13r r~ r N)r# r� r� r r0 r rO r r� r r r r� r� r
rC rk r% r% r& �+test_statusCodeRequestedRangeNotSatisfiable� s �z6RangeTests.test_statusCodeRequestedRangeNotSatisfiablec C s� | j j�dd� | j�| j � | �| j jtj� | �d� | j j
�d� | �| j j � d�d d� | �| j j � d�d t
dt| j�f �� d S )
a
If a range is unsatisfiable due to the start not being less than the
length of the resource, the response is 416 (Requested range not
satisfiable) and no data is written to the response body (RFC 2616,
section 14.35.1).
r� s bytes=67-108r r~ r r� r� z
bytes */%dN)r# r� r� r r0 r rO r r� r r r� r� r rC r rk r% r% r& �test_invalidStartBytePos� s ���z#RangeTests.test_invalidStartBytePosN)r2 r3 r4 r5 r r! r$ r( r* r+ r, r. r/ r0 r2 r4 rE rY r[ r_ ra rd re rf rg r% r% r% r& r g s,
r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Ze�� srde_dd� Zdd � Zd d � Zd S ) �DirectoryListerTestsz.
Tests for L{static.DirectoryLister}.
c C s t dg�}||_|S r )r r� )r$ r� r# r% r% r& �_request� s
z DirectoryListerTests._requestc C sL t | �� �}|�� t�|j�}|�| �d��}| �d|� | �d|� dS )zm
L{static.DirectoryLister} prints the request uri as header of the
rendered content.
r s" <h1>Directory listing for foo</h1>s( <title>Directory listing for foo</title>N� r rB rQ r
rX rJ r0 ri r� �r$ rJ �listerr+ r% r% r& �test_renderHeader� s z&DirectoryListerTests.test_renderHeaderc C sL t | �� �}|�� t�|j�}|�| �d��}| �d|� | �d|� dS )zW
L{static.DirectoryLister} unquote the request uri before printing it.
s foo%20bars& <h1>Directory listing for foo bar</h1>s, <title>Directory listing for foo bar</title>Nrj rk r% r% r& �test_renderUnquoteHeader� s z-DirectoryListerTests.test_renderUnquoteHeaderc C sL t | �� �}|�� t�|j�}|�| �d��}| �d|� | �d|� dS )zl
L{static.DirectoryLister} escape "&", "<" and ">" after unquoting the
request uri.
s foo%26bars* <h1>Directory listing for foo&bar</h1>s0 <title>Directory listing for foo&bar</title>Nrj rk r% r% r& �test_escapeHeader s z&DirectoryListerTests.test_escapeHeaderc C sh t | �� �}|�� |�d��d� |�d��dd � t�|j�}|�| � d��}d}| �
||� dS ) ze
L{static.DirectoryLister} is able to list all the files inside a
directory.
�file1s content1�file2s content2i� r s� <tr class="odd">
<td><a href="file1">file1</a></td>
<td>8B</td>
<td>[text/html]</td>
<td></td>
</tr>
<tr class="even">
<td><a href="file2">file2</a></td>
<td>7K</td>
<td>[text/html]</td>
<td></td>
</tr>N) r rB rQ rU rI r
rX rJ r0 ri r� �r$ rJ rl r+ r@ r% r% r& �test_renderFiles s z%DirectoryListerTests.test_renderFilesc C s` t | �� �}|�� |�d��� |�d��� t�|j�}|�| �d��}d}| � ||� dS )zk
L{static.DirectoryLister} is able to list all the directories inside
a directory.
�dir1zdir2 & 3r s� <tr class="odd">
<td><a href="dir1/">dir1/</a></td>
<td></td>
<td>[Directory]</td>
<td></td>
</tr>
<tr class="even">
<td><a href="dir2%20%26%203/">dir2 & 3/</a></td>
<td></td>
<td>[Directory]</td>
<td></td>
</tr>N�
r rB rQ rU r
rX rJ r0 ri r� rr r% r% r& �test_renderDirectories0 s z+DirectoryListerTests.test_renderDirectoriesc C sv t | �� �}|�� |�d��� |�d��� |�d��� tj|jddgd�}|�| �d��}d}| � ||� dS )z�
L{static.DirectoryLister} takes an optional C{dirs} argument that
filter out the list of directories and files printed.
rt Zdir2Zdir3)�dirsr s� <tr class="odd">
<td><a href="dir1/">dir1/</a></td>
<td></td>
<td>[Directory]</td>
<td></td>
</tr>
<tr class="even">
<td><a href="dir3/">dir3/</a></td>
<td></td>
<td>[Directory]</td>
<td></td>
</tr>Nru rr r% r% r& �test_renderFilteredK s z(DirectoryListerTests.test_renderFilteredc C s� t �d�}dd� td�D �}|�|�}| �t|�d� | �|d �d�� | �|d �d�� | �|d �d�� | �|d
�d�� | �|d �d�� dS ) zu
L{static.DirectoryLister} gives an alternate class for each odd and
even rows in the table.
Nc S s g | ]}d d d d d d��qS )r )�href�textr� r� r� r% )rO �ir% r% r& rR l s ��z8DirectoryListerTests.test_oddAndEven.<locals>.<listcomp>r� r z<tr class="odd">r: z<tr class="even">r r r )r
rX r3 Z_buildTableContentr rC r� r� )r$ rl �elementsr� r% r% r& �test_oddAndEvenf s
�
z$DirectoryListerTests.test_oddAndEvenc C sP t | �� �}|�� t�|j�}| �d�}|�|� | �|j �
d�d d� dS )z�
L{static.DirectoryLister} produces a MIME-type that indicates that it is
HTML, and includes its charset (UTF-8).
r r� r s text/html; charset=utf-8N) r rB rQ r
rX rJ ri r0 r r� r� )r$ rJ rl Zreqr% r% r& �test_contentTypex s
�z%DirectoryListerTests.test_contentTypec C s� t | �� �}|�� |�d��d� |�d��d� |�d��d� |�d��d� t�|j�}|�� d d
d d d
�}t j
|j|d�}|� |�\}}| � |g � | � |dddddd�dddddd�dddddd�dddddd�g� dS )zl
L{static.DirectoryLister} is able to detect mimetype and encoding of
listed files.
z file1.txt� file1zfile2.pys pythonz
file3.conf.gzs conf compressedzfile4.diff.bz2s diff compressedr� z text/pythonztext/configurationz text/diff)z.txtz.pyz.confz.diff)� contentTypesr Z5Bz [text/plain])r� ry r� rz r� Z6Bz
[text/python]z[gzip]Z15Bz[text/configuration]z[bzip2]z [text/diff]N)
r rB rQ rU rI �os�listdirrJ �sortr
rX �_getFilesAndDirectoriesr )r$ rJ � directoryr� rl rw �filesr% r% r& �test_mimeTypeAndEncodings� sT � �����z.DirectoryListerTests.test_mimeTypeAndEncodingsc C s� t | �� �}|�� |�d�}|�d� |�|�d�� |�� t�|j �}t
� |j �}|� � |�
|�\}}| �|g � | �|g � dS )z�
If on the file in the listing points to a broken symlink, it should not
be returned by L{static.DirectoryLister._getFilesAndDirectories}.
rp r rq N)r rB rQ rU rI ZlinkTo�remover
rX rJ r� r� r� r� r )r$ rJ rp rl r� rw r� r% r% r& �test_brokenSymlink� s
z'DirectoryListerTests.test_brokenSymlinkzNo symlink supportc s\ t ��� �}|�� t�|j�}��d�� t�|� �}t |� �}� �fdd�}|�
|� |S )zu
Any child resource of L{static.DirectoryLister} renders an HTTP
I{NOT FOUND} response code.
r c s �� � jtj� d S r7 )r rO r Z NOT_FOUNDr r" r% r& r' � s z>DirectoryListerTests.test_childrenNotFound.<locals>.cbRendered) r rB rQ r
rX rJ ri r rR r r* )r$ rJ rl rU ry r' r% r" r& �test_childrenNotFound� s
z*DirectoryListerTests.test_childrenNotFoundc C sL t | �� �}t�|j�}| �t|�d|jf � | �t|�d|jf � dS )zR
L{static.DirectoryLister.__repr__} gives the path of the lister.
z<DirectoryLister of %r>N)r rB r
rX rJ r �reprr� )r$ rJ rl r% r% r& � test_repr� s
�
�z DirectoryListerTests.test_reprc C sp | � t�d�d� | � t�d�d� | � t�d�d� | � t�d�d� | � t�d �d
� | � t�d �d � d
S )zi
L{static.formatFileSize} format an amount of bytes into a more readable
format.
r Z0Br� Z123Bi� Z4Ki�͇ Z8Mi�X�IZ1Gl P�c}Z1149GN)r r
ZformatFileSizerk r% r% r& �test_formatFileSize� s z(DirectoryListerTests.test_formatFileSizeN)r2 r3 r4 r5 ri rm rn ro rs rv rx r} r~ r� r� r Z_supportsSymlinksr� r� r� r� r% r% r% r& rh � s"
/ rh c @ s> e Zd ZdZe� Zdd� Zdd� Zdd� Zdd � Z d
d � Z
d S )
�LoadMimeTypesTestsz�
Tests for the MIME type loading routine.
@cvar UNSET: A sentinel to signify that C{self.paths} has not been set by
the mock init.
c C s | j | _d S r7 )�UNSET�pathsrk r% r% r& r � s zLoadMimeTypesTests.setUpc C s
|| _ dS )z�
A mock L{mimetypes.init} that records the value of the passed C{paths}
argument.
@param paths: The paths that will be recorded.
N)r� �r$ r� r% r% r& � _fakeInit s z LoadMimeTypesTests._fakeInitc C s t j| jd� | �| jd� dS )zE
By default, L{None} is passed to C{mimetypes.init}.
��initN�r
�
loadMimeTypesr� r r� rk r% r% r& �test_defaultArgumentIsNone
s z-LoadMimeTypesTests.test_defaultArgumentIsNonec C s, dddg}t j|| jd� | �| j|� dS )zI
Passed MIME type files are passed to C{mimetypes.init}.
�x�y�zr� Nr� r� r% r% r& �test_extraLocationsWork s
z*LoadMimeTypesTests.test_extraLocationsWorkc C sd t tdd�r0t�tj�}| �|jd jtj � n0t�
tj�\}}}}||� d� }| �|tj � dS )z:
By default, C{mimetypes.init} is called.
� signatureNr� ) �getattr�inspectr� r
r� ZassertIsZ
parameters�default� mimetypesr� Z
getargspec�index)r$ r� �argsr� �defaultsZ defaultInitr% r% r& �test_usesGlobalInitFunction s �z.LoadMimeTypesTests.test_usesGlobalInitFunctionN) r2 r3 r4 r5 �objectr� r r� r� r� r� r% r% r% r& r� � s
r� c @ s e Zd Zdd� ZdS )�StaticDeprecationTestsc C sP ddl m} |tdg�� | �| jg�}| �t|�d� | �|d d d� dS )z?
L{twisted.web.static.addSlash} is deprecated.
r )�addSlashr r: r" z<twisted.web.static.addSlash was deprecated in Twisted 16.0.0N)Ztwisted.web.staticr� r Z
flushWarnings�test_addSlashDeprecatedr rC )r$ r� r? r% r% r& r� 1 s �z.StaticDeprecationTests.test_addSlashDeprecatedN)r2 r3 r4 r� r% r% r% r& r� / s r� )4r5 rn r� r� r� r� r� r? �ior r� Zzope.interface.verifyr Ztwisted.internetr r Ztwisted.python.runtimer Ztwisted.python.filepathr Ztwisted.pythonr r Ztwisted.python.compatr
r Ztwisted.trial.unittestr Z twisted.webr
r r r Ztwisted.web.serverr Z twisted.web.test.requesthelperr Ztwisted.web.test._utilr Ztwisted.web._responsesr r r6 r� r� r r r r rh r� r� r% r% r% r& �<module> sP } ; EEN } 9