HEX
Server: Apache
System: Linux scp1.abinfocom.com 5.4.0-216-generic #236-Ubuntu SMP Fri Apr 11 19:53:21 UTC 2025 x86_64
User: confeduphaar (1010)
PHP: 8.1.33
Disabled: exec,passthru,shell_exec,system
$ pwd: /proc/self/root/lib/python3/dist-packages/twisted/web/test/__pycache__/
Upload Files
File: //proc/self/root/lib/python3/dist-packages/twisted/web/test/__pycache__/test_http2.cpython-38.pyc
U

s�@g���@sZdZddlmZmZddlZddlmZmZddlm	Z	m
Z
mZmZddl
mZddlmZddlmZmZdd	lmZdd
lmZddlmZddlmZmZmZmZmZm Z m!Z!dd
l"m#Z#dZ$zHddl%m&Z&ddl'Z'ddl(Z'ddl)Z'ddl*Z*ddl+Z+ddl,m-Z-m.Z.Wne/k
�r&dZ$YnXGdd�de0�Z1Gdd�de0�Z2d2dd�Z3d3dd�Z4dd�Z5Gdd�dej6�Z7e e7�Z8Gdd�dej6�Z9e e9�Z:Gd d!�d!e9�Z;e e;�Z<Gd"d#�d#ej6�Z=e e=�Z>Gd$d%�d%e0�Z?e e?�Z@Gd&d'�d'e0�ZAGd(d)�d)ejBeA�ZCGd*d+�d+ejBeA�ZDGd,d-�d-ejBeA�ZEGd.d/�d/ejBeA�ZFGd0d1�d1ejBeA�ZGdS)4z
Test HTTP/2 support.
�)�absolute_import�divisionN)�
providedBy�directlyProvides)�defer�reactor�task�error)�failure)�	iterbytes)�StringTransport�MemoryReactorClock)�
DummyProducer)�unittest)�http)�DummyHTTPHandler�DummyHTTPHandlerProxy�DelayedHTTPHandler�DelayedHTTPHandlerProxy�DummyPullProducerHandlerProxy�_makeRequestProxyFactory�._IDeprecatedHTTPChannelToRequestInterfaceProxy)�IPv4Address)�H2Connection)�Encoder�DecoderzHTTP/2 support not enabledc@s|eZdZdZdd�Zdd�Zdd�Zgdfd	d
�Zddd
�Zddd�Z	dd�Z
ddd�Zd dd�Zd!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.
    cCst�|_dS�N�r�encoder��self�r"�=/usr/lib/python3/dist-packages/twisted/web/test/test_http2.py�__init__7szFrameFactory.__init__cCst�|_dSrrr r"r"r#�refreshEncoder;szFrameFactory.refreshEncodercCsdS)NsPRI * HTTP/2.0

SM

r"r r"r"r#�clientConnectionPreface?sz$FrameFactory.clientConnectionPreface�c	Ks^tj�|�}|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#�buildHeadersFrameCszFrameFactory.buildHeadersFrameNcCs2|dk	rt|�nt�}tj�|�}||_||_|S)zD
        Builds a single data frame out of a chunk of data.
        N)�setr)r*�	DataFramer-r.)r!r-r.r3r4r"r"r#�buildDataFrameWs
zFrameFactory.buildDataFrameFcCs&tj�d�}|r|j�d�||_|S)z1
        Builds a single settings frame.
        rZACK)r)r*Z
SettingsFramer.r/�settings)r!r<Zackr4r"r"r#�buildSettingsFramebs
zFrameFactory.buildSettingsFramecCstj�|�}||_|S)z5
        Builds a single WindowUpdate frame.
        )r)r*�WindowUpdateFrameZwindow_increment)r!r3�	incrementr4r"r"r#�buildWindowUpdateFramensz#FrameFactory.buildWindowUpdateFramer�cCs"tj�d�}||_||_||_|S)z/
        Builds a single GOAWAY frame.
        r)r)r*�GoAwayFrame�
error_code�last_stream_idZadditional_data)r!�lastStreamID�	errorCodeZadditionalDatar4r"r"r#�buildGoAwayFramews
zFrameFactory.buildGoAwayFramecCstj�|�}||_|S)z3
        Builds a single RST_STREAM frame.
        )r)r*�RstStreamFramerC)r!r3rFr4r"r"r#�buildRstStreamFrame�sz FrameFactory.buildRstStreamFramecCs"tj�|�}||_||_||_|S)z1
        Builds a single priority frame.
        )r)r*Z
PriorityFrameZ
depends_on�
stream_weight�	exclusive)r!r3�weight�	dependsOnrKr4r"r"r#�buildPriorityFrame�s
zFrameFactory.buildPriorityFramecCs:tj�|�}||_|j�|�|_t|�|_|j�	d�|S)z5
        Builds a single Push Promise frame.
        r()
r)r*ZPushPromiseFrameZpromised_stream_idrr,r-r9r.r/)r!r3�promisedStreamIDr2r.r4r"r"r#�buildPushPromiseFrame�s
z"FrameFactory.buildPushPromiseFrame)Nr')F)rrA)r)rF)�__name__�
__module__�__qualname__�__doc__r$r%r&r8r;r=r@rGrIrNrPr"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)NrA)r�decoder�_datar r"r"r#r$�szFrameBuffer.__init__cCs|j|7_dSr)rW)r!r-r"r"r#�receiveData�szFrameBuffer.receiveDatacCs|Srr"r r"r"r#�__iter__�szFrameBuffer.__iter__cCs�t|j�dkrt��tjj�|jdd��\}}t|j�|dkrHt��|�t|jdd|���|jd|d�|_t	|tjj
�r�|jj|j
dd�|_
|S)N�	T)�raw)�lenrW�
StopIterationr)r*ZFrameZparse_frame_headerZ
parse_body�
memoryview�
isinstancer+rV�decoder-)r!r*Zlengthr"r"r#�next�s�zFrameBuffer.nextN)	rQrRrSrTr$rXrYra�__next__r"r"r"r#rU�srUr'csT�dkrt��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�r2r3c3s|]}�j|�d�VqdS)�r3N)r;)�.0�chunk��frameFactoryr3r"r#�	<genexpr>�sz%buildRequestFrames.<locals>.<genexpr>����
END_STREAM)r�appendr8�extendr.r/�r2r-rhr3�framesr"rgr#�buildRequestFrames�s��rpcCs"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}
    rAcss|]}|��VqdSr��	serialize�rer4r"r"r#risz$buildRequestBytes.<locals>.<genexpr>)rp�joinrnr"r"r#�buildRequestBytes�srucCst�}|�|�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.
    )rUrX�list)r-�bufferr"r"r#�framesFromBytess
rxc@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>
    shello world!cCsDt|j�d�d�}|�d�t|�D]}|�|j�q&|��dS)N�/rj��)�intZuri�split�setResponseCode�range�write�	chunkData�finish)r!Zchunks�_r"r"r#�process5s

zChunkedHTTPHandler.processN)rQrRrSrTr�r�r"r"r"r#ry,sryc@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.
    cOs.tjj|f|�|�|j��d|_d|_dS)NF)r�Requestr$�channel�pauseProducing�_requestReceivedrW�r!�args�kwargsr"r"r#r$Is
zConsumerDummyHandler.__init__cCs|j��dS)z&
        Start the data pipe.
        N)r��resumeProducingr r"r"r#�
acceptDataRszConsumerDummyHandler.acceptDatacOsd|_tjj|f|�|�S)NT)r�rr��requestReceivedr�r"r"r#r�Ysz$ConsumerDummyHandler.requestReceivedcCs0|�d�|j��|_d}|�|�|��dS)Nr{s0this is a response from a consumer dummy handler)r~Zcontent�readrWr�r�)r!Z
returnDatar"r"r#r�^s


zConsumerDummyHandler.processN)rQrRrSrTr$r�r�r�r"r"r"r#r�Bs
	r�c@seZdZdZdd�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��
stopProducingr r"r"r#r�ps
z'AbortingConsumerDummyHandler.acceptDataN)rQrRrSrTr�r"r"r"r#r�isr�c@seZdZdZdd�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.
    cCs|�d�|�t�d�dS)Nr{T)r~�registerProducerrr r"r"r#r��s
zDummyProducerHandler.processN)rQrRrSrTr�r"r"r"r#r�|sr�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!ZwrappedFactoryZ	interfacer"r"r#r$�sz NotifyingRequestFactory.__init__cOs$|j||�}|j�|���t|�Sr)r�r�rlZnotifyFinishr)r!r�r��reqr"r"r#�__call__�sz NotifyingRequestFactory.__call__N)rQrRrSrTr$r�r"r"r"r#r��sr�c@s eZdZdZereZdd�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�priorityZ
DeadlockErrorra)r!�
connectionr"r"r#�assertAllStreamsBlocked�sz(HTTP2TestHelpers.assertAllStreamsBlockedN)rQrRrSrT�skipH2�skipr�r"r"r"r#r��sr�c@sfeZdZdddddddgZddd	ddd
gZddd
dgZdddddgZdZdddddgZdZ	dd�Z
dd�Zdd�Zdd �Z
d!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zd7d8�Zd9d:�Zd;d<�Zd=d>�Zd?d@�ZdAdB�ZdCdD�ZdEdF�Z dGdH�Z!dIdJ�Z"dKdL�Z#dMdN�Z$dOdP�Z%dQdR�Z&dSdT�Z'dUdV�Z(dWdX�Z)dYdZ�Z*d[S)\�HTTP2ServerTests��:method�GET�s
:authoritys	localhost��:pathrz�s:schemeshttps�s
user-agentstwisted-test-code��
custom-header�1�r��2�r��POST�r��/post_endpoint��content-lengths25�hello �world, �it's �http/2!��:statuss200)�requestrz)�commandr��sversionsHTTP/2)r�s13�
'''
None
'''
�r�r��r�r�)r�s36�$'''
25
hello world, it's http/2!'''
cCsNt�}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}
        )rrr&ru�makeConnectionr�dataReceived)r!r�r2Zbodyrh�	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.
        cs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����t|dtjj
����t|dtjj
����t|dj�t�j
����|dj�j���|djd���d|djk�dS)	N�css|]}|jdkVqdS�r'N��	stream_idrsr"r"r#riszGHTTP2ServerTests.test_basicRequest.<locals>.validate.<locals>.<genexpr>r'��rArk)rx�value�assertEqualr\�
assertTrue�allr_r)r*r+r:�dictr-�getResponseHeaders�getResponseDatar.�r3ro�r!r�r"r#�validates ��z4HTTP2ServerTests.test_basicRequest.<locals>.validater')rr�requestFactoryr��getRequestHeaders�_streamCleanupCallbacks�addCallback�r!r�r�r�r"r�r#�test_basicRequests�z"HTTP2ServerTests.test_basicRequestcs@t�}t|_��|�j�j�\}���fdd�}|jd�|�S)zV
        Send a POST request and confirm that the data is safely transferred.
        cs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����t|dtjj
����t|dtjj
����t|dj�t�j
����|dj�j���|djd���d|djk�dS)	Nr�css|]}|jdkVqdSr�r�rsr"r"r#ri=szFHTTP2ServerTests.test_postRequest.<locals>.validate.<locals>.<genexpr>������rjrArk)rxr�r�r\r�r�r_r)r*r+r:r�r-�postResponseHeaders�postResponseDatar.r�r�r"r#r�8s ��z3HTTP2ServerTests.test_postRequest.<locals>.validater')rrr�r��postRequestHeaders�postRequestDatar�r�r�r"r�r#�test_postRequest.s�z!HTTP2ServerTests.test_postRequestcsddddddg�d�dd��jD�}t�}t|_��||�j�\}�����fd	d
�}|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")re�x�yr"r"r#�
<listcomp>^s�z=HTTP2ServerTests.test_postRequestNoLength.<locals>.<listcomp>cs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����t|dtjj
����t|dtjj
����t|dj�t�����|dj����|djd���d|dj
k�dS)	Nr�css|]}|jdkVqdSr�r�rsr"r"r#rinszNHTTP2ServerTests.test_postRequestNoLength.<locals>.validate.<locals>.<genexpr>r�r�rjrArk)rxr�r�r\r�r�r_r)r*r+r:r�r-r.r��r�r�r!r�r"r#r�is ��z;HTTP2ServerTests.test_postRequestNoLength.<locals>.validater')r�rrr�r�r�r�r�)r!r�r�r�r�r"r�r#�test_postRequestNoLengthOs&���z)HTTP2ServerTests.test_postRequestNoLengthcs�d}t��t��t�}t|_ttd|dd�����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�csg|]}t�j�j�|��qSr")rpr�r��rer3)r4r!r"r#r��s��z=HTTP2ServerTests.test_interleavedRequests.<locals>.<listcomp>rAcss|]}|��VqdSrrq�rer*r"r"r#ri�sz<HTTP2ServerTests.test_interleavedRequests.<locals>.<genexpr>cs�t����}��t|�d��D]|��fdd�|D�}��t|�d���t|dj�t�j����|dj�j���|djd���d	|dj	k�q dS)
N�ycs(g|] }|j�krt|tjj�s|�qSr")r�r_r)r*r>rsrdr"r#r��s
�zOHTTP2ServerTests.test_interleavedRequests.<locals>.validate.<locals>.<listcomp>r�rr'r�rArk)
rxr�r�r\r�r-r�r�r�r.)r�roZstreamFrames)�br!�	streamIDsrdr#r��s
��z;HTTP2ServerTests.test_interleavedRequests.<locals>.validate)rrrrr�rvrr&�	itertools�chain�
from_iterable�ziprtr�rr�r�DeferredListr��valuesr�)r!Z
REQUEST_COUNT�aror�r�r�r")r�r4r!r�r#�test_interleavedRequests�s*�
��z)HTTP2ServerTests.test_interleavedRequestscst��t��t�}t|_�j�d�d<��fdd�dD�}|ddj�d�d|dd_|d	dj�d�d
|d	d_�j	ddd	d
d�}|d�
d|�tj�
|�}���}|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�csg|]}t�g�|��qSr")rpr�)r4r�r"r#r��s�zAHTTP2ServerTests.test_sendAccordingToPriority.<locals>.<listcomp>)r'r��rZPRIORITY�@r'� r��T)r3rLrMrKrAcss|]}|��VqdSrrqr�r"r"r#ri�sz@HTTP2ServerTests.test_sendAccordingToPriority.<locals>.<genexpr>cs\t����}��t|�d�dd�|D�}dddddddddddddddg}��||�dS)N�cSs g|]}t|tjj�r|j�qSr")r_r)r*r:r�rsr"r"r#r��s�zSHTTP2ServerTests.test_sendAccordingToPriority.<locals>.validate.<locals>.<listcomp>r'r�r�)rxr�r�r\)r�ror�Z
expectedOrder�r�r!r"r#r��s�"z?HTTP2ServerTests.test_sendAccordingToPriority.<locals>.validate)rrr�ChunkedHTTPHandlerProxyr�r�r.r/rJrN�insertr�r�r�r&rtr�rr�rr�rvr�r�r�)r!r�roZ
priorityFramer�r�r�r")r�r4r�r!r#�test_sendAccordingToPriority�s@��
��z-HTTP2ServerTests.test_sendAccordingToPrioritycCs�t�}t�}t�}t|_|��}|t|jg|�7}||jdd|jdgd��	�7}|�
|�t|�D]}|�|�|j
rbq|qbt|���}|�t|�d�|�t|dtjj��|�|j
�dS)zR
        A protocol error from the remote peer terminates the connection.
        r'r�r()r3rOr2r.r�rjN)rrrrr�r&rur�rPrrr�rr��
disconnectingrxr�r�r\r�r_r)r*rB)r!r4r�r�r�r�ror"r"r#�&test_protocolErrorTerminatesConnections.�

�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�cs4t����}��t|�d���d|djk�dS�Nr�rkrj�rxr�r�r\r�r.r�r�r"r#r�Tsz;HTTP2ServerTests.test_streamProducingData.<locals>.validate)r�ConsumerDummyHandlerProxyr�r�r�r��streams�_request�original�assertFalser�rxr�r�r\r�r�rWr�r�)r!r�r��requestror�r"r�r#�test_streamProducingData1s$�z)HTTP2ServerTests.test_streamProducingDatac	s�t�}t��t�}t|_t�j�j|�}t�|d_	|�
�}|d�dd�|D��7}|���t
|�D]}|�|�qf|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.
        rjrAcss|]}|��VqdSrrqrsr"r"r#riqszAHTTP2ServerTests.test_abortStreamProducingData.<locals>.<genexpr>r'csJt����}��t|�d���t|dtjj����|dj	d�dS)Nr�rjr')
rxr�r�r\r�r_r)r*rHr�r�r�r"r#r��s�z@HTTP2ServerTests.test_abortStreamProducingData.<locals>.validate)rrr�!AbortingConsumerDummyHandlerProxyr�rpr�r�r9r.r&rtr�rr�rrrr	r�r�r�r�)	r!r4r�ror�r�r
�cleanupCallbackr�r"r�r#�test_abortStreamProducingData_s*�

z.HTTP2ServerTests.test_abortStreamProducingDatacs�t�}t|_��|�jg�\}�|jdjj}|�d�|�d�|j	d}|�
|jddd������
|j���
|jdk���t|jd���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)rFNsthird chunkcsJt����}��t|�d���|djd���t|dtjj	��dS�Nr�r'�
rxr�r�r\r�r�r_r)r*r+r�r�r"r#r��s�z9HTTP2ServerTests.test_terminatedRequest.<locals>.validate)r�DummyProducerHandlerProxyr�r�r�rrrr�r�r�rIrrr��
_disconnectedr�r��AttributeErrorr��r!r�rhr
r
r�r"r�r#�test_terminatedRequest�s&	�


�
z'HTTP2ServerTests.test_terminatedRequestcs�t�}t|_��|�jg�\}�|jdjj}|�d�|�d�|j	d}|�
|jdd������
|j���
|jdk���|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'rrr�rENcsJt����}��t|�d���|djd���t|dtjj	��dSrrr�r�r"r#r��s�z<HTTP2ServerTests.test_terminatedConnection.<locals>.validate)rrr�r�r�rrrr�r�r�rGrrr�rr�r	�_stillProducingr�rr"r�r#�test_terminatedConnection�s&	�


�
z*HTTP2ServerTests.test_terminatedConnectioncsHt�}t|_�jdg}��||g�\}���fdd�}|jd�|�S)zq
        Requests containing Expect: 100-continue cause provisional 100
        responses to be emitted.
        )sexpects100-continuecs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����|dj
dg���d|djk�dS)Nr�css|]}|jdkVqdSr�r�rsr"r"r#riszQHTTP2ServerTests.test_respondWith100Continue.<locals>.validate.<locals>.<genexpr>r')r�s100rkrj)rxr�r�r\r�r�r_r)r*r+r-r.r�r�r"r#r�s ��z>HTTP2ServerTests.test_respondWith100Continue.<locals>.validater')rrr�r�r�r�r�)r!r�r2r�r�r"r�r#�test_respondWith100Continuesz,HTTP2ServerTests.test_respondWith100Continuecsxt�}t|_��|�jg�\}�|jd}|jj}|jd}|�	���
|j���
|jdk���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
k�dS)Nr�r')r�s400rkrj)rxr�r�r\r�r_r)r*r+r-r.r�r�r"r#r�?s��z6HTTP2ServerTests.test_respondWith400.<locals>.validate)rrr�r�r�rrrr��!_respondToBadRequestAndDisconnectr�rr�r�)r!r�r��streamr
r
r�r"r�r#�test_respondWith400!s�


z$HTTP2ServerTests.test_respondWith400cs~t�}t|_��|�jg�\}�|jd}|jj}ddddddg��D]}|�|�qF|�	����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
k�dd�|D�}��|�d	g�dS)
NrZcss|]}|jdkVqdSr�r�rsr"r"r#rioszQHTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate.<locals>.<genexpr>r'rkrjcSs g|]}t|tjj�r|j�qSr"�r_r)r*r:r-rsr"r"r#r�vs�zRHTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate.<locals>.<listcomp>rA�rxr�r�r\r�r�r_r)r*r+r.�r3roZreceivedDataChunks��
dataChunksr!r�r"r#r�js ���z>HTTP2ServerTests.test_loseH2StreamConnection.<locals>.validate)rrr�r�r�rrrr�ZloseConnectionr�r�)r!r�r�rr
rfr�r"r(r#�test_loseH2StreamConnectionOs�
z,HTTP2ServerTests.test_loseH2StreamConnectioncCsDt�}t|_|�||jg�|jd}|jj}|�t	|j
|d�dS)zK
        The L{H2Stream} object forbids registering two producers.
        r'TN)rrr�r�r�rrrr��
ValueErrorr�)r!r�rr
r"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�)r
r"r#�<lambda>�rAz;HTTP2ServerTests.test_handlesPullProducer.<locals>.<lambda>csTt����}��d|djk�dd�|D�}��|ddddd	d
ddd
ddg�dS)NrkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r��s�zOHTTP2ServerTests.test_handlesPullProducer.<locals>.validate.<locals>.<listcomp>�0r�r��3�4�5�6�7�8�9rA�rxr�r�r.r��r3ror)r�r"r#r��s(���z;HTTP2ServerTests.test_handlesPullProducer.<locals>.validate)rrr�r�r�rrrZ_actualProducer�resultr�r�)r!r�r�rZproducerCompleter�r")r
r!r�r#�test_handlesPullProducer�s�
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�rrrr	ZisSecure�abortConnection)r!r�r
r"r"r#�test_isSecureWorksProperly�sz+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�?cs4t����}��t|�d���d|djk�dS)Nr�rkrjr�r�ror�r"r#�validateComplete�szCHTTP2ServerTests.test_lateCompletionWorks.<locals>.validateComplete)
rrr�r�r�rrrr�	callLaterr�r�r�)r!r�r�r
r>r"r�r#�test_lateCompletionWorks�s�
�z)HTTP2ServerTests.test_lateCompletionWorkscsvt�}t|_��|�jg�\}�|jd}|jj}|�d�|�	dddg�|�
�|jd}��fdd�}|�|�S)zW
        L{H2Stream} objects can send a series of frames via C{writeSequence}.
        r'r{�Hello�,�world!csFt����}��d|djk�dd�|D�}��|ddddg�dS)	NrkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r��s�zTHTTP2ServerTests.test_writeSequenceForChannels.<locals>.validate.<locals>.<listcomp>rArBrCrAr6r7r�r"r#r��s���z@HTTP2ServerTests.test_writeSequenceForChannels.<locals>.validate)
rrr�r�r�rrrr~Z
writeSequencer�r�r�)r!r�r�rr
ZcompletionDeferredr�r"r�r#�test_writeSequenceForChannels�s�


z.HTTP2ServerTests.test_writeSequenceForChannelscs�t��t��t��t�_���}|��tjj	j
di���7}|t�j
g��7}����t|�D]}��|�q`�jd}|jj���d�ddddg���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'�fiverr!r"r#r$cs �D]}��|�q���dSr)r�r�)rf)r)r
r"r#�write_chunks+sz7HTTP2ServerTests.test_delayWrites.<locals>.write_chunksr<cs���jddd����S)Nr'�2�r3r?)r�r@rr�r��r�r4r"r#r-3s�z3HTTP2ServerTests.test_delayWrites.<locals>.<lambda>cs�t����}��t|�d���tdd�|dd�D�����t|dtjj	����d|dj
k�dd�|D�}��|d	g�d
g�dS)NrZcss|]}|jdkVqdSr�r�rsr"r"r#ri?szFHTTP2ServerTests.test_delayWrites.<locals>.validate.<locals>.<genexpr>r�rkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r�Fs�zGHTTP2ServerTests.test_delayWrites.<locals>.validate.<locals>.<listcomp>rErAr&r')r�r)r!r"r#r�:s ���z3HTTP2ServerTests.test_delayWrites.<locals>.validate)rrrrr�r&r=�h2r<�SettingCodes�INITIAL_WINDOW_SIZErrrur�r�rr�rrrr�r�
deferLaterrr�r�)r!r�r�rrF�dr�r")r�r�r)r4r
r!r#�test_delayWrites
s6�
�


�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.
        �r2r-rhr'rdN)rrrrr�r&rur�rIrrr�r�rxr�r�r\ZassertNotInr�)r!rhr�r�r�ror"r"r#�test_resetAfterBodyRs$��

z$HTTP2ServerTests.test_resetAfterBodycs�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.
        cseZdZ�fdd�ZdS)zTHTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.SuperRequestcs,tj|f|�|���|jj|jjf�dSr)rr$�callbackr��site�factoryr��rOr"r#r$tsz]HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.SuperRequest.__init__N)rQrRrSr$r"rVr"r#�SuperRequestssrWcs(|\}}��|�j���|�j�dSr)�assertIsrTrU)r�rTrU)r�r!r"r#�validateFactoryAndSite�sz^HTTP2ServerTests.test_RequestRequiringFactorySiteInConstructor.<locals>.validateFactoryAndSiter')rZDeferredrrrZHTTPFactoryrr�rU�objectrTr�r�r�r��
gatherResults)r!rWZhttpFactoryrYr
r")r�rOr!r#�-test_RequestRequiringFactorySiteInConstructorls


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)ZassertIsNone)r8r r"r#r��sz?HTTP2ServerTests.test_notifyOnCompleteRequest.<locals>.validater)
rr�rr�r�r�r�r�r\r�r�rr[)r!r�r�r��	deferredsr�rOr
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'cs��d�dS�Nz#Didn't errback, called back instead�Zfail�Zignr r"r#rS�sz;HTTP2ServerTests.test_notifyOnResetStream.<locals>.callbackcs"��|tj���|jtj�dSr��assertIsInstancer
�FailurerX�typer	ZConnectionLost��reasonr r"r#�errback�sz:HTTP2ServerTests.test_notifyOnResetStream.<locals>.errbackrrd)
rr�rr�r�r�r�r�r\�addCallbacksrIrrr�)	r!r�rhr�r]rSrhrO�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�rgr�cs��d�dSr_r`rar r"r#rS�sz=HTTP2ServerTests.test_failWithProtocolError.<locals>.callbackcs$��|tj���|jtjj�dSr)rcr
rdr�rK�
exceptionsZ
ProtocolErrorrfr r"r#rh�sz<HTTP2ServerTests.test_failWithProtocolError.<locals>.errback�yo���r-r3)rr�rr�r�r�rur�r�r�r\rir;rrrr[�
r!r�rhr��
secondRequestr]rSrhrOrjr"r r#�test_failWithProtocolError�s4	
��
�

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�rgr�cs��d�dSr_r`rar r"r#rSsz4HTTP2ServerTests.test_failOnGoaway.<locals>.callbackcs"��|tj���|jtj�dSrrbrfr r"r#rhsz3HTTP2ServerTests.test_failOnGoaway.<locals>.errbackr)rr�rr�r�r�rur�r�r�r\rirGrrrr[rpr"r r#�test_failOnGoaway�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�rgr�cs��d�dSr_r`rar r"r#rS=sz;HTTP2ServerTests.test_failOnStopProducing.<locals>.callbackcs"��|tj���|jtj�dSrrbrfr r"r#rh@sz:HTTP2ServerTests.test_failOnStopProducing.<locals>.errback)rr�rr�r�r�rur�r�r�r\rir�rr[)	r!r�rhr�rqr]rSrhrOr"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'cs��d�dSr_r`rar r"r#rS_sz7HTTP2ServerTests.test_notifyOnFast400.<locals>.callbackcs"��|tj���|jtj�dSrrbrfr r"r#rhbsz6HTTP2ServerTests.test_notifyOnFast400.<locals>.errbackr)rr�rr�r�r�r�r�r\rirr)	r!r�rhr�r]rSrhrOrr"r r#�test_notifyOnFast400Ns 
�
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'rdrN)r
rr?rr�rrr&r�r�rur�r��_maxBufferedControlFrameBytesrr��disconnected)r!�
memoryReactorr�r3rhr�r&r"r"r#�test_fast400WithCircuitBreakervs,

��
z/HTTP2ServerTests.test_fast400WithCircuitBreakercCs�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�dr'�eN)rrr�rrr&r�r�r�rr=rrrxr�r�r\r�)r!r�rhr�r&r�ror"r"r#� test_bufferingAutomaticFrameData�s

z1HTTP2ServerTests.test_bufferingAutomaticFrameDatacCs�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.
        rzr�N)rrr�rrr&r�r�r�rvr	rrr=rrr�rw)r!r�rhr�r&r�r"r"r#�2test_bufferingAutomaticFrameDataWithCircuitBreaker�s

zCHTTP2ServerTests.test_bufferingAutomaticFrameDataWithCircuitBreakercCs
Gdd�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@seZdZdd�ZdS)zeHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWrite.<locals>.AutoPausingStringTransportc_s tj|f|�|�|j��dSr)rr��producerr�r�r"r"r#r��szkHTTP2ServerTests.test_bufferingContinuesIfProducerIsPausedOnWrite.<locals>.AutoPausingStringTransport.writeN)rQrRrSr�r"r"r"r#�AutoPausingStringTransport�sr�Tr'rr}�cr��ZN)rrrr�rr�r&r�r�ZassertIsNotNoneZ_consumerBlockedrxr�r�r\Z_bufferedControlFrameBytesrr=rrr�)r!r�r�rhr�r&ror�r"r"r#�0test_bufferingContinuesIfProducerIsPausedOnWrite�s,

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.
        rrmrnroN)r
rr?rrr&r�r�r�rvr;rrr�rw)r!rxr�rhr�r&rjr"r"r#�+test_circuitBreakerAbortsAfterProtocolErrors 

�

z<HTTP2ServerTests.test_circuitBreakerAbortsAfterProtocolErrorN)+rQrRrSr�r�r�r�r�r�r�r�r�r�r�r�rrrrrrrrr*r,r9r;r@rDrPrRr\r^rkrrrsrtruryr|r~r�r�r"r"r"r#r��s~��
�	�	#!1>I*.377!.3)
(H""/+*(* #-r�c@s�eZdZdZdddddgZdZddd	ddd
gZddd
dgZdZdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�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�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'rHcsFt����}��d|djk�d�dd�|D��}���j|�dS)NrkrjrAcss"|]}t|tjj�r|jVqdSrr%rsr"r"r#ri{s�zMH2FlowControlTests.test_bufferExcessData.<locals>.validate.<locals>.<genexpr>)rxr�r�r.rtr�r��r3roZactualResponseDatar�r"r#r�ts
�z:H2FlowControlTests.test_bufferExcessData.<locals>.validate)rrrrr�r&r=rKr<rLrMrrrur�r�rr�r\r�rr@r�r�)	r!r4r�r�r�ZbonusFramesr�r*r�r"r�r#�test_bufferExcessDataSs,�
�

z(H2FlowControlTests.test_bufferExcessDatacs�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���|jjdddg�|�|jddd������|j���|jjddddg�|��|����fd	d
�}|jd�|�S)z�
        L{Request} objects that have registered producers get blocked and
        unblocked according to HTTP/2 flow control.
        r�r'�
helloworld�pauserHr�resumerGcsDt����}��d|djk�dd�|D�}��|dddg�dS)NrkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r��s�zXH2FlowControlTests.test_producerBlockingUnblocking.<locals>.validate.<locals>.<listcomp>r�rAr6r7r�r"r#r��s��zDH2FlowControlTests.test_producerBlockingUnblocking.<locals>.validate) rrrrr�r&r=rKr<rLrMrrrur�r�rr�rrrr��_producerProducingr�r	r�r�eventsr@�unregisterProducerr�r�r��r!r4r�r�r�rr
r�r"r�r#�test_producerBlockingUnblocking�sh�
�


���
�
�z2H2FlowControlTests.test_producerBlockingUnblockingcst��t��t��t�_���}|��tjj	j
di���7}|t�j
g��7}����t|�D]}��|�q`�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.
        r�r'r�r��hcsL���jddd�������j����jjddg������	�dS)Nr'rGrHr�r�)
r�r@rrr�r�r�rr�r�r�r")r�r4r
r!rr"r#�window_opens��z=H2FlowControlTests.test_flowControlExact.<locals>.window_openrcsFt����}��d|djk�dd�|D�}��|ddddg�dS)	NrkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r�&s�zNH2FlowControlTests.test_flowControlExact.<locals>.validate.<locals>.<listcomp>rr r�rAr6r7r�r"r#r�s�z:H2FlowControlTests.test_flowControlExact.<locals>.validate)"rrrrr�r&r=rKr<rLrMrrrur�r�rr�rrrr�r�r�r	r�rr�rrNrr�r�rr�)r!r�r�r�ZwindowDeferr�Z
validateDeferr")r�r�r4r
r!rr#�test_flowControlExact�s8�
�




z(H2FlowControlTests.test_flowControlExactcs�t�}t��t�}t|_|��}||�tjj	j
di���7}|t�j
g|�7}|���t|�D]}|�|�q`|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.
        r�r'r�rrGrHcsDt����}��d|djk�dd�|D�}��|dddg�dS)NrkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r�es�zQH2FlowControlTests.test_endingBlockedStream.<locals>.validate.<locals>.<listcomp>rr rAr6r7r�r"r#r�^s��z=H2FlowControlTests.test_endingBlockedStream.<locals>.validate)rrrrr�r&r=rKr<rLrMrrrur�r�rr�rrrr�r�r�r�r��finishedrr?r@r�r�r�r"r�r#�test_endingBlockedStream0s<�
�


�z+H2FlowControlTests.test_endingBlockedStreamc	s�t�}t��t�}t|_|��}|t�jg|�7}|���t	|�D]}|�
|�qD|jd}|jj
}|jd}|��|����|j���fdd�}|�|�S)z<
        We safely handle responses without bodies.
        r'csPt����}��t|�d���d|djk�dd�|D�}��|dg�dS)Nr�rkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r��s�zQH2FlowControlTests.test_responseWithoutBody.<locals>.validate.<locals>.<listcomp>rArr7r�r"r#r��s��z=H2FlowControlTests.test_responseWithoutBody.<locals>.validate)rrrrr�r&rur�r�rr�rrrr�r�r�r�r�r��	r!r4r�r�r�rr
r
r�r"r�r#�test_responseWithoutBodyqs*�


z+H2FlowControlTests.test_responseWithoutBodyc	s�t�}t��t�}t|_|��}|t�jg|�7}|���t	|�D]}|�
|�qD|jd}|jj
}|jd}|��|����|j�|�
|jddd������fdd�}|�|�S)zk
        WindowUpdate frames received after we've completed the stream are
        safely handled.
        r'rGrHcsPt����}��t|�d���d|djk�dd�|D�}��|dg�dS)Nr�rkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r��s�z[H2FlowControlTests.test_windowUpdateForCompleteStream.<locals>.validate.<locals>.<listcomp>rArr7r�r"r#r��s��zGH2FlowControlTests.test_windowUpdateForCompleteStream.<locals>.validate)rrrrr�r&rur�r�rr�rrrr�r�r�r�r�r@rrr�r�r"r�r#�"test_windowUpdateForCompleteStream�s0�


