ruạṛ
a �u:j�L � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl m Z d dl mZmZmZ d dlZd dlZd dlmZ d dlmZmZmZmZ d dlm Z ej dkr�d dl m Z n d dl m Z e j g e j!ej"j# f Z$e j%e j&e'e'f ej(j)f Z*e j%e+e j&e'e'f f Z,e j%e+e j&e'e'f e j-f Z.e j%e/e j0e/e/f f Z1e j&e'e j-f Z2e j%e j0e'e'f ej3j4e j ej5gej5f f Z6e j eege j7d f Z8e j ege8f Z9e j eeege j7d f Z:G d d � d ej;j<j=j>�Z?G d d � d �Z@G d d� deA�ZBeCe'd�dd�ZDe j%e9e:f eEd�dd�ZFG dd� d�ZGG dd� de �ZHG dd� dejIjJ�ZKG dd � d �ZLG d d � d ejM�ZNdS ) � N)�Future)�unquote�urljoin�urlsplit)�StapledObjectStream)�Message�Receive�Scope�Send)�WebSocketDisconnect)� � )� TypedDictc @ s e Zd Zeeed�dd�ZdS )� _HeaderDict)�key�default�returnc C s | � |�S �N)� getheaders)�selfr r � r �G/opt/bart/bart_venv/lib/python3.9/site-packages/starlette/testclient.py�get_all6 s z_HeaderDict.get_allN)�__name__� __module__� __qualname__�strr r r r r r 5 s r c @ s> e Zd ZdZejejeef dd�dd�Ze d�dd�Z dS ) �_MockOriginalResponsezi We have to jump through some hoops to present the response as if it was made using urllib3. N)�headersr c C s t |�| _d| _d S )NF)r �msg�closed)r r r r r �__init__@ s z _MockOriginalResponse.__init__�r c C s | j S r )r �r r r r �isclosedD s z _MockOriginalResponse.isclosed) r r r �__doc__�typing�List�Tuple�bytesr! �boolr$ r r r r r : s r c @ s e Zd Zddd�dd�ZdS )�_Upgrade�WebSocketTestSessionN)�sessionr c C s || _ d S r )r- )r r- r r r r! I s z_Upgrade.__init__)r r r r! r r r r r+ H s r+ )� status_coder c C s( zt �| �jW S ty" Y dS 0 d S )N� )�http� HTTPStatus�phrase� ValueError)r. r r r �_get_reason_phraseM s r4 ��appr c C s> t �| �rt| d�S t �| �r(t�| �S t| dd �}t�|�S )N� __await__�__call__)�inspect�isclass�hasattr� isfunction�asyncio�iscoroutinefunction�getattr)r6 �callr r r � _is_asgi3T s rA c @ s4 e Zd ZdZedd�dd�Zeeedd�dd�Z dS ) � _WrapASGI2z7 Provide an ASGI3 interface onto an ASGI2 app. Nr5 c C s || _ d S r �r6 )r r6 r r r r! b s z_WrapASGI2.__init__)�scope�receive�sendr c � s | � |�}|||�I d H d S r rC )r rD rE rF �instancer r r r8 e s z_WrapASGI2.__call__) r r r r% �ASGI2Appr! r r r r8 r r r r rB ] s rB c @ s* e Zd ZU eed< ejeejf ed< dS )� _AsyncBackend�backend�backend_optionsN)r r r r �__annotations__r& �Dict�Anyr r r r rI j s rI c @ s@ e Zd Zd eeeedd�dd�Zej e j e j ej d�dd �Z dS ) � _ASGIAdapterTr/ N)r6 �portal_factory�raise_server_exceptions� root_pathr c C s || _ || _|| _|| _d S r )r6 rQ rR rP )r r6 rP rQ rR r r r r! p s z_ASGIAdapter.__init__)�request�args�kwargsr c s� dd� t �j�D �\}}}}}ddddd�| } d|v rR|�dd�\} } t| �} n|} | } d�jv rjg } n0| | kr�d | �� fg} nd | � d| � ��� fg} | d d � �j�� D �7 } |d v �r4�j�d d �}|d u r�g }ndd � |�d�D �}dt|�|�� | j ||�� | ddg| | g|d� }t | j || j �}t |��dd�jt|�|�� | j ||�� | ddg| | gdi id� }d�d�dt�� i�d �d � td����fdd � }td d �� �����fd d � }zL| � � �.}|�tj��|�| j |||� W d � n1 �s�0 Y W n2 t�y< } z| j�r(|�W Y d }~n d }~0 0 | j�rV��sxJ d ��n"��sxd!d"d#g dtg �t�� d$��tjjjf i ���}| � �|�}�d u�r��|_ � |_ |S )%Nc s s | ]}t |�V qd S r )r )�.0�itemr r r � <genexpr> s z$_ASGIAdapter.send.<locals>.<genexpr>�P i� )r0 �ws�https�wss�:� �hosts hostc S s$ g | ] \}}|� � �� |�� f�qS r )�lower�encode�rV r �valuer r r � <listcomp>� s �z%_ASGIAdapter.send.<locals>.<listcomp>> rZ r\ �sec-websocket-protocolc S s g | ] }|� � �qS r )�strip)rV rc r r r rd � � �,Z websocket� testclientiP� ) �type�path�raw_pathrR �scheme� query_stringr �client�server� subprotocolsr0 z1.1�http.response.template) rj � http_version�methodrk rl rR rm rn r ro rp � extensionsF�bodyr" c � s� �r"�� � s��� I d H ddiS � j} t| t�r>| �d�}nn| d u rLd}n`t| tj�r�z,| �d �}t|t�rx|�d�}d|dd�W S t y� d�ddd� Y S 0 n| }d�d|d�S ) Nrj zhttp.disconnect�utf-8rg z http.requestT)rj rv � more_body)rj rv ) �is_set�waitrv � isinstancer ra �types� GeneratorTyperF � StopIteration)rv Z body_bytes�chunk)rS �request_complete�response_completer r rE � s* z"_ASGIAdapter.send.<locals>.receive��messager c � s | d dkrt�rJ d��d�d< | d �d< t | d ��d< dd � | �d g �D ��d < d �d < t�d ��d < d�n�| d dkr�s�J d����� r�J d��| �dd�}| �dd �}�jdkr̈d �|� |s�d �d� ��� n | d dk�r| d �| d � d S )Nrj zhttp.response.startz1Received multiple "http.response.start" messages.� �version�status�reasonc S s g | ]\}}|� � |� � f�qS r )�decoderb r r r rd � s �z3_ASGIAdapter.send.<locals>.send.<locals>.<listcomp>r F�preload_content�original_responseTzhttp.response.bodyz<Received "http.response.body" without "http.response.start".z7Received "http.response.body" after response completed.rv rg rx �HEADr rr �template�context)r4 �getr ry rt �write�seek�set)r� rv rx )r� � raw_kwargsrS r� �response_startedr� r r rF � sF �� � � ���� z _ASGIAdapter.send.<locals>.sendz(TestClient did not receive any response.r� i� zInternal Server Error)r� r� r� r r� r� rv ) r �url�split�intr ra �itemsr� r rR r, r6 rP r+ rt �io�BytesIOr r@ �anyio�Event� BaseExceptionrQ r �requests�packages�urllib3� HTTPResponse�build_responser� r� )r rS rT rU rm �netlocrk �query�fragment� default_portr_ Z port_string�portr � subprotocolrq rD r- rE rF �portal�exc�raw�responser )r� r� rS r� r� r� r� r rF | s� � � � � % 6� z_ASGIAdapter.send)Tr/ )r r r �ASGI3App�_PortalFactoryTyper* r r! r� �PreparedRequestr& rN �ResponserF r r r r rO o s �� �rO c @ s e Zd Zeeedd�dd�Zd d�dd�Zej dd�d d �Z dd�d d �Z e d�d d�Z e dd�dd�Ze dd�dd�Ze dd�dd�Zedd�dd�Zedd�dd�Zd,ej edd �d d �Zd-edd �d!d"�Ze d�d#d$�Zed�d%d&�Zed�d'd(�Zd.eej d)�d*d+�ZdS )/r, N)r6 rD rP r c C s6 || _ || _d | _d | _|| _t�� | _t�� | _d S r ) r6 rD �accepted_subprotocol� extra_headersrP �queue�Queue�_receive_queue� _send_queue)r r6 rD rP r r r r! & s z WebSocketTestSession.__init__r" c C s� t �� | _| j�| �� �| _z2| j�| j�}| �ddi� | � � }| � |� W n t yl | j� � � Y n0 |� dd �| _|� dd �| _| S )Nrj zwebsocket.connectr� r )� contextlib� ExitStack� exit_stack� enter_contextrP r� �start_task_soon�_runrF rE �_raise_on_close� Exception�closer� r� r� )r �_r� r r r � __enter__4 s z WebSocketTestSession.__enter__�rT r c G sN z| � d� W | j� � n | j� � 0 | j�� sJ| j�� }t|t�r&|�q&d S )N�� )r� r� r� �emptyr� r{ r� )r rT r� r r r �__exit__D s z WebSocketTestSession.__exit__c � sb | j }| j}| j}z| �|||�I dH W n2 ty\ } z| j�|� � W Y d}~n d}~0 0 dS )zE The sub-thread in which the websocket session runs. N)rD � _asgi_receive� _asgi_sendr6 r� r� �put)r rD rE rF r� r r r r� N s zWebSocketTestSession._runc � s& | j �� r t�d�I d H q | j �� S )Nr )r� r� r� �sleepr� r# r r r r� [ s z"WebSocketTestSession._asgi_receiver� c � s | j �|� d S r )r� r� �r r� r r r r� ` s z WebSocketTestSession._asgi_sendc C s* |d dkr&t |�dd�|�dd���d S )Nrj zwebsocket.close�coder� r� r/ )r r� r� r r r r� c s �z$WebSocketTestSession._raise_on_closec C s | j �|� d S r )r� r� r� r r r rF i s zWebSocketTestSession.send)�datar c C s | � d|d�� d S )N�websocket.receive�rj �text�rF �r r� r r r � send_textl s z WebSocketTestSession.send_textc C s | � d|d�� d S )Nr� �rj r) r� r� r r r � send_byteso s z WebSocketTestSession.send_bytesr� )r� �moder c C sJ |dv s J �t �|�}|dkr0| �d|d�� n| �d|�d�d�� d S )N�r� �binaryr� r� r� rw r� )�json�dumpsrF ra )r r� r� r� r r r � send_jsonr s z WebSocketTestSession.send_jsonr� )r� r c C s | � d|d�� d S )Nzwebsocket.disconnect)rj r� r� )r r� r r r r� z s zWebSocketTestSession.closec C s | j �� }t|t�r|�|S r )r� r� r{ r� r� r r r rE } s z WebSocketTestSession.receivec C s | � � }| �|� |d S )Nr� �rE r� r� r r r � receive_text� s z!WebSocketTestSession.receive_textc C s | � � }| �|� |d S )Nr) r� r� r r r � receive_bytes� s z"WebSocketTestSession.receive_bytes)r� r c C sH |dv s J �| � � }| �|� |dkr0|d }n|d �d�}t�|�S )Nr� r� r) rw )rE r� r� r� �loads)r r� r� r� r r r � receive_json� s z!WebSocketTestSession.receive_json)r� )r� )r� ) r r r r� r r� r! r� r& rN r� r� r r� r� r� rF r r� r) r� r� r� r� rE r� r� r� r r r r r, % s( � r, c sn e Zd ZU dZded< dZejej j ed< d!ej e e f eeeeejejeejf dd �� fd d � Zejejej j ddf d �dd��Zd"eeeeejeef eee e eejeef ejeej eef ej eej eef f eje j d�� fdd� Z!d#eej"e ejejd�� fdd� Z#d d �dd�Z$ejdd�dd�Z%dd �dd �Z&dd �d d �Z'dd �d d �Z(� Z)S )$� TestClientFz Future[None]�taskNr� �http://testserverTr/ r= )r6 �base_urlrQ rR rJ rK r c s� t � �� t||pi d�| _t|�r6t�t|�}|}nt�t|�}t |�}t || j ||d�}| � d|� | � d|� | � d|� | � d|� | j �ddi� || _|| _d S ) N)rJ rK )rP rQ rR zhttp://zhttps://zws://zwss://z user-agentri )�superr! rI � async_backendrA r& �castr� rH rB rO �_portal_factory�mountr �updater6 r� ) r r6 r� rQ rR rJ rK Zasgi_app�adapter�� __class__r r r! � s, � � zTestClient.__init__r" c c sP | j d ur| j V n8tjf i | j���}|V W d � n1 sB0 Y d S r )r� r� �start_blocking_portalr� )r r� r r r r� � s zTestClient._portal_factory)rt r� �paramsr� r �cookies�files�auth�timeout�allow_redirects�proxies�hooks�stream�verify�certr� r c s8 t | j|�}t� j||||||||| | | | | |||d�S )N)r� r� r r� r� r� r� r� r� r� r� r� r� r� )r r� r� rS )r rt r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r r rS � s& �zTestClient.request)r� rq rU r c s� t d|�}|�di �}|�dd� |�dd� |�dd� |d urT|�d d �|�� ||d< zt� jd |fi |�� W n* ty� } z|j}W Y d }~nd }~0 0 td ��|S ) Nzws://testserverr � connection�upgradezsec-websocket-keyz testserver==zsec-websocket-versionZ13re z, �GETzExpected WebSocket upgrade) r r� � setdefault�joinr� rS r+ r- � RuntimeError)r r� rq rU r r� r- r� r r �websocket_connect� s z TestClient.websocket_connectc s� t �� ��}|�tjf i �j��� �_� |jd d��fdd� �}tt� t j �� �_ tt� t j �� �_ � ��j��_� ��j� |jd d�� �fdd� �}|�� �_W d � n1 s�0 Y �S )Nr" c s d � _ d S r )r� r r# r r � reset_portal s z*TestClient.__enter__.<locals>.reset_portalc s � � �j� d S r )r@ � wait_shutdownr �r� r r r r s z+TestClient.__enter__.<locals>.wait_shutdown)r� r� r� r� r� r� r� �callbackr Zcreate_memory_object_stream�math�inf� stream_send�stream_receiver� �lifespanr� r@ � wait_startup�pop_allr� )r �stackr� r r r r r� s$ � � � (zTestClient.__enter__r� c G s | j �� d S r )r� r� )r rT r r r r� s zTestClient.__exit__c � sT ddi}z2| � || jj| jj�I d H W | j�d �I d H n| j�d �I d H 0 d S )Nrj r )r6 r rE r rF )r rD r r r r ! s zTestClient.lifespanc � sb � j �ddi�I d H tjd�� fdd� }|� I d H }|d dv sFJ �|d dkr^|� I d H d S )Nrj zlifespan.startupr" c � s& � j �� I d H } | d u r"� j�� | S r �r rE r� �result�r� r# r r rE + s z(TestClient.wait_startup.<locals>.receive)zlifespan.startup.complete�lifespan.startup.failedr )r rF r& rN �r rE r� r r# r r ( s zTestClient.wait_startupc � s� t jd�� fdd� }� j4 I d H �` � j�ddi�I d H |� I d H }|d dv sVJ �|d dkrn|� I d H W d �I d H q�1 I d H s�0 Y d S )Nr" c � s& � j �� I d H } | d u r"� j�� | S r r r r# r r rE : s z)TestClient.wait_shutdown.<locals>.receiverj zlifespan.shutdown)zlifespan.shutdown.complete�lifespan.shutdown.failedr )r& rN r r rF r r r# r r 9 s zTestClient.wait_shutdown)r� Tr/ r= N)NNNNNNNNNNNNNN)N)*r r r Z__test__rL r� r& �Optionalr� �abc�BlockingPortal�UnionrH r� r r* rM rN r! r� �contextmanager� Generatorr� �Params�DataType�MutableMapping�Cookies�FileType�AuthType�TimeOutr( r� r� rS �Sequencer� r� r� r r r � __classcell__r r r� r r� � s~ � �!� � �( ��r� )Or= r� r0 r9 r� r� r r� �sysr| r& �concurrent.futuresr � urllib.parser r r Z anyio.abcr� r� Zanyio.streams.stapledr Zstarlette.typesr r r r Zstarlette.websocketsr � version_infor Ztyping_extensions�Callable�ContextManagerr r r� r r r r� �RequestsCookieJarr r) r �IOr �floatr( r r r� �AuthBaser� r � AwaitableZ ASGIInstancerH r� r� r� � _collections�HTTPHeaderDictr r r� r+ r� r4 r* rA rB rI �adapters� HTTPAdapterrO r, �Sessionr� r r r r �<module> sd �� �� 7s
cải xoăn