File: //usr/lib/python3/dist-packages/certbot/plugins/__pycache__/standalone_test.cpython-38.pyc
U
�]� � @ s� d Z ddlZddlmZ ddlZddlZddlZddlZddl Z
ddl m Z ddl m
Z ddlmZmZmZ ddlmZ ddlmZ dd lmZ dd
lmZ G d d � d ej�Zd
d� Z G dd� dej�Z e dkr�e� � dS )z%Tests for certbot.plugins.standalone.� N��errno)�
challenges)�
standalone)�Dict�Tuple�Set)� achallenges)�errors)� acme_util)�utilc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�ServerManagerTestz3Tests for certbot.plugins.standalone.ServerManager.c C s, ddl m} i | _i | _|| j| j�| _d S )Nr )�
ServerManager)�certbot.plugins.standaloner �certs�http_01_resources�mgr)�selfr � r �A/usr/lib/python3/dist-packages/certbot/plugins/standalone_test.py�setUp s zServerManagerTest.setUpc C s, | � | jj| jk� | � | jj| jk� d S �N)�
assertTruer r r �r r r r � test_init" s �zServerManagerTest.test_initc C sZ | j jd|d�}|�� d d }| �| j �� ||i� | j j|d� | �| j �� i � d S )Nr ��port�challenge_type� )r )r �run� getsocknames� assertEqual�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 sz | j jdtjd�}|�� d d }| j j|tjd�}| �| j �� ||i� | �||k� | j �|� | �| j �� i � d S )Nr r r ) r r r r& r r! r" r r# )r r$ r �server2r r r �test_run_idempotent1 s z%ServerManagerTest.test_run_idempotentc C s� t � t j�}|�d� |�� d }t � � }z|�d|f� W n t jk
rR Y nX | jtj| jj |t
j d� | � | j�
� i � |�� |�� d S )N�� r r r+ )r )�socketZAF_INET6�bind� getsockname�error� assertRaisesr
�StandaloneBindErrorr r r r& r! r" �close)r Z some_serverr Zmaybe_another_serverr r r �test_run_bind_error: s"
�z%ServerManagerTest.test_run_bind_errorN)
�__name__�
__module__� __qualname__�__doc__r r r% r' r) r3 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. r2 )Z open_socketr r r r �
get_open_portK s
r8 c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Ze� � d
d � �Z
e� � d d
� �Z dd� Z dd� Z
dd� Zdd� Zedd� �Zdd� ZdS )�AuthenticatorTestz3Tests for certbot.plugins.standalone.Authenticator.c C s< ddl m} tjt� d�| _|| jdd�| _t�� | j_d S )Nr )�
Authenticator)Z http01_portr )�name)r r: �mock� MagicMockr8 Zconfig�auth�servers)r r: r r r r W s zAuthenticatorTest.setUpc C s | � t| j�� tj�� d S r )r �
isinstancer> Z more_info�sixZ string_typesr r r r �test_more_info^ s z AuthenticatorTest.test_more_infoc C s | � | jjd d�tjg� d S )N)�domain)r! r> Zget_chall_prefr r& r r r r �test_get_chall_prefa s �z%AuthenticatorTest.test_get_chall_prefc C s2 | � � }| j�|�}dd� |D �}| �||� d S )Nc S s g | ]}|� |j��qS r )�response� account_key)�.0Zachallr r r �
<listcomp>i s z2AuthenticatorTest.test_perform.<locals>.<listcomp>)� _get_achallsr> �performr! )r ZachallsrE Zexpectedr r r � test_performe s z AuthenticatorTest.test_performc C s^ |� }t j}t�tj|d�d�}|gdt�� g | jjj_ |j
}d|_ | � � | �
|� d S )Nr ���� T)�
socket_errors�
EADDRINUSEr
r1 r< r= r> r? r � side_effect�yesno� return_valuerK �_assert_correct_yesno_call)r �mock_get_utility� mock_utilityr r/ �
mock_yesnor r r � test_perform_eaddrinuse_retryl s z/AuthenticatorTest.test_perform_eaddrinuse_retryc C s8 |� }|j }d|_tj}| �tj| j|� | �|� d S )NF) rQ rR rN rO r0 r
� PluginError�
_fail_performrS )r rT rU rV r r r r � test_perform_eaddrinuse_no_retryx s z2AuthenticatorTest.test_perform_eaddrinuse_no_retryc C s2 |j \}}| �d|d k� | �|�dd�� d S )Nzin user �defaultT)Z call_argsr Z assertFalse�get)r rV Z
yesno_argsZ yesno_kwargsr r r rS � s
z,AuthenticatorTest._assert_correct_yesno_callc C s t j}| �tj| j|� d S r )rN ZEACCESr0 r
rX rY �r r r r r �test_perform_eacces� s z%AuthenticatorTest.test_perform_eaccesc C s t j}| �tj| j|� d S r )rN ZENOTCONNr0 r
r1 rY 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 )Nr rL )
r
r1 r< r= r> r? r rP rJ rI )r r r/ r r r rY � s z AuthenticatorTest._fail_performc C s. d}t j�t�d��}tjtj||d�}|gS )Ns localhostzrsa512_key.pem)ZchallbrC rF ) �joseZJWK�load� test_utilZ load_vectorr Z"KeyAuthorizationAnnotatedChallenger ZHTTP01_P)�clsrC �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 � t ddg�d�� | j jj
� d� d di| j jj_| j �dg� | �| j jt � t dg�d�� | �d| j jj
j � | j �dg� | �| j jt � t g �d�� | j jj
�
d � d S )
N�server1r( )r rM Zchall1Zchall2Zchall3)re r( r rM )r> r? r" rR Zserved�add�updateZcleanupr! �setr# Zassert_called_once_withZ
call_countZassert_called_withr r r r � test_cleanup� s4 �
� �
�
�z AuthenticatorTest.test_cleanupN)r4 r5 r6 r7 r rB rD rK rb Zpatch_get_utilityrW rZ rS r^ r_ rY � classmethodrI ri r r r r r9 T s
r9 �__main__) r7 r, r rN ZunittestZjosepyr` r< rA ZOpenSSL.cryptoZOpenSSLZacmer r Zacme_standaloneZacme.magic_typingr r r Zcertbotr r
Z
certbot.testsr r rb ZTestCaser
r8 r9 r4 �mainr r r r �<module> s& 3 e