HEX

File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_http2.cpython-310.pyc
o

�bF��@s>dZddlZddlmZmZddlmZmZmZm	Z	ddl
mZddlm
Z
mZddlmZddlmZdd	lmZdd
lmZddlmZddlmZmZmZmZmZmZm Z dZ!z$ddl"Z"ddl#Z"ddl$Z"ddl%Z%ddl&Z&dd
l'm(Z(m)Z)ddl*m+Z+Wne,y�dZ!YnwGdd�d�Z-Gdd�d�Z.d1dd�Z/d1dd�Z0dd�Z1Gdd�dej2�Z3e e3�Z4Gdd�dej2�Z5e e5�Z6Gdd �d e5�Z7e e7�Z8Gd!d"�d"ej2�Z9e e9�Z:Gd#d$�d$�Z;e e;�Z<Gd%d&�d&�Z=Gd'd(�d(ej>e=�Z?Gd)d*�d*ej>e=�Z@Gd+d,�d,ej>e=�ZAGd-d.�d.ej>e=�ZBGd/d0�d0ej>e=�ZCdS)2z
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|eZdZdZdd�Zdd�Zdd�Zgdfd	d
�Zddd
�Zddd�Z	dd�Z
ddd�Zd dd�Zd!dd�Z
gfdd�ZdS)"�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.
    cC�t�|_dS�N�r�encoder��self�r!�=/usr/lib/python3/dist-packages/twisted/web/test/test_http2.py�__init__8�zFrameFactory.__init__cCrrrrr!r!r"�refreshEncoder;r$zFrameFactory.refreshEncodercCsdS)NsPRI * HTTP/2.0

SM

r!rr!r!r"�clientConnectionPreface>�z$FrameFactory.clientConnectionPreface�c	Ks^tj�|�}|j�|�|_|j�d�|D]}|j�|�q|��D]
\}}t	|||�q"|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/�streamID�priorityKwargs�f�flag�k�vr!r!r"�buildHeadersFrameAszFrameFactory.buildHeadersFrameNcCs2|durt|�nt�}tj�|�}||_||_|S)zD
        Builds a single data frame out of a chunk of data.
        N)�setr*r+�	DataFramer.r/)r r.r/r4r6r!r!r"�buildDataFramePs
zFrameFactory.buildDataFrameFcCs&tj�d�}|r|j�d�||_|S)z1
        Builds a single settings frame.
        r�ACK)r*r+�
SettingsFramer/r0�settings)r r@�ackr6r!r!r"�buildSettingsFrameZs
zFrameFactory.buildSettingsFramecC�tj�|�}||_|S)z5
        Builds a single WindowUpdate frame.
        )r*r+�WindowUpdateFrame�window_increment)r r4�	incrementr6r!r!r"�buildWindowUpdateFramee�z#FrameFactory.buildWindowUpdateFramer�cCs"tj�d�}||_||_||_|S)z/
        Builds a single GOAWAY frame.
        r)r*r+�GoAwayFrame�
error_code�last_stream_id�additional_data)r �lastStreamID�	errorCode�additionalDatar6r!r!r"�buildGoAwayFramem�
zFrameFactory.buildGoAwayFramecCrC)z3
        Builds a single RST_STREAM frame.
        )r*r+�RstStreamFramerK)r r4rOr6r!r!r"�buildRstStreamFramewrHz FrameFactory.buildRstStreamFramecCs"tj�|�}||_||_||_|S)z1
        Builds a single priority frame.
        )r*r+�
PriorityFrame�
depends_on�
stream_weight�	exclusive)r r4�weight�	dependsOnrXr6r!r!r"�buildPriorityFramerRzFrameFactory.buildPriorityFramecCs:tj�|�}||_|j�|�|_t|�|_|j�	d�|S)z5
        Builds a single Push Promise frame.
        r))
r*r+�PushPromiseFrame�promised_stream_idrr-r.r;r/r0)r r4�promisedStreamIDr3r/r6r!r!r"�buildPushPromiseFrame�s
z"FrameFactory.buildPushPromiseFrame�Nr()F)rrI)r)rF)�__name__�
__module__�__qualname__�__doc__r#r%r&r:r=rBrGrQrTr[r_r!r!r!r"r1s







rc@s4eZdZdZdd�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.
    cCst�|_d|_dS)NrI)r�decoder�_datarr!r!r"r#�s
zFrameBuffer.__init__cCs|j|7_dSr)rg)r r.r!r!r"�receiveData�szFrameBuffer.receiveDatacCs|Srr!rr!r!r"�__iter__�r'zFrameBuffer.__iter__cCs�t|j�dkr
t��tjj�|jdd��\}}t|j�|dkr$t��|�t|jdd|���|jd|d�|_t	|tjj
�rM|jj|j
dd�|_
|S)N�	T)�raw)�lenrg�
StopIterationr*r+�Frame�parse_frame_header�
parse_body�
memoryview�
isinstancer,rf�decoder.)r r+�lengthr!r!r"�next�szFrameBuffer.nextN)	rarbrcrdr#rhriru�__next__r!r!r!r"re�srer(csT�durt��g}|��j|�d��|���fdd�|D��|dj�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�r3r4c3s�|]
}�j|�d�VqdS)�r4N)r=)�.0�chunk��frameFactoryr4r!r"�	<genexpr>�s�
�z%buildRequestFrames.<locals>.<genexpr>����
END_STREAM)r�appendr:�extendr/r0�r3r.r|r4�framesr!r{r"�buildRequestFrames�s�r�cCs"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}
    rIcs��|]}|��VqdSr��	serialize�ryr6r!r!r"r}���z$buildRequestBytes.<locals>.<genexpr>)r��joinr�r!r!r"�buildRequestBytes�sr�cCst�}|�|�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.
    )rerh�list)r.�bufferr!r!r"�framesFromBytes�s
r�c@seZdZdZdZdd�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>
    shello world!cCsDt|j�d�d�}|�d�t|�D]}|�|j�q|��dS)N�/r~��)�int�uri�split�setResponseCode�range�write�	chunkData�finish)r �chunks�_r!r!r"�processs

zChunkedHTTPHandler.processN)rarbrcrdr�r�r!r!r!r"r�sr�c@s0eZdZdZdd�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.
    cOs4tjj|g|�Ri|��|j��d|_d|_dS)NF)r�Requestr#�channel�pauseProducing�_requestReceivedrg�r �args�kwargsr!r!r"r#*s

zConsumerDummyHandler.__init__cCs|j��dS)z&
        Start the data pipe.
        N)r��resumeProducingrr!r!r"�
acceptData2szConsumerDummyHandler.acceptDatacOs d|_tjj|g|�Ri|��S)NT)r�rr��requestReceivedr�r!r!r"r�8sz$ConsumerDummyHandler.requestReceivedcCs0|�d�|j��|_d}|�|�|��dS)Nr�s0this is a response from a consumer dummy handler)r��content�readrgr�r�)r �
returnDatar!r!r"r�<s


zConsumerDummyHandler.processN)rarbrcrdr#r�r�r�r!r!r!r"r�"sr�c@�eZdZdZdd�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.
    cCs|j��|j��dS)z@
        Start and then immediately stop the data pipe.
        N)r�r��
stopProducingrr!r!r"r�Os
z'AbortingConsumerDummyHandler.acceptDataN)rarbrcrdr�r!r!r!r"r�Gsr�c@r�)�DummyProducerHandlerz�
    An HTTP request handler that registers a dummy producer to serve the body.

    The owner must call C{finish} to complete the response.
    cCs|�d�|�t�d�dS)Nr�T)r��registerProducerr
rr!r!r"r�cs
zDummyProducerHandler.processN)rarbrcrdr�r!r!r!r"r�\�r�c@s eZdZdZdd�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.
    cCs*g|_||_t|j�D]}t||�qdSr)�results�_wrappedFactoryrr)r �wrappedFactory�	interfacer!r!r"r#ss
�z NotifyingRequestFactory.__init__cOs(|j|i|��}|j�|���t|�Sr)r�r�r��notifyFinishr)r r�r��reqr!r!r"�__call__}sz NotifyingRequestFactory.__call__N)rarbrcrdr#r�r!r!r!r"r�ks
r�c@s eZdZdZer
eZdd�ZdS)�HTTP2TestHelpersz]
    A superclass that contains no tests but provides test helpers for HTTP/2
    tests.
    cCs|�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�priority�
DeadlockErrorru)r �
connectionr!r!r"�assertAllStreamsBlocked�sz(HTTP2TestHelpers.assertAllStreamsBlockedN)rarbrcrd�skipH2�skipr�r!r!r!r"r��s
r�c@sDeZdZgd�Zgd�Zgd�Zgd�ZdZgd�ZdZ	dd	�Z
d
d�Zdd
�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�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��:pathr��s:schemeshttps�s
user-agentstwisted-test-code)�
custom-header�1)r��2�)r��POSTr�)r��/post_endpointr�r�)�content-lengths25�shello sworld, sit's shttp/2!)��:statuss200)�requestr�)�commandr��sversionsHTTP/2)r�s13�
'''
None
'''
)r��r�r��r�r�r�)r�s36�$'''
25
hello world, it's http/2!'''
cCsNt�}t�}|��}|t|||�7}|�|�t|�D]}|�|�q||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}
        )rr
r&r��makeConnectionr�dataReceived)r r�r3�bodyr|�	transport�requestBytes�byter!r!r"�connectAndReceive�s
z"HTTP2ServerTests.connectAndReceivecs>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��t����}��t|�d���tdd�|dd�D�����t|dtjj	����t|dtjj
����t|dtjj
����t|dj�t�j
����|dj�j���|djd���d|djv�dS)	N�cs��|]}|jdkVqdS�r(N��	stream_idr�r!r!r"r}���zGHTTP2ServerTests.test_basicRequest.<locals>.validate.<locals>.<genexpr>r(��rIr)r��value�assertEqualrl�
assertTrue�allrrr*r+r,r<�dictr.�getResponseHeaders�getResponseDatar/�r4r��r r�r!r"�validate�s z4HTTP2ServerTests.test_basicRequest.<locals>.validater()rr�requestFactoryr��getRequestHeaders�_streamCleanupCallbacks�addCallback�r r�r�r�r!r�r"�test_basicRequest�s
z"HTTP2ServerTests.test_basicRequestcs@t�}t|_��|�j�j�\}���fdd�}|jd�|�S)zV
        Send a POST request and confirm that the data is safely transferred.
        cr�)	Nr�csr�r�r�r�r!r!r"r}	r�zFHTTP2ServerTests.test_postRequest.<locals>.validate.<locals>.<genexpr>������r~rIr)r�r�r�rlr�r�rrr*r+r,r<r�r.�postResponseHeaders�postResponseDatar/r�r�r!r"r�s z3HTTP2ServerTests.test_postRequest.<locals>.validater()rrr�r��postRequestHeaders�postRequestDatar�r�r�r!r�r"�test_postRequest�s
�z!HTTP2ServerTests.test_postRequestcs^gd��d�dd��jD�}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�s38s&'''
None
hello world, it's http/2!'''
cSs g|]\}}|dkr||f�qS)r�r!)ry�x�yr!r!r"�
<listcomp>%sz=HTTP2ServerTests.test_postRequestNoLength.<locals>.<listcomp>cs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����t|dtjj
����t|dtjj
����t|dj�t�����|dj����|djd���d|dj
v�dS)	Nr�csr�r�r�r�r!r!r"r}4r�zNHTTP2ServerTests.test_postRequestNoLength.<locals>.validate.<locals>.<genexpr>r�r�r~rIr)r�r�r�rlr�r�rrr*r+r,r<r�r.r/r��r�r�r r�r!r"r�/s z;HTTP2ServerTests.test_postRequestNoLength.<locals>.validater()rrrr�r�rr�r�)r rr�r�r�r!rr"�test_postRequestNoLengths��z)HTTP2ServerTests.test_postRequestNoLengthcs�d}t��t��t�}t|_ttd|dd�����fdd��D�}���}tj	�
t|��}|d�dd�|D��7}|�
��t|�D]}|�|�qC���fd	d
�}t�t|j�����|�S)zi
        Many interleaved POST requests all get received and responded to
        appropriately.
        �(r(r�csg|]}t�j�j�|��qSr!)r�rr�ryr4)r6r r!r"rPs���z=HTTP2ServerTests.test_interleavedRequests.<locals>.<listcomp>rIcsr�rr��ryr+r!r!r"r}\r�z<HTTP2ServerTests.test_interleavedRequests.<locals>.<genexpr>cs�t����}��t|�d��D]>��fdd�|D�}��t|�d���t|dj�t�j����|dj�j���|djd���d	|dj	v�qdS)
N�ycs(g|]}|j�krt|tjj�s|�qSr!)r�rrr*r+rDr�rxr!r"rms
��zOHTTP2ServerTests.test_interleavedRequests.<locals>.validate.<locals>.<listcomp>r�rr(r�rIr)
r�r�r�rlr�r.r�r�r�r/)r�r��streamFrames)�br �	streamIDsrxr"r�bs
���z;HTTP2ServerTests.test_interleavedRequests.<locals>.validate)rr
rrr�r�r�r&�	itertools�chain�
from_iterable�zipr�r�rr�r�DeferredListr��valuesr�)r �
REQUEST_COUNT�ar�r�r�r�r!)r
r6r rr"�test_interleavedRequestsAs&�
�z)HTTP2ServerTests.test_interleavedRequestscst��t��t�}t|_�j�d�d<��fdd�dD�}|ddj�d�d|dd_|d	dj�d�d
|d	d_�j	ddd	d
d�}|d�
d|�tj�
|�}���}|d�dd�|D��7}|���t|�D]}|�|�qo��fdd�}t�t|j�����|�S)zR
        Data in responses is interleaved according to HTTP/2 priorities.
        )z:pathz
/chunked/4r�csg|]	}t�g�|��qSr!)r�r	)r6r�r!r"r�s��zAHTTP2ServerTests.test_sendAccordingToPriority.<locals>.<listcomp>)r(r��r�PRIORITY�@r(� r�T)r4rYrZrXrIcsr�rr�r
r!r!r"r}�r�z@HTTP2ServerTests.test_sendAccordingToPriority.<locals>.<genexpr>csBt����}��t|�d�dd�|D�}gd�}��||�dS)N�cS� g|]}t|tjj�r|j�qSr!)rrr*r+r<r�r�r!r!r"r��
��zSHTTP2ServerTests.test_sendAccordingToPriority.<locals>.validate.<locals>.<listcomp>)r(r�r(r(r�r(r(r�rr�rr�rrr)r�r�r�rl)r�r�r�
expectedOrder�r
r r!r"r��s�z?HTTP2ServerTests.test_sendAccordingToPriority.<locals>.validate)rr
r�ChunkedHTTPHandlerProxyr�r�r/r0rWr[�insertrrrr&r�r�rr�rrr�r�rr�)r rr��
priorityFramer�r�r�r!)r
r6r�r r"�test_sendAccordingToPriority�s<��

�z-HTTP2ServerTests.test_sendAccordingToPrioritycCs�t�}t�}t�}t|_|��}|t|jg|�7}||jdd|jdgd��	�7}|�
|�t|�D]}|�|�|j
r=nq1t|���}|�t|�d�|�t|dtjj��|�|j
�dS)zR
        A protocol error from the remote peer terminates the connection.
        r(r�r))r4r^r3r/r�r~N)rr
rrr�r&r�r�r_r�r�rr��
disconnectingr�r�r�rlr�rrr*r+rJ)r r6r
rr�r�r�r!r!r"�&test_protocolErrorTerminatesConnection�s0��

�z7HTTP2ServerTests.test_protocolErrorTerminatesConnectioncs�t�}t|_��|�j�j�\}�|jdjj}��	|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(shello world, it's http/2!r�c�4t����}��t|�d���d|djv�dS�Nr�rr~�r�r�r�rlr�r/r�r�r!r"r�sz;HTTP2ServerTests.test_streamProducingData.<locals>.validate)r�ConsumerDummyHandlerProxyr�r�rr�streams�_request�original�assertFalser�r�r�r�rlr�r�rgr�r�)r r�r��requestr�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]}|�|�q3|jdjj}��|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.
        r~rIcsr�rr�r�r!r!r"r},r�zAHTTP2ServerTests.test_abortStreamProducingData.<locals>.<genexpr>r(csJt����}��t|�d���t|dtjj����|dj	d�dS)Nr�r~r()
r�r�r�rlr�rrr*r+rSr�r�r!r!r"r�?sz@HTTP2ServerTests.test_abortStreamProducingData.<locals>.validate)rr
r�!AbortingConsumerDummyHandlerProxyr�r�rrr;r/r&r�r�rr�r,r-r.r/r�r�r�r�)	r r6rr�r�r�r0�cleanupCallbackr�r!r!r"�test_abortStreamProducingDatas"


	z.HTTP2ServerTests.test_abortStreamProducingDatacs�t�}t|_��|�jg�\}�|jdjj}|�d�|�d�|j	d}|�
|jddd������
|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)rONsthird chunkc�Jt����}��t|�d���|djd���t|dtjj	��dS�Nr�r(�
r�r�r�rlr�r�rrr*r+r,r�r�r!r"r�u�z9HTTP2ServerTests.test_terminatedRequest.<locals>.validate)r�DummyProducerHandlerProxyr�r�r�r,r-r.r�r�r�rTr�r��
_disconnectedr�r��r r�r|r0r3r�r!r�r"�test_terminatedRequestJs"	�


�

z'HTTP2ServerTests.test_terminatedRequestcs�t�}t|_��|�jg�\}�|jdjj}|�d�|�d�|j	d}|�
|jdd������
|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(r5r6r�rNNcr7r8r9r�r�r!r"r��r:z<HTTP2ServerTests.test_terminatedConnection.<locals>.validate)rr;r�r�r�r,r-r.r�r�r�rQr�r�r<r�r/�_stillProducingr�r=r!r�r"�test_terminatedConnections"	�


�
z*HTTP2ServerTests.test_terminatedConnectioncsHt�}t|_�jdg}��||g�\}���fdd�}|jd�|�S)zq
        Requests containing Expect: 100-continue cause provisional 100
        responses to be emitted.
        )sexpects100-continuecs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����|dj
dg���d|djv�dS)Nrcsr�r�r�r�r!r!r"r}�r�zQHTTP2ServerTests.test_respondWith100Continue.<locals>.validate.<locals>.<genexpr>r()r�s100rr~)r�r�r�rlr�r�rrr*r+r,r.r/r�r�r!r"r��s z>HTTP2ServerTests.test_respondWith100Continue.<locals>.validater()rrr�r�r�r�r�)r r�r3r�r�r!r�r"�test_respondWith100Continue�s
z,HTTP2ServerTests.test_respondWith100Continuecsxt�}t|_��|�jg�\}�|jd}|jj}|jd}|�	���
|j���
|jdu���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(Ncs`t����}��t|�d���t|dtjj����|dj	dg���d|dj
v�dS)Nr�r()r�s400rr~)r�r�r�rlr�rrr*r+r,r.r/r�r�r!r"r��s
z6HTTP2ServerTests.test_respondWith400.<locals>.validate)rr;r�r�r�r,r-r.r��!_respondToBadRequestAndDisconnectr�r<r�r�)r r�r��streamr0r3r�r!r�r"�test_respondWith400�s


	z$HTTP2ServerTests.test_respondWith400csvt�}t|_��|�jg�\}�|jd}|jj}gd���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�writescs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����d|dj
v�dd�|D�}��|�d	g�dS)
Nrjcsr�r�r�r�r!r!r"r}r�zQHTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate.<locals>.<genexpr>r(rr~cSrr!�rrr*r+r<r.r�r!r!r"rrzRHTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate.<locals>.<listcomp>rI�r�r�r�rlr�r�rrr*r+r,r/�r4r��receivedDataChunks��
dataChunksr r�r!r"r�s ��z>HTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate)rr;r�r�r�r,r-r.r��loseConnectionr�r�)r r�r�rDr0rzr�r!rPr"�test_loseH2StreamConnection�s
z,HTTP2ServerTests.test_loseH2StreamConnectioncCsDt�}t|_|�||jg�|jd}|jj}|�t	|j
|d�dS)zK
        The L{H2Stream} object forbids registering two producers.
        r(TN)rr;r�r�r�r,r-r.r��
ValueErrorr�)r r�rDr0r!r!r"�test_cannotRegisterTwoProducers#s
z0HTTP2ServerTests.test_cannotRegisterTwoProducerscsjt�}t|_��|�jg�\}�|jd}|jj��jj	}|�
�fdd����fdd�}|jd�
|�S)z�
        L{Request} objects that have registered pull producers get blocked and
        unblocked according to HTTP/2 flow control.
        r(cs���Sr)r�)r)r0r!r"�<lambda>?sz;HTTP2ServerTests.test_handlesPullProducer.<locals>.<lambda>c�Bt����}��d|djv�dd�|D�}��|gd��dS)Nrr~cSrr!rLr�r!r!r"rIrzOHTTP2ServerTests.test_handlesPullProducer.<locals>.validate.<locals>.<listcomp>)�0r�r��3�4�5�6�7�8�9rI�r�r�r�r/r��r4r�rQr�r!r"r�Bs��z;HTTP2ServerTests.test_handlesPullProducer.<locals>.validate)rrr�r�r�r,r-r.�_actualProducer�resultr�r�)r r�r�rD�producerCompleter�r!)r0r r�r"�test_handlesPullProducer1s
z)HTTP2ServerTests.test_handlesPullProducercCsJt�}t|_|�||jg�|jdjj}|�|�	��|jd�
�dS)zJ
        L{Request} objects can correctly ask isSecure on HTTP/2.
        r(N)rrr�r�r�r,r-r.r/�isSecure�abortConnection)r r�r0r!r!r"�test_isSecureWorksProperlySsz+HTTP2ServerTests.test_isSecureWorksProperlycsZt�}t|_��|�jg�\}�|jdjj}t�	d|j
���fdd�}|jd�|�S)zL
        L{H2Connection} correctly unblocks when a stream is ended.
        r(�{�G�z�?cr()Nr�rr~r*�r�r�r�r!r"�validateCompletel�zCHTTP2ServerTests.test_lateCompletionWorks.<locals>.validateComplete)
rrr�r�r�r,r-r.r�	callLaterr�r�r�)r r�r�r0rkr!r�r"�test_lateCompletionWorks_sz)HTTP2ServerTests.test_lateCompletionWorkscstt�}t|_��|�jg�\}�|jd}|jj}|�d�|�	gd��|�
�|jd}��fdd�}|�|�S)zW
        L{H2Stream} objects can send a series of frames via C{writeSequence}.
        r(r�)�Hello�,�world!crW)Nrr~cSrr!rLr�r!r!r"r�rzTHTTP2ServerTests.test_writeSequenceForChannels.<locals>.validate.<locals>.<listcomp>)rorprqrIr`rar�r!r"r����z@HTTP2ServerTests.test_writeSequenceForChannels.<locals>.validate)
rrr�r�r�r,r-r.r��
writeSequencer�r�r�)r r�r�rDr0�completionDeferredr�r!r�r"�test_writeSequenceForChannelsus



z.HTTP2ServerTests.test_writeSequenceForChannelscs�t��t��t��t�_���}|��tjj	j
di���7}|t�j
g��7}����t|�D]}��|�q0�jd}|jj���d�gd����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.
        rr(�fiver)rHrIrJrKcs �D]}��|�q���dSr)r�r�)rz)rQr0r!r"�write_chunks�sz7HTTP2ServerTests.test_delayWrites.<locals>.write_chunksrics���jddd����S)Nr(�2�r4rF)r�rGr��r��rr6r!r"rV�s�z3HTTP2ServerTests.test_delayWrites.<locals>.<lambda>cs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����d|dj
v�dd�|D�}��|d	g�d
g�dS)Nrjcsr�r�r�r�r!r!r"r}�r�zFHTTP2ServerTests.test_delayWrites.<locals>.validate.<locals>.<genexpr>r�rr~cSrr!rLr�r!r!r"r�rzGHTTP2ServerTests.test_delayWrites.<locals>.validate.<locals>.<listcomp>rvrIrMrN)r
rQr r!r"r��s ��z3HTTP2ServerTests.test_delayWrites.<locals>.validate)rr
rrr�r&rB�h2r@�SettingCodes�INITIAL_WINDOW_SIZEr�r�r�r�rr�r,r-r.r�r�
deferLaterrr�r�)r r�r�rDrw�dr�r!)rr
rQr6r0r r"�test_delayWrites�s2��


�z!HTTP2ServerTests.test_delayWritescCs�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.
        �r3r.r|r(rxN)rr
rrr�r&r�r�rTr�r�r�r�r�r�rl�assertNotInr�)r r|r�rr�r�r!r!r"�test_resetAfterBody�s�

z$HTTP2ServerTests.test_resetAfterBodycs�t���G�fdd�dt�}t��t��}t|��_|�_t	��_
����jg���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.
        cseZdZ�fdd�ZdS)zTHTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.SuperRequestcs2tj|g|�Ri|����|jj|jjf�dSr)rr#�callbackr��site�factoryr��r�r!r"r#�sz]HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.SuperRequest.__init__N)rarbrcr#r!r�r!r"�SuperRequest�sr�cs(|\}}��|�j���|�j�dSr)�assertIsr�r�)r�r�r�)r�r r!r"�validateFactoryAndSitesz^HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.validateFactoryAndSiter()r�Deferredrrr�HTTPFactoryrr�r��objectr�r�r�r�r��
gatherResults)r r��httpFactoryr�r3r!)r�r�r r"�-test_RequestRequiringFactorySiteInConstructor�s


z>HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructorcsrt�}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(cs��|�dSr)�assertIsNone)rcrr!r"r��z?HTTP2ServerTests.test_notifyOnCompleteRequest.<locals>.validater)
rr�rr�r�r�r�r�rlr�r�rr�)r r�r�r��	deferredsr�r�r3r!rr"�test_notifyOnCompleteRequests


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���d�dS�Nz#Didn't errback, called back instead��fail��ignrr!r"r�6r�z;HTTP2ServerTests.test_notifyOnResetStream.<locals>.callbackc�"��|tj���|jtj�dSr��assertIsInstancer�Failurer��typer�ConnectionLost��reasonrr!r"�errback9�z:HTTP2ServerTests.test_notifyOnResetStream.<locals>.errbackrrx)
rr�rr�r�r�r�r�rl�addCallbacksrTr�r�)	r r�r|r�r�r�r�r��invalidDatar!rr"�test_notifyOnResetStream&s
�
z)HTTP2ServerTests.test_notifyOnResetStreamc
s�t�}tt�|_��|�jg�\}}t�jg|dd�}|�|�|jj}��	t
|�d��fdd�}�fdd�}|D]}|�||�q:|jdd	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�r{r�cr�r�r�r�rr!r"r�ar�z=HTTP2ServerTests.test_failWithProtocolError.<locals>.callbackcs$��|tj���|jtjj�dSr)r�rr�r�r|�
exceptions�
ProtocolErrorr�rr!r"r�dsz<HTTP2ServerTests.test_failWithProtocolError.<locals>.errback�yo���r.r4)rr�rr�r�r�r�r�r�r�rlr�r=r�rr��
r r�r|r��
secondRequestr�r�r�r�r�r!rr"�test_failWithProtocolErrorGs$	
�
�


z+HTTP2ServerTests.test_failWithProtocolErrorc
s�t�}tt�|_��|�jg�\}}t�jg|dd�}|�|�|jj}��	t
|�d��fdd�}�fdd�}|D]}|�||�q:|jdd��
�}	|�|	�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�r{r�cr�r�r�r�rr!r"r��r�z4HTTP2ServerTests.test_failOnGoaway.<locals>.callbackcr�rr�r�rr!r"r��r�z3HTTP2ServerTests.test_failOnGoaway.<locals>.errbackr?)rr�rr�r�r�r�r�r�r�rlr�rQr�rr�r�r!rr"�test_failOnGoawayss$	
�
�


z"HTTP2ServerTests.test_failOnGoawayc	s�t�}tt�|_��|�jg�\}}t�jg|dd�}|�|�|jj}��	t
|�d��fdd�}�fdd�}|D]}|�||�q:|��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�r{r�cr�r�r�r�rr!r"r��r�z;HTTP2ServerTests.test_failOnStopProducing.<locals>.callbackcr�rr�r�rr!r"r��r�z:HTTP2ServerTests.test_failOnStopProducing.<locals>.errback)rr�rr�r�r�r�r�r�r�rlr�r�rr�)	r r�r|r�r�r�r�r�r�r!rr"�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(cr�r�r�r�rr!r"r��r�z7HTTP2ServerTests.test_notifyOnFast400.<locals>.callbackcr�rr�r�rr!r"r��r�z6HTTP2ServerTests.test_notifyOnFast400.<locals>.errbackr)rr�rr�r�r�r�r�rlr�r,rC)	r r�r|r�r�r�r�r�rDr!rr"�test_notifyOnFast400�s
�
z%HTTP2ServerTests.test_notifyOnFast400cCs�t�}t|�}|j|_t|_d}t�}t�}|��}|�|�|�	|�|�	t
|jg||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(rxrN)r	rrmrr�rr
r&r�r�r�r�r��_maxBufferedControlFrameBytesrCr��disconnected)r �
memoryReactorr�r4r|r�r&r!r!r"�test_fast400WithCircuitBreaker�s&


��	
z/HTTP2ServerTests.test_fast400WithCircuitBreakercCs�t�}t|_t�}t�}|��}|�|�|�|�|��t	dd�D]}|�|�
i����q#t|�
��}|�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�dr(�eN)rrr�rr
r&r�r�r�r�rBr�r�r�r�rlr�)r r�r|r�r&r�r�r!r!r"� test_bufferingAutomaticFrameDatas

z1HTTP2ServerTests.test_bufferingAutomaticFrameDatacCs�t�}t|_t�}t�}|��}|�|�|�|�|��d|_	|�
|j�tdd�D]}|�|�
i����q,|�
|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.
        r�r�N)rrr�rr
r&r�r�r�r�r/r&r�rBr�r�r�)r r�r|r�r&r�r!r!r"�2test_bufferingAutomaticFrameDataWithCircuitBreaker6s

zCHTTP2ServerTests.test_bufferingAutomaticFrameDataWithCircuitBreakercCs
Gdd�dt�}t�}t|_t�}|�}|�|d�|��}|�|�|�|�|�	|j
�t|���}|�
t|�d�|�
|jd�tdd�D]}|�|�i����qHt|���}|�
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@seZdZdd�ZdS)zeHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWrite.<locals>.AutoPausingStringTransportc_s&tj|g|�Ri|��|j��dSr)r
r��producerr�r�r!r!r"r�_szkHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWrite.<locals>.AutoPausingStringTransport.writeN)rarbrcr�r!r!r!r"�AutoPausingStringTransport^sr�Tr(rr��cr��ZN)r
rrr�rr�r&r�r��assertIsNotNone�_consumerBlockedr�r�r�rl�_bufferedControlFrameBytesr�rBr�r�)r r�r�r|r�r&r�r�r!r!r"�0test_bufferingContinuesIfProducerIsPausedOnWriteXs,

zAHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWritecCsxt�}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.
        rr�r�r�N)r	rrmrr
r&r�r�r�r�r=r�r�r�)r r�r�r|r�r&r�r!r!r"�+test_circuitBreakerAbortsAfterProtocolError�s


z<HTTP2ServerTests.test_circuitBreakerAbortsAfterProtocolErrorN)+rarbrcr�rrr�r�r�r�r�r�rrrr%r'r1r4r>rArBrErSrUrerhrnrur�r�r�r�r�r�r�r�r�r�r�r�r�r�r!r!r!r"r��sR
	+@G'-.54'-"B"!,*)'*"-r�c@s�eZdZdZgd�ZdZgd�Zgd�ZdZdd�Z	d	d
�Z
dd�Zd
d�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �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]}|�|�q0t�j�d}t|�D]}|jddd�}|�|���qC��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.
        rr(ryc�Ft����}��d|djv�d�dd�|D��}���j|�dS)Nrr~rIcs�$�|]
}t|tjj�r|jVqdSrrLr�r!r!r"r}����
�zMH2FlowControlTests.test_bufferExcessData.<locals>.validate.<locals>.<genexpr>)r�r�r�r/r�r�r��r4r��actualResponseDatar!r!r"r���
�z:H2FlowControlTests.test_bufferExcessData.<locals>.validate)rr
rrr�r&rBr|r@r}r~r�r�r�r�rr�rlr�r�rGr�r�)	r r6rr�r��bonusFramesr�r+r�r!r!r"�test_bufferExcessData�s(��
z(H2FlowControlTests.test_bufferExcessDatacs�t�}t��t�}t|_|��}||�tjj	j
di���7}|t�j
g|�7}|���t|�D]}|�|�q0|jd}|jj}��|j�|�d���|j���|jjdg�|�|jddd������|j���|jjdg�|�|jddd������|j���|jjdg�|�|jddd������|j���|jjddg�|�d���|j���|jjgd��|�|jdd	d������|j���|jjgd
��|��|����fdd�}|jd�|�S)
z�
        L{Request} objects that have registered producers get blocked and
        unblocked according to HTTP/2 flow control.
        rr(�
helloworld�pauseryr�resume)r�r�r�rx)r�r�r�r�c�Bt����}��d|djv�dd�|D�}��|gd��dS)Nrr~cSrr!rLr�r!r!r"r:rzXH2FlowControlTests.test_producerBlockingUnblocking.<locals>.validate.<locals>.<listcomp>)r�r�rIr`rar!r!r"r�3rrzDH2FlowControlTests.test_producerBlockingUnblocking.<locals>.validate) rr
rr;r�r&rBr|r@r}r~r�r�r�r�rr�r,r-r.r��_producerProducingr�r/r�r��eventsrG�unregisterProducerr�r�r��r r6rr�r�rDr0r�r!r!r"�test_producerBlockingUnblocking�sR��



�z2H2FlowControlTests.test_producerBlockingUnblockingcst��t��t��t�_���}|��tjj	j
di���7}|t�j
g��7}����t|�D]}��|�q0�jd��jj����j���d����j����jjdg���d������fdd�}t�td|�}��fd	d
�}�jd�|�}t �!||g�S)zQ
        Exactly filling the flow control window still blocks producers.
        rr(r�r��hcsL���jddd�������j����jjddg������	�dS)Nr(rxryr�r�)
r�rGr�r�r�r�r�r�r�r�r!)rr6r0r rDr!r"�window_openhs�z=H2FlowControlTests.test_flowControlExact.<locals>.window_openrcr�)Nrr~cSrr!rLr�r!r!r"r{rzNH2FlowControlTests.test_flowControlExact.<locals>.validate.<locals>.<listcomp>)rFrGr�rIr`rar!r!r"r�trrz:H2FlowControlTests.test_flowControlExact.<locals>.validate)"rr
rr;r�r&rBr|r@r}r~r�r�r�r�rr�r,r-r.r�r�r�r/r�r�r�rrrr�r�rr)r r�r�r��windowDeferr��
validateDeferr!)rr
r6r0r rDr"�test_flowControlExactAs4��



	z(H2FlowControlTests.test_flowControlExactcs�t�}t��t�}t|_|��}||�tjj	j
di���7}|t�j
g|�7}|���t|�D]}|�|�q0|jd}|jj}��|j�|�d�|��|����|j�t�d|j|jddd������fdd�}|jd�|�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.
        rr(r�rrxrycr�)Nrr~cSrr!rLr�r!r!r"r�rzQH2FlowControlTests.test_endingBlockedStream.<locals>.validate.<locals>.<listcomp>)rFrGrIr`rar!r!r"r��rrz=H2FlowControlTests.test_endingBlockedStream.<locals>.validate)rr
rr;r�r&rBr|r@r}r~r�r�r�r�rr�r,r-r.r�r�r�r�r��finishedrrmrGr�r�r�r!r!r"�test_endingBlockedStream�s8��


�z+H2FlowControlTests.test_endingBlockedStreamc	s�t�}t��t�}t|_|��}|t�jg|�7}|���t	|�D]}|�
|�q"|jd}|jj
}|jd}|��|����|j���fdd�}|�|�S)z<
        We safely handle responses without bodies.
        r(c�Pt����}��t|�d���d|djv�dd�|D�}��|dg�dS)Nr�rr~cSrr!rLr�r!r!r"r�rzQH2FlowControlTests.test_responseWithoutBody.<locals>.validate.<locals>.<listcomp>rIr*rar!r!r"r�����z=H2FlowControlTests.test_responseWithoutBody.<locals>.validate)rr
rr;r�r&r�r�r�rr�r,r-r.r�r�r�r�r�r��	r r6rr�r�rDr0r3r�r!r!r"�test_responseWithoutBody�s"



z+H2FlowControlTests.test_responseWithoutBodyc	s�t�}t��t�}t|_|��}|t�jg|�7}|���t	|�D]}|�
|�q"|jd}|jj
}|jd}|��|����|j�|�
|jddd������fdd�}|�|�S)zk
        WindowUpdate frames received after we've completed the stream are
        safely handled.
        r(rxrycr�)Nr�rr~cSrr!rLr�r!r!r"r"rz[H2FlowControlTests.test_windowUpdateForCompleteStream.<locals>.validate.<locals>.<listcomp>rIr*rar!r!r"r�r�zGH2FlowControlTests.test_windowUpdateForCompleteStream.<locals>.validate)rr
rr;r�r&r�r�r�rr�r,r-r.r�r�r�r�r�rGr�r�r�r!r!r"�"test_windowUpdateForCompleteStream�s$



z5H2FlowControlTests.test_windowUpdateForCompleteStreamcst�}t��t�}t|_|��}||�tjj	j
di���7}|t�j
g|�7}|���t|�D]}|�|�q0|jd}|jj}��|j�|�d���|j���|jjg�|�|jddd������|j���|jjg�|��|����fdd�}|jd�|�S)z�
        L{Request} objects that have registered producers that are not blocked
        behind flow control do not have their producer notified.
        rr(�wordrycsBt����}��d|djv�dd�|D�}��|ddg�dS)Nrr~cSrr!rLr�r!r!r"rarzOH2FlowControlTests.test_producerUnblocked.<locals>.validate.<locals>.<listcomp>r�rIr`rar!r!r"r�Zrrz;H2FlowControlTests.test_producerUnblocked.<locals>.validate)rr
rr;r�r&rBr|r@r}r~r�r�r�r�rr�r,r-r.r�r�r�r�r�r�rGr�r�r�r�r�r!r!r"�test_producerUnblocked,s6��


z)H2FlowControlTests.test_producerUnblockedcs�t�}t��t�}t|_t�j�j|�}|�d|j	ddd��|�
�}|d�dd�|D��7}|���t
|�D]}|�|�q8��fdd	�}|jd�|�S)
z�
        When a WindowUpdate frame is received for the whole connection but no
        data is currently waiting, nothing exciting happens.
        r(rrryrIcsr�rr�r�r!r!r"r}vr�zBH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.<genexpr>cr�)Nrr~rIcsr�rrLr�r!r!r"r}�r�zTH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.validate.<locals>.<genexpr>)r�r�r�r/r�r�r�r�r!r!r"r�}r�zAH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.validate)rr
rrr�r�rrr#rGr&r�r�rr�r�r�)r r6rr�r�r�r�r!r!r"�test_unnecessaryWindowUpdatehs
z/H2FlowControlTests.test_unnecessaryWindowUpdatecCs�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(rwrryrIcsr�rr�r�r!r!r"r}�r�zKH2FlowControlTests.test_unnecessaryWindowUpdateForStream.<locals>.<genexpr>N)rr
rrr�r�r:rrGr&r�r�r�r�)r r6r��connr�r.r!r!r"�%test_unnecessaryWindowUpdateForStream�s