�z5H2FlowControlTests.test_windowUpdateForCompleteStreamcst�}t��t�}t|_|��}||�tjj	j
di���7}|t�j
g|�7}|���t|�D]}|�|�q`|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.
        r�r'�wordrHcsBt����}��d|djk�dd�|D�}��|ddg�dS)NrkrjcSs g|]}t|tjj�r|j�qSr"r%rsr"r"r#r�#	s�zOH2FlowControlTests.test_producerUnblocked.<locals>.validate.<locals>.<listcomp>r�rAr6r7r�r"r#r�	s��z;H2FlowControlTests.test_producerUnblocked.<locals>.validate)rrrrr�r&r=rKr<rLrMrrrur�r�rr�rrrr�r�r�r�rr�r@r�r�r�r�r�r"r�r#�test_producerUnblocked�s>�
�


�z)H2FlowControlTests.test_producerUnblockedcs�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'rr�rHrAcss|]}|��VqdSrrqrsr"r"r#ri?	szBH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.<genexpr>csFt����}��d|djk�d�dd�|D��}���j|�dS)NrkrjrAcss"|]}t|tjj�r|jVqdSrr%rsr"r"r#riM	s�zTH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.validate.<locals>.<genexpr>)rxr�r�r.rtr�r�r�r�r"r#r�F	s
�zAH2FlowControlTests.test_unnecessaryWindowUpdate.<locals>.validate)rrrrr�rpr�r�r�r@r&rtr�rr�r�r�)r!r4r�ror�r�r�r"r�r#�test_unnecessaryWindowUpdate/	s"�

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'rcr�rHrAcss|]}|��VqdSrrqrsr"r"r#rij	szKH2FlowControlTests.test_unnecessaryWindowUpdateForStream.<locals>.<genexpr>N)rrrrr�rlr8r�r@r&rtr�r�r�)r!r4r��connror-r"r"r#�%test_unnecessaryWindowUpdateForStreamV	s�

z8H2FlowControlTests.test_unnecessaryWindowUpdateForStreamcCs�t�}t�}t�}t|_t|j|j|�}|��}|d�	dd�|D��7}|�
|�t|�D]}|�|�qZ|j
d��|jddd�}|�|���t|���}|�t|dtjj��dS)	zq
        When a WindowUpdate frame is received for a stream that has been
        aborted it is ignored.
        rAcss|]}|��VqdSrrqrsr"r"r#ri�	szEH2FlowControlTests.test_windowUpdateAfterTerminate.<locals>.<genexpr>r'r�rHrjN)rrrrr�rpr�r�r&rtr�rr�rr:r@rrrxr�r�r_r)r*rH)r!r4r�r�ror�r��windowUpdateFramer"r"r#�test_windowUpdateAfterTerminater	s*�
�z2H2FlowControlTests.test_windowUpdateAfterTerminatecs�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.
        rAcss|]}|��VqdSrrqrsr"r"r#ri�	szDH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.<genexpr>cs �jddd�}��|���dS)Nr'r�rH)r@r�rr)r�r�rJr"r#�
update_window�	s
�zHH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.update_windowcs"t����}��d|dj�dS)Nrkrj)rxr�ZassertInr.r=r�r"r#r��	szCH2FlowControlTests.test_windowUpdateAfterComplete.<locals>.validater')rrrrr�rpr�r�r&rtr�rr�r�r�)r!ror�r�r�r�rOr")r�r�r4r!r#�test_windowUpdateAfterComplete�	s$�
z1H2FlowControlTests.test_windowUpdateAfterCompletecCst�}t�}t�}t|_ddgd}d�tdd�|D���}|jdd�d	|fg}t|||d
�}|d=|�	|j
dtjj
jd��|��}|d
�dd�|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�z{}css|]}t|�VqdSr)r\)rer-r"r"r#ri�	sz;H2FlowControlTests.test_dataAndRstStream.<locals>.<genexpr>Nrjzcontent-lengthrQr')r3rFrAcss|]}|��VqdSrrqrsr"r"r#ri�	scSs g|]}t|tjj�r|j�qSr")r_r)r*r>r�rsr"r"r#r��	s�z<H2FlowControlTests.test_dataAndRstStream.<locals>.<listcomp>rcSsg|]}t|tjj�r|�qSr")r_r)r*r+rsr"r"r#r��	scSsg|]}t|tjj�r|�qSr")r_r)r*r:rsr"r"r#r��	s)rrrrr��format�sumr�rprlrIrK�errors�
ErrorCodesZINTERNAL_ERRORr&rtr�r�rxr�r�r	)r!rhr�r�Z	frameDataZ
bodyLengthr2ror�ZwindowUpdateFrameIDsZ
headersFramesZ
dataFramesr"r"r#�test_dataAndRstStream�	sJ����

���
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'rcrN)r
rr?rr�rrr&r�r�r8r�rrr�rv�clearZabortRequestr	r�r�rw)r!rxr�rhr�r&r3ZheadersFrameDatar"r"r#�#test_abortRequestWithCircuitBreaker�	s*

�


z6H2FlowControlTests.test_abortRequestWithCircuitBreakerN)rQrRrSrTr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r"r"r"r#r�3s<�	�
1cIA8AE'%'=r�c@s>eZdZdddddddgZdd	�Zd
d�Zdd
�Zdd�ZdS)�HTTP2TransportCheckingr�r�r�r�r�r�r�cCs2t�}t�}t|_|�|d�|�|j|k�dS)zU
        L{H2Connection} can be registered with the transport as a producer.
        TN)rrrr�r�r�r)r!r�r�r"r"r#�"test_registerProducerWithTransport4
s
z9HTTP2TransportChecking.test_registerProducerWithTransportcs�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.
        rAcss|]}|��VqdSrrqrsr"r"r#riM
szNHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.<genexpr>Tr'cs`t����}��t|�d���t|dtjj����	���	���	���	���	��S)Nr�rj)
rxr�r�r\r	r_r)r*r:r�r=�r�r�r
r!r"r#�validateNotSentd
s�zTHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.validateNotSentcs4t����}��t|�d���d|djk�dSrrr=r�r"r#r>t
szUHTTP2TransportChecking.test_pausingProducerPreventsDataSend.<locals>.validateCompleter<)rrrrr�rpr�r&rtr�r�rr�r�r�rrNrr�)r!r4ror�r�r�r>rOr"r�r#�$test_pausingProducerPreventsDataSend@
s$



z;HTTP2TransportChecking.test_pausingProducerPreventsDataSendcCs�t�}t�}t�}t|_t|jg|�}|��}|d�dd�|D��7}|�	|�|�
|d�t|�D]}|�|�qd|�
�t|���}|�t|�d�|�t|dtjj��|�|j�dS)zo
        L{H2Connection} can be stopped by its producer. That causes it to lose
        its transport.
        rAcss|]}|��VqdSrrqrsr"r"r#ri�
sz<HTTP2TransportChecking.test_stopProducing.<locals>.<genexpr>Tr�rjN)rrrrr�rpr�r&rtr�r�rr�r�rxr�r�r\r	r_r)r*r:r)r!r4r�r�ror�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]}|�|�qv|j
d
���������������|jd
}����fdd�}|�|�S)
z{
        A L{H2Stream} object correctly passes through host and peer information
        from its L{H2Connection}.
        ZTCPz
17.52.24.8i�z17.188.0.12i})�hostAddress�peerAddressrAcss|]}|��VqdSrrqr�r"r"r#ri�
szEHTTP2TransportChecking.test_passthroughHostAndPeer.<locals>.<genexpr>r'cs$��������������dSr)r��getHost�getPeerrI�r�r�r!rr"r#r��
szDHTTP2TransportChecking.test_passthroughHostAndPeer.<locals>.validate)rrrrrr�r�rpr�r&rtrr�rr�r�r�r�r�)	r!rhr�r�ror�r�r
r�r"r�r#�test_passthroughHostAndPeer�
s*�


