a
}|g @ s d Z ddlZddlZddlZddlmZ ddlZddlZddlZddlZddl m
Z
ddl m Z ddl m
Z
ddlm mZ ejedZG dd d ejZG d
d d eZG d d
d
eZG dd dejZe dkre e ej dd e g dS )z Tests for certbot._internal.eff. N)mock)messages)account) constantszrsa512_key.pemc s e Zd ZdZ fddZ ZS )SubscriptionTestz&Abstract class for subscription tests.c sb t tjtjd t ddttjjdt j ddddddt
j d d
d | _d | j _
d | j _d S )
Nhi)ZuribodyZnew_authzr_uriztest.certbot.orgi
)Ztzinfo)Z
creation_hostZ creation_dt)Zregrkeymetacertbot@example.org)supersetUpr ZAccountr ZRegistrationResourceZ Registration_KEYZMetadatetimepytzZUTCconfigemail eff_emailself __class__ D/usr/lib/python3.9/site-packages/certbot/_internal/tests/eff_test.pyr s
zSubscriptionTest.setUp)__name__
__module__ __qualname____doc__r
__classcell__r r r r r s r c @ s e Zd ZdZdd Ze eddd Z e dd Z
e d d
Z e d d Z e d
d Z
dd Zdd ZdS )PrepareSubscriptionTestz5Tests for certbot._internal.eff.prepare_subscription.c C s ddl m} || j| j d S )Nr )prepare_subscription)certbot._internal.effr$ r r )r r$ r r r _call' s z PrepareSubscriptionTest._call)certbot._internal.eff.display_util.notifyc C s. d | j _d| j _| |jd d }d}d S )NTr z,because you didn't provide an e-mail address)r r r r& call_args)r mock_notifymock_get_utilityactual
expected_partr r r test_failure+ s z$PrepareSubscriptionTest.test_failurec C s d| j _| | | d S NFr r r& _assert_no_get_utility_callsr r* r r r &test_will_not_subscribe_with_no_prompt6 s
z>PrepareSubscriptionTest.test_will_not_subscribe_with_no_promptc C s d| j _| | | d S NTr/ r1 r r r "test_will_subscribe_with_no_prompt= s
z:PrepareSubscriptionTest.test_will_subscribe_with_no_promptc C s d| j _| | | d S r. yesno return_valuer& _assert_correct_yesno_callr1 r r r #test_will_not_subscribe_with_promptD s
z;PrepareSubscriptionTest.test_will_not_subscribe_with_promptc C s d| j _| | | d S r3 r5 r1 r r r test_will_subscribe_with_promptL s
z7PrepareSubscriptionTest.test_will_subscribe_with_promptc C s d S Nr r1 r r r r0 T s z4PrepareSubscriptionTest._assert_no_get_utility_callsc C s | j j\}}|d }d}d S )Nr z Electronic Frontier Foundation)r6 r( )r r* r( call_kwargsr+ r, r r r r8 X s z2PrepareSubscriptionTest._assert_correct_yesno_callN)r r r r! r& test_utilpatch_display_utilr patchr- r2 r4 r9 r: r0 r8 r r r r r# % s
r# c @ s< e Zd ZdZdd Zeddd Zeddd Zd S )
HandleSubscriptionTestz4Tests for certbot._internal.eff.handle_subscription.c C s ddl m} || j| j d S )Nr )handle_subscription)r% rA r r )r rA r r r r& c s z HandleSubscriptionTest._callz certbot._internal.eff.subscribec C s | d S r; r& r Zmock_subscriber r r test_no_subscribeg s z(HandleSubscriptionTest.test_no_subscribec C s$ | j jj| jjd| j _| d S )N)Zregister_to_eff)r r updater r r& rC r r r test_subscribel s z%HandleSubscriptionTest.test_subscribeN) r r r r! r& r r? rD rF r r r r r@ a s
r@ c @ sj e Zd ZdZdd Zeddd Zdd Zd d
Z d d Z
d
d Z dd Z dd Z
e dd ZdS )
SubscribeTestz*Tests for certbot._internal.eff.subscribe.c C sN d| _ ddi| _tjdd| _| j| jj_td}| | _| |j
d S )Nr statusT)okr' ) r jsonr ZMockresponser7 r? startr) Z
addCleanupstop)r Zpatcherr r r r v s
zSubscribeTest.setUpz#certbot._internal.eff.requests.postc C s, | j |_ddlm} || j | | d S )Nr ) subscribe)rK r7 r% rN r _check_post_call)r mock_postrN r r r r& s
zSubscribeTest._callc C s |j \}}|d}d S )Ndata)r( get)r rP r( r< rQ r r r rO s
z SubscribeTest._check_post_callc C s" d| j d< | | }d}d S )NFrH z2because your e-mail address appears to be invalid.)rJ r& _get_reported_messager r+ r, r r r test_bad_status s
z SubscribeTest.test_bad_statusc C s. d| j _tjj| j j_| | }d}d S )NFZbecause) rK rI requests
exceptionsZ HTTPErrorZraise_for_status side_effectr& rS )r r+ Zunexpected_partr r r test_not_ok s zSubscribeTest.test_not_okc C s$ t | jj_| | }d}d S NZproblem)
ValueErrorrK rJ rX r& rS rT r r r test_response_not_json s
z$SubscribeTest.test_response_not_jsonc C s" | j | | }d}d S rZ )rJ clearr& rS rT r r r )test_response_json_missing_status_element s
z7SubscribeTest.test_response_json_missing_status_elementc C s | j jd d S )Nr )r) r( r r r r rS s z#SubscribeTest._get_reported_messagec C s | d S r; rB r1 r r r rF s z SubscribeTest.test_subscribeN)r r r r! r r r? r& rO rU rY r\ r^ rS r= r> rF r r r r rG t s
rG __main__ )!r! r sysZunittestr ZjosepyZpytestr rV Zacmer Zcertbot._internalr r Zcertbot.tests.utilZtestsutilr= ZJWKRSAloadZ load_vectorr ZConfigTestCaser r# r@ ZTestCaserG r exitmainargv__file__r r r r s& <C