z8H2FlowControlTests.test_unnecessaryWindowUpdateForStreamcCs�t�}t�}t�}t|_t|j|j|�}|��}|d�	dd�|D��7}|�
|�t|�D]}|�|�q-|j
d��|jddd�}|�|���t|���}|�t|dtjj��dS)	zq
        When a WindowUpdate frame is received for a stream that has been
        aborted it is ignored.
        rIcsr�rr�r�r!r!r"r}�r�zEH2FlowControlTests.test_windowUpdateAfterTerminate.<locals>.<genexpr>r(rryr~N)rr
rrr�r�rrr&r�r�rr�r,rgrGr�r�r�r�rrr*r+rS)r r6r
rr�r�r��windowUpdateFramer!r!r"�test_windowUpdateAfterTerminate�s
z2H2FlowControlTests.test_windowUpdateAfterTerminatecs�t��t��t��t�_t�j�j��}���}|d�	dd�|D��7}��
��t|�D]}��|�q-��fdd�}��fdd�}�j
d�|�}|�|�S)	zs
        When a WindowUpdate frame is received for a stream that has been
        completed it is ignored.
        rIcsr�rr�r�r!r!r"r}�r�zDH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.<genexpr>cs �jddd�}��|���dS)Nr(rry)rGr�r�)r�r�r{r!r"�
update_window�szHH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.update_windowcs"t����}��d|dj�dS)Nrr~)r�r��assertInr/rjr!r!r"r��szCH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.validater()rr
rrr�r�rrr&r�r�rr�r�r�)r r�r�r�r�r�r�r!)rr
r6r r"�test_windowUpdateAfterComplete�s

z1H2FlowControlTests.test_windowUpdateAfterCompletecCst�}t�}t�}t|_ddgd}tdd�|D���}|jdd�d|fg}t|||d	�}|d=|�|j	d
t
jjj
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�css�|]}t|�VqdSr)rl)ryr.r!r!r"r}�r�z;H2FlowControlTests.test_dataAndRstStream.<locals>.<genexpr>Nr~zcontent-lengthr�r()r4rOrIcsr�rr�r�r!r!r"r}	r�cSrr!)rrr*r+rDr�r�r!r!r"r
	s��z<H2FlowControlTests.test_dataAndRstStream.<locals>.<listcomp>rcS�g|]}t|tjj�r|�qSr!)rrr*r+r,r�r!r!r"r	s
��cSr�r!)rrr*r+r<r�r!r!r"r	s)rr
rrr��sumrr�r�rTr|�errors�
ErrorCodes�INTERNAL_ERRORr&r�r�r�r�r�r�r/)r r|r�r�	frameData�
bodyLengthr3r�r��windowUpdateFrameIDs�
headersFrames�
dataFramesr!r!r"�test_dataAndRstStream�s>�
��

��
z(H2FlowControlTests.test_dataAndRstStreamcCs�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(rwrN)r	rrmrr�rr
r&r�r�r:rr�r�r��clear�abortRequestr/r�r�r�)r r�r�r|r�r&r4�headersFrameDatar!r!r"�#test_abortRequestWithCircuitBreaker	s,

��

z6H2FlowControlTests.test_abortRequestWithCircuitBreakerN)rarbrcrdr�r�rrr�r�r�r�r�r�r�r�r�r�r�r�r�r�r!r!r!r"r��s(	-SB:4;<# "7r�c@s4eZdZgd�Zdd�Zdd�Zdd�Zdd	�Zd
S)�HTTP2TransportCheckingr�cCs2t�}t�}t|_|�|d�|�|j|u�dS)zU
        L{H2Connection} can be registered with the transport as a producer.
        TN)r
rrr�r�r�r�)r r
rr!r!r"�"test_registerProducerWithTransportT	s
z9HTTP2TransportChecking.test_registerProducerWithTransportcs�t�}t��t��t�_t�jg|�}|��}|d�dd�|D��7}��	����
�d�t|�D]}��|�q2��
��jd�����fdd�}��fdd	�}t�td
|�}|�|�|S)z|
        L{H2Connection} can be paused by its consumer. When paused it stops
        sending data to the transport.
        rIcsr�rr�r�r!r!r"r}l	r�zNHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.<genexpr>Tr(cs`t����}��t|�d���t|dtjj����	���	���	���	���	��S)Nr�r~)
r�r�r�rlr/rrr*r+r<r�rj�rr
r3r r!r"�validateNotSent�	szTHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.validateNotSentcs4t����}��t|�d���d|djv�dSr)r*rjr!r!r"rk�	rlzUHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.validateCompleteri)rr
rrr�r�r�r&r�r�r�rr�r�r�rrrr�)r r6r�r�r�rrkr�r!rr"�$test_pausingProducerPreventsDataSend_	s$



z;HTTP2TransportChecking.test_pausingProducerPreventsDataSendcCs�t�}t�}t�}t|_t|jg|�}|��}|d�dd�|D��7}|�	|�|�
|d�t|�D]}|�|�q2|�
�t|���}|�t|�d�|�t|dtjj��|�|j�dS)zo
        L{H2Connection} can be stopped by its producer. That causes it to lose
        its transport.
        rIcsr�rr�r�r!r!r"r}�	r�z<HTTP2TransportChecking.test_stopProducing.<locals>.<genexpr>Tr�r~N)rr
rrr�r�r�r&r�r�r�rr�r�r�r�r�rlr/rrr*r+r<r@)r r6r
rr�r�r�r!r!r"�test_stopProducing�	s 
z)HTTP2TransportChecking.test_stopProducingc	s�tddd��tddd��t�}t��d�}t�}t|_|�|�t�jg|�}|�	�}|d�
dd	�|D��7}t|�D]}|�|�q;|j
d
���������������|jd
}����fdd�}|�|�S)
z{
        A L{H2Stream} object correctly passes through host and peer information
        from its L{H2Connection}.
        �TCPz
17.52.24.8i�z17.188.0.12i})�hostAddress�peerAddressrIcsr�rr�r
r!r!r"r}�	r�zEHTTP2TransportChecking.test_passthroughHostAndPeer.<locals>.<genexpr>r(cs$��������������dSr)r��getHost�getPeerrz�rrr rDr!r"r��	szDHTTP2TransportChecking.test_passthroughHostAndPeer.<locals>.validate)rrr
rrr�r�r�r�r&r�rr�r,r�rrr�r�)	r r|r�r�r�r�r�r3r�r!r	r"�test_passthroughHostAndPeer�	s$



z2HTTP2TransportChecking.test_passthroughHostAndPeerN)rarbrcr�r�rrr
r!r!r!r"r�I	s
>r�c@r�)�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.
    cCs|t��}t|�}|��}|�t|�d�|d}|�|���|�|jd�|�|j	|j
�|�|jd�|�|ji�dS)z}
        When a H2Connection is established it schedules one call to be run as
        soon as the reactor has time.
        r(rr!N)
r�Clockr�getDelayedCallsr�rlr��active�time�func�_sendPrioritisedDatar��kw)r rr�calls�callr!r!r"�"test_initiallySchedulesOneDataCall�	sz7HTTP2SchedulingTests.test_initiallySchedulesOneDataCallN)rarbrcrdrr!r!r!r"r�	r�rc@s�eZdZdZgd�Ze�Zdd�Zdd�Zdd�Z	efd	d
�Z
dd�Zd
d�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�ZdS) �HTTP2TimeoutTestsz@
    The L{H2Connection} object times out idle connections.
    r�cCs|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)rm)r r�rr!r!r"�patch_TimeoutMixin_clock
s
z*HTTP2TimeoutTests.patch_TimeoutMixin_clockcCs^t��}t|�}d|_|�||�t�}t|�|_|�|�t	|�D]}|�
|�q"|||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.
        r�)rrr�timeOutrr
rr�r�rr�)r �initialDatar�rr�r�r�r!r!r"�initiateH2Connection
s



z&HTTP2TimeoutTests.initiateH2ConnectioncCsXt|�}|�t|�|�|�t|dtjj��|�|dj|�|�|dj	|�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.
        r~N)
r�r�rlr�rrr*r+rJrKrL)r r.�
frameCountrOrNr�r!r!r"�assertTimedOut9
s
z HTTP2TimeoutTests.assertTimedOutcCs�||jurtj}t�}|��}|j|td�\}}}||_|�d�|j|�	�dt
jjj
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�r�r�r�rrOrN)�_DEFAULTr�abortTimeoutrr&rr�advancerr�r|r�r��NO_ERRORr�r&r/r�)r r r|rrr�r�r!r!r"�prepareAbortTestF
s&
�
�
z"HTTP2TimeoutTests.prepareAbortTestcCs�t�}|��}|j|td�\}}}|��}|�d�|�||���|�|j�|�d�|j	|��dt
jjj
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.
        rr�r�rrN)rr&rrr�r!r�r/r&rr|r�r�r"r�)r r|rrr�r��preambler!r!r"�test_timeoutAfterInactivityl
s$�

�z-HTTP2TimeoutTests.test_timeoutAfterInactivitycCs�t�}d}|j|td�\}}}t|���D]}|�|�|�d�|�|j�q|�d�|j	|�
�dtjj
jdd�|�|j�dS)zM
        When a L{H2Connection} receives data, the timeout is reset.
        rIrr�r�rrN)rrrrr&r�r!r/r&rr�r|r�r�r"r�)r r|rrr�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
�qF|�d�|j|��d
tjjjdd�dS)zJ
        When a L{H2Connection} sends data, the timeout is reset.
        rIcsr�rr�r�r!r!r"r}�
r�zDHTTP2TimeoutTests.test_timeoutResetByResponseData.<locals>.<genexpr>cst||d�}��|�|S)N)�queued)rr�)rDr'r���requestsr!r"�saveRequest�
s
zFHTTP2TimeoutTests.test_timeoutResetByResponseData.<locals>.saveRequestrr�r(�
rs
some bytesr��
rN)rr�r�r&r�rr�r!�assertEqualsrlr�r�r/r&rr�r|r�r��PROTOCOL_ERROR)	r r|rr�r*rr�r�rr!r(r"�test_timeoutResetByResponseData�
s2�



�z1HTTP2TimeoutTests.test_timeoutResetByResponseDatacCs~t�}t|jg|�}|��}|d�dd�|D��7}|j|td�\}}}|�d�|j|�	�dt
jjj
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}.
        rIcsr�rr�r�r!r!r"r}�
r�zOHTTP2TimeoutTests.test_timeoutWithProtocolErrorIfStreamsOpen.<locals>.<genexpr>rr�r�r(rN)rr�r�r&r�rr�r!rr�r|r�r�r.r�r&)r r|r�rrr�r�r!r!r"�*test_timeoutWithProtocolErrorIfStreamsOpen�
s �
�z<HTTP2TimeoutTests.test_timeoutWithProtocolErrorIfStreamsOpenc
Cs�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.
        rIcsr�rr�r�r!r!r"r}�
r�zCHTTP2TimeoutTests.test_noTimeoutIfConnectionLost.<locals>.<genexpr>rr�r(r�N)
rr�r�r&r�rr�r�rlr
�connectionLostr�r!)
r r|r�rrr�r��sentData�oldCallCount�currentCallCountr!r!r"�test_noTimeoutIfConnectionLost�
s�

z0HTTP2TimeoutTests.test_noTimeoutIfConnectionLostcCsV|��\}}}|�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/r��r rr�r�r!r!r"�,test_timeoutEventuallyForcesConnectionCloseds

z>HTTP2TimeoutTests.test_timeoutEventuallyForcesConnectionClosedcCsH|��\}}}|�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.
        r6Nr()r#r!r1r�r&r/r�r8r!r!r"�$test_losingConnectionCancelsTheAbort!s


z6HTTP2TimeoutTests.test_losingConnectionCancelsTheAbortcCs8|jdd�\}}}|�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 lr7r8r!r!r"�'test_losingConnectionWithNoAbortTimeOut1s
z9HTTP2TimeoutTests.test_losingConnectionWithNoAbortTimeOutcCs@|��\}}}|�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&r�r1r�ConnectionDoner8r!r!r"�"test_connectionLostAfterForceClose>s

z4HTTP2TimeoutTests.test_connectionLostAfterForceClosecCs�t�}t|�}|j|_d|_t�}t�}|��}|�|�|�|�t	|j|j
�D]}|�|�d����|�
d�q*|�|j�dS)zR
        A client that sends only invalid frames is eventually timed out.
        �<r(N)r	rrmrrr
r&r�r�r�r rTr�r!r�r�)r r�r�r|r�r&r�r!r!r"�,test_timeOutClientThatSendsOnlyInvalidFramesOs

z>HTTP2TimeoutTests.test_timeOutClientThatSendsOnlyInvalidFramesN)rarbrcrdr�r�rrrrr#r%r&r/r0r5r9r:r;r>r@r!r!r!r"r�	s$
&"",
rr`)Drdr�zope.interfacerr�twisted.internetrrrr�twisted.internet.addressr�twisted.internet.testingr	r
�twisted.pythonr�twisted.python.compatr�twisted.test.test_internetr
�
twisted.trialr�twisted.webr�twisted.web.test.test_httprrrrrrrr�r|�	h2.errors�
h2.exceptionsr*r��hpack.hpackrr�twisted.web._http2r�ImportErrorrrer�r�r�r�r�r"r�r+r�r2r�r;r��NotifyingRequestFactoryProxyr��TestCaser�r�r�rrr!r!r!r"�<module>s~$
�d
*
!"�)