← Back
Editing: eff_test.cpython-38.pyc
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
Save File
Cancel