File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_distrib.cpython-38.pyc
U
W[pG � @ sX d Z ddlmZ ddlmZ z ddlZW n ek
r@ dZY nX ddlmZ ddl m
Z
m Z ddl m
Z
mZ ddlmZ dd lmZ dd
lmZ dd lmZmZmZmZmZ dd lm Z m Z dd
l m Z ddl m!Z! ddl"m#Z# ddl$m%Z% G dd� dej&�Z'G dd� dej(�Z(G dd� de)�Z*G dd� dej+�Z,G dd� d�Z-G dd � d ej+�Z.dS ) z#
Tests for L{twisted.web.distrib}.
� )�abspath)� parseStringN)� verifyObject)�filepath�failure)�reactor�defer)�unittest)�pb��
SIZE_LIMIT)�distrib�client�resource�static�server)� DummyRequest� DummyChannel)�_render)�
proto_helpers)�Headers)�globalLogPublisherc @ 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 , 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% 2 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� | jdk r>| jjdk r>| jj�� fdd�� n� d �d� | jdk r�| jjdk r�| jjj�� fdd�� | jjjj �
� n� d �d� | j dk r�� � | j �
� � | jdk rΈ � | 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 s � d � d �S �Nr ��callbackr �Zdlr r �<lambda>F � z'DistribTests.tearDown.<locals>.<lambda>r c s � d � d �S )N� r( r r* r r r+ K r, r- )r �Deferred�f1r! �notifyOnDisconnectr) �subZ publisherZbrokerZ transportZloseConnection�port1�append�
stopListening�port2Z
gatherResults�r" r r* r �tearDown? 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 |�}|�tj� |�| jd� |S )
Ns theres rootz
text/plainr � 127.0.0.1s herez http://127.0.0.1:{}/here/there�ascii� GET) r �Resource�putChildr ZDatar �Siter r
�ResourcePublisherr/ r � listenTCPr2 �ResourceSubscription�getHost�portr1 r r5 r �Agent�format�encode�request� addCallback�readBody� assertEqual)r" Zr1Zsite1Zr2�f2�agent�url�dr r r � testDistribV 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 r8 )Z interface)r r; r< r r= r r
r> r/ r r? Z
addCleanupr4 rA r@ �hostrB r1 ) r" �childZ distribRootZ distribSiteZdistribFactoryZ distribPortr# ZmainRootZmainSite�mainPort�mainAddrr r r �_setupDistribServerm s,
�
� �
z DistribTests._setupDistribServerc K sR | � |�\}}t�t�}d|j|jf }|�d�}|jd|f|�}|�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.
zhttp://%s:%s/childr9 r: )
rS r rC r rO rB rE rF rG rH )r" rP �kwargsrQ rR rK rL rM r r r � _requestTest� s
zDistribTests._requestTestc K sT | � |�\}}d�|j|j�}|�d�}t�t�jd|f|�}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.
zhttp://{}:{}/childr9 r: c s8 t �� � ��� � t�� }� _|�� �fdd�� |S )Nc s � �fS r r )�_�Zprotocol�responser r r+ � r, zGDistribTests._requestAgentTest.<locals>.cbCollectBody.<locals>.<lambda>)r ZAccumulatingProtocolZ deliverBodyr r. ZclosedDeferredrG )rX rM r rW r �
cbCollectBody� s
z5DistribTests._requestAgentTest.<locals>.cbCollectBody)
rS rD rO rB rE r rC r rF rG )r" rP rT rQ rR rL rM rY 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 r, )�update�dict�requestHeadersZgetAllRawHeaders�r" rF ��reqr] r r �render� s
�zEDistribTests.test_requestHeaders.<locals>.ReportRequestHeaders.renderN�r r r ra r r_ r r �ReportRequestHeaders� s rc 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 ZremoveObserver)Zmsgs)� logObserverr` r" r r �
check_logs� s �z4DistribTests.test_requestHeaders.<locals>.check_logsZfooZbar)Zheadersc s$ �j j�� � ���d dg� d S )Ns Foos bar)r/ r! r0 rI ��result)rj r] r" r r � cbRequested� s z5DistribTests.test_requestHeaders.<locals>.cbRequested) r �EventLoggingObserverr Z addObserverr r; rU r rG )r" rc rF rm r )rj ri r` r] r" r �test_requestHeaders� s
�
z DistribTests.test_requestHeadersc s8 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 @ s e Zd Zdd� ZdS )z>DistribTests.test_requestResponseCode.<locals>.SetResponseCodec S s |� d� dS )N�� � �ZsetResponseCoder^ r r r ra � s
zEDistribTests.test_requestResponseCode.<locals>.SetResponseCode.renderNrb r r r r �SetResponseCode� s rs c s: � � | d jd� � � | d jd� � � | d jd� d S )Nr r, r- rp s OK�rI �data�code�phraserk r6 r r rm � s z:DistribTests.test_requestResponseCode.<locals>.cbRequested�r r; rZ rG �r" rs rF rm r r6 r �test_requestResponseCode� s
z%DistribTests.test_requestResponseCodec s8 G dd� dt j�}� �|� �}� fdd�}|�|� |S )z�
The response code and message can be set by the request object passed to
a distributed resource's C{render} method.
c @ s e Zd Zdd� ZdS )zEDistribTests.test_requestResponseCodeMessage.<locals>.SetResponseCodec S s |� dd� dS )Nrp � some-messagerq rr r^ r r r ra � s zLDistribTests.test_requestResponseCodeMessage.<locals>.SetResponseCode.renderNrb r r r r rs � s rs c s: � � | d jd� � � | d jd� � � | d jd� d S )Nr r, r- rp r{ rt rk r6 r r rm � s zADistribTests.test_requestResponseCodeMessage.<locals>.cbRequestedrx ry r r6 r � test_requestResponseCodeMessage� s
z,DistribTests.test_requestResponseCodeMessagec C s8 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 @ s e Zd Zdd� ZdS )z0DistribTests.test_largeWrite.<locals>.LargeWritec S s |� dt d � |�� tjS �N� x� y)�writer Zfinishr Z NOT_DONE_YETr^ r r r ra s z7DistribTests.test_largeWrite.<locals>.LargeWrite.renderNrb r r r r �
LargeWrite
s r� r~ r �r r; rU rG rI r )r" r� rF r r r �test_largeWrite s z DistribTests.test_largeWritec C s8 G dd� dt j�}| �|� �}|�| jdt d � |S )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 @ s e Zd Zdd� ZdS )z2DistribTests.test_largeReturn.<locals>.LargeReturnc S s dt d S r} r r^ r r r ra s z9DistribTests.test_largeReturn.<locals>.LargeReturn.renderNrb r r r r � LargeReturn s r� r~ r r� )r" r� rF 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 r8 r, c
s\ �� � jd� ��tj�}�� t|�d� ddddddd d
d dg
}�� d �|�g� j� d S )
Ni� r- r, 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>�
)rI � responseCode�flushLoggedErrorsr
Z NoSuchMethod�len�join�written)�ignored�errorsZexpected�rF r" r r �
cbRendered4 s �z4DistribTests.test_connectionLost.<locals>.cbRendered)r r
ZRootr/ r r? r2 r
r@ rA rB r1 r r rG )r" Z
serverFactoryZ
serverPortZ subscriptionrM 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.
r, r- zFailure instancer rd N)r rn �createWithCleanupr r �Failurer% r r
ZIssueZfailed� assertEqualsr� rh )r" ri �frF Zissuer r r �test_logFailedQ 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.
r- r Z log_failureN)r rn r� r r% r r� r
ZRequestr Zfailr� r� r� ZassertIs)r" ri �errr� r` r r r �test_requestFaile s �
z DistribTests.test_requestFail)r r r r2 r5 r1 r/ r7 rN rS rU rZ ro rz r| r� r� r� r� r� r r r r r& 9 s" &+r& c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_PasswordDatabasec C s
|| _ d S r )�_users)r" Zusersr r r �__init__y s z_PasswordDatabase.__init__c C s
t | j�S r )�iterr� r6 r r r �getpwall} s z_PasswordDatabase.getpwallc C s* | j D ]}|d |kr| S qt� �d S r' )r� �KeyError)r" Zusername�userr r r �getpwnam� s
z_PasswordDatabase.getpwnamN)r r r r� r� r� r r r r r� x s r� c @ sn e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
edkrjde
_dS )�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,,,)Zmktempr� r� r� Zdatabaser
�
UserDirectory� directoryr6 r r r �setUp� s zUserDirectoryTests.setUpc C s | � ttj| j�� dS )zK
L{UserDirectory} instances provide L{resource.IResource}.
N)Z
assertTruer r Z IResourcer� r6 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� )rI r� )r� r� r r r� � s z/UserDirectoryTests._404Test.<locals>.cbRendered)r r� �getChildr rG )r" �namerl rM r� r r� r �_404Test� s
zUserDirectoryTests._404Testc C s
| � 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.
Zcarol�r� r6 r r r �test_getInvalidUser� s z&UserDirectoryTests.test_getInvalidUserc C s
| � d�S )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� r6 r r r �test_getUserWithoutResource� s 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� rP �makedirsr r� r� �assertIsInstancer ZFilerI �path)r" �homer� rF rl 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.twistdZunixN)r r� r� r� rP r r� r� r� r
r@ rI rO r rB r� )r" r� �webrF rl 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.
rq ZPOSTN)r �methodZ assertRaisesr ZUnsupportedMethodr� ra r^ 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� r, rq c s� t d�� j��}|�d�\}}��|jjd� ��|j�d�d� ��|jjjd� ��|jjd� ��|j�d�d� ��|jjjd� d S ) Nr, Zli�aZhrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) r r� r� ZgetElementsByTagNamerI Z
firstChildZtagNameZ getAttributeru )r� Zdocumentr� r� r� r r r� � s z2UserDirectoryTests.test_render.<locals>.cbRendered) r r� r� rP r� r� Z
setContentr r r� rG )r" r� r� rl r� r r� r � test_render� s
z UserDirectoryTests.test_renderc 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� ZassertIdenticalZ_pwd�pwd)r" r� r r r �test_passwordDatabase s z(UserDirectoryTests.test_passwordDatabaseNzpwd module required)r r r r$ r� r� r� r� r� r� r� r� r� r� r� �skipr r r r r� � s r� )/r$ �os.pathr Zxml.dom.minidomr r� � ImportErrorZzope.interface.verifyr Ztwisted.pythonr r Ztwisted.internetr r Z
twisted.trialr Ztwisted.spreadr
Ztwisted.spread.bananar Z twisted.webr
r r r r Ztwisted.web.test.test_webr r Ztwisted.web.test._utilr Z twisted.testr Ztwisted.web.http_headersr Ztwisted.loggerr r= r r � Exceptionr% ZTestCaser& r� r� r r r r �<module> s4
A