a
}|ägÀ7 ã @ s d Z ddlZddlZddlZddlmZ ddlZddlZddlZddl m
Z
ddl m Z ddl
mZ ddl
mZ ddl
mZ ddlm mZ ej e d ¡¡ZG d
d „ d ejƒZG d d
„ d
ejƒZG dd„ dejƒZ G dd„ dej ƒZ e dkre e !ej"dd… e#g ¡¡ dS )z$Tests for certbot._internal.account.é N)Úmock)Úmessages)Úerrors)Ú
filesystem)Úmisc)Úoszrsa512_key.pemc @ s8 e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd S )
Ú AccountTestz,Tests for certbot._internal.account.Account.c C sÚ ddl m} t ¡ | _|jdtjddddddtjd d
| _ || jt
| j ƒ| _ tjd d | j_ t
d
¡^}d|j_t
d¡,}| j j|jj_|| jt
ƒ| _W d ƒ n1 s®0 Y W d ƒ n1 sÌ0 Y d S )Nr ©ÚAccountztest.certbot.orgiß é é é é
©Ztzinfo©Z
creation_hostÚ creation_dtZ i_am_a_regr)Ú return_valuez certbot._internal.account.socketz"certbot._internal.account.datetime)Úcertbot._internal.accountr
r Ú MagicMockÚregrÚMetaÚdatetimeÚpytzÚUTCÚmetaÚKEYÚaccÚ__repr__ÚpatchZgetfqdnr r ZnowZ acc_no_meta)Úselfr
Z mock_socketZmock_dt© r úH/usr/lib/python3.9/site-packages/certbot/_internal/tests/account_test.pyÚsetUp s
ÿþ zAccountTest.setUpc C s d S ©Nr ©r r r r! Ú test_init( s zAccountTest.test_initc C s d S r# r r$ r r r! Útest_id- s zAccountTest.test_idc C s d S r# r r$ r r r! Ú test_slug0 s zAccountTest.test_slugc C s d S r# r r$ r r r! Ú test_repr3 s zAccountTest.test_reprN) Ú__name__Ú
__module__Ú __qualname__Ú__doc__r" r% r& r' r( r r r r! r s r c @ s e Zd ZdZdd„ Zdd„ ZdS )ÚMetaTestz)Tests for certbot._internal.account.Meta.c C s ddl m} |j d¡}d S )Nr r zU{ "creation_dt": "2020-06-13T07:46:45Z", "creation_host": "hyperion.localdomain"}©r r
r Z
json_loads©r r
r r r r! Útest_deserialize_partial: s
ÿz!MetaTest.test_deserialize_partialc C s ddl m} |j d¡}d S )Nr r zq{ "creation_dt": "2020-06-13T07:46:45Z", "creation_host": "hyperion.localdomain", "register_to_eff": "bar"}r. r/ r r r! Útest_deserialize_fullE s
ÿz MetaTest.test_deserialize_fullN)r) r* r+ r, r0 r1 r r r r! r- 8 s r- c @ s e Zd ZdZdd„ Zdd„ ZdS )ÚAccountMemoryStorageTestz9Tests for certbot._internal.account.AccountMemoryStorage.c C s ddl m} |ƒ | _d S )Nr )ÚAccountMemoryStorage)r r3 Ústorage)r r3 r r r! r" U s z AccountMemoryStorageTest.setUpc C sd t jdd}t tj¡ | j d¡ W d ƒ n1 s:0 Y | j |d ¡ | j |d ¡ d S )NÚx)Úid) r ZMockÚpytestÚraisesr ÚAccountNotFoundr4 ÚloadÚsave)r Úaccountr r r! Útest_itY s *z AccountMemoryStorageTest.test_itN)r) r* r+ r, r" r= r r r r! r2 R s r2 c s e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd„ Zd d „ Zd d „ Zd d!„ Zd"d#„ Ze d$¡d%d&„ ƒZd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Z d1d2„ Z d3d4„ Z d5d6„ Z d7d8„ Z d9d:„ Z!d;d<„ Z"d=d>„ Z#d?d@„ Z$‡ Z%S )AÚAccountFileStorageTestz7Tests for certbot._internal.account.AccountFileStorage.c s~ t ƒ ¡ ddlm} || jƒ| _ddlm} |jdtjddddd d
t j
d d }|t j d t
¡ d
t|d| _t ¡ | _d S )Nr ©ÚAccountFileStorager ztest.example.orgiå é é r
r r r r )ZuriÚbody)r Úkeyr )Úsuperr" r r@ Úconfigr4 r
r r r r r ZRegistrationResourceZ Registrationr r r r Ú mock_client)r r@ r
r ©Ú __class__r r! r" h s$
ÿþÿüz AccountFileStorageTest.setUpc C s d S r# r r$ r r r! Útest_init_creates_dirz s z,AccountFileStorageTest.test_init_creates_dirc C sF | j | j| j¡ tj | jj| jj ¡}dD ]}q,| j
| jj ¡}d S )N)z regr.jsonz meta.jsonzprivate_key.json) r4 r; r rG r ÚpathÚjoinrF Ú accounts_dirr6 r: )r Ú account_pathÚ file_nameZloadedr r r! Útest_save_and_restore~ s z,AccountFileStorageTest.test_save_and_restorec C s( | j | j¡ tj | jj| jj¡}d S r# ) r4 Z update_regrr r rK rL rF rM r6 ©r rN r r r! Útest_update_regr‹ s z'AccountFileStorageTest.test_update_regrc C s( | j | j¡ tj | jj| jj¡}d S r# ) r4 Z update_metar r rK rL rF rM r6 rQ r r r! Útest_update_meta” s z'AccountFileStorageTest.test_update_metac C s | j | j| j¡ d S r# )r4 r; r rG r$ r r r! Ú
test_find_all s z$AccountFileStorageTest.test_find_allc C s d S r# r r$ r r r! Ú test_find_all_none_empty_list¡ s z4AccountFileStorageTest.test_find_all_none_empty_listc C s t | jj¡ d S r# )r ÚrmdirrF rM r$ r r r! Ú!test_find_all_accounts_dir_absent¤ s z8AccountFileStorageTest.test_find_all_accounts_dir_absentc C sP t jdtjdgd| j_t d¡}g d¢|_W d ƒ n1 sB0 Y d S )Nr5 Úz©Ú side_effectz$certbot._internal.account.os.listdir)r5 ÚyrX )r r r ÚAccountStorageErrorr4 Ú_load_for_server_pathr r )r Z mock_listdirr r r! Útest_find_all_load_skips¨ s
ÿ
z/AccountFileStorageTest.test_find_all_load_skipsc C s< t tj¡ | j d¡ W d ƒ n1 s.0 Y d S )NÚmissing)r7 r8 r r9 r4 r: r$ r r r! Ú#test_load_non_existent_raises_error° s z:AccountFileStorageTest.test_load_non_existent_raises_errorc C s$ || j _ddlm} || j ƒ| _d S )Nr r? )rF Úserverr r@ r4 )r ra r@ r r r! Ú _set_server´ s z"AccountFileStorageTest._set_serverc C s | d¡ d S ©Nú6https://acme-staging-v02.api.letsencrypt.org/directory)rb r$ r r r! Ú test_find_all_neither_exists¹ s
z3AccountFileStorageTest.test_find_all_neither_existsc C s$ | d¡ | j | j| j¡ d}d S ©Nrd ú2https://acme-staging.api.letsencrypt.org/directory©rb r4 r; r rG ©r Zprev_server_pathr r r! Ú test_find_all_find_before_save¿ s
z5AccountFileStorageTest.test_find_all_find_before_savec C s$ | d¡ | j | j| j¡ d}d S rf rh ri r r r! Ú test_find_all_save_before_findÊ s
z5AccountFileStorageTest.test_find_all_save_before_findc C s* | d¡ | j | j| j¡ | d¡ d S rf rh r$ r r r! Ú test_find_all_server_downgradeÔ s
z5AccountFileStorageTest.test_find_all_server_downgradec C s* | d¡ | j | j| j¡ | d¡ d S ©Nrg rd rh r$ r r r! Ú test_upgrade_version_stagingÝ s
z3AccountFileStorageTest.test_upgrade_version_stagingc C s* | d¡ | j | j| j¡ | d¡ d S )Nz.https://acme-v01.api.letsencrypt.org/directoryz.https://acme-v02.api.letsencrypt.org/directoryrh r$ r r r! Ú test_upgrade_version_productionã s
z6AccountFileStorageTest.test_upgrade_version_productionzcertbot.compat.os.rmdirc C sB | d¡ | j | j| j¡ t|_tjt j
d| j_ | d¡ d S )Nrg rY rd ) rb r4 r; r rG ÚOSErrorrZ r r r r\ r] )r Z
mock_rmdirr r r! Útest_corrupted_accounté s
ÿ
z-AccountFileStorageTest.test_corrupted_accountc C sJ | d¡ | j | j| j¡ | j | jj¡}| d¡ | j | jj¡}d S rm )rb r4 r; r rG r: r6 ©r Z prev_accountr< r r r! Útest_upgrade_loadô s
z(AccountFileStorageTest.test_upgrade_loadc C sJ | d¡ | j | j| j¡ | j | jj¡}| d¡ | j | jj¡}d S rm )rb r4 r; r rG r: r6 Ú _set_server_and_stop_symlinkrr r r r! Ú test_upgrade_load_single_accountü s
z7AccountFileStorageTest.test_upgrade_load_single_accountc C sŒ | j | j| j¡ t ¡ }t|_t d|¡L t
t j ¡ | j
| jj¡ W d ƒ n1 s`0 Y W d ƒ n1 s~0 Y d S ©Nz
builtins.open)r4 r; r rG r Ú mock_openÚIOErrorrZ r r7 r8 r r\ r: r6 ©r rw r r r! Útest_load_ioerror s z(AccountFileStorageTest.test_load_ioerrorc C s| t ¡ }t|_t d|¡N t tj¡" | j
| j | j ¡ W d ƒ n1 sP0 Y W d ƒ n1 sn0 Y d S rv )
r rw rx rZ r r7 r8 r r\ r4 r; r rG ry r r r! Útest_save_ioerrors s
z)AccountFileStorageTest.test_save_ioerrorsc C sb | j | j| j¡ | j | jj¡ t tj ¡ | j
| jj¡ W d ƒ n1 sT0 Y d S r# ) r4 r; r rG Údeleter6 r7 r8 r r9 r: r$ r r r! Ú test_delete s z"AccountFileStorageTest.test_deletec C s@ t tj¡ | j | jj¡ W d ƒ n1 s20 Y d S r# )r7 r8 r r9 r4 r| r r6 r$ r r r! Útest_delete_no_account s z-AccountFileStorageTest.test_delete_no_accountc C s’ | d¡ | j | j| j¡ t di ¡V | d¡ t t j
¡ | j | jj ¡ W d ƒ n1 sf0 Y W d ƒ n1 s„0 Y d S )Nrg z,certbot._internal.constants.LE_REUSE_SERVERSrd )
rb r4 r; r rG r r r7 r8 r r9 r: r6 r$ r r r! Ú!_assert_symlinked_account_removed s
z8AccountFileStorageTest._assert_symlinked_account_removedc C sà | d¡ | j | j| j¡ | d¡ | j | jj¡ | |¡ | j | jj¡ | d¡ t t
j ¡ | j | jj¡ W d ƒ n1 sŒ0 Y | d¡ t t
j ¡ | j | jj¡ W d ƒ n1 sÒ0 Y d S rm ) rb r4 r; r rG r: r6 r| r7 r8 r r9 )r Z
server_urlr r r! Ú_test_delete_folders' s
.
z+AccountFileStorageTest._test_delete_foldersc C s | d¡ | ¡ d S )Nrg ©r€ r r$ r r r! Útest_delete_folders_up: s
z-AccountFileStorageTest.test_delete_folders_upc C s | d¡ | ¡ d S rc r r$ r r r! Útest_delete_folders_down> s
z/AccountFileStorageTest.test_delete_folders_downc C sP | |¡ ttj | jjd¡dƒ}| d¡ W d ƒ n1 sB0 Y d S )NZfooÚwZbar)rb Úopenr rK rL rF rM Úwrite)r Z server_pathÚfr r r! rt B s
z3AccountFileStorageTest._set_server_and_stop_symlinkc C s | d¡ | d¡ d S rf ©rt r€ r$ r r r! Ú test_delete_shared_account_upG s
z4AccountFileStorageTest.test_delete_shared_account_upc C s | d¡ | d¡ d S rc rˆ r$ r r r! Ú test_delete_shared_account_downK s
z6AccountFileStorageTest.test_delete_shared_account_down)&r) r* r+ r, r" rJ rP rR rS rT rU rW r^ r` rb re rj rk rl rn ro r r rq rs ru rz r{ r} r~ r r€ r‚ rƒ rt r‰ rŠ Ú
__classcell__r r rH r! r> e sB
r> Ú__main__rA )$r, r ÚsysZunittestr ZjosepyZjoser7 r Zacmer Zcertbotr Zcertbot.compatr r r Zcertbot.tests.utilZtestsÚutilZ test_utilZJWKRSAr: Z load_vectorr ZTestCaser r- r2 ZConfigTestCaser> r) ÚexitÚmainÚargvÚ__file__r r r r! Ú s* # l