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 sL d | j _d| j _| |jd d }d}||v s6J | jjjd u sHJ d S )NTr z,because you didn't provide an e-mail address)r r r r& call_argsr r register_to_eff)r mock_notifymock_get_utilityactual
expected_partr r r test_failure+ s z$PrepareSubscriptionTest.test_failurec C s0 d| j _| | | | jjjd u s,J d S NF)r r r& _assert_no_get_utility_callsr r r) r r+ r r r &test_will_not_subscribe_with_no_prompt6 s
z>PrepareSubscriptionTest.test_will_not_subscribe_with_no_promptc C s4 d| j _| | | | jjj| j jks0J d S NT)r r r& r0 r r r) r r1 r r r "test_will_subscribe_with_no_prompt= s
z:PrepareSubscriptionTest.test_will_subscribe_with_no_promptc C s@ d| j _| | jjr J | | | jjjd u s rT r r r rQ s
z SubscribeTest._check_post_callc C s. d| j d< | | }d}||v s*J d S )NFrJ z2because your e-mail address appears to be invalid.)rL 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}||vs6J d S )NFZbecause) rM rK requests
exceptionsZ HTTPErrorZraise_for_status side_effectr& rU )r r, Zunexpected_partr r r test_not_ok s zSubscribeTest.test_not_okc C s0 t | jj_| | }d}||v s,J d S NZproblem)
ValueErrorrM rL rZ r& rU rV r r r test_response_not_json s
z$SubscribeTest.test_response_not_jsonc C s. | j | | }d}||v s*J d S r\ )rL clearr& rU rV r r r )test_response_json_missing_status_element s
z7SubscribeTest.test_response_json_missing_status_elementc C s | j js J | j jd d S )Nr )r* r8 r( r r r r rU s z#SubscribeTest._get_reported_messagec C s | |jdu sJ d S r/ rD r1 r r r rH s z SubscribeTest.test_subscribeN)r r r r! r r rA r& rQ rW r[ r^ r` rU r? r@ rH r r r r rI t s
rI __main__rS )!r! r sysZunittestr ZjosepyZpytestr rX Zacmer Zcertbot._internalr r Zcertbot.tests.utilZtestsutilr? ZJWKRSAloadZ load_vectorr ZConfigTestCaser r# rB ZTestCaserI r exitmainargv__file__r r r r s& <C