Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/twisted/web/test/__pycache__/test_http2.cpython-39.pyc
Ðазад
a k�h.� � @ s@ d Z ddlZddlmZmZ ddlmZmZmZm Z ddl m Z ddl m Z mZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZmZm Z m Z m Z m Z m Z dZ!zHddl"Z"ddl#Z"ddl$Z"ddl%Z%ddl&Z&dd l'm(Z(m)Z) ddl*m+Z+ W n e,�y dZ!Y n0 G dd� d�Z-G dd� d�Z.d1dd�Z/d2dd�Z0dd� Z1G dd � d ej2�Z3e e3�Z4G d d � d ej2�Z5e e5�Z6G d d � d e5�Z7e e7�Z8G d!d"� d"ej2�Z9e e9�Z:G d#d$� d$�Z;e e;�Z<G d%d&� d&�Z=G d'd(� d(ej>e=�Z?G d)d*� d*ej>e=�Z@G d+d,� d,ej>e=�ZAG d-d.� d.ej>e=�ZBG d/d0� d0ej>e=�ZCdS )3z Test HTTP/2 support. � N)�directlyProvides� providedBy)�defer�error�reactor�task)� IPv4Address)�MemoryReactorClock�StringTransport)�failure)� iterbytes)� DummyProducer)�unittest)�http)�DelayedHTTPHandler�DelayedHTTPHandlerProxy�DummyHTTPHandler�DummyHTTPHandlerProxy� DummyPullProducerHandlerProxy�._IDeprecatedHTTPChannelToRequestInterfaceProxy�_makeRequestProxyFactory)�Decoder�Encoder)� H2ConnectionzHTTP/2 support not enabledc @ s| e Zd ZdZdd� Zdd� Zdd� Zg dfd d �Zd d d �Zd dd�Z dd� Z d dd�Z d dd�Z d!dd�Z g fdd �Zd S )"� FrameFactoryz� A class containing lots of helper methods and state to build frames. This allows test cases to easily build correct HTTP/2 frames to feed to hyper-h2. c C s t � | _d S �N�r �encoder��self� r ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/twisted/web/test/test_http2.py�__init__8 s zFrameFactory.__init__c C s t � | _d S r r r r r r! �refreshEncoder; s zFrameFactory.refreshEncoderc C s dS )Ns PRI * HTTP/2.0 SM r r r r r! �clientConnectionPreface> s z$FrameFactory.clientConnectionPreface� c K s^ t j�|�}| j�|�|_|j�d� |D ]}|j�|� q*|�� D ]\}}t |||� qD|S )zS Builds a single valid headers frame out of the contained headers. � END_HEADERS) � hyperframe�frame� HeadersFramer �encode�data�flags�add�items�setattr) r �headersr, �streamIDZpriorityKwargs�f�flag�k�vr r r! �buildHeadersFrameA s z FrameFactory.buildHeadersFrameNc C s2 |durt |�nt � }tj�|�}||_||_|S )zD Builds a single data frame out of a chunk of data. N)�setr' r( � DataFramer+ r, )r r+ r, r1 r2 r r r! �buildDataFrameP s zFrameFactory.buildDataFrameFc C s& t j�d�}|r |j�d� ||_|S )z1 Builds a single settings frame. r ZACK)r' r( Z SettingsFramer, r- �settings)r r: Zackr2 r r r! �buildSettingsFrameZ s z FrameFactory.buildSettingsFramec C s t j�|�}||_|S )z5 Builds a single WindowUpdate frame. )r' r( �WindowUpdateFrameZwindow_increment)r r1 � incrementr2 r r r! �buildWindowUpdateFramee s z#FrameFactory.buildWindowUpdateFramer � c C s" t j�d�}||_||_||_|S )z/ Builds a single GOAWAY frame. r )r' r( � GoAwayFrame� error_code�last_stream_idZadditional_data)r � lastStreamID� errorCodeZadditionalDatar2 r r r! �buildGoAwayFramem s z FrameFactory.buildGoAwayFramec C s t j�|�}||_|S )z3 Builds a single RST_STREAM frame. )r' r( �RstStreamFramerA )r r1 rD r2 r r r! �buildRstStreamFramew s z FrameFactory.buildRstStreamFramec C s" t j�|�}||_||_||_|S )z1 Builds a single priority frame. )r' r( Z PriorityFrameZ depends_on� stream_weight� exclusive)r r1 �weight� dependsOnrI r2 r r r! �buildPriorityFrame s z FrameFactory.buildPriorityFramec C s: t j�|�}||_| j�|�|_t|�|_|j� d� |S )z5 Builds a single Push Promise frame. r& ) r' r( ZPushPromiseFrameZpromised_stream_idr r* r+ r7 r, r- )r r1 �promisedStreamIDr0 r, r2 r r r! �buildPushPromiseFrame� s z"FrameFactory.buildPushPromiseFrame)Nr% )F)r r? )r )r F)�__name__� __module__� __qualname__�__doc__r" r# r$ r6 r9 r; r> rE rG rL rN r r r r! r 1 s r c @ s4 e Zd ZdZdd� Zdd� Zdd� Zdd � ZeZd S ) � FrameBuffera� A test object that converts data received from Twisted's HTTP/2 stack and turns it into a sequence of hyperframe frame objects. This is primarily used to make it easier to write and debug tests: rather than have to serialize the expected frames and then do byte-level comparison (which can be unclear in debugging output), this object makes it possible to work with the frames directly. It also ensures that headers are properly decompressed. c C s t � | _d| _d S )Nr? )r �decoder�_datar r r r! r" � s zFrameBuffer.__init__c C s | j |7 _ d S r )rU )r r+ r r r! � receiveData� s zFrameBuffer.receiveDatac C s | S r r r r r r! �__iter__� s zFrameBuffer.__iter__c C s� t | j�dk rt� �tjj�| jd d� �\}}t | j�|d k rHt� �|�t| jdd| � �� | jd| d � | _t |tjj �r�| j j |j dd�|_ |S )N� T)�raw)�lenrU � StopIterationr' r( �FrameZparse_frame_headerZ parse_body� memoryview� isinstancer) rT �decoder+ )r r( �lengthr r r! �next� s zFrameBuffer.nextN) rO rP rQ rR r" rV rW ra �__next__r r r r! rS � s rS r% c sT � du rt � � g }|�� j| �d�� |�� �fdd�|D �� |d j�d� |S )a� Provides a sequence of HTTP/2 frames that encode a single HTTP request. This should be used when you want to control the serialization yourself, e.g. because you want to interleave other frames with these. If that's not necessary, prefer L{buildRequestBytes}. @param headers: The HTTP/2 headers to send. @type headers: L{list} of L{tuple} of L{bytes} @param data: The HTTP data to send. Each list entry will be sent in its own frame. @type data: L{list} of L{bytes} @param frameFactory: The L{FrameFactory} that will be used to construct the frames. @type frameFactory: L{FrameFactory} @param streamID: The ID of the stream on which to send the request. @type streamID: L{int} N�r0 r1 c 3 s | ]}� j |�d �V qdS )�r1 N)r9 )�.0�chunk�� frameFactoryr1 r r! � <genexpr>� s z%buildRequestFrames.<locals>.<genexpr>���� END_STREAM)r �appendr6 �extendr, r- �r0 r+ rh r1 �framesr rg r! �buildRequestFrames� s �rp c C s" t | |||�}d�dd� |D ��S )a: Provides the byte sequence for a collection of HTTP/2 frames representing the provided request. @param headers: The HTTP/2 headers to send. @type headers: L{list} of L{tuple} of L{bytes} @param data: The HTTP data to send. Each list entry will be sent in its own frame. @type data: L{list} of L{bytes} @param frameFactory: The L{FrameFactory} that will be used to construct the frames. @type frameFactory: L{FrameFactory} @param streamID: The ID of the stream on which to send the request. @type streamID: L{int} r? c s s | ]}|� � V qd S r �� serialize�re r2 r r r! ri � r? z$buildRequestBytes.<locals>.<genexpr>)rp �joinrn r r r! �buildRequestBytes� s ru c C s t � }|�| � t|�S )a� Given a sequence of bytes, decodes them into frames. Note that this method should almost always be called only once, before making some assertions. This is because decoding HTTP/2 frames is extremely stateful, and this function doesn't preserve any of that state between calls. @param data: The serialized HTTP/2 frames. @type data: L{bytes} @returns: A list of HTTP/2 frames. @rtype: L{list} of L{hyperframe.frame.Frame} subclasses. )rS rV �list)r+ �bufferr r r! �framesFromBytes� s rx c @ s e Zd ZdZdZdd� ZdS )�ChunkedHTTPHandlerz� A HTTP request object that writes chunks of data back to the network based on the URL. Must be called with a path /chunked/<num_chunks> s hello world!c C sD t | j�d�d �}| �d� t|�D ]}| �| j� q&| �� d S )N� /rj �� )�int�uri�split�setResponseCode�range�write� chunkData�finish)r �chunks�_r r r! �process s zChunkedHTTPHandler.processN)rO rP rQ rR r� r� r r r r! ry s ry c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ConsumerDummyHandlera This is a HTTP request handler that works with the C{IPushProducer} implementation in the L{H2Stream} object. No current IRequest object does that, but in principle future implementations could: that codepath should therefore be tested. c O s4 t jj| g|�R i |�� | j�� d| _d | _d S )NF)r �Requestr" �channel�pauseProducing�_requestReceivedrU �r �args�kwargsr r r! r" * s z ConsumerDummyHandler.__init__c C s | j �� dS )z& Start the data pipe. N)r� �resumeProducingr r r r! � acceptData2 s z ConsumerDummyHandler.acceptDatac O s d| _ tjj| g|�R i |��S )NT)r� r r� �requestReceivedr� r r r! r� 8 s z$ConsumerDummyHandler.requestReceivedc C s0 | � d� | j�� | _d}| �|� | �� d S )Nr{ s0 this is a response from a consumer dummy handler)r �content�readrU r� r� )r Z returnDatar r r! r� < s z ConsumerDummyHandler.processN)rO rP rQ rR r" r� r� r� r r r r! r� " s r� c @ s e Zd ZdZdd� ZdS )� AbortingConsumerDummyHandlerz� This is a HTTP request handler that works with the C{IPushProducer} implementation in the L{H2Stream} object. The difference between this and the ConsumerDummyHandler is that after resuming production it immediately aborts it again. c C s | j �� | j �� dS )z@ Start and then immediately stop the data pipe. N)r� r� � stopProducingr r r r! r� O s z'AbortingConsumerDummyHandler.acceptDataN)rO rP rQ rR r� r r r r! r� G s r� c @ s e Zd ZdZdd� ZdS )�DummyProducerHandlerz� An HTTP request handler that registers a dummy producer to serve the body. The owner must call C{finish} to complete the response. c C s | � d� | �t� d� d S )Nr{ T)r �registerProducerr r r r r! r� c s z DummyProducerHandler.processN)rO rP rQ rR r� r r r r! r� \ s r� c @ s e Zd ZdZdd� Zdd� ZdS )�NotifyingRequestFactorya( A L{http.Request} factory that calls L{http.Request.notifyFinish} on all L{http.Request} objects before it returns them, and squirrels the resulting L{defer.Deferred} away on the class for later use. This is done as early as possible to ensure that we always see the result. c C s* g | _ || _t| j�D ]}t| |� qd S r )�results�_wrappedFactoryr r )r ZwrappedFactoryZ interfacer r r! r" s s z NotifyingRequestFactory.__init__c O s( | j |i |��}| j�|�� � t|�S r )r� r� rl Z notifyFinishr )r r� r� �reqr r r! �__call__} s z NotifyingRequestFactory.__call__N)rO rP rQ rR r" r� r r r r! r� k s r� c @ s e Zd ZdZereZdd� ZdS )�HTTP2TestHelpersz] A superclass that contains no tests but provides test helpers for HTTP/2 tests. c C s | � tjt|j� dS )z� Confirm that all streams are blocked: that is, the priority tree believes that none of the streams have data ready to send. N)� assertRaises�priorityZ DeadlockErrorra )r � connectionr r r! �assertAllStreamsBlocked� s z(HTTP2TestHelpers.assertAllStreamsBlockedN)rO rP rQ rR �skipH2�skipr� r r r r! r� � s r� c @ sD e Zd Zg d�Zg d�Zg d�Zg d�ZdZg d�ZdZ dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$d>d?� Z%d@dA� Z&dBdC� Z'dDdE� Z(dFdG� Z)dHdI� Z*dJS )K�HTTP2ServerTests��� :method� GET�s :authoritys localhost�� :pathrz �s :schemes https�s user-agents twisted-test-code)� custom-header� 1)r� � 2�)r� � POSTr� )r� � /post_endpointr� r� )� content-lengths 25�s hello s world, s it's s http/2!)�� :statuss 200)� requestrz )� commandr� �s versions HTTP/2)r� s 13� ''' None ''' )r� �r� r� �r� r� r� )r� s 36�$ ''' 25 hello world, it's http/2!''' c C sN t � }t� }|�� }|t|||�7 }|�|� t|�D ]}|�|� q6||fS )a! Takes a single L{H2Connection} object and connects it to a L{StringTransport} using a brand new L{FrameFactory}. @param connection: The L{H2Connection} object to connect. @type connection: L{H2Connection} @param headers: The headers to send on the first request. @type headers: L{Iterable} of L{tuple} of C{(bytes, bytes)} @param body: Chunks of body to send, if any. @type body: L{Iterable} of L{bytes} @return: A tuple of L{FrameFactory}, L{StringTransport} )r r r$ ru �makeConnectionr � dataReceived)r r� r0 �bodyrh � transport� requestBytes�byter r r! �connectAndReceive� s z"HTTP2ServerTests.connectAndReceivec s> t � }t|_� �|� jg �\}�� �fdd�}|jd �|�S )z� Send request over a TCP connection and confirm that we get back the expected data in the order and style we expect. c s� t ��� �}� �t|�d� � �tdd� |dd � D ��� � �t|d tjj �� � �t|d tjj �� � �t|d tjj �� � �t |d j �t � j �� � �|d j � j� � �|d j d� � �d|d jv � d S ) N� c s s | ]}|j d kV qdS �r% N�� stream_idrs r r r! ri � r? zGHTTP2ServerTests.test_basicRequest.<locals>.validate.<locals>.<genexpr>r% � � r? rk )rx �value� assertEqualrZ � assertTrue�allr^ r' r( r) r8 �dictr+ �getResponseHeaders�getResponseDatar, �r1 ro �r r� r r! �validate� s z4HTTP2ServerTests.test_basicRequest.<locals>.validater% )r r �requestFactoryr� �getRequestHeaders�_streamCleanupCallbacks� addCallback�r r� r� r� r r� r! �test_basicRequest� s z"HTTP2ServerTests.test_basicRequestc s@ t � }t|_� �|� j� j�\}�� �fdd�}|jd �|�S )zV Send a POST request and confirm that the data is safely transferred. c s� t ��� �}� �t|�d� � �tdd� |dd � D ��� � �t|d tjj �� � �t|d tjj �� � �t|d tjj �� � �t |d j �t � j �� � �|d j � j� � �|d j d� � �d|d jv � d S ) Nr� c s s | ]}|j d kV qdS r� r� rs r r r! ri r? zFHTTP2ServerTests.test_postRequest.<locals>.validate.<locals>.<genexpr>������rj r? rk )rx r� r� rZ r� r� r^ r' r( r) r8 r� r+ �postResponseHeaders�postResponseDatar, r� r� r r! r� s z3HTTP2ServerTests.test_postRequest.<locals>.validater% )r r r� r� �postRequestHeaders�postRequestDatar� r� r� r r� r! �test_postRequest� s �z!HTTP2ServerTests.test_postRequestc s^ g d��d� dd� �j D �}t� }t|_��||�j�\}�� ���fdd�}|jd �|�S )zm Send a POST request without length and confirm that the data is safely transferred. )r� r� r� r� )r� s 38s& ''' None hello world, it's http/2!''' c S s g | ]\}}|d kr||f�qS )r� r )re �x�yr r r! � <listcomp>% s z=HTTP2ServerTests.test_postRequestNoLength.<locals>.<listcomp>c s� t ��� �}��t|�d� ��tdd� |dd � D ��� ��t|d tjj �� ��t|d tjj �� ��t|d tjj �� ��t |d j �t ��� ��|d j � � ��|d j d� ��d|d j v � d S ) Nr� c s s | ]}|j d kV qdS r� r� rs r r r! ri 4 r? zNHTTP2ServerTests.test_postRequestNoLength.<locals>.validate.<locals>.<genexpr>r� r� rj r? rk )rx r� r� rZ r� r� r^ r' r( r) r8 r� r+ r, r� �r� r� r r� r r! r� / s z;HTTP2ServerTests.test_postRequestNoLength.<locals>.validater% )r� r r r� r� r� r� r� )r r� r� r� r� r r� r! �test_postRequestNoLength s ��z)HTTP2ServerTests.test_postRequestNoLengthc s� d}t � �t� � t� }t|_ttd|d d�����fdd��D �}��� }tj � t |� �}|d� dd� |D ��7 }|� � � t|�D ]}|�|� q�� ��fd d �}t�t|j�� ���|�S ) zi Many interleaved POST requests all get received and responded to appropriately. �( r% r� c s g | ]}t �j�j� |��qS r )rp r� r� �re r1 )r2 r r r! r� P s � �z=HTTP2ServerTests.test_interleavedRequests.<locals>.<listcomp>r? c s s | ]}|� � V qd S r rq �re r( r r r! ri \ r? z<HTTP2ServerTests.test_interleavedRequests.<locals>.<genexpr>c s� t ��� �}��t|�d� �D ]|� � fdd�|D �}��t|�d� ��t|d j�t�j�� ��|d j�j� ��|d jd� ��d |d j v � q d S ) N�y c s( g | ] }|j � krt|tjj�s|�qS r )r� r^ r' r( r< rs rd r r! r� m s �zOHTTP2ServerTests.test_interleavedRequests.<locals>.validate.<locals>.<listcomp>r� r r% r� r? rk ) rx r� r� rZ r� r+ r� r� r� r, )r� ro Z streamFrames)�br � streamIDsrd r! r� b s ��z;HTTP2ServerTests.test_interleavedRequests.<locals>.validate)r r r r r� rv r� r$ � itertools�chain� from_iterable�ziprt r� r r� r � DeferredListr� �valuesr� )r Z REQUEST_COUNT�aro r� r� r� r )r� r2 r r� r! �test_interleavedRequestsA s& � �z)HTTP2ServerTests.test_interleavedRequestsc s t � �t� � t� }t|_�j�d�d< ��fdd�dD �}|d d j�d� d|d d _|d d j�d� d |d d _�j d d d d d�}|d � d|� t j � |�}��� }|d�dd� |D ��7 }|�� � t|�D ]}|�|� qއ �fdd�}t�t|j�� ���|�S )zR Data in responses is interleaved according to HTTP/2 priorities. )z:pathz /chunked/4r� c s g | ]}t �g � |��qS r )rp r� )r2 r� r r! r� � s �zAHTTP2ServerTests.test_sendAccordingToPriority.<locals>.<listcomp>)r% r� � r ZPRIORITY�@ r% � r� � T)r1 rJ rK rI r? c s s | ]}|� � V qd S r rq r� r r r! ri � r? z@HTTP2ServerTests.test_sendAccordingToPriority.<locals>.<genexpr>c sB t � �� �}��t|�d� dd� |D �}g d�}��||� d S )N� c S s g | ]}t |tjj�r|j�qS r )r^ r' r( r8 r� rs r r r! r� � s zSHTTP2ServerTests.test_sendAccordingToPriority.<locals>.validate.<locals>.<listcomp>)r% r� r% r% r� r% r% r� r� r� r� r� r� r� r� )rx r� r� rZ )r� ro r� Z expectedOrder�r� r r r! r� � s �z?HTTP2ServerTests.test_sendAccordingToPriority.<locals>.validate)r r r �ChunkedHTTPHandlerProxyr� r� r, r- rH rL �insertr� r� r� r$ rt r� r r� r r� rv r� r� r� )r r� ro Z priorityFramer� r� r� r )r� r2 r� r r! � test_sendAccordingToPriority� s< �� �z-HTTP2ServerTests.test_sendAccordingToPriorityc C s� t � }t� }t� }t|_|�� }|t| jg |�7 }||jdd| jdgd�� � 7 }|� |� t |�D ]}|� |� |j rb q|qbt|�� �}| �t|�d� | �t|d tjj�� | �|j � dS )zR A protocol error from the remote peer terminates the connection. r% r� r& )r1 rM r0 r, r� rj N)r r r r r� r$ ru r� rN rr r� r r� � disconnectingrx r� r� rZ r� r^ r' r( r@ )r r2 r� r� r� r� ro r r r! �&test_protocolErrorTerminatesConnection� s* � z7HTTP2ServerTests.test_protocolErrorTerminatesConnectionc s� t � }t|_� �|� j� j�\}�|jd jj}� � |j � t �� � �}� � t|�d� |�� � �|j � � �|jd� t �� � �}� � t|�d� � �fdd�}|jd �|�S )z� The H2Stream data implements IPushProducer, and can have its data production controlled by the Request if the Request chooses to. r% s hello world, it's http/2!r� c s4 t ��� �}� �t|�d� � �d|d jv � d S �Nr� rk rj �rx r� r� rZ r� r, r� r� r r! r� s z;HTTP2ServerTests.test_streamProducingData.<locals>.validate)r �ConsumerDummyHandlerProxyr� r� r� r� �streams�_request�original� assertFalser� rx r� r� rZ r� r� rU r� r� )r r� r� �requestro r� r r� r! �test_streamProducingData� s � z)HTTP2ServerTests.test_streamProducingDatac s� t � }t� � t� }t|_t�j�j|�}t� |d _ |� � }|d� dd� |D ��7 }|� � � t |�D ]}|�|� qf|jd jj}��|j� |jd }|�� � �fdd�}|�|�S )z� The H2Stream data implements IPushProducer, and can have its data production controlled by the Request if the Request chooses to. When the production is stopped, that causes the stream connection to be lost. rj r? c s s | ]}|� � V qd S r rq rs r r r! ri , r? zAHTTP2ServerTests.test_abortStreamProducingData.<locals>.<genexpr>r% c sJ t � �� �}��t|�d� ��t|d tjj�� ��|d j d� d S )Nr� rj r% ) rx r� r� rZ r� r^ r' r( rF r� r� r� r r! r� ? s z@HTTP2ServerTests.test_abortStreamProducingData.<locals>.validate)r r r �!AbortingConsumerDummyHandlerProxyr� rp r� r� r7 r, r$ rt r� r r� r r r r r� r� r� r� ) r r2 r� ro r� r� r �cleanupCallbackr� r r� r! � test_abortStreamProducingData s" z.HTTP2ServerTests.test_abortStreamProducingDatac s� t � }t|_� �|� jg �\}�|jd jj}|�d� |�d� |j d }|� |j ddd�� � � � � |j� � � |jdu � |�d� � �fdd�}|�|�S ) z� When a RstStream frame is received, the L{H2Connection} and L{H2Stream} objects tear down the L{http.Request} and swallow all outstanding writes. r% � first chunk� second chunk)rD Ns third chunkc sJ t ��� �}� �t|�d� � �|d jd� � �t|d tjj �� d S �Nr� r% � rx r� r� rZ r� r� r^ r' r( r) r� r� r r! r� u s z9HTTP2ServerTests.test_terminatedRequest.<locals>.validate)r �DummyProducerHandlerProxyr� r� r� r r r r� r� r� rG rr r� � _disconnectedr� r� �r r� rh r r r� r r� r! �test_terminatedRequestJ s" � � z'HTTP2ServerTests.test_terminatedRequestc s� t � }t|_� �|� jg �\}�|jd jj}|�d� |�d� |j d }|� |j dd�� � � � � |j� � � |jdu � � �|j� � �fdd�}|�|�S ) z� When a GoAway frame is received, the L{H2Connection} and L{H2Stream} objects tear down all outstanding L{http.Request} objects and stop all writing. r% r r r �rC Nc sJ t ��� �}� �t|�d� � �|d jd� � �t|d tjj �� d S r r r� r� r r! r� � s z<HTTP2ServerTests.test_terminatedConnection.<locals>.validate)r r r� r� r� r r r r� r� r� rE rr r� r r� r �_stillProducingr� r r r� r! �test_terminatedConnection s" � � z*HTTP2ServerTests.test_terminatedConnectionc sH t � }t|_� jdg }� �||g �\}�� �fdd�}|jd �|�S )zq Requests containing Expect: 100-continue cause provisional 100 responses to be emitted. )s expects 100-continuec s� t ��� �}� �t|�d� � �tdd� |dd � D ��� � �t|d tjj �� � �|d j dg� � �d|d j v � d S )Nr� c s s | ]}|j d kV qdS r� r� rs r r r! ri � r? zQHTTP2ServerTests.test_respondWith100Continue.<locals>.validate.<locals>.<genexpr>r% )r� s 100rk rj ) rx r� r� rZ r� r� r^ r' r( r) r+ r, r� r� r r! r� � s z>HTTP2ServerTests.test_respondWith100Continue.<locals>.validater% )r r r� r� r� r� r� )r r� r0 r� r� r r� r! �test_respondWith100Continue� s z,HTTP2ServerTests.test_respondWith100Continuec sx t � }t|_� �|� jg �\}�|jd }|jj}|jd }|� � � � |j � � � |j du � � �fdd�}|� |�S )z� Triggering the call to L{H2Stream._respondToBadRequestAndDisconnect} leads to a 400 error being sent automatically and the stream being torn down. r% Nc s` t ��� �}� �t|�d� � �t|d tjj�� � �|d j dg� � �d|d j v � d S )Nr� r% )r� s 400rk rj ) rx r� r� rZ r� r^ r' r( r) r+ r, r� r� r r! r� � s z6HTTP2ServerTests.test_respondWith400.<locals>.validate)r r r� r� r� r r r r� �!_respondToBadRequestAndDisconnectr� r r� r� )r r� r� �streamr r r� r r� r! �test_respondWith400� s z$HTTP2ServerTests.test_respondWith400c sv t � }t|_��|�jg �\}�|jd }|jj}g d�� � D ]}|�|� q>|� � � ��fdd�}|j d � |�S )z� Calling L{Request.loseConnection} causes all data that has previously been sent to be flushed, and then the stream cleanly closed. r% )� hello� world� here� are� some� writesc s� t ��� �}��t|�d� ��tdd� |dd � D ��� ��t|d tjj �� ��d|d j v � dd� |D �}��|� d g � d S ) NrX c s s | ]}|j d kV qdS r� r� rs r r r! ri r? zQHTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate.<locals>.<genexpr>r% rk rj c S s g | ]}t |tjj�r|j�qS r �r^ r' r( r8 r+ rs r r r! r� s zRHTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate.<locals>.<listcomp>r? � rx r� r� rZ r� r� r^ r' r( r) r, �r1 ro ZreceivedDataChunks�� dataChunksr r� r r! r� s ��z>HTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate) r r r� r� r� r r r r� ZloseConnectionr� r� )r r� r� r r rf r� r r% r! �test_loseH2StreamConnection� s z,HTTP2ServerTests.test_loseH2StreamConnectionc C sD t � }t|_| �|| jg � |jd }|jj}| �t |j |d� dS )zK The L{H2Stream} object forbids registering two producers. r% TN) r r r� r� r� r r r r� � ValueErrorr� )r r� r r r r r! � test_cannotRegisterTwoProducers# s z0HTTP2ServerTests.test_cannotRegisterTwoProducersc sj t � }t|_��|�jg �\}�|jd }|jj� � jj }|� � fdd�� ��fdd�}|j d � |�S )z� L{Request} objects that have registered pull producers get blocked and unblocked according to HTTP/2 flow control. r% c s � � � S r )r� )r� )r r r! �<lambda>? r? z;HTTP2ServerTests.test_handlesPullProducer.<locals>.<lambda>c sB t ��� �}� �d|d jv � dd� |D �}� �|g d�� d S )Nrk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� I s zOHTTP2ServerTests.test_handlesPullProducer.<locals>.validate.<locals>.<listcomp>) � 0r� r� � 3� 4� 5� 6� 7� 8� 9r? �rx r� r� r, r� �r1 ro r& r� r r! r� B s ��z;HTTP2ServerTests.test_handlesPullProducer.<locals>.validate) r r r� r� r� r r r Z_actualProducer�resultr� r� )r r� r� r ZproducerCompleter� r )r r r� r! �test_handlesPullProducer1 s z)HTTP2ServerTests.test_handlesPullProducerc C sJ t � }t|_| �|| jg � |jd jj}| �|� � � |jd � � dS )zJ L{Request} objects can correctly ask isSecure on HTTP/2. r% N) r r r� r� r� r r r r ZisSecure�abortConnection)r r� r r r r! �test_isSecureWorksProperlyS s z+HTTP2ServerTests.test_isSecureWorksProperlyc sZ t � }t|_� �|� jg �\}�|jd jj}t� d|j � � �fdd�}|j d � |�S )zL L{H2Connection} correctly unblocks when a stream is ended. r% �{�G�z�?c s4 t ��� �}� �t|�d� � �d|d jv � d S )Nr� rk rj r �r� ro r� r r! �validateCompletel s zCHTTP2ServerTests.test_lateCompletionWorks.<locals>.validateComplete) r r r� r� r� r r r r � callLaterr� r� r� )r r� r� r r; r r� r! �test_lateCompletionWorks_ s z)HTTP2ServerTests.test_lateCompletionWorksc st t � }t|_� �|� jg �\}�|jd }|jj}|�d� |� g d�� |� � |j d }� �fdd�}|� |�S )zW L{H2Stream} objects can send a series of frames via C{writeSequence}. r% r{ )� Hello� ,� world!c sB t ��� �}� �d|d jv � dd� |D �}� �|g d�� d S )Nrk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� � s zTHTTP2ServerTests.test_writeSequenceForChannels.<locals>.validate.<locals>.<listcomp>)r> r? r@ r? r3 r4 r� r r! r� � s �z@HTTP2ServerTests.test_writeSequenceForChannels.<locals>.validate) r r r� r� r� r r r r Z writeSequencer� r� r� )r r� r� r r ZcompletionDeferredr� r r� r! � test_writeSequenceForChannelsu s z.HTTP2ServerTests.test_writeSequenceForChannelsc s� t � �t� �t� � t� _��� }|��tjj j di�� � 7 }|t �j g ��7 }� ��� t|�D ]}� �|� q`� jd }|jj���d� g d����fdd�}t�td|�}|�� �fdd �� ���fd d �}� jd �|�S ) z� Delaying writes from L{Request} causes the L{H2Connection} to block on sending until data is available. However, data is *not* sent if there's no room in the flow control window. r� r% � fiver)r r r r! c s � D ]} �� | � q��� d S r )r� r� )rf )r& r r r! � write_chunks� s z7HTTP2ServerTests.test_delayWrites.<locals>.write_chunksr9 c s � � �jddd��� �S )Nr% �2 �r1 r= )r� r> rr �r� �r� r2 r r! r* � s �z3HTTP2ServerTests.test_delayWrites.<locals>.<lambda>c s� t � �� �}��t|�d� ��tdd� |dd � D ��� ��t|d tjj �� ��d|d j v � dd� |D �}��|d g� d g � d S ) NrX c s s | ]}|j d kV qdS r� r� rs r r r! ri � r? zFHTTP2ServerTests.test_delayWrites.<locals>.validate.<locals>.<genexpr>r� rk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� � s zGHTTP2ServerTests.test_delayWrites.<locals>.validate.<locals>.<listcomp>rB r? r# r$ )r� r& r r r! r� � s ��z3HTTP2ServerTests.test_delayWrites.<locals>.validate)r r r r r� r$ r; �h2r: � SettingCodes�INITIAL_WINDOW_SIZErr ru r� r� r r� r r r r� r � deferLaterr r� r� )r r� r� r rC �dr� r )r� r� r& r2 r r r! �test_delayWrites� s. � �z!HTTP2ServerTests.test_delayWritesc C s� t � }t� }t� }t|_|�� }|t| jg |d�7 }||jdd�� � 7 }|� |� |� |� t |� � �}| �t|�d� | �d|j� dS )zu A client that immediately resets after sending the body causes Twisted to send no response. �r0 r+ rh r% rd N)r r r r r� r$ ru r� rG rr r� r� rx r� r� rZ Z assertNotInr� )r rh r� r� r� ro r r r! �test_resetAfterBody� s � z$HTTP2ServerTests.test_resetAfterBodyc s� t �� �G �fdd�dt�}t� � t�� }t|�� _|� _t � � _ �� � �j g � � �fdd�}�� |� � jd }t ��|g�S )z� A custom L{Request} subclass that requires the site and factory in the constructor is able to get them. c s e Zd Z� fdd�ZdS )zTHTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.SuperRequestc s2 t j| g|�R i |�� � �| jj| jjf� d S r )r r" �callbackr� �site�factoryr� �rL r r! r" � s z]HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.SuperRequest.__init__N)rO rP rQ r" r rS r r! � SuperRequest� s rT c s( | \}}�� |� j� �� |� j� d S r )�assertIsrQ rR )r� rQ rR )r� r r r! �validateFactoryAndSite s z^HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.validateFactoryAndSiter% )r ZDeferredr r r Z HTTPFactoryr r� rR �objectrQ r� r� r� r� � gatherResults)r rT Z httpFactoryrV r r )r� rL r r! �-test_RequestRequiringFactorySiteInConstructor� s z>HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructorc sr t � }tt�|_� �|� jg �\}}|jj}� �t|�d� � fdd�}|d }|� |� |j d }t � ||g�S )z� A request sent to a HTTP/2 connection fires the L{http.Request.notifyFinish} callback with a L{None} value. r% c s � � | � d S r )Z assertIsNone)r5 r r r! r� s z?HTTP2ServerTests.test_notifyOnCompleteRequest.<locals>.validater ) r r� r r� r� r� r� r� rZ r� r� r rX )r r� r� r� � deferredsr� rL r r r r! � test_notifyOnCompleteRequest s z-HTTP2ServerTests.test_notifyOnCompleteRequestc s� t � }tt�|_� �|� jg �\}}|jj}� �t|�d� � fdd�}� fdd�}|d }|� ||� |j dd�� � }|� |� |S )zw A HTTP/2 reset stream fires the L{http.Request.notifyFinish} deferred with L{ConnectionLost}. r% c s � � d� d S �Nz#Didn't errback, called back instead�Zfail�Zignr r r! rP 6 s z;HTTP2ServerTests.test_notifyOnResetStream.<locals>.callbackc s" � � | tj� � �| jtj� d S r ��assertIsInstancer �FailurerU �typer ZConnectionLost��reasonr r r! �errback9 s z:HTTP2ServerTests.test_notifyOnResetStream.<locals>.errbackr rd ) r r� r r� r� r� r� r� rZ � addCallbacksrG rr r� ) r r� rh r� rZ rP re rL � invalidDatar r r! �test_notifyOnResetStream& s � z)HTTP2ServerTests.test_notifyOnResetStreamc s� t � }tt�|_� �|� jg �\}}t� jg |dd�}|�|� |jj}� � t |�d� � fdd�}� fdd�}|D ]}|� ||� qt|j dd d �� � } |�| � t�|�S ) z� A HTTP/2 protocol error triggers the L{http.Request.notifyFinish} deferred for all outstanding requests with a Failure that contains the underlying exception. r� rg r� c s � � d� d S r\ r] r^ r r r! rP a s z=HTTP2ServerTests.test_failWithProtocolError.<locals>.callbackc s$ � � | tj� � � | jtjj� d S r )r` r ra r� rH � exceptions� ProtocolErrorrc r r r! re d s z<HTTP2ServerTests.test_failWithProtocolError.<locals>.errback� yo�� �r+ r1 )r r� r r� r� r� ru r� r� r� rZ rf r9 rr r rX � r r� rh r� � secondRequestrZ rP re rL rg r r r! �test_failWithProtocolErrorG s$ � � z+HTTP2ServerTests.test_failWithProtocolErrorc s� t � }tt�|_� �|� jg �\}}t� jg |dd�}|�|� |jj}� � t |�d� � fdd�}� fdd�}|D ]}|� ||� qt|j dd�� � } |�| � t�|�S ) z� A HTTP/2 GoAway triggers the L{http.Request.notifyFinish} deferred for all outstanding requests with a Failure that contains a RemoteGoAway error. r� rg r� c s � � d� d S r\ r] r^ r r r! rP � s z4HTTP2ServerTests.test_failOnGoaway.<locals>.callbackc s" � � | tj� � �| jtj� d S r r_ rc r r r! re � s z3HTTP2ServerTests.test_failOnGoaway.<locals>.errbackr )r r� r r� r� r� ru r� r� r� rZ rf rE rr r rX rn r r r! �test_failOnGoaways s$ � � z"HTTP2ServerTests.test_failOnGoawayc s� t � }tt�|_� �|� jg �\}}t� jg |dd�}|�|� |jj}� � t |�d� � fdd�}� fdd�}|D ]}|� ||� qt|� � t �|�S )z� The transport telling the HTTP/2 connection to stop producing will fire all L{http.Request.notifyFinish} errbacks with L{error.} r� rg r� c s � � d� d S r\ r] r^ r r r! rP � s z;HTTP2ServerTests.test_failOnStopProducing.<locals>.callbackc s" � � | tj� � �| jtj� d S r r_ rc r r r! re � s z:HTTP2ServerTests.test_failOnStopProducing.<locals>.errback)r r� r r� r� r� ru r� r� r� rZ rf r� r rX ) r r� rh r� ro rZ rP re rL r r r! �test_failOnStopProducing� s" � � z)HTTP2ServerTests.test_failOnStopProducingc s~ t � }tt�|_� �|� jg �\}}|jj}� �t|�d� � fdd�}� fdd�}|d }|� ||� |j d }|� � |S )z� A HTTP/2 stream that has had _respondToBadRequestAndDisconnect called on it from a request handler calls the L{http.Request.notifyFinish} errback with L{ConnectionLost}. r% c s � � d� d S r\ r] r^ r r r! rP � s z7HTTP2ServerTests.test_notifyOnFast400.<locals>.callbackc s" � � | tj� � �| jtj� d S r r_ rc r r r! re � s z6HTTP2ServerTests.test_notifyOnFast400.<locals>.errbackr ) r r� r r� r� r� r� r� rZ rf r r ) r r� rh r� rZ rP re rL r r r r! �test_notifyOnFast400� s � z%HTTP2ServerTests.test_notifyOnFast400c C s� t � }t|�}|j|_t|_d}t� }t� }|�� }|�|� |� |� |� t | j g ||d�� |� � d|_ |�|� | �|j� dS )z� A HTTP/2 stream that has had _respondToBadRequestAndDisconnect called on it does not write control frame data if its transport is paused and its control frame limit has been reached. r% rd r N)r r r< r r� r r r$ r� r� ru r� r� � _maxBufferedControlFrameBytesr r� � disconnected)r � memoryReactorr� r1 rh r� r$ r r r! � test_fast400WithCircuitBreaker� s& �� z/HTTP2ServerTests.test_fast400WithCircuitBreakerc C s� t � }t|_t� }t� }|�� }|�|� |�|� |�� t dd�D ]}|�|� i �� � � qFt |� � �}| �t|�d� |�� t |� � �}| �t|�d� dS )z� If a the L{H2Connection} has been paused by the transport, it will not write automatic frame data triggered by writes. r �d r% �e N)r r r� r r r$ r� r� r� r� r; rr rx r� r� rZ r� )r r� rh r� r$ r� ro r r r! � test_bufferingAutomaticFrameData s z1HTTP2ServerTests.test_bufferingAutomaticFrameDatac C s� t � }t|_t� }t� }|�� }|�|� |�|� |�� d|_ | � |j � t dd�D ]}|�|� i ��� � qX| � |j � |�|� i ��� � | �|j� dS )z� If the L{H2Connection} has been paused by the transport, it will not write automatic frame data triggered by writes. If this buffer gets too large, the connection will be dropped. rx r � N)r r r� r r r$ r� r� r� rt r r� r� r; rr r� ru )r r� rh r� r$ r� r r r! �2test_bufferingAutomaticFrameDataWithCircuitBreaker6 s zCHTTP2ServerTests.test_bufferingAutomaticFrameDataWithCircuitBreakerc C s G dd� dt �}t� }t|_t� }|� }|�|d� |�� }|�|� |�|� | � |j � t |� � �}| � t|�d� | � |jd� tdd�D ]}|�|�i ��� � q�t |� � �}| � t|�d� | � |jd� |�� t |� � �}| � t|�d� | � |jd � d S ) z� If the L{H2Connection} has buffered control frames, is unpaused, and then paused while unbuffering, it persists the buffer and stops trying to write. c @ s e Zd Zdd� ZdS )zeHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWrite.<locals>.AutoPausingStringTransportc _ s& t j| g|�R i |�� | j�� d S r )r r� �producerr� r� r r r! r� _ s zkHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWrite.<locals>.AutoPausingStringTransport.writeN)rO rP rQ r� r r r r! �AutoPausingStringTransport^ s r~ Tr% r r{ �c r� �Z N)r r r r� r r� r$ r� r� ZassertIsNotNoneZ_consumerBlockedrx r� r� rZ Z_bufferedControlFrameBytesr� r; rr r� )r r~ r� rh r� r$ ro r� r r r! �0test_bufferingContinuesIfProducerIsPausedOnWriteX s, zAHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWritec C sx t � }t|�}|j|_t� }t� }|�� }|�|� |�|� |�� d|_ |j ddd�� � }|�|� | � |j � dS )z� A client that triggers a L{h2.exceptions.ProtocolError} over a paused connection that's reached its buffered control frame limit causes that connection to be aborted. r rk rl rm N)r r r< r r r$ r� r� r� rt r9 rr r� ru )r rv r� rh r� r$ rg r r r! �+test_circuitBreakerAbortsAfterProtocolError� s z<HTTP2ServerTests.test_circuitBreakerAbortsAfterProtocolErrorN)+rO rP rQ r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r' r) r6 r8 r= rA rM rO rY r[ rh rp rq rr rs rw rz r| r� r� r r r r! r� � sP +@G'-.54 '-" B"!,*)'* "-r� c @ s� e Zd ZdZg d�ZdZg d�Zg d�ZdZdd� Z d d � Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�H2FlowControlTestszT Tests that ensure that we handle HTTP/2 flow control limits appropriately. )r� r� r� r� r� r� r� r� r� c s� t � }t� � t� }t|_|�� }||�tjj j di�� � 7 }|t �j g |�7 }|�� � t|�D ]}|�|� q`t�j�d }t|�D ] }|jddd�}|�|� � � q�� �fdd�}|jd �|�S )z� When a L{Request} object is not using C{IProducer} to generate data and so is not having backpressure exerted on it, the L{H2Stream} object will buffer data until the flow control window is opened. r� r% rE c sF t � �� �}��d|d jv � d�dd� |D ��}���j|� d S )Nrk rj r? c s s" | ]}t |tjj�r|jV qd S r r" rs r r r! ri � s zMH2FlowControlTests.test_bufferExcessData.<locals>.validate.<locals>.<genexpr>)rx r� r� r, rt r� r� �r1 ro ZactualResponseDatar� r r! r� � s �z:H2FlowControlTests.test_bufferExcessData.<locals>.validate)r r r r r� r$ r; rH r: rI rJ rr ru r� r� r r� rZ r� r� r> r� r� ) r r2 r� r� r� Z bonusFramesr� r( r� r r� r! �test_bufferExcessData� s$ � z(H2FlowControlTests.test_bufferExcessDatac s� t � }t� � t� }t|_|�� }||�tjj j di�� � 7 }|t �j g |�7 }|�� � t|�D ]}|�|� q`|jd }|jj}��|j� |�d� ��|j� ��|jjdg� |�|jddd�� � � ��|j� ��|jjdg� |�|jddd�� � � ��|j� ��|jjdg� |�|jddd�� � � ��|j� ��|jjddg� |�d� ��|j� ��|jjg d�� |�|jdd d�� � � ��|j� ��|jjg d �� |� � |� � � �fd d �}|j d � |�S ) z� L{Request} objects that have registered producers get blocked and unblocked according to HTTP/2 flow control. r� r% � helloworld�pauserE r �resume)r� r� r� rD )r� r� r� r� c sB t � �� �}��d|d jv � dd� |D �}��|g d�� d S )Nrk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� : s zXH2FlowControlTests.test_producerBlockingUnblocking.<locals>.validate.<locals>.<listcomp>)r� r� r? r3 r4 r� r r! r� 3 s �zDH2FlowControlTests.test_producerBlockingUnblocking.<locals>.validate) r r r r r� r$ r; rH r: rI rJ rr ru r� r� r r� r r r r� �_producerProducingr� r r� r} �eventsr> �unregisterProducerr� r� r� �r r2 r� r� r� r r r� r r� r! � test_producerBlockingUnblocking� sN � � z2H2FlowControlTests.test_producerBlockingUnblockingc s t � �t� �t� � t� _��� }|��tjj j di�� � 7 }|t �j g ��7 }� ��� t|�D ]}� �|� q`� jd ��jj����j� ��d� ���j� ���jjdg� ��d� � ����fdd�}t� t d|�}��fd d �}� j d � |�}t �!||g�S ) zQ Exactly filling the flow control window still blocks producers. r� r% r� r� � hc sL � � �jddd��� � ���j� ���jjddg� ��� �� � d S )Nr% rD rE r� r� ) r� r> rr r� r� r� r} r� r� r� r )r� r2 r r r r r! � window_openh s � z=H2FlowControlTests.test_flowControlExact.<locals>.window_openr c sB t � �� �}��d|d jv � dd� |D �}��|g d�� d S )Nrk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� { s zNH2FlowControlTests.test_flowControlExact.<locals>.validate.<locals>.<listcomp>)r r r� r? r3 r4 r� r r! r� t s �z:H2FlowControlTests.test_flowControlExact.<locals>.validate)"r r r r r� r$ r; rH r: rI rJ rr ru r� r� r r� r r r r� r� r� r r� r} r� r rK r r� r� r r� )r r� r� r� Z windowDeferr� Z validateDeferr )r� r� r2 r r r r! �test_flowControlExactA s0 � z(H2FlowControlTests.test_flowControlExactc s� t � }t� � t� }t|_|�� }||�tjj j di�� � 7 }|t �j g |�7 }|�� � t|�D ]}|�|� q`|jd }|jj}��|j� |�d� |�� |�� ��|j� t�d|j|j ddd�� � � � �fdd�}|j d � |�S ) z� L{Request} objects that end a stream that is currently blocked behind flow control can still end the stream and get cleaned up. r� r% r� r rD rE c sB t � �� �}��d|d jv � dd� |D �}��|g d�� d S )Nrk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� � s zQH2FlowControlTests.test_endingBlockedStream.<locals>.validate.<locals>.<listcomp>)r r r? r3 r4 r� r r! r� � s �z=H2FlowControlTests.test_endingBlockedStream.<locals>.validate) r r r r r� r$ r; rH r: rI rJ rr ru r� r� r r� r r r r� r� r� r� r� �finishedr r< r> r� r� r� r r� r! �test_endingBlockedStream� s4 � � z+H2FlowControlTests.test_endingBlockedStreamc s� t � }t� � t� }t|_|�� }|t�jg |�7 }|�� � t |�D ]}|� |� qD|j d }|j j }|jd }|�� |�� ��|j� � �fdd�}|�|�S )z< We safely handle responses without bodies. r% c sP t � �� �}��t|�d� ��d|d jv � dd� |D �}��|dg� d S )Nr� rk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� � s zQH2FlowControlTests.test_responseWithoutBody.<locals>.validate.<locals>.<listcomp>r? r r4 r� r r! r� � s ��z=H2FlowControlTests.test_responseWithoutBody.<locals>.validate)r r r r r� r$ ru r� r� r r� r r r r� r� r� r� r� r� � r r2 r� r� r� r r r r� r r� r! �test_responseWithoutBody� s" z+H2FlowControlTests.test_responseWithoutBodyc s� t � }t� � t� }t|_|�� }|t�jg |�7 }|�� � t |�D ]}|� |� qD|j d }|j j }|jd }|�� |�� ��|j� |� |jddd��� � � �fdd�}|�|�S )zk WindowUpdate frames received after we've completed the stream are safely handled. r% rD rE c sP t � �� �}��t|�d� ��d|d jv � dd� |D �}��|dg� d S )Nr� rk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� " s z[H2FlowControlTests.test_windowUpdateForCompleteStream.<locals>.validate.<locals>.<listcomp>r? r r4 r� r r! r� s ��zGH2FlowControlTests.test_windowUpdateForCompleteStream.<locals>.validate)r r r r r� r$ ru r� r� r r� r r r r� r� r� r� r� r> rr r� r� r r� r! �"test_windowUpdateForCompleteStream� s$ z5H2FlowControlTests.test_windowUpdateForCompleteStreamc s t � }t� � t� }t|_|�� }||�tjj j di�� � 7 }|t �j g |�7 }|�� � t|�D ]}|�|� q`|jd }|jj}��|j� |�d� ��|j� ��|jjg � |�|jddd�� � � ��|j� ��|jjg � |�� |� � � �fdd�}|j d � |�S )z� L{Request} objects that have registered producers that are not blocked behind flow control do not have their producer notified. r� r% � wordrE c sB t � �� �}��d|d jv � dd� |D �}��|ddg� d S )Nrk rj c S s g | ]}t |tjj�r|j�qS r r" rs r r r! r� a s zOH2FlowControlTests.test_producerUnblocked.<locals>.validate.<locals>.<listcomp>r� r? r3 r4 r� r r! r� Z s �z;H2FlowControlTests.test_producerUnblocked.<locals>.validate) r r r r r� r$ r; rH r: rI rJ rr ru r� r� r r� r r r r� r� r� r� r} r� r> r� r� r� r� r� r r� r! �test_producerUnblocked, s2 � z)H2FlowControlTests.test_producerUnblockedc s� t � }t� � t� }t|_t�j�j|�}|�d|j ddd�� |� � }|d� dd� |D ��7 }|� � � t |�D ]}|�|� qp� �fdd �}|jd �|�S ) z� When a WindowUpdate frame is received for the whole connection but no data is currently waiting, nothing exciting happens. r% r r� rE r? c s s | ]}|� � V qd S r rq rs r r r! ri v r? zBH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.<genexpr>c sF t � �� �}��d|d jv � d�dd� |D ��}���j|� d S )Nrk rj r? c s s" | ]}t |tjj�r|jV qd S r r" rs r r r! ri � s zTH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.validate.<locals>.<genexpr>)rx r� r� r, rt r� r� r� r� r r! r� } s �zAH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.validate)r r r r r� rp r� r� r� r> r$ rt r� r r� r� r� )r r2 r� ro r� r� r� r r� r! � test_unnecessaryWindowUpdateh s z/H2FlowControlTests.test_unnecessaryWindowUpdatec C s� t � }t� }t� }t|_g }|�|j| jdd�� |�|jddd�� |� � }|d� dd� |D ��7 }|� |� |� |� | � |� dS ) z� When a WindowUpdate frame is received for a stream but no data is currently waiting, that stream is not marked as unblocked and the priority tree continues to assert that no stream can progress. r% rc r� rE r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? zKH2FlowControlTests.test_unnecessaryWindowUpdateForStream.<locals>.<genexpr>N)r r r r r� rl r6 r� r> r$ rt r� r� r� )r r2 r� �connro r+ r r r! �%test_unnecessaryWindowUpdateForStream� s z8H2FlowControlTests.test_unnecessaryWindowUpdateForStreamc C s� t � }t� }t� }t|_t| j| j|�}|�� }|d� dd� |D ��7 }|� |� t |�D ]}|� |� qZ|j d �� |jddd�}|� |�� � t|�� �}| �t|d tjj�� dS ) zq When a WindowUpdate frame is received for a stream that has been aborted it is ignored. r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? zEH2FlowControlTests.test_windowUpdateAfterTerminate.<locals>.<genexpr>r% r� rE rj N)r r r r r� rp r� r� r$ rt r� r r� r r7 r> rr rx r� r� r^ r' r( rF )r r2 r� r� ro r� r� �windowUpdateFramer r r! � test_windowUpdateAfterTerminate� s z2H2FlowControlTests.test_windowUpdateAfterTerminatec s� t � �t� �t� � t� _t�j�j��}��� }|d� dd� |D ��7 }� � �� t |�D ]}� � |� qZ� �fdd�}��fdd�}� j d �|�}|�|�S ) zs When a WindowUpdate frame is received for a stream that has been completed it is ignored. r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? zDH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.<genexpr>c s �j ddd�}� �|�� � d S )Nr% r� rE )r> r� rr )r� r� rG r r! � update_window� s zHH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.update_windowc s" t � �� �}��d|d j� d S )Nrk rj )rx r� ZassertInr, r: r� r r! r� � s zCH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.validater% )r r r r r� rp r� r� r$ rt r� r r� r� r� )r ro r� r� r� r� rL r )r� r� r2 r r! � test_windowUpdateAfterComplete� s z1H2FlowControlTests.test_windowUpdateAfterCompletec C s t � }t� }t� }t|_dd gd }tdd� |D ��� }| jdd� d|fg }t|||d �}|d= |�|j d t j j j d �� |�� }|d �d d� |D ��7 }|�|� |�|� t|�� �}dd� |D �} | �dg| � dd� |D �} dd� |D �} | �| � | �| � dS )z� When a DATA frame is received at the same time as RST_STREAM, Twisted does not send WINDOW_UPDATE frames for the stream. � i @ r� c s s | ]}t |�V qd S r )rZ )re r+ r r r! ri � r? z;H2FlowControlTests.test_dataAndRstStream.<locals>.<genexpr>Nrj zcontent-lengthrN r% )r1 rD r? c s s | ]}|� � V qd S r rq rs r r r! ri r? c S s g | ]}t |tjj�r|j�qS r )r^ r' r( r< r� rs r r r! r� s �z<H2FlowControlTests.test_dataAndRstStream.<locals>.<listcomp>r c S s g | ]}t |tjj�r|�qS r )r^ r' r( r) rs r r r! r� s c S s g | ]}t |tjj�r|�qS r )r^ r' r( r8 rs r r r! r� r? )r r r r r� �sumr� rp rl rG rH �errors� ErrorCodes�INTERNAL_ERRORr$ rt r� r� rx r� r� r ) r rh r� r� Z frameDataZ bodyLengthr0 ro r� ZwindowUpdateFrameIDsZ headersFramesZ dataFramesr r r! �test_dataAndRstStream� s> � �� �� z(H2FlowControlTests.test_dataAndRstStreamc C s� t � }t|�}|j|_t|_t� }t� }|�� }|�|� |� |� d}|j | j |d�� � }|� |� |� � d|_|�� |�|� | �|�� � | �|j� dS )z� Aborting a request associated with a paused connection that's reached its buffered control frame limit causes that connection to be aborted. r% rc r N)r r r< r r� r r r$ r� r� r6 r� rr r� rt �clearZ abortRequestr r� r� ru )r rv r� rh r� r$ r1 ZheadersFrameDatar r r! �#test_abortRequestWithCircuitBreaker s( � z6H2FlowControlTests.test_abortRequestWithCircuitBreakerN)rO rP rQ rR r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� � s& -SB:4;<# "7r� c @ s4 e Zd Zg d�Zdd� Zdd� Zdd� Zdd � Zd S ) �HTTP2TransportCheckingr� c C s2 t � }t� }t|_|�|d� | �|j|u � dS )zU L{H2Connection} can be registered with the transport as a producer. TN)r r r r� r� r� r} )r r� r� r r r! �"test_registerProducerWithTransportT s z9HTTP2TransportChecking.test_registerProducerWithTransportc s� t � }t� �t� � t� _t�jg |�}|�� }|d�dd� |D ��7 }� � �� �� � d� t |�D ]}� � |� qd� � � � jd �� ���fdd�}��fdd �}t�td |�}|�|� |S ) z| L{H2Connection} can be paused by its consumer. When paused it stops sending data to the transport. r? c s s | ]}|� � V qd S r rq rs r r r! ri l r? zNHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.<genexpr>Tr% c s` t ��� �}��t|�d� ��t|d tjj�� � � � � � � � � � � � � � � � �S )Nr� rj ) rx r� r� rZ r r^ r' r( r8 r� r: �r� r� r r r r! �validateNotSent� s zTHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.validateNotSentc s4 t � �� �}��t|�d� ��d|d jv � d S r r r: r� r r! r; � s zUHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.validateCompleter9 )r r r r r� rp r� r$ rt r� r� r r� r� r� r rK r r� )r r2 ro r� r� r� r; rL r r� r! �$test_pausingProducerPreventsDataSend_ s$ z;HTTP2TransportChecking.test_pausingProducerPreventsDataSendc C s� t � }t� }t� }t|_t| jg |�}|�� }|d�dd� |D ��7 }|� |� |� |d� t |�D ]}|� |� qd|� � t|�� �}| �t|�d� | �t|d tjj�� | �|j� dS )zo L{H2Connection} can be stopped by its producer. That causes it to lose its transport. r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? z<HTTP2TransportChecking.test_stopProducing.<locals>.<genexpr>Tr� rj N)r r r r r� rp r� r$ rt r� r� r r� r� rx r� r� rZ r r^ r' r( r8 r )r r2 r� r� ro r� r� r r r! �test_stopProducing� s z)HTTP2TransportChecking.test_stopProducingc s� t ddd�� t ddd��t� }t� �d�}t� }t|_|�|� t�jg |�}|� � }|d� dd � |D ��7 }t |�D ]}|� |� qv|j d ������ � � ����� �� |jd }� ���fd d �}|�|�S ) z{ A L{H2Stream} object correctly passes through host and peer information from its L{H2Connection}. ZTCPz 17.52.24.8i� z 17.188.0.12i} )� hostAddress� peerAddressr? c s s | ]}|� � V qd S r rq r� r r r! ri � r? zEHTTP2TransportChecking.test_passthroughHostAndPeer.<locals>.<genexpr>r% c s$ �� ��� � � �� ��� �� d S r )r� �getHost�getPeerrF �r� r� r r r r! r� � s zDHTTP2TransportChecking.test_passthroughHostAndPeer.<locals>.validate)r r r r r r� r� rp r� r$ rt r r� r r� r� r� r� r� ) r rh r� r� ro r� r� r r� r r� r! �test_passthroughHostAndPeer� s$ z2HTTP2TransportChecking.test_passthroughHostAndPeerN)rO rP rQ r� r� r� r� r� r r r r! r� I s > r� c @ s e Zd ZdZdd� ZdS )�HTTP2SchedulingTestsz� The H2Connection object schedules certain events (mostly its data sending loop) using callbacks from the reactor. These tests validate that the calls are scheduled correctly. c C s| t �� }t|�}|�� }| �t|�d� |d }| �|�� � | �|jd� | �|j |j � | �|j d� | �|j i � dS )z} When a H2Connection is established it schedules one call to be run as soon as the reactor has time. r% r r N) r �Clockr �getDelayedCallsr� rZ r� Zactive�time�funcZ_sendPrioritisedDatar� �kw)r r r� Zcalls�callr r r! �"test_initiallySchedulesOneDataCall� s z7HTTP2SchedulingTests.test_initiallySchedulesOneDataCallN)rO rP rQ rR r� r r r r! r� � s r� c @ s� e Zd ZdZg d�Ze� Zdd� Zdd� Zdd� Z efd d �Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �HTTP2TimeoutTestsz@ The L{H2Connection} object times out idle connections. r� c C s |j |_ dS )a� Unfortunately, TimeoutMixin does not allow passing an explicit reactor to test timeouts. For that reason, we need to monkeypatch the method set up by the TimeoutMixin. @param connection: The HTTP/2 connection object to patch. @type connection: L{H2Connection} @param reactor: The reactor whose callLater method we want. @type reactor: An object implementing L{twisted.internet.interfaces.IReactorTime} N)r< )r r� r r r r! �patch_TimeoutMixin_clock s z*HTTP2TimeoutTests.patch_TimeoutMixin_clockc C s^ t �� }t|�}d|_| �||� t� }t|�|_|�|� t |�D ]}|� |� qD|||fS )a� Performs test setup by building a HTTP/2 connection object, a transport to back it, a reactor to run in, and sending in some initial data as needed. @param initialData: The initial HTTP/2 data to be fed into the connection after setup. @type initialData: L{bytes} @param requestFactory: The L{Request} factory to use with the connection. rx ) r r� r �timeOutr� r r r� r� r r� )r � initialDatar� r r� r� r� r r r! �initiateH2Connection s z&HTTP2TimeoutTests.initiateH2Connectionc C sX t |�}| �t|�|� | �t|d tjj�� | �|d j|� | �|d j |� dS )z� Confirm that the data that was sent matches what we expect from a timeout: namely, that it ends with a GOAWAY frame carrying an appropriate error code and last stream ID. rj N) rx r� rZ r� r^ r' r( r@ rA rB )r r+ � frameCountrD rC ro r r r! �assertTimedOut9 s z HTTP2TimeoutTests.assertTimedOutc C s� || j u rtj}t� }|�� }| j|td�\}}}||_|�d� | j|� � dt j j j dd� | �|j� | �|j� |||fS )a� Does the common setup for tests that want to test the aborting functionality of the HTTP/2 stack. @param abortTimeout: The value to use for the abortTimeout. Defaults to whatever is set on L{H2Connection.abortTimeout}. @type abortTimeout: L{int} or L{None} @return: A tuple of the reactor being used for the connection, the connection itself, and the transport. �r� rx r� r �r� rD rC )�_DEFAULTr � abortTimeoutr r$ r� r �advancer� r� rH r� r� �NO_ERRORr� r� r ru )r r� rh r� r r� r� r r r! �prepareAbortTestF s& � � z"HTTP2TimeoutTests.prepareAbortTestc C s� t � }|�� }| j|td�\}}}|�� }|�d� | �||�� � | �|j� |�d� | j |�� dt j j j dd� | �|j� dS )z� When a L{H2Connection} does not receive any data for more than the time out interval, it closes the connection cleanly. r� r r� r r� N)r r$ r� r r� r� r� r r� r� rH r� r� r� r� )r rh r� r r� r� �preambler r r! �test_timeoutAfterInactivityl s$ � �z-HTTP2TimeoutTests.test_timeoutAfterInactivityc C s� t � }d}| j|td�\}}}t|�� �D ]$}|�|� |�d� | �|j� q*|�d� | j |� � dt j j jdd� | �|j� dS )zM When a L{H2Connection} receives data, the timeout is reset. r? r� r r� r r� N)r r� r r r$ r� r� r r� r� r� rH r� r� r� r� )r rh r� r r� r� r� r r r! � test_timeoutResetByRequestData� s$ � �z0HTTP2TimeoutTests.test_timeoutResetByRequestDatac s� t � }d}g � t| jg |�}|�� }|d�dd� |D ��7 }� fdd�}| j||d�\}}}|�|�� � |�d� | �t � �d� t d �D ](}� d � d � |�d� | � |j � q�|�d � | j|�� d tjjjdd� dS )zJ When a L{H2Connection} sends data, the timeout is reset. r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? zDHTTP2TimeoutTests.test_timeoutResetByResponseData.<locals>.<genexpr>c s t | |d�}� �|� |S )N)�queued)r rl )r r� r� ��requestsr r! � saveRequest� s zFHTTP2TimeoutTests.test_timeoutResetByResponseData.<locals>.saveRequestr� r r% � r s some bytesr� � r� N)r rp r� r$ rt r� r� r� Z assertEqualsrZ r� r� r r� r� r� rH r� r� �PROTOCOL_ERROR) r rh r� ro r� r r� r� r� r r� r! � test_timeoutResetByResponseData� s2 � �z1HTTP2TimeoutTests.test_timeoutResetByResponseDatac C s~ t � }t| jg |�}|�� }|d�dd� |D ��7 }| j|td�\}}}|�d� | j|� � dt j j j dd� | �|j� d S ) z� When a L{H2Connection} times out with active streams, the error code returned is L{h2.errors.ErrorCodes.PROTOCOL_ERROR}. r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? zOHTTP2TimeoutTests.test_timeoutWithProtocolErrorIfStreamsOpen.<locals>.<genexpr>r� ry r� r% r� N)r rp r� r$ rt r� r� r� r� r� rH r� r� r� r� r� )r rh ro r� r r� r� r r r! �*test_timeoutWithProtocolErrorIfStreamsOpen� s � �z<HTTP2TimeoutTests.test_timeoutWithProtocolErrorIfStreamsOpenc C s� t � }t| jg |�}|�� }|d�dd� |D ��7 }| j|td�\}}}|�� }t|� � �}|� d� t|� � �} | � |d | � |� d� | � |�� |� dS ) zU When a L{H2Connection} loses its connection it cancels its timeout. r? c s s | ]}|� � V qd S r rq rs r r r! ri � r? zCHTTP2TimeoutTests.test_noTimeoutIfConnectionLost.<locals>.<genexpr>r� rd r% ry N) r rp r� r$ rt r� r� r� rZ r� �connectionLostr� r� ) r rh ro r� r r� r� ZsentDataZ oldCallCountZcurrentCallCountr r r! � test_noTimeoutIfConnectionLost� s � z0HTTP2TimeoutTests.test_noTimeoutIfConnectionLostc C sV | � � \}}}|�d� | �|j� | �|j� |�d� | �|j� | �|j� dS )z� When a L{H2Connection} has timed the connection out, and the transport doesn't get torn down within 15 seconds, it gets forcibly closed. � r% N�r� r� r� r� r ru �r r r� r� r r r! �,test_timeoutEventuallyForcesConnectionClosed s z>HTTP2TimeoutTests.test_timeoutEventuallyForcesConnectionClosedc C sH | � � \}}}|�d� |�d� |�d� | �|j� | �|j� dS )z� When a L{H2Connection} has timed the connection out, getting C{connectionLost} called on it cancels the forcible connection close. r� Nr% )r� r� r� r� r� r ru r� r r r! �$test_losingConnectionCancelsTheAbort! s z6HTTP2TimeoutTests.test_losingConnectionCancelsTheAbortc C s8 | j dd�\}}}|�d� | �|j� | �|j� dS )z� When a L{H2Connection} has timed the connection out but the C{abortTimeout} is set to L{None}, the connection is never aborted. N)r� l r� r� r r r! �'test_losingConnectionWithNoAbortTimeOut1 s z9HTTP2TimeoutTests.test_losingConnectionWithNoAbortTimeOutc C s@ | � � \}}}|�d� | �|j� | �|j� |�tj� dS )z} If a timed out transport doesn't close after 15 seconds, the L{HTTPChannel} will forcibly close it. � N)r� r� r� r� ru r� r ZConnectionDoner� r r r! �"test_connectionLostAfterForceClose> s z4HTTP2TimeoutTests.test_connectionLostAfterForceClosec C s� t � }t|�}|j|_d|_t� }t� }|�� }|�|� |�|� t |j|j �D ]"}|�|� d�� � � |� d� qT| �|j� dS )zR A client that sends only invalid frames is eventually timed out. �< r% N)r r r< r� r r r$ r� r� r� r� rG rr r� r� ru )r rv r� rh r� r$ r� r r r! �,test_timeOutClientThatSendsOnlyInvalidFramesO s z>HTTP2TimeoutTests.test_timeOutClientThatSendsOnlyInvalidFramesN)rO rP rQ rR r� rW r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� � s" &"", r� )Nr% )Nr% )DrR r� Zzope.interfacer r Ztwisted.internetr r r r Ztwisted.internet.addressr Ztwisted.internet.testingr r Ztwisted.pythonr Ztwisted.python.compatr Ztwisted.test.test_internetr Z twisted.trialr Z twisted.webr Ztwisted.web.test.test_httpr r r r r r r r� rH Z h2.errorsZ h2.exceptionsr' r� Z hpack.hpackr r Ztwisted.web._http2r � ImportErrorr rS rp ru rx r� ry r� r� r r� r r� r r� Z NotifyingRequestFactoryProxyr� ZTestCaser� r� r� r� r� r r r r! �<module> sz $ d* ! "� )
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка