|
Server : Apache/2.4.41 (Ubuntu) System : Linux vmi1525618.contaboserver.net 5.4.0-105-generic #119-Ubuntu SMP Mon Mar 7 18:49:24 UTC 2022 x86_64 User : www-data ( 33) PHP Version : 8.2.12 Disable Function : NONE Directory : /lib/python3/dist-packages/certbot/tests/__pycache__/ |
Upload File : |
U
�]� � @ sr d Z ddlZddlZddlZddlmZ ddlm mZ G dd� de j
�Z G dd� dej �Z
edkrne�� dS ) zTests for certbot.eff.� N)� constantsc s� e Zd ZdZ� fdd�Zdd� Ze�� e� d�dd� ��Z
e� d�d d
� �Z e�� e� d�d d � ��Z d
d� Z
e�� e� d�dd� ��Zdd� Ze�� e� d�dd� ��Zdd� Z� ZS )�HandleSubscriptionTestz*Tests for certbot.eff.handle_subscription.c s* t t| ��� d| _| j| j_d | j_d S )N�certbot@example.org)�superr �setUp�email�config� eff_email��self�� __class__� �8/usr/lib/python3/dist-packages/certbot/tests/eff_test.pyr
s
z HandleSubscriptionTest.setUpc C s ddl m} || j�S )Nr )�handle_subscription)� certbot.effr r )r r r r r �_call s z HandleSubscriptionTest._callzcertbot.eff.subscribec C s\ d | j _d| j _| �� | �|j� | �|� jj� |� jjd d }d}| � ||k� d S )NTr z,because you didn't provide an e-mail address)
r r r r � assertFalse�called�yesno� add_message� call_args�
assertTrue)r �mock_subscribe�mock_get_utility�actual�
expected_partr r r � test_failure s z#HandleSubscriptionTest.test_failurec C s> d| j _t�� �}| �� W 5 Q R X | �|j� | �|� d S �NF)r r � test_util�patch_get_utilityr r r � _assert_no_get_utility_calls�r r r r r r � test_no_subscribe_with_no_prompt# s
z7HandleSubscriptionTest.test_no_subscribe_with_no_promptc C s( d| j _| �� | �|� | �|� d S �NT)r r r �_assert_subscribedr! r" r r r � test_subscribe_with_no_prompt+ s
z4HandleSubscriptionTest.test_subscribe_with_no_promptc C s$ | � |� jj� | � |� jj� d S �N)r r r r �r r r r r r! 3 s z3HandleSubscriptionTest._assert_no_get_utility_callsc C s: d|� j _| �� | �|� | �|� jj� | �|� d S r$ )r � return_valuer r% r r r �_assert_correct_yesno_callr" r r r �test_subscribe_with_prompt7 s
z1HandleSubscriptionTest.test_subscribe_with_promptc C s( | � |j� | �|jd d | j� d S �Nr )r r � assertEqualr r )r r r r r r% @ s z)HandleSubscriptionTest._assert_subscribedc C s< d|� j _| �� | �|j� | �|� jj� | �|� d S r )r r) r r r r r* r" r r r � test_no_subscribe_with_promptD s
z4HandleSubscriptionTest.test_no_subscribe_with_promptc C sN | � |� jj� |� jj\}}|d }d}| � ||k� | �|�dd�� d S )Nr z Electronic Frontier Foundation�defaultT)r r r r r �get)r r r � call_kwargsr r r r r r* M s z1HandleSubscriptionTest._assert_correct_yesno_call)�__name__�
__module__� __qualname__�__doc__r r r r �mock�patchr r# r& r! r+ r% r. r* �
__classcell__r r r r r s(
r c @ s� e Zd ZdZdd� Ze�d�dd� �Zdd� Ze �
� d d
� �Z e �
� d d � �Z e �
� d
d� �Z
e �
� dd� �Zdd� Ze �
� dd� �ZdS )�
SubscribeTestz Tests for certbot.eff.subscribe.c C s. d| _ ddi| _tjdd�| _| j| jj_d S )Nr �statusT)�ok)r �jsonr6 ZMock�responser) r
r r r r X s
zSubscribeTest.setUpzcertbot.eff.requests.postc C s, | j |_ddlm} || j� | �|� d S )Nr )� subscribe)r= r) r r> r �_check_post_call)r � mock_postr> r r r r ^ s
zSubscribeTest._callc C sZ | � |jd� |j\}}| � |d tj� |�d�}| �|d k� | � |�d�| j� d S )N� r �datar )r- Z
call_countr r ZEFF_SUBSCRIBE_URIr0 r r )r r@ r r1 rB r r r r? f s
z SubscribeTest._check_post_callc C s2 d| j d< | �� | �|�}d}| �||k� d S )NFr: z2because your e-mail address appears to be invalid.)r< r �_get_reported_messager �r r r r r r r �test_bad_statuso s
z SubscribeTest.test_bad_statusc C s> d| j _tjj| j j_| �� | �|�}d}| � ||k� d S )NFZbecause)
r= r; �requests�
exceptionsZ HTTPErrorZraise_for_status� side_effectr rC r )r r r Zunexpected_partr r r � test_not_okw s
zSubscribeTest.test_not_okc C s4 t � | jj_| �� | �|�}d}| �||k� d S �NZproblem)�
ValueErrorr= r< rH r rC r rD r r r �test_response_not_json� s
z$SubscribeTest.test_response_not_jsonc C s2 | j �� | �� | �|�}d}| �||k� d S rJ )r< �clearr rC r rD r r r �)test_response_json_missing_status_element� s
z7SubscribeTest.test_response_json_missing_status_elementc C s" | � |� jj� |� jjd d S r, )r r r r r( r r r rC � s z#SubscribeTest._get_reported_messagec C s | � � | �|j� d S r' )r r r r( r r r �test_subscribe� s z SubscribeTest.test_subscribeN)r2 r3 r4 r5 r r6 r7 r r? r r rE rI rL rN rC rO r r r r r9 V s
r9 �__main__)r5 Zunittestr6 rF Zcertbotr Zcertbot.tests.utilZtests�utilr ZConfigTestCaser ZTestCaser9 r2 �mainr r r r �<module> s KD