JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/acme/_internal/tests/__pycache__/client_test.cpython-39.pyc
a
}|�g� � @ s� d Z ddlZddlZddlmZ ddlZddlZddlm Z ddl
Z
ddl
m Z ddl Z
ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ e� d
�Z e� d�Z e� d�Z e
j �!e� d��Z"e�#ddddej#�$� d��Z%G dd� de
j&�Z'G dd� de
j(�Z)G dd� de
j&�Z*G d d � d e
j&�Z+e,d k�r�e�-e�.ej/d d� e0g �� dS ) zTests for acme.client.� N)�Dict)�mock)�
challenges)�errors)�jws)�messages)�
messages_test)� test_util)�
ClientNetwork)�ClientV2z cert-san.pemz
csr-mixed.pemzcsr-nosans.pemzrsa512_key.pemz1https://www.letsencrypt-demo.org/acme/new-account�/https://www.letsencrypt-demo.org/acme/new-noncez/https://www.letsencrypt-demo.org/acme/new-orderz1https://www.letsencrypt-demo.org/acme/revoke-cert)�
newAccount�newNonceZnewOrder�
revokeCert�metac @ sT e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
�d�dd� �Ze
�d�dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!e
�d�d;d<� �Z"e
�d�d=d>� �Z#e
�d�d?d@� �Z$e
�d�dAdB� �Z%dCdD� Z&dES )F� ClientV2Testz Tests for acme.client.ClientV2.c C s� t jdtji i d�| _t �� | _| j| jj_| j| jj_t j
t j dd�| _ d| _
t j| j
t�� d�}t|�}t jf i |��| _t j|dd�| _d }t j|d
t jtjt�d �d �d
�}t j ||d�| _ t j t j
t j dd�|fd�| _ t j | j |d�| _!d| _"t#| _$t%| j$| j�| _&| jj'dd�| _d| _(| j j't j
t j dd�t j)d�| _*t j | j*| j(d�| _+t j,| j j | j*j ft j)| j!j-| j(fdd�| _.t j/| j.d| j!| j+gt0d�| _1t j/| j.d| j!| j+gt2d�| _3d S )NT)�ok� status_code�headers�linksz example.com)�typ�value)z mailto:cert-admin@example.comztel:+12025551212)�contact�key�+https://www.letsencrypt-demo.org/acme/reg/1��body�uriz-https://www.letsencrypt-demo.org/acme/authz/1z/1z+evaGxfADs6pSRb2LAv9IZf17Dt3juxGJ-PCt92wr-oA)�token)r �statusZchall)r �
authzr_uri)�
identifierr � )Zterms_of_service_agreedz-https://www.letsencrypt-demo.org/acme/authz/2zwww.example.com)r! r z=https://www.letsencrypt-demo.org/acme/acct/1/order/1/finalize)Z identifiersr �authorizations�finalizez4https://www.letsencrypt-demo.org/acme/acct/1/order/1)r r r# Zcsr_pem)4r � MagicMock� http_client�OK�response�net�post� return_value�getr Z
IdentifierZIDENTIFIER_FQDNr! r Z Registration�KEYZ
public_key�dictZNewRegistration�new_regZRegistrationResource�regrZ
ChallengeBody� STATUS_VALIDr ZDNS�jose� b64decodeZChallengeResource�challrZ
Authorization�authz�AuthorizationResource�authzr�rsn� DIRECTORY_V2� directoryr �client�update� authzr_uri2ZSTATUS_PENDING�authz2�authzr2ZOrderr �orderZ
OrderResource�
CSR_MIXED_PEM�orderr�CSR_NO_SANS_PEM�orderr2)�selfZregZthe_argr �challb� rG �D/usr/lib/python3.9/site-packages/acme/_internal/tests/client_test.py�setUp* s�
�
�
��
������
��� � � �zClientV2Test.setUpc C sH t j| j_| jj�� | jj_| jj | jj
d< | j| j � | j
�ksDJ �d S �N�Location)r& �CREATEDr( r r0 r �to_json�jsonr+ r r r; � new_accountr/ �rE rG rG rH �test_new_accountj s
z ClientV2Test.test_new_accountc C s^ t j| j_| jj�� | jj_| jj | jj
d< | jj � dddii� | j
�| j�jdksZJ �d S )NrK zterms-of-service�urlz$https://www.letsencrypt-demo.org/tos)r& rL r( r r0 r rM rN r+ r r r r< r; rO r/ Zterms_of_servicerP rG rG rH �test_new_account_tos_linkq s
��z&ClientV2Test.test_new_account_tos_linkc C sX t j| j_| jj| jjd< t�t j
�� | j � | j
� W d � n1 sJ0 Y d S rJ )r& r' r( r r0 r r �pytest�raisesr �
ConflictErrorr; rO r/ rP rG rG rH �test_new_account_conflict} s
z&ClientV2Test.test_new_account_conflictc C s^ | j j| j jjdd�d�}|j�� | jj_tj| j_ | j j
| jj d< | j �
| j �|ksZJ �d S )NZ deactivated�r �r rK )r0 r< r rM r( rN r+ r& r' r r r r; Zdeactivate_registration)rE Zdeactivated_regrrG rG rH �test_deactivate_account� s �
z$ClientV2Test.test_deactivate_accountc C s� | j j| j jjtjd�d�}|j�� | jj_| j �
| j �}|j|jksJJ �| j j j j
dks^J �| j j| j j jd d v s|J �d S )NrX rY r" r )r7 r< r r ZSTATUS_DEACTIVATEDrM r( rN r+ r; Zdeactivate_authorizationr) r* �
call_countr �call_args_list)rE Zdeactivated_authzr7 rG rG rH � test_deactivate_authorization� s �z*ClientV2Test.test_deactivate_authorizationc C s t �| j�}tj|_| j�� |j_ | j
j |j d< || j
j_ t �| j�}| j�� |j_ | jj |j d< | j}| j�� |j_ | jj |j d< t�d��0}||f|_| j�t�| j
ks�J �W d � n1 s�0 Y t�d��0}||f|_| j�t�| jks�J �W d � n1 �s0 Y d S )NrK z!acme.client.ClientV2._post_as_get)�copy�deepcopyr( r& rL r r@ rM rN r+ rB r r r) r* r5 r7 r> r? r �patch� side_effectr; Z new_orderrA rC rD )rE Zorder_responseZauthz_responseZauthz_response2Zmock_post_as_getrG rG rH �test_new_order� s"
4
zClientV2Test.test_new_orderc C s� d| j ji| jjd< | j j�� | jj_tj d d�}| j
� | j j|� t �
tj��* | j
� | j jjdd�|� W d � n1 s�0 Y d S )NrR Zup�Z
validation�foo)r )r4 r r( r r rM rN r+ r � DNSResponser; �answer_challengerT rU r �UnexpectedUpdater< )rE Zchall_responserG rG rH �test_answer_challege� s z!ClientV2Test.test_answer_challegec C sJ t �tj��* | j�| jjtj d d�� W d � n1 s<0 Y d S )Nrc )
rT rU r � ClientErrorr; rf r4 r r re rP rG rG rH �"test_answer_challenge_missing_next� s z/ClientV2Test.test_answer_challenge_missing_nextzacme.client.datetimec C s� t � ddd�|j j_t j|_|j �� t jdd� }tj| jd�| j_tj| jd�| j_ | j�
| j�| jksnJ �| jj� | j|� | jj � | j|� d S )N�� � � �Z �Zseconds�r+ ) �datetime�nowr+ � timedeltar ZMockrB r; �poll_authorizations�finalize_orderZpoll_and_finalize�assert_called_once_with)rE �
mock_datetimeZexpected_deadlinerG rG rH �test_poll_and_finalize� s z#ClientV2Test.test_poll_and_finalizec C s� t � ddd�t � ddd�t � ddd�g}||j j_| j�� | j�� | j�� g| jj_t� t
j ��$ | j �
| j|d � W d � n1 s�0 Y d S )Nrk rl rm � � r" )rq rr ra r5 rM r> r( rN rT rU r � TimeoutErrorr; rt rB )rE rw Znow_side_effectrG rG rH � test_poll_authorizations_timeout� s �
�
z-ClientV2Test.test_poll_authorizations_timeoutc C s� t � ddd�}| jjjtjtj�d�d�}| jjtj|fd�}|� � | j
j _ t
�tj�� | j�| j|� W d � n1 s~0 Y d S )N�' � � unauthorized)r �error)r r )rq r4 r r< r �STATUS_INVALID�Error� with_coder5 rM r( rN r+ rT rU r ZValidationErrorr; rt rB )rE �deadlinerF r5 rG rG rH � test_poll_authorizations_failure� s
�z-ClientV2Test.test_poll_authorizations_failurec C s~ t � ddd�}| jjtjd�}tj|| jd�}| jj| j|gd�}| j �
� | j�
� |�
� f| j j _
| j�| j|�|kszJ �d S )Nr} r~ rX r )r# )rq r> r< r r1 r6 r= rB r7 r5 rM r( rN ra r; rt )rE r� Zupdated_authz2Zupdated_authzr2�updated_orderrrG rG rH � test_poll_authorizations_success� s ��
z-ClientV2Test.test_poll_authorizations_successc C sd | j j| jjdd�d�}|�� | jj_t�t j
�� | j � | j
� W d � n1 sV0 Y d S )Nrd �r )r! )r5 r< r! rM r( rN r+ rT rU r rg r; �pollr7 )rE Z
updated_authzrG rG rH �test_poll_unexpected_update� s z(ClientV2Test.test_poll_unexpected_updatec C sb | j jdtjd�}| jj|td�}|�� | jj_ t| j_
t � ddd�}| j �
| j|�|ks^J �d S )N�+https://www.letsencrypt-demo.org/acme/cert/�Z certificater )r �
fullchain_pemr} r~ )r@ r< r r1 rB � CERT_SAN_PEMrM r( rN r+ �textrq r; ru )rE �
updated_orderr� r� rG rG rH �test_finalize_order_success� s �z(ClientV2Test.test_finalize_order_successc C sv | j jtj�d�tjd�}|�� | jj_ t
�
ddd�}t � t
j�� | j�| j|� W d � n1 sh0 Y d S )Nr �r� r r} r~ )r@ r< r r� r� r� rM r( rN r+ rq rT rU r Z
IssuanceErrorr; ru rB )rE r� r� rG rG rH �test_finalize_order_error� s
�z&ClientV2Test.test_finalize_order_errorc C sn | j jd tjd�}|�� | jj_tj t
j dd��* | j �
| jt�ddd�� W d � n1 s`0 Y d S )Nr� z The certificate order failed)�matchr} r~ )r@ r< r r� rM r( rN r+ rT rU r r� r; ru rB rq )rE r@ rG rG rH �"test_finalize_order_invalid_status s z/ClientV2Test.test_finalize_order_invalid_statusc C sV t j �� t jdd� }t�tj�� | j�| j |� W d � n1 sH0 Y d S )N�< ro )
rq rr rs rT rU r r{ r; ru rB )rE r� rG rG rH �test_finalize_order_timeout s z(ClientV2Test.test_finalize_order_timeoutc C s� | j jdtjd�}| jj|tttgd�}|�� | jj_ t| j_
d| jj d< t � ddd�}| j
j| j|dd �}| jjjd
tjtjd � | jjjd tjtjd � ||ks�J �| jj d= | j
j| j|dd �}||jg d
�ks�J �d S )Nr� r� )r r� �alternative_fullchains_pemz�<https://example.com/acme/cert/1>;rel="alternate", <https://example.com/dir>;rel="index", <https://example.com/acme/cert/2>;title="foo";rel="alternate"ZLinkr} r~ T)Zfetch_alternative_chainsz https://example.com/acme/cert/1��
new_nonce_urlz https://example.com/acme/cert/2)r� )r@ r< r r1 rB r� rM r( rN r+ r� r rq r; ru r) r* Zassert_any_callr �ANY)rE r� r� r� ZresprG rG rH � test_finalize_order_alt_chains
s2 ���
�
�
z+ClientV2Test.test_finalize_order_alt_chainsc C s6 | j �tj| j� | jjj| jd t j
t d d� d S )Nr r r� ) r; �revoker �CERTr8 r) r* rv r: r r� r9 rP rG rG rH � test_revoke( s �zClientV2Test.test_revokec C sL t j| j_t�tj��" | j� t
j | j � W d � n1 s>0 Y d S �N)
r& �METHOD_NOT_ALLOWEDr( r rT rU r ri r; r� r r� r8 rP rG rG rH �#test_revoke_bad_status_raises_error- s
�z0ClientV2Test.test_revoke_bad_status_raises_errorc C s� | j j| jjd< | j j�� | jj_| j | j� | j �ks:J �| jj
j d usLJ �| jj
j j
dks`J �tj| j
j jd d v s|J �| j jjdd��� | jj_d S )NrK rl r rG )r )r0 r r( r r rM rN r+ r; Zupdate_registrationr) �accountr* r[ r9 r
r\ r< rP rG rG rH �test_update_registration3 s �z%ClientV2Test.test_update_registrationc C s. t �dt jjdd�i�| j_| j�� s*J �d S )Nr T��external_account_required�r � Directory�Metar; r: r� rP rG rG rH �#test_external_account_required_true? s �
z0ClientV2Test.test_external_account_required_truec C s. t �dt jjdd�i�| j_| j�� r*J �d S )Nr Fr� r� rP rG rG rH �$test_external_account_required_falseF s �
z1ClientV2Test.test_external_account_required_falsec C s | j �� rJ �d S r� )r; r� rP rG rG rH �&test_external_account_required_defaultM s z3ClientV2Test.test_external_account_required_defaultc C s: | j j�� | jj_d| jjd< | j | j�| j �ks6J �d S )Nr rK ) r0 r rM r( rN r+ r r; Zquery_registrationrP rG rG rH � test_query_registration_clientP s z+ClientV2Test.test_query_registration_clientc C sl t �d��N}| j|_| j�| j� | jjjj| jj d dd� | jjj
� � W d � n1 s^0 Y d S )Nz*acme.client.ClientV2._authzr_from_responser r� ) r r` r? r+ r; r� r) r* rv r r, Zassert_not_called)rE Z mock_clientrG rG rH �test_post_as_getU s
�z ClientV2Test.test_post_as_getc C s: d| j jd< t�dddddd�| jj| j dd �ks6J �d S )
Nz Fri, 31 Dec 1999 23:59:59 GMT� Retry-Afteri� � � � �; �
�r( �default)r( r rq r; � retry_afterrP rG rG rH �test_retry_after_date` s �z"ClientV2Test.test_retry_after_datec C sV t � ddd�|j j_t j|_d| jjd< t � dddddd�| jj| jdd�ksRJ �d S ) N�� � � Zfoooor� r r� r� �rq rr r+ rs r( r r; r� �rE Zdt_mockrG rG rH �test_retry_after_invalide s �z%ClientV2Test.test_retry_after_invalidc C s` t � ddd�|j j_t j|_t j |j _d| jjd< t � dddddd�| jj| jdd�ks\J �d S ) Nr� r� r� z Tue, 116 Feb 2016 11:50:00 MSTr� r r� r� ) rq rr r+ rs ra r( r r; r� r� rG rG rH �test_retry_after_overflown s
�z&ClientV2Test.test_retry_after_overflowc C sV t � ddd�|j j_t j|_d| jjd< t � dddddd�| jj| jdd �ksRJ �d S )
Nr� r� r� Z50r� r �2 r� r� r� r� rG rG rH �test_retry_after_secondsx s �z%ClientV2Test.test_retry_after_secondsc C sJ t � ddd�|j j_t j|_t � dddddd�| jj| jdd�ksFJ �d S )Nr� r� r� r r� r� )rq rr r+ rs r; r� r( r� rG rG rH �test_retry_after_missing� s
�z%ClientV2Test.test_retry_after_missingc C s0 t �� | jj_t �� t�d| j��� ks,J �d S )Nzhttps://example.com/dir) r9 rM r( rN r+ �to_partial_jsonr Z
get_directoryr) rP rG rG rH �test_get_directory� s �z ClientV2Test.test_get_directoryN)'�__name__�
__module__� __qualname__�__doc__rI rQ rS rW rZ r] rb rh rj r r` rx r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rG rG rG rH r ' sP @
r c @ s( e Zd Zdd� Zdd� Zedd� �ZdS )�MockJSONDeSerializablec C s
|| _ d S r� r� )rE r rG rG rH �__init__� s z MockJSONDeSerializable.__init__c C s
d| j iS )Nrd r� rP rG rG rH r� � s z&MockJSONDeSerializable.to_partial_jsonc C s d S r� rG )�clsZjobjrG rG rH � from_json� s z MockJSONDeSerializable.from_jsonN)r� r� r� r� r� � classmethodr� rG rG rG rH r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
�d�dd� �Ze
�d�dd� �Zdd� Zdd � Zd d � Ze
�d�d d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd3d*d+�Zd,d-� Ze
�d.�d/d0� �Zd1d2� Z d)S )4�ClientNetworkTestz$Tests for acme.client.ClientNetwork.c C sX t �� | _t jt jjd�| _tttj | jdd�| _
t jdt j d�| _
i | j
_i | j
_d S )Nrp �acme-python-test)r �alg�
verify_sslZ
user_agentT�r r )r r% r� �sentinel�wrappedZ wrap_in_jwsr
r- r2 ZRS256r) r& r' r( r r rP rG rG rH rI � s
�zClientNetworkTest.setUpc C s | j j| ju sJ �d S r� )r) r� rP rG rG rH � test_init� s zClientNetworkTest.test_initc C sT | j jtd�ddd�}tj�|�}t�|j� � �ddiks>J �|j
j j dksPJ �d S )Nrd � TgrR ��noncerR )
r) � _wrap_in_jwsr� �acme_jws�JWS�
json_loadsrN �loads�payload�decode� signature�combinedr� �rE Zjws_dumpr rG rG rH �test_wrap_in_jws� s
� z"ClientNetworkTest.test_wrap_in_jwsc C s� ddi| j _| j jtd�ddd�}tj�|�}t�|j �
� �ddiksJJ �|j j j
dks\J �|j j jdksnJ �|j j jdks�J �d S )Nr zacct-urird r� rR r� )r) r� r� r� r� r� r� rN r� r� r� r� r� r� ZkidrR r� rG rG rH �test_wrap_in_jws_v2� s
� z%ClientNetworkTest.test_wrap_in_jws_v2c C s� d| j _i | j j_t�d��R}tj|_t �
t j �� | j
�| j � W d � n1 sV0 Y W d � n1 st0 Y d S )NFz$acme.client.messages.Error.from_json)r( r rN r+ r r` r2 ZDeserializationErrorra rT rU r ri r) �_check_response)rE r� rG rG rH �(test_check_response_not_ok_jobj_no_error� s
z:ClientNetworkTest.test_check_response_not_ok_jobj_no_errorc C sb d| j _tjjdddd��� | j j_t� tj�� | j
� | j � W d � n1 sT0 Y d S )NFZserverInternalrd z
some title)Zdetail�title) r( r r r� r� rM rN r+ rT rU r) r� rP rG rG rH �%test_check_response_not_ok_jobj_error� s �z7ClientNetworkTest.test_check_response_not_ok_jobj_errorc C sP d| j _t| j j_t�tj�� | j �
| j � W d � n1 sB0 Y d S )NF) r( r �
ValueErrorrN ra rT rU r ri r) r� rP rG rG rH �"test_check_response_not_ok_no_jobj� s
z4ClientNetworkTest.test_check_response_not_ok_no_jobjc C sn t | jj_| jjdfD ]R}|| jjd< t�t j
��& | jj | j| jjd� W d � q1 s^0 Y qd S )Nrd � Content-Type�� content_type) r� r( rN ra r) �JSON_CONTENT_TYPEr rT rU r ri r� �rE Z response_ctrG rG rH �*test_check_response_ok_no_jobj_ct_required� s
�z<ClientNetworkTest.test_check_response_ok_no_jobj_ct_requiredc C sD t | jj_| jjdfD ](}|| jjd< | j| j�| j�ksJ �qd S �Nrd r� )r� r( rN ra r) r� r r� r� rG rG rH �$test_check_response_ok_no_jobj_no_ct� s
z6ClientNetworkTest.test_check_response_ok_no_jobj_no_ctzacme.client.loggerc C sf i | j j_d| j jd< | j | jj| j dd�ks2J �z|j�dd� W n tyX Y d S 0 td��d S )Nz application/json; charset=utf-8r� �application/jsonr� �<Ignoring wrong Content-Type (%r) for JSON decodable responsez5Expected Content-Type warning to not have been logged) r( rN r+ r r) r� �debug�assert_called_with�AssertionError�rE Z mock_loggerrG rG rH �&test_check_response_ok_ct_with_charset� s
� � z8ClientNetworkTest.test_check_response_ok_ct_with_charsetc C sD i | j j_d| j jd< | j | jj| j dd�ks2J �|j�dd� d S )Nz
text/plainr� r� r� r� )r( rN r+ r r) r� r� r� r� rG rG rH � test_check_response_ok_bad_ct� s
� �z/ClientNetworkTest.test_check_response_ok_bad_ctc C sN d| j _d| j _t�tj�� | j�| j � W d � n1 s@0 Y d S )NFi� ) r( r r rT rU r rV r) r� rP rG rG rH � test_check_response_conflict s z.ClientNetworkTest.test_check_response_conflictc C sD i | j j_| jjdfD ](}|| j jd< | j | j�| j �ksJ �qd S r� )r( rN r+ r) r� r r� r� rG rG rH �test_check_response_jobj
s
z*ClientNetworkTest.test_check_response_jobjc C s` t �� | j_| j| jjj_| j| jjddddd�ks8J �| jjjjdddt j t j t j dd� d S )N�HEAD�http://example.com/rd �baz��bar)r �verify�timeoutr �
r r% r) �sessionr( �requestr+ �
_send_requestrv r� rP rG rG rH �test_send_request s
�
�z#ClientNetworkTest.test_send_requestc C sZ t �� | j_t jdtjdd�| jjj_| jjdddt j ddd id
� |j
� d d d
d� d S )NTs hi)r r Zcontentr� r� rd r� ZAcceptzapplication/pkix-cert)r r r z!Received response:
HTTP %d
%s
%s�� � s aGk=) r r% r) r r& r' r r+ r r� r� r� r� rG rG rH �test_send_request_get_der s � ��z+ClientNetworkTest.test_send_request_get_derc
C sd t �� | j_| j| jjj_| j| jjdddddd�ks:J �| jjjjdddt j t j t j ddd� d S )NZPOSTr� rd Zquxr� )�datar )r r r r r r rP rG rG rH �test_send_request_post+ s
�
�z(ClientNetworkTest.test_send_request_postc C sf dD ]\}t �� | j_| j| jjj_|| j_| j| j�dd�ksBJ �| jjjj dd|t j
t j
d� qd S )N)TF�GETr� �r r r ) r r% r) r r( r r+ r� r rv r� )rE r rG rG rH � test_send_request_verify_ssl5 s �
�z.ClientNetworkTest.test_send_request_verify_sslc C s� t �� | j_| jjddddid� | jjjjddt jt jddd�d� | jjddd d
id� | jjjjddt jt jd d
id� d S ) Nr r� r r� �r r� )�
User-Agentr r r Zfoo2) r r% r) r r r rv r� r� rP rG rG rH � test_send_request_user_agentB s
�
�
�
�z.ClientNetworkTest.test_send_request_user_agentc C sH t �� | j_| jjddddid� | jjjjt jt jt jt jdd� d S )Nr r� r r� r �- )r r r )r r% r) r r r rv r� rP rG rG rH �test_send_request_timeoutR s
�
�z+ClientNetworkTest.test_send_request_timeoutNc C s2 t �� }|d ur||j_|| j_| `|j�� d S r� )r r% �closera r) r rv )rE Zclose_exceptionZsessrG rG rH �test_del[ s zClientNetworkTest.test_delc C s | � t� d S r� )r �ReferenceErrorrP rG rG rH �test_del_errore s z ClientNetworkTest.test_del_errorzacme.client.requestsc C sT t j|_t jj|j_t�t jj�� | j�dd� W d � n1 sF0 Y d S )Nr r ) �requests�
exceptions�RequestExceptionr ra rT rU r) r )rE Z
mock_requestsrG rG rH � test_requests_error_passthroughh s z1ClientNetworkTest.test_requests_error_passthroughc
C s� z| j �dd� W nv tyF } z dt|�ks2J �W Y d }~nLd }~0 tjjy� } z(dt|�v stdt|�v stJ �W Y d }~n
d }~0 0 d S )Nr z&http://localhost:19123/nonexistent.txtz4Requesting localhost/nonexistent: Connection refusedz'Connection aborted.'z[WinError 10061])r) r r� �strr r �ConnectionError)rE �y�zrG rG rH �test_urllib_errorp s � z#ClientNetworkTest.test_urllib_error)N) r� r� r� r� rI r� r� r� r� r� r� r� r� r r` r� r� r� r� r r r
r r r r r r r" rG rG rG rH r� � s8
r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd S )!�#ClientNetworkWithMockedResponseTestz<Tests for acme.client.ClientNetwork which mock out response.c s� t d d d�� _tjdtjd�� _i � j_i � j_d� j_ tjdtj
d�� _ i � j _t�� � _ t�� � _
tjj� _t�d�t�d�t�d�g� _� jd d � � _� fdd �}tj|d
� � j_� _� j� j_tj� j
d �� j_d S ) N)r r� Tr� Fs Nonces Nonce2s Nonce3c sd d|vs J �| d }| d }|dkr4|dkr4� j }n� j}� jrZ� jj� j�� �� i|_ni |_|S )Nr� r r" r� �
new_nonce_uri)�acmev1_nonce_responser( �available_noncesr) ZREPLAY_NONCE_HEADER�popr� r )�args�kwargs�methodr r( rP rG rH � send_request� s �z?ClientNetworkWithMockedResponseTest.setUp.<locals>.send_request)ra rp )r
r) r r% r& r' r( r r �checkedr� r% �objZ wrapped_objr� r� r2 � b64encode�
all_noncesr& r r+ �check_responser� r� )rE r+ rG rP rH rI � s, �
� �
z)ClientNetworkWithMockedResponseTest.setUpc C s6 | j |ksJ �| j|ks J �| j js(J �d| j _| j S )NT)r( r� r r, )rE r( r� rG rG rH r0 � s
z2ClientNetworkWithMockedResponseTest.check_responsec C s4 | j | jjdddd�ks J �| jjddddd� d S )Nr� rd r� r r� )r% r) �headr+ rv rP rG rG rH � test_head� s
� �z-ClientNetworkWithMockedResponseTest.test_headc C s4 | j | jjdddd�ks J �| jjddddd� d S )Nr$ rd r� r r� )r( r) r1 r+ rv rP rG rG rH � test_head_v2� s
� �z0ClientNetworkWithMockedResponseTest.test_head_v2c C s@ | j | jjd| jdd�ks J �| j js*J �| jjdddd� d S )Nr� r� )r� r r r )r( r) r, r� r, r+ rv rP rG rG rH �test_get� s
� �z,ClientNetworkWithMockedResponseTest.test_getc C s4 | j j| _| j| j �d| j�ks$J �| jjs0J �d S )Nr )r) ZJOSE_CONTENT_TYPEr� r( r* r- r, rP rG rG rH �test_post_no_content_type� s
z=ClientNetworkWithMockedResponseTest.test_post_no_content_typec C s� | j | jjd| j| jd�ks J �| j js,J �| jj�| jt� | j
� � �d� g | _ t
�tj��& | jjd| j| jd� W d � n1 s�0 Y | jj�| jt� | j
� � �d� d S )Nr r� )r( r) r* r- r� r, r� rv r2 r3 r/ r' r& rT rU r Z MissingNoncer� rP rG rG rH � test_post� s
� �4�z-ClientNetworkWithMockedResponseTest.test_postc C sV dt �d�g| _t�tj��& | jjd| j | j
d� W d � n1 sH0 Y d S )N� f� goodr r� � r2 r. r& rT rU r ZBadNoncer) r* r- r� rP rG rG rH � test_post_wrong_initial_nonce� s
�zAClientNetworkWithMockedResponseTest.test_post_wrong_initial_noncec C sV t �d�dg| _t�tj��& | jjd| j | j
d� W d � n1 sH0 Y d S )Nr8 r7 r r� r9 rP rG rG rH �#test_post_wrong_post_response_nonce� s
�zGClientNetworkWithMockedResponseTest.test_post_wrong_post_response_noncec C sd t �� }tj�d�|_|| j_t� tj��& | jj
d| j | j d� W d � n1 sV0 Y d S �NZbadNoncer r� )
r r% r r� r� ra r) r� rT rU r* r- r� �rE r0 rG rG rH �test_post_failed_retry� s �z:ClientNetworkWithMockedResponseTest.test_post_failed_retryc C sj t �� }tj�d�| jg|_|| j_t �
tj��& | jj d| j | j
d� W d � n1 s\0 Y d S )NZ malformedr r� )r r% r r� r� r( ra r) r� rT rU r* r- r� r= rG rG rH �test_post_not_retried� s
��z9ClientNetworkWithMockedResponseTest.test_post_not_retriedc C s@ t �� }tj�d�| jg|_| j| jjd| j | j
d�ks<J �d S r< ) r r% r r� r� r( ra r) r* r- r� )rE Z post_oncerG rG rH �test_post_successful_retry� s
�
�z>ClientNetworkWithMockedResponseTest.test_post_successful_retryc C s� t jj| j_| jj| jjfD ]<}t� t jj�� |dd� W d � q 1 sN0 Y q t� t jj��" | jj
d| j d� W d � n1 s�0 Y d S )Nr r )r- ) r r r r+ ra r) r1 r, rT rU r* r- )rE r* rG rG rH �$test_head_get_post_error_passthrough s *zHClientNetworkWithMockedResponseTest.test_head_get_post_error_passthroughc C s� t jdtjd�}t �� | j_|| jj_d | _t �� }|| j_t �
t j ��( | jj
d| j| jdd� W d � n1 sv0 Y |jdks�J �d S )NFr� r r$ �r� r� r" )r r% r& ZSERVICE_UNAVAILABLEr) r r+ r� r� rT rU r ri r* r- r[ )rE Z bad_responser0 rG rG rH �test_post_bad_nonce_head s
�$z<ClientNetworkWithMockedResponseTest.test_post_bad_nonce_headc C s d | _ | jjd| jd dd� d S )Nr r$ rB )r� r) r* r- rP rG rG rH �test_new_nonce_uri_removed s z>ClientNetworkWithMockedResponseTest.test_new_nonce_uri_removedN)r� r� r� r� rI r0 r2 r3 r4 r5 r6 r: r; r> r? r@ rA rC rD rG rG rG rH r# � s +
r# �__main__r" )1r� r^ rq Z http.clientr; r& rN �sys�typingr Zunittestr Zjosepyr2 rT r Zacmer r r r� r Zacme._internal.testsr r Z acme.clientr
r Z load_vectorr� rA rC ZJWKRSA�loadr- r� r� r9 ZTestCaser ZJSONDeSerializabler� r� r# r� �exit�main�argv�__file__rG rG rG rH �<module> sP
� j
e #
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg