/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 �certs�http_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( | j j| ju sJ �| j j| ju s$J �d S �N)r r r �r r r r � test_init s zServerManagerTest.test_initc C sZ | j jd|d�}|�� d d }| j �� ||iks6J �| j j|d� | j �� i ksVJ �d S )Nr ��port�challenge_type� )r )r �run� getsocknames�running�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 sx | j jdtjd�}|�� d d }| j j|tjd�}| j �� ||iksJJ �||u sVJ �| j �|� | j �� i kstJ �d S )Nr r 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 | j � � i ks�J �|�� |�� d S )N�� r r r( )r )�socketZAF_INET6�bind� getsockname�OSError�pytest�raisesr �StandaloneBindErrorr r 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& r1 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+ r0 )Z open_socketr r r r � get_open_portH s r6 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 r8 r � MagicMockr6 �config�auth�servers)r r8 r r r r T s zAuthenticatorTest.setUpc C s t | j�� t�sJ �d S r )� isinstancer= Z more_info�strr r r r �test_more_info[ s z AuthenticatorTest.test_more_infoc C s | j jd d�tjgksJ �d S )N)�domain)r= Zget_chall_prefr r# r r r r �test_get_chall_pref^ s �z%AuthenticatorTest.test_get_chall_prefc C s2 | � � }| j�|�}dd� |D �}||ks.J �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 ZachallsrD 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)rM � EADDRINUSEr r/ r r; r= r> r � side_effect�yesno� return_valuerK �_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) rR rS rM rP r- r. r � PluginError� _fail_performrT )r rU rV rY rW r r r � test_perform_eaddrinuse_no_retryu s (z2AuthenticatorTest.test_perform_eaddrinuse_no_retryc C s. |j \}}d|d v sJ �|�dd�r*J �d S )Nzin user �defaultT)Z call_args�get)r rY Z yesno_argsZ yesno_kwargsr r r rT � s z,AuthenticatorTest._assert_correct_yesno_callc C s@ t j}t�tj�� | �|� W d � n1 s20 Y d S r )rM ZEACCESr- r. r r[ r\ �r rW 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 )rM ZENOTCONNr- r. r r/ r\ r` 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 )NrL rN ) r r/ r r; r= r> r rQ rJ rI )r rW rX r r r r\ � s z AuthenticatorTest._fail_performc C s. d}t j�t�d��}tjtj||d�}|gS )Ns localhostzrsa512_key.pem)ZchallbrB rE ) �joseZJWK�load� test_utilZ load_vectorr Z"KeyAuthorizationAnnotatedChallenger ZHTTP01_P)�clsrB �keyZhttp_01r r r rI � s �z AuthenticatorTest._get_achallsc C s� ddd�| j jj_| j jd �d� | j jd �ddg� | j �dg� | j jt� ddhd�ksdJ �| j jj � d� d di| j jj_| j �dg� | j jt� dhd�ks�J �d| j jj j ks�J �| j �dg� | j jt� t� d�ks�J �| j jj � d � d S ) N�server1r% )r rO Zchall1Zchall2Zchall3)rh r% r rO ) r= r> r rS Zserved�add�updateZcleanup�setr Zassert_called_once_withZ call_countZassert_called_withr r r r � test_cleanup� s, � � � � � z AuthenticatorTest.test_cleanupc C sD d| j _d | j _d| j�g �v s$J �d| j _d| j�g �v s@J �d S )NZ80z on port 80z 127.0.0.1zon 127.0.0.1:80)r< r9 Zhttp01_addressr= Z auth_hintr r r r �test_auth_hint� s z AuthenticatorTest.test_auth_hintN)r2 r3 r4 r5 r rA rC rK re Zpatch_display_utilrZ r] rT ra rb r\ � classmethodrI rl rm r r r r r7 Q s" r7 �__main__r )#r5 rM r) �sys�typingr r r Zunittestr Zjosepyrc ZOpenSSL.cryptoZOpenSSLr- Zacmer r Zacme_standaloneZcertbotr r Z certbot.testsr r re ZTestCaser r6 r7 r2 �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