← Back
Editing: 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
Save File
Cancel