← Back
Editing: account_test.cpython-38.pyc
U �]&7 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlm Z ddl m m Z ddlmZ ddlmZ ddlmZ ddlmZ ej�e �d��ZG d d � d ej�ZG d d � d e j�ZG d d� dej�Z G dd� de j�Z e dkr�e� � dS )zTests for certbot.account.� N)�messages)�errors)� filesystem)�misc)�oszrsa512_key.pemc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � AccountTestz"Tests for certbot.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 5 Q R X W 5 Q R X d S )Nr ��Accountztest.certbot.orgi� � � � � )Ztzinfo)Z creation_host� creation_dtZ i_am_a_regr)� return_valuezcertbot.account.socketzcertbot.account.datetime)�certbot.accountr �mock� MagicMock�regrZMeta�datetime�pytzZUTC�meta�KEY�acc�__repr__�patchZgetfqdnr r Znow� acc_no_meta)�selfr Z mock_socketZmock_dt� r �</usr/lib/python3/dist-packages/certbot/tests/account_test.py�setUp s* �� zAccountTest.setUpc C s8 | � | j| jj� | � t| jj� | � | j| jj� d S �N)� assertEqualr r r �keyr r �r r r r � test_init( s zAccountTest.test_initc C s | � | jjd� d S )NZ 7adac10320f585ddf118429c0c4af2cd)r! r �idr# r r r �test_id- s �zAccountTest.test_idc C s | � | jjd� d S )Nz,test.certbot.org@2015-07-04T14:04:10Z (7ada))r! r Zslugr# r r r � test_slug1 s �zAccountTest.test_slugc C s | � t| j��d�� d S )Nz=<Account(i_am_a_regr, 7adac10320f585ddf118429c0c4af2cd, Meta()� assertTrue�reprr � startswithr# r r r � test_repr5 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� Ze�d�dd� �Ze�d�dd� �Zd S ) �ReportNewAccountTestz-Tests for certbot.account.report_new_account.c C s ddl m} || j� d S )Nr )�report_new_account)r r1 �config)r r1 r r r �_call< s zReportNewAccountTest._callz+certbot.account.zope.component.queryUtilityc C s d |_ | �� d S r )r r3 )r � mock_zoper r r �test_no_reporter@ s z%ReportNewAccountTest.test_no_reporterc C s4 | � � |� jj}| �| jj|d d d k� d S )Nr )r3 Z add_messageZcall_args_listr( r2 Z config_dir)r r4 Z call_listr r r �test_itE s z ReportNewAccountTest.test_itN) r, r- r. r/ r3 r r r5 r6 r r r r r0 9 s r0 c @ s e Zd ZdZdd� Zdd� ZdS )�AccountMemoryStorageTestz/Tests for certbot.account.AccountMemoryStorage.c C s ddl m} |� | _d S )Nr )�AccountMemoryStorage)r r8 �storage)r r8 r r r r O s z AccountMemoryStorageTest.setUpc C s� t jdd�}| �g | j�� � | �tj| jjd� | j� |d � | �|g| j�� � | �|| j�d�� | j� |d � | �|g| j�� � d S )N�x)r% ) r ZMockr! r9 �find_all� assertRaisesr �AccountNotFound�load�save)r �accountr r r r6 S s z AccountMemoryStorageTest.test_itN)r, r- r. r/ r r6 r r r r r7 L s r7 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�AccountFileStorageTestz-Tests for certbot.account.AccountFileStorage.c sl t t| ��� ddlm} || j�| _ddlm} d}|tj d t� � |d�t d�| _ t �� | _|| jj_d S )Nr ��AccountFileStorager �hi)ZuriZbody�new_authzr_uri)r r" )�superrA r r rC r2 r9 r r ZRegistrationResourceZ Registrationr r r r � mock_clientZ directoryZ new_authz)r rC r rE �� __class__r r r b s �� z AccountFileStorageTest.setUpc C s | � tj�t�| jj��� d S r )r( r �path�isdirr Z.underscores_for_unsupported_characters_in_pathr2 � accounts_dirr# r r r �test_init_creates_dirr s �z,AccountFileStorageTest.test_init_creates_dirc C s� | j �| j| j� tj�| jj| jj �}| � tj� |�� dD ] }| � tj� tj�||��� q>| � t � tj�|d�d�� | j �| jj �}| �| j|� d S )N)� regr.json� meta.json�private_key.jsonrP � )r9 r? r rG r rJ �joinr2 rL r% r( �existsr Z check_moder> r! )r � account_path� file_nameZloadedr r r �test_save_and_restorev s ��z,AccountFileStorageTest.test_save_and_restorec C s\ | j �| j| j� tj�| jj| jj d�}t |d��}t � |�}W 5 Q R X | � d|k� dS )z=Saved regr should include a new_authzr_uri for older CertbotsrN �rrE N)r9 r? r rG r rJ rR r2 rL r% �open�jsonr> r( )r rJ �fr r r r �!test_save_and_restore_old_version� s z8AccountFileStorageTest.test_save_and_restore_old_versionc C s� | j �| j| j� tj�| jj| jj �}| � tj� |�� | � tj� tj�|d��� dD ] }| � tj� tj�||��� qZd S )NrN )rO rP ) r9 Z save_regrr rG r rJ rR r2 rL r% r( rS � assertFalse)r rT rU r r r �test_save_regr� s � �z%AccountFileStorageTest.test_save_regrc C s, | j �| j| j� | �| jg| j �� � d S r )r9 r? r rG r! r; r# r r r � test_find_all� s z$AccountFileStorageTest.test_find_allc C s | � g | j�� � d S r )r! r9 r; r# r r r � test_find_all_none_empty_list� s z4AccountFileStorageTest.test_find_all_none_empty_listc C s$ t �| jj� | �g | j�� � d S r )r �rmdirr2 rL r! r9 r; r# r r r �!test_find_all_accounts_dir_absent� s z8AccountFileStorageTest.test_find_all_accounts_dir_absentc C sT t jdtjdgd�| j_t �d��(}dddg|_| �ddg| j� � � W 5 Q R X d S )Nr: �z�� side_effectzcertbot.account.os.listdir�y) r r r �AccountStorageErrorr9 �_load_for_server_pathr r r! r; )r Z mock_listdirr r r �test_find_all_load_skips� s � z/AccountFileStorageTest.test_find_all_load_skipsc C s | � tj| jjd� d S )NZmissing)r< r r= r9 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 rB )r2 �serverr rC r9 )r rj rC r r r � _set_server� s z"AccountFileStorageTest._set_serverc C sH | � d� | �g | j�� � | �g | j�� � | �tj�| jj �� d S �N�6https://acme-staging-v02.api.letsencrypt.org/directory) rk r! r9 r; r\ r rJ �islinkr2 rL r# r r r � test_find_all_neither_exists� s z3AccountFileStorageTest.test_find_all_neither_existsc C s� | � d� | �g | j�� � | j�| j| j� | �| jg| j�� � | �| jg| j�� � | �tj � | j j �� d}| �tj � | j �|��� d S �Nrm �2https://acme-staging.api.letsencrypt.org/directory)rk r! r9 r; r? r rG r\ r rJ rn r2 rL rK � accounts_dir_for_server_path�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� | �| jg| j�� � | �| jg| j�� � | �tj � | j j �� | � tj �| j j �� d}| �tj �| j �|��� d S rp )rk r9 r? r rG r! r; r\ r rJ rn r2 rL r( rK rr rs r r r � test_find_all_save_before_find� s z5AccountFileStorageTest.test_find_all_save_before_findc C sd | � d� | �g | j�� � | j�| j| j� | �| jg| j�� � | � d� | �g | j�� � d S rp )rk r! r9 r; r? r rG r# r r r � test_find_all_server_downgrade� s z5AccountFileStorageTest.test_find_all_server_downgradec C s@ | � d� | j�| j| j� | � d� | �| jg| j�� � d S �Nrq rm �rk r9 r? r rG r! r; r# r r r � test_upgrade_version_staging� s z3AccountFileStorageTest.test_upgrade_version_stagingc C s@ | � d� | j�| j| j� | � d� | �| jg| j�� � d S )Nz.https://acme-v01.api.letsencrypt.org/directoryz.https://acme-v02.api.letsencrypt.org/directoryrx r# r r r � test_upgrade_version_production� s z6AccountFileStorageTest.test_upgrade_version_productionzcertbot.compat.os.rmdirc C sT | � d� | j�| j| j� t|_tjt j d�| j_ | � d� | � g | j� � � d S )Nrq rc rm )rk r9 r? r rG �OSErrorrd r r r rf rg r! r; )r Z mock_rmdirr r r �test_corrupted_account� s � z-AccountFileStorageTest.test_corrupted_accountc C sV | � d� | j�| j| j� | j�| jj�}| � d� | j�| jj�}| �||� d S rw )rk r9 r? r rG r> r% r! �r Z prev_accountr@ r r r �test_upgrade_load� s z(AccountFileStorageTest.test_upgrade_loadc C sV | � d� | j�| j| j� | j�| jj�}| �d� | j�| jj�}| �||� d S rw ) rk r9 r? r rG r> r% � _set_server_and_stop_symlinkr! r} r r r � test_upgrade_load_single_account� s z7AccountFileStorageTest.test_upgrade_load_single_accountc C sT | j �| j| j� t�� }t|_t�d|�� | � t j | j j | jj � W 5 Q R X d S �Nzsix.moves.builtins.open)r9 r? r rG r � mock_open�IOErrorrd r r< r rf r> r% �r r� r r r �test_load_ioerror� s �z(AccountFileStorageTest.test_load_ioerrorc C sD t �� }t|_t �d|�� | �tj| jj | j | j � W 5 Q R X d S r� ) r r� r� rd r r< r rf r9 r? r rG r� r r r �test_save_ioerrors s �z)AccountFileStorageTest.test_save_ioerrorsc C s> | j �| j| j� | j �| jj� | �tj| j j | jj� d S r ) r9 r? r rG �deleter% r< r r= r> r# r r r � test_delete s z"AccountFileStorageTest.test_deletec C s | � tj| jj| jj� d S r )r< r r= r9 r� r r% r# r r r �test_delete_no_account s z-AccountFileStorageTest.test_delete_no_accountc C sZ | � d� | j�| j| j� t�di ��( | � d� | �tj | jj | jj � W 5 Q R X d S )Nrq z"certbot.constants.LE_REUSE_SERVERSrm ) rk r9 r? r rG r r r< r r= r> r% 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 j | jj| jj� | � d� | �t j | jj| jj� d S rw ) rk r9 r? r rG r> r% r� r< r r= )r Z server_urlr r r �_test_delete_folders s z+AccountFileStorageTest._test_delete_foldersc C s | � d� | �� d S )Nrq �r� r� r# r r r �test_delete_folders_up. s z-AccountFileStorageTest.test_delete_folders_upc C s | � d� | �� d S rl r� r# r r r �test_delete_folders_down2 s z/AccountFileStorageTest.test_delete_folders_downc C s< | � |� ttj�| jjd�d��}|�d� W 5 Q R X d S )NZfoo�wZbar)rk rX r rJ rR r2 rL �write)r Z server_pathrZ r r r r 6 s z3AccountFileStorageTest._set_server_and_stop_symlinkc C s | � d� | �d� d S rp �r r� r# r r r � test_delete_shared_account_up; s z4AccountFileStorageTest.test_delete_shared_account_upc C s | � d� | �d� d S rl r� r# r r r � test_delete_shared_account_down? s z6AccountFileStorageTest.test_delete_shared_account_down)&r, r- r. r/ r rM rV r[ r] r^ r_ ra rh ri rk ro rt ru rv ry rz r r r| r~ r� r� r� r� r� r� r� r� r� r r� r� � __classcell__r r rH r rA ^ sB rA �__main__) r/ r rY ZunittestZjosepyZjoser r Zacmer Zcertbot.tests.utilZtests�utilZ test_utilZcertbotr Zcertbot.compatr r r ZJWKRSAr> Z load_vectorr ZTestCaser ZConfigTestCaser0 r7 rA r, �mainr r r r �<module> s( $ g
Save File
Cancel