z2HTTP2TransportChecking.test_passthroughHostAndPeerN)rQrRrSr�r�r�r�r�r"r"r"r#r�(
s�A"r�c@seZdZdZdd�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.
    cCs|t��}t|�}|��}|�t|�d�|d}|�|���|�|jd�|�|j	|j
�|�|jd�|�|ji�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�r\r�Zactive�time�funcZ_sendPrioritisedDatar��kw)r!rr�Zcalls�callr"r"r#�"test_initiallySchedulesOneDataCall�
sz7HTTP2SchedulingTests.test_initiallySchedulesOneDataCallN)rQrRrSrTr�r"r"r"r#r��
sr�c@s�eZdZdZdddddddgZe�Zd	d
�Zdd�Zd
d�Z	efdd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%S)&�HTTP2TimeoutTestsz@
    The L{H2Connection} object times out idle connections.
    r�r�r�r�r�r�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)r?)r!r�rr"r"r#�patch_TimeoutMixin_clock�
s
z*HTTP2TimeoutTests.patch_TimeoutMixin_clockcCs^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.
        rz)rr�r�timeOutr�rrr�r�rr�)r!�initialDatar�rr�r�r�r"r"r#�initiateH2Connections


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.
        rjN)
rxr�r\r�r_r)r*rBrCrD)r!r-�
frameCountrFrEror"r"r#�assertTimedOut%s�z HTTP2TimeoutTests.assertTimedOutcCs�||jkrtj}t�}|��}|j|td�\}}}||_|�d�|j|�	�dt
jj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�rzr�r�r�rFrE)�_DEFAULTr�abortTimeoutrr&r�r�advancer�r�rKr�r��NO_ERRORr�rr	rw)r!r�rhr�rr�r�r"r"r#�prepareAbortTest5s&
�
�z"HTTP2TimeoutTests.prepareAbortTestcCs�t�}|��}|j|td�\}}}|��}|�d�|�||���|�|j�|�d�|j	|��dt
jj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�rr�N)rr&r�rr�r�r�r	rr�rKr�r�r�r�)r!rhr�rr�r�Zpreambler"r"r#�test_timeoutAfterInactivity[s$�

�z-HTTP2TimeoutTests.test_timeoutAfterInactivitycCs�t�}d}|j|td�\}}}t|���D]$}|�|�|�d�|�|j�q*|�d�|j	|�
�dtjj
jdd�|�|j�dS)zM
        When a L{H2Connection} receives data, the timeout is reset.
        rAr�r�r�rr�N)rr�rrr&r�r�r	rr�r�rKr�r�r�r�)r!rhr�rr�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.
        rAcss|]}|��VqdSrrqrsr"r"r#ri�szDHTTP2TimeoutTests.test_timeoutResetByResponseData.<locals>.<genexpr>cst||d�}��|�|S)N)�queued)rrl)rr�r��Zrequestsr"r#�saveRequest�s
zFHTTP2TimeoutTests.test_timeoutResetByResponseData.<locals>.saveRequestr�r�r'�
rs
some bytesr��
r�N)rrpr�r&rtr�r�r�ZassertEqualsr\rr�r	rr�r�rKr�r��PROTOCOL_ERROR)	r!rhr�ror�rr�r�r�r"r�r#�test_timeoutResetByResponseData�s2�


�z1HTTP2TimeoutTests.test_timeoutResetByResponseDatacCs~t�}t|jg|�}|��}|d�dd�|D��7}|j|td�\}}}|�d�|j|�	�dt
jj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}.
        rAcss|]}|��VqdSrrqrsr"r"r#ri�szOHTTP2TimeoutTests.test_timeoutWithProtocolErrorIfStreamsOpen.<locals>.<genexpr>r�r{r�r'r�N)rrpr�r&rtr�r�r�r�r�rKr�r�r�r�r)r!rhror�rr�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.
        rAcss|]}|��VqdSrrqrsr"r"r#ri�szCHTTP2TimeoutTests.test_noTimeoutIfConnectionLost.<locals>.<genexpr>r�rgr'r{N)
rrpr�r&rtr�r�r�r\r��connectionLostr�r�)
r!rhror�rr�r�ZsentDataZoldCallCountZcurrentCallCountr"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�rr	rw�r!rr�r�r"r"r#�,test_timeoutEventuallyForcesConnectionCloseds

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.
        r�Nr')r�r�r�r�rr	rwr�r"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�lr�r�r"r"r#�'test_losingConnectionWithNoAbortTimeOut"s
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�rrwr�r	ZConnectionDoner�r"r"r#�"test_connectionLostAfterForceClose0s

z4HTTP2TimeoutTests.test_connectionLostAfterForceClosecCs�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
rr?r�rrr&r�r�rr�rIrrr�r�rw)r!rxr�rhr�r&r�r"r"r#�,test_timeOutClientThatSendsOnlyInvalidFramesBs

z>HTTP2TimeoutTests.test_timeOutClientThatSendsOnlyInvalidFramesN)rQrRrSrTr�rZr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r"r"r"r#r��
s0�&"",r�)Nr')Nr')HrTZ
__future__rrr�Zzope.interfacerrZtwisted.internetrrrr	Ztwisted.pythonr
Ztwisted.python.compatrZtwisted.test.proto_helpersrr
Ztwisted.test.test_internetrZ
twisted.trialrZtwisted.webrZtwisted.web.test.test_httprrrrrrrZtwisted.internet.addressrr�Ztwisted.web._http2rrKZ	h2.errorsZ
h2.exceptionsr)r�Zhpack.hpackrr�ImportErrorrZrrUrprurxr�ryr�r�rr�rr�rr�ZNotifyingRequestFactoryProxyr�ZTestCaser�r�r�r�r�r"r"r"r#�<module>s�$
z0
$
$��z#
@ Telegram