File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_distrib.cpython-310.pyc
o
�btF � @ sb d Z ddlmZ ddlmZ zddlZW n ey dZY nw eZddlm Z ddl
m Z ddl m
Z
mZ ddlmZ dd lmZmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddl m Z m Z m Z m Z m!Z! ddl"m#Z# ddl$m%Z% ddl&m'Z'm(Z( G dd� de j)�Z*G dd� dej+�Z+G dd� de,�Z-G dd� de�Z.G dd� d�Z/G d d � d e�Z0dS ) z#
Tests for L{twisted.web.distrib}.
� )�abspath)� parseStringN)�skipIf)� verifyObject)�defer�reactor)�globalLogPublisher)�failure�filepath)�pb��
SIZE_LIMIT)�
proto_helpers)�TestCase)�client�distrib�resource�server�static)�Headers)�_render)� DummyChannel� DummyRequestc @ s e Zd ZdS )�MySiteN)�__name__�
__module__� __qualname__� r r �?/usr/lib/python3/dist-packages/twisted/web/test/test_distrib.pyr # s r c @ s e Zd ZdZdZdd� ZdS )�PBServerFactoryz�
A PB server factory which keeps track of the most recent protocol it
created.
@ivar proto: L{None} or the L{Broker} instance most recently returned
from C{buildProtocol}.
Nc C s t j�| |�| _| jS �N)r r �
buildProtocol�proto)�self�addrr r r r! 2 s z PBServerFactory.buildProtocol)r r r �__doc__r" r! r r r r r ' s r c @ s e Zd ZdZdS )�ArbitraryErrorz%
An exception for this test.
N)r r r r% r r r r r& 7 s r& c @ s� e Zd ZdZdZdZdZdd� Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) � DistribTestsNc s� t �� t �� g� | jdur | jjdur | jj�� fdd�� n� d �d� | jdurF| jjdurF| jjj�� fdd�� | jjjj �
� n� d �d� | j durZ� � | j �
� � | jdurg� � | j�
� � t �� �S )z�
Clean up all the event sources left behind by either directly by
test methods or indirectly via some distrib API.
Nc � � d � d �S �Nr ��callbackr ��dlr r �<lambda>J � z'DistribTests.tearDown.<locals>.<lambda>r c r( )N� r* r r, r r r. N r/ r0 )r �Deferred�f1r" �notifyOnDisconnectr+ �sub� publisher�broker� transport�loseConnection�port1�append�
stopListening�port2�
gatherResults�r# r r, r �tearDownC s
zDistribTests.tearDownc C s� t �� }|�dt�dd�� t�|�}tt� |��| _
t � d| j
�| _
t�d| j
�� j�| _t �� }|�d| j� t|�}t � d|�| _t�t �}d| j�� j� d�}|�d �}|�d
|�}|�tj� |�| jd� |S ) Ns theres rootz
text/plainr � 127.0.0.1s herezhttp://127.0.0.1:z /here/there�ascii� GET)r �Resource�putChildr �Datar �Siter r �ResourcePublisherr2 r � listenTCPr9 �ResourceSubscription�getHost�portr4 r r< r �Agent�encode�request� addCallback�readBody� assertEqual)r# �r1�site1�r2�f2�agent�url�dr r r � testDistribX s"
zDistribTests.testDistribc C s� t �� }|�d|� t�|�}tt�|�� | _}t j
d|dd�}| � |j � |�
� }t�|j|j� | _}t�|�}t j
d|dd�} | � | j � | �
� }
| |
fS )z�
Set up a resource on a distrib site using L{ResourcePublisher}.
@param child: The resource to publish using distrib.
@return: A tuple consisting of the host and port on which to contact
the created site.
s childr r@ )� interface)r rC rD r rF r r rG r2 r rH �
addCleanupr; rJ rI �hostrK r4 ) r# �child� distribRoot� distribSite�distribFactory� distribPortr$ �mainRoot�mainSite�mainPort�mainAddrr r r �_setupDistribServerl s
�
z DistribTests._setupDistribServerc K s\ | � |�\}}t�t�}d|j� d|j� d�}|�d�}|jd|fi |��}|�tj � |S )a�
Set up a resource on a distrib site using L{ResourcePublisher} and
then retrieve it from a L{ResourceSubscription} via an HTTP client.
@param child: The resource to publish using distrib.
@param **kwargs: Extra keyword arguments to pass to L{Agent.request} when
requesting the resource.
@return: A L{Deferred} which fires with the result of the request.
�http://�:�/childrA rB )
rf r rL r r\ rK rM rN rO rP )r# r] �kwargsrd re rV rW rX r r r � _requestTest� s
zDistribTests._requestTestc K s^ | � |�\}}d|j� d|j� d�}|�d�}t�t�jd|fi |��}dd� }|�|� |S )a;
Set up a resource on a distrib site using L{ResourcePublisher} and
then retrieve it from a L{ResourceSubscription} via an HTTP client.
@param child: The resource to publish using distrib.
@param **kwargs: Extra keyword arguments to pass to L{Agent.request} when
requesting the resource.
@return: A L{Deferred} which fires with a tuple consisting of a
L{twisted.test.proto_helpers.AccumulatingProtocol} containing the
body of the response and an L{IResponse} with the response itself.
rg rh ri rA rB c s8 t �� � ��� � t�� }� _|�� �fdd�� |S )Nc s � �fS r r )�_��protocol�responser r r. � s zGDistribTests._requestAgentTest.<locals>.cbCollectBody.<locals>.<lambda>)r �AccumulatingProtocol� deliverBodyr r1 �closedDeferredrO )ro rX r rm r �
cbCollectBody� s
z5DistribTests._requestAgentTest.<locals>.cbCollectBody) rf r\ rK rM r rL r rN rO )r# r] rj rd re rW rX rs r r r �_requestAgentTest� s
z DistribTests._requestAgentTestc s| i �t �� �t��� dg�G ��fdd�dtj�}���fdd�� �j|� tddgi�d�}� ��fd d
�}|�|� |S ) z�
The request headers are available on the request object passed to a
distributed resource's C{render} method.
Nc s e Zd Z� �fdd�ZdS )z>DistribTests.test_requestHeaders.<locals>.ReportRequestHeadersc s |� d< �� t|j�� �� dS )Nr � )�update�dict�requestHeaders�getAllRawHeaders�r# rN ��reqrx r r �render� s zEDistribTests.test_requestHeaders.<locals>.ReportRequestHeaders.renderN�r r r r} r r{ r r �ReportRequestHeaders� s r c sD dd� � D �} �� d| � �� d| � �� �d | � t�� � d S )Nc S s g | ]}|d �qS )�
log_formatr )�.0�er r r �
<listcomp>� s zHDistribTests.test_requestHeaders.<locals>.check_logs.<locals>.<listcomp>zconnected to publisherz3could not connect to distributed web service: {msg}r )�assertInr �removeObserver)�msgs)� logObserverr| r# r r �
check_logs� s
z4DistribTests.test_requestHeaders.<locals>.check_logs�foo�bar)�headersc s$ �j j�� � ���d dg� d S )Ns Foos bar)r2 r" r3 rQ ��result)r� rx r# r r � cbRequested� s z5DistribTests.test_requestHeaders.<locals>.cbRequested) r �EventLoggingObserverr � addObserverr rC rk r rO )r# r rN r� r )r� r� r| rx r# r �test_requestHeaders� s
�
z DistribTests.test_requestHeadersc �8 G dd� dt j�}� �|� �}� fdd�}|�|� |S )z�
The response code can be set by the request object passed to a
distributed resource's C{render} method.
c @ � e Zd Zdd� ZdS )z>DistribTests.test_requestResponseCode.<locals>.SetResponseCodec S s |� d� dS )N�� � ��setResponseCoderz r r r r} � s
zEDistribTests.test_requestResponseCode.<locals>.SetResponseCode.renderNr~ r r r r �SetResponseCode� � r� c �: � � | d jd� � � | d jd� � � | d jd� d S )Nr ru r0 r� s OK�rQ �data�code�phraser� r> r r r� � � z:DistribTests.test_requestResponseCode.<locals>.cbRequested�r rC rt rO �r# r� rN r� r r> r �test_requestResponseCode� �
z%DistribTests.test_requestResponseCodec r� )z�
The response code and message can be set by the request object passed to
a distributed resource's C{render} method.
c @ r� )zEDistribTests.test_requestResponseCodeMessage.<locals>.SetResponseCodec S s |� dd� dS )Nr� � some-messager� r� rz r r r r} � s zLDistribTests.test_requestResponseCodeMessage.<locals>.SetResponseCode.renderNr~ r r r r r� � r� r� c r� )Nr ru r0 r� r� r� r� r> r r r� � r� zADistribTests.test_requestResponseCodeMessage.<locals>.cbRequestedr� r� r r> r � test_requestResponseCodeMessage� r� z,DistribTests.test_requestResponseCodeMessagec C �8 G dd� dt j�}| �|� �}|�| jdt d � |S )z�
If a string longer than the Banana size limit is passed to the
L{distrib.Request} passed to the remote resource, it is broken into
smaller strings to be transported over the PB connection.
c @ r� )z0DistribTests.test_largeWrite.<locals>.LargeWritec S s |� dt d � |�� tjS �N� x� y)�writer
�finishr � NOT_DONE_YETrz r r r r} s z7DistribTests.test_largeWrite.<locals>.LargeWrite.renderNr~ r r r r �
LargeWrite
r� r� r� r� �r rC rk rO rQ r
)r# r� rN r r r �test_largeWrite s z DistribTests.test_largeWritec C r� )z�
Like L{test_largeWrite}, but for the case where C{render} returns a
long string rather than explicitly passing it to L{Request.write}.
c @ r� )z2DistribTests.test_largeReturn.<locals>.LargeReturnc S s dt d S r� r rz r r r r} s z9DistribTests.test_largeReturn.<locals>.LargeReturn.renderNr~ r r r r � LargeReturn r� r� r� r� r� )r# r� rN r r r �test_largeReturn s z DistribTests.test_largeReturnc sl t t�� � �_}t�d|� �_}t�d|� � j
� �_ }t dg�� t
|� �}� �fdd�}|�|� |S )zz
If there is an error issuing the request to the remote publisher, an
error response is returned.
r r@ ru c sL �� � jd� ��tj�}�� t|�d� g d�}�� d�|�g� j� d S )Ni� r0 )
ru s <html>s: <head><title>500 - Server Connection Lost</title></head>s <body>s# <h1>Server Connection Lost</h1>s� <p>Connection to distributed server lost:<pre>[Failure instance: Traceback from remote host -- twisted.spread.flavors.NoSuchMethod: No such method: remote_requests ]</pre></p>s </body>s </html>ru �
)rQ � responseCode�flushLoggedErrorsr � NoSuchMethod�len�join�written)�ignored�errors�expected�rN r# r r �
cbRendered2 s
z4DistribTests.test_connectionLost.<locals>.cbRendered)r r �Rootr2 r rH r9 r rI rJ rK r4 r r rO )r# �
serverFactory�
serverPort� subscriptionrX r� r r� r �test_connectionLost" s
�
z DistribTests.test_connectionLostc C s` t j�| t�}t�t� �}tdg�}t� |�}|�
|� | � dt |�� | �
d|d d � dS )zQ
When a request fails, the string form of the failure is logged.
ru r0 zFailure instancer r� N)r r� �createWithCleanupr r �Failurer& r r �Issue�failed� assertEqualsr� r� )r# r� �frN �issuer r r �test_logFailedN s �
zDistribTests.test_logFailedc C sf t j�| t�}t� }t�|�}t�t � �}|�
|� | � t� | � dt
|�� | �|d d |� dS )zf
When L{twisted.web.distrib.Request}'s fail is called, the failure
is logged.
r0 r � log_failureN)r r� r� r r& r r� r �Requestr �failr� r� r� �assertIs)r# r� �errr� r| r r r �test_requestFail] s �
z DistribTests.test_requestFail)r r r r9 r< r4 r2 r? rY rf rk rt r� r� r� r� r� r� r� r� r r r r r' = s$ ", r' c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_PasswordDatabasec C s
|| _ d S r )�_users)r# �usersr r r �__init__o �
z_PasswordDatabase.__init__c C s
t | j�S r )�iterr� r> r r r �getpwallr r� z_PasswordDatabase.getpwallc C s& | j D ] }|d |kr| S qt� �r) )r� �KeyError)r# �username�userr r r �getpwnamu s
�z_PasswordDatabase.getpwnamN)r r r r� r� r� r r r r r� n s r� c @ sl e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
e d�dd� �ZdS )�UserDirectoryTestszj
Tests for L{UserDirectory}, a resource for listing all user resources
available on a system.
c C sT ddddd| � � df| _dddd d
| � � df| _t| j| jg�| _t�| j�| _d S ) N�alice�x�{ i� zAlice,,,z/bin/sh�bob�� i7 zBob,,,)�mktempr� r� r� �databaser �
UserDirectory� directoryr> r r r �setUp� s zUserDirectoryTests.setUpc C s | � ttj| j�� dS )zK
L{UserDirectory} instances provide L{resource.IResource}.
N)�
assertTruer r � IResourcer� r> r r r �test_interface� s z!UserDirectoryTests.test_interfacec s> t |g�� �j�|� �}t|� �}� �fdd�}|�|� |S )zr
Verify that requesting the C{name} child of C{self.directory} results
in a 404 response.
c s �� � jd� d S )Ni� )rQ r� )r� r� r r r� � s z/UserDirectoryTests._404Test.<locals>.cbRendered)r r� �getChildr rO )r# �namer� rX r� r r� r �_404Test� s
zUserDirectoryTests._404Testc C �
| � d�S )z�
L{UserDirectory.getChild} returns a resource which renders a 404
response when passed a string which does not correspond to any known
user.
�carol�r� r> r r r �test_getInvalidUser� �
z&UserDirectoryTests.test_getInvalidUserc C r� )z�
L{UserDirectory.getChild} returns a resource which renders a 404
response when passed a string which corresponds to a known user who has
neither a user directory nor a user distrib socket.
r� r� r> r r r �test_getUserWithoutResource� r� z.UserDirectoryTests.test_getUserWithoutResourcec C s\ t �| jd �}|�d�}|�� tdg�}| j�d|�}| �|t j
� | � |j |j � dS )z�
L{UserDirectory.getChild} returns a L{static.File} instance when passed
the name of a user with a home directory containing a I{public_html}
directory.
���� public_htmlr� N)
r
�FilePathr� r] �makedirsr r� r� �assertIsInstancer �FilerQ �path)r# �homer� rN r� r r r �test_getPublicHTMLChild� s
z*UserDirectoryTests.test_getPublicHTMLChildc C sn t �| jd �}|�� |�d�}tdg�}| j�d|�}| �|t j
� | � |j d� | � t
|j�|j� dS )z�
L{UserDirectory.getChild} returns a L{ResourceSubscription} instance
when passed the name of a user suffixed with C{".twistd"} who has a
home directory containing a I{.twistd-web-pb} socket.
r� �.twistd-web-pbr� z
bob.twistd�unixN)r
r� r� r� r] r r� r� r� r rI rQ r\ r rK r� )r# r� �webrN r� r r r �test_getDistribChild� s
z'UserDirectoryTests.test_getDistribChildc C s( t dg�}d|_| �tj| jj|� dS )zr
L{UserDirectory.render} raises L{UnsupportedMethod} in response to a
non-I{GET} request.
r� �POSTN)r �method� assertRaisesr �UnsupportedMethodr� r} rz r r r �test_invalidMethod� s
z%UserDirectoryTests.test_invalidMethodc sx t ��jd ��d�}|�� t ��jd �}|�� |�d��d� tdg�� t�j � �}� �fdd�}|�
|� |S )z}
L{UserDirectory} renders a list of links to available user content
in response to a I{GET} request.
r� r� r� ru r� c s� t d�� j��}|�d�\}}��|jjd� ��|j�d�d� ��|jjjd� ��|jjd� ��|j�d�d� ��|jjjd� d S ) Nru �li�a�hrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) r r� r� �getElementsByTagNamerQ �
firstChild�tagName� getAttributer� )r� �documentr� r� r� r r r� � s z2UserDirectoryTests.test_render.<locals>.cbRendered) r
r� r� r] r� r� �
setContentr r r� rO )r# r� r� r� r� r r� r � test_render� s
z UserDirectoryTests.test_renderzpwd module requiredc C s t �� }| �|jt� dS )z�
If L{UserDirectory} is instantiated with no arguments, it uses the
L{pwd} module as its password database.
N)r r� �assertIdentical�_pwd�pwd)r# r� r r r �test_passwordDatabase� s z(UserDirectoryTests.test_passwordDatabaseN)r r r r% r� r� r� r� r� r� r� r r r r r r r r r r� | s
r� )1r% �os.pathr �xml.dom.minidomr r r � ImportError�unittestr �zope.interface.verifyr �twisted.internetr r �twisted.loggerr �twisted.pythonr r
�twisted.spreadr �twisted.spread.bananar
� twisted.testr �twisted.trial.unittestr � twisted.webr r r r r �twisted.web.http_headersr �twisted.web.test._utilr �twisted.web.test.test_webr r rF r r � Exceptionr& r' r� r� r r r r �<module> s< � 3