/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/__pycache__
a }|�g� � @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlZddlm Z ddl Z ddl Z ddlZddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ G d d� dej�Zdd� Z G dd� dej�Z e dk�r e� e� ej!dd� e"g �� dS )z/Tests for certbot._internal.plugins.standalone.� N)�Dict)�Set)�Tuple)�mock)� challenges)� standalone)� achallenges)�errors)� acme_util)�utilc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�ServerManagerTestz=Tests for certbot._internal.plugins.standalone.ServerManager.c C s, ddl m} i | _i | _|| j| j�| _d S )Nr )� ServerManager)�$certbot._internal.plugins.standaloner ZcertsZhttp_01_resources�mgr)�selfr � r �S/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/standalone_test.py�setUp s zServerManagerTest.setUpc C s d S �Nr �r r r r � test_init s zServerManagerTest.test_initc C s2 | j jd|d�}|�� d d }| j j|d� d S )Nr ��port�challenge_type� )r )r �run� getsocknames�stop)r r �serverr r r r �_test_run_stop$ s z ServerManagerTest._test_run_stopc C s | � tj� d S r )r r �HTTP01r r r r �test_run_stop_http_01+ s z'ServerManagerTest.test_run_stop_http_01c C sD | j jdtjd�}|�� d d }| j j|tjd�}| j �|� d S )Nr r r )r r r r r r )r r r �server2r r r �test_run_idempotent. s z%ServerManagerTest.test_run_idempotentc C s� t � t j�}|�d� |�� d }t � � }z|�d|f� W n tyN Y n0 t�tj��" | j j |t j d� W d � n1 s�0 Y |� � |� � d S )N�� r r r% )r )�socketZAF_INET6�bind� getsockname�OSError�pytest�raisesr �StandaloneBindErrorr r r r �close)r Z some_serverr Zmaybe_another_serverr r r �test_run_bind_error7 s �$z%ServerManagerTest.test_run_bind_errorN) �__name__� __module__� __qualname__�__doc__r r r r! r# r. r r r r r s r c C s4 t � t jt jd�} | �d� | �� d }| �� |S )z%Gets an open port number from the OS.r r$ r )r& ZAF_INETZ SOCK_STREAMr' r( r- )Z open_socketr r r r � get_open_portH s r3 c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Ze� � d d � �Z e� � d d � �Z dd� Z dd� Z dd� Zdd� Zedd� �Zdd� Zdd� Zd S ) �AuthenticatorTestz=Tests for certbot._internal.plugins.standalone.Authenticator.c C s< ddl m} tjt� d�| _|| jdd�| _t�� | j_d S )Nr )� Authenticator)� http01_portr )�name)r r5 r � MagicMockr3 �config�auth�servers)r r5 r r r r T s zAuthenticatorTest.setUpc C s d S r r r r r r �test_more_info[ s z AuthenticatorTest.test_more_infoc C s d S r r r r r r �test_get_chall_pref^ s z%AuthenticatorTest.test_get_chall_prefc C s& | � � }| j�|�}dd� |D �}d S )Nc S s g | ]}|� |j��qS r )�response� account_key)�.0Zachallr r r � <listcomp>f � z2AuthenticatorTest.test_perform.<locals>.<listcomp>)� _get_achallsr: �perform)r Zachallsr> Zexpectedr r r � test_performb s z AuthenticatorTest.test_performc C s^ |� }t j}t�tj|d�d�}|gdt�� g | jjj_ |j }d|_ | � � | � |� d S )N��errno���� T)rG � EADDRINUSEr r, r r8 r: r; r � side_effect�yesno� return_valuerE �_assert_correct_yesno_call)r �mock_get_utility� mock_utility�encountered_errno�error� mock_yesnor r r � test_perform_eaddrinuse_retryi s z/AuthenticatorTest.test_perform_eaddrinuse_retryc C s\ |� }|j }d|_tj}t�tj�� | �|� W d � n1 sD0 Y | � |� d S )NF) rL rM rG rJ r* r+ r � PluginError� _fail_performrN )r rO rP rS rQ r r r � test_perform_eaddrinuse_no_retryu s (z2AuthenticatorTest.test_perform_eaddrinuse_no_retryc C s |j \}}d S r )Z call_args)r rS Z yesno_argsZ yesno_kwargsr r r rN � s z,AuthenticatorTest._assert_correct_yesno_callc C s@ t j}t�tj�� | �|� W d � n1 s20 Y d S r )rG ZEACCESr* r+ r rU rV �r rQ r r r �test_perform_eacces� s z%AuthenticatorTest.test_perform_eaccesc C s@ t j}t�tj�� | �|� W d � n1 s20 Y d S r )rG ZENOTCONNr* r+ r r, rV rX r r r �$test_perform_unexpected_socket_error� s z6AuthenticatorTest.test_perform_unexpected_socket_errorc C s4 t �tj|d�d�}|| jjj_| j�| � � � d S )NrF rH ) r r, r r8 r: r; r rK rD rC )r rQ rR r r r rV � s z AuthenticatorTest._fail_performc C s. d}t j�t�d��}tjtj||d�}|gS )Ns localhostzrsa512_key.pem)Zchallb�domainr? ) �joseZJWK�load� test_utilZ load_vectorr Z"KeyAuthorizationAnnotatedChallenger ZHTTP01_P)�clsr[ �keyZhttp_01r r r rC � s �z AuthenticatorTest._get_achallsc C s� ddd�| j jj_| j jd �d� | j jd �ddg� | j �dg� | j jj� d� ddi| j jj_| j �dg� | j �dg� | j jj� d� d S ) NZserver1r" )r rI Zchall1Zchall2Zchall3r rI ) r: r; ZrunningrM Zserved�add�updateZcleanupr Zassert_called_once_withZassert_called_withr r r r � test_cleanup� s �� z AuthenticatorTest.test_cleanupc C s d| j _d | j _d| j _d S )NZ80z 127.0.0.1)r9 r6 Zhttp01_addressr r r r �test_auth_hint� s z AuthenticatorTest.test_auth_hintN)r/ r0 r1 r2 r r< r= rE r^ Zpatch_display_utilrT rW rN rY rZ rV � classmethodrC rc rd r r r r r4 Q s" r4 �__main__r )#r2 rG r& �sys�typingr r r Zunittestr Zjosepyr\ ZOpenSSL.cryptoZOpenSSLr* Zacmer r Zacme_standaloneZcertbotr r Z certbot.testsr r r^ ZTestCaser r3 r4 r/ �exit�main�argv�__file__r r r r �<module> s, 1 n
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
common_test.cpython-39.opt-1.pyc
Edit
common_test.cpython-39.pyc
Edit
disco_test.cpython-39.opt-1.pyc
Edit
disco_test.cpython-39.pyc
Edit
dns_common_test.cpython-39.opt-1.pyc
Edit
dns_common_test.cpython-39.pyc
Edit
enhancements_test.cpython-39.opt-1.pyc
Edit
enhancements_test.cpython-39.pyc
Edit
manual_test.cpython-39.opt-1.pyc
Edit
manual_test.cpython-39.pyc
Edit
null_test.cpython-39.opt-1.pyc
Edit
null_test.cpython-39.pyc
Edit
selection_test.cpython-39.opt-1.pyc
Edit
selection_test.cpython-39.pyc
Edit
standalone_test.cpython-39.opt-1.pyc
Edit
standalone_test.cpython-39.pyc
Edit
storage_test.cpython-39.opt-1.pyc
Edit
storage_test.cpython-39.pyc
Edit
util_test.cpython-39.opt-1.pyc
Edit
util_test.cpython-39.pyc
Edit
webroot_test.cpython-39.opt-1.pyc
Edit
webroot_test.cpython-39.pyc
Edit