Your IP : 216.73.216.190
U
�]�T � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
m Z ddl
mZ ddlmZ ddlmZ G dd� dej�ZG d d
� d
ej�ZG d d � d e j�ZG d
d� de j�ZG dd� de j�ZG dd� de j�ZzeZW n$ e k
�r ddl Z e j ZY nX G dd� de j�Z G dd� de j�Z G dd� dej�Z!G dd� dej�Z"G dd � d ej�Z#G d d � d ej�Z$G d d � d ej�Z%G d!d"� d"ej�Z&G d#d$� d$ej�Z'e(d%k�r�e�)� dS )&zTests for certbot.util.� N)�
reload_module)�errors)�os)�
filesystemc @ sR e Zd ZdZedd� �Ze�d�dd� �Ze�d�dd� �Z e�d�d d
� �Z
d S ) �
RunScriptTestz"Tests for certbot.util.run_script.c C s ddl m} ||�S )Nr )�
run_script)� certbot.utilr )�clsZparamsr � r
�9/usr/lib/python3/dist-packages/certbot/tests/util_test.py�_call s zRunScriptTest._call� certbot.util.subprocess.Popenc C s> d|� _ d|� j_| �dg�\}}| �|d� | �|d� dS )z.These will be changed soon enough with reload.r )�stdout�stderr�testr r N)�
returncode� communicate� return_valuer � assertEqual)�self�
mock_popen�out�errr
r
r � test_default s
zRunScriptTest.test_defaultc C s t |_| �tj| jdg� d S )Nr )�OSError� side_effect� assertRaisesr �SubprocessErrorr �r r r
r
r �test_bad_process" s z RunScriptTest.test_bad_processc C s* d|� j _d|� _| �tj| jdg� d S )N)� r � r )r r r r r r r r r
r
r � test_failure( s
zRunScriptTest.test_failureN) �__name__�
__module__� __qualname__�__doc__� classmethodr �mock�patchr r r" r
r
r
r r s
r c @ s, e Zd ZdZedd� �Zdd� Zdd� ZdS ) �
ExeExistsTestz"Tests for certbot.util.exe_exists.c C s ddl m} ||�S )Nr )�
exe_exists)r r+ )r �exer+ r
r
r r 3 s zExeExistsTest._callc C s. t jddd�� | �| �d�� W 5 Q R X d S )N�%certbot.util.filesystem.is_executableT�r � /path/to/exe)r( r) �
assertTruer �r r
r
r �test_exe_exists8 s z ExeExistsTest.test_exe_existsc C s. t jddd�� | �| �d�� W 5 Q R X d S )Nr- Fr. r/ )r( r) � assertFalser r1 r
r
r �test_exe_not_exists< s z!ExeExistsTest.test_exe_not_existsN)r# r$ r% r& r' r r2 r4 r
r
r
r r* 0 s
r* c sH e Zd ZdZedd� �Z� fdd�Ze�d�e�d�dd � ��Z � Z
S )
�LockDirUntilExitz+Tests for certbot.util.lock_dir_until_exit.c O s ddl m} |||�S )Nr )�lock_dir_until_exit)r r6 )r �args�kwargsr6 r
r
r r C s zLockDirUntilExit._callc s$ t t| ��� dd l}t|j� d S �Nr )�superr5 �setUpr r �util)r �certbot�� __class__r
r r; H s zLockDirUntilExit.setUpzcertbot.util.loggerz certbot.util.atexit_registerc C s� t j�| jd�}t�|� | �| j� | �|� | �|� | �|jd� |j d d }ddl
m } | � t
|j�dk� |� | �|jjd� d S )N�subdirr! r )r< � )r �path�join�tempdirr �mkdirr r �
call_count� call_argsr= r< r0 �lenZ_LOCKS�debug)r Z
mock_registerZ mock_loggerr@ Zregistered_funcr< r
r
r �test_itN s
zLockDirUntilExit.test_it) r# r$ r% r& r' r r; r( r) rJ �
__classcell__r
r
r> r r5 A s
r5 c @ s< e Zd ZdZdd� Ze�d�dd� �Ze�d�dd � �Zd
S ) �SetUpCoreDirTestz/Tests for certbot.util.make_or_verify_core_dir.c O s ddl m} |||�S )Nr )�set_up_core_dir)r rM )r r7 r8 rM r
r
r r g s zSetUpCoreDirTest._callz certbot.util.lock_dir_until_exitc C sB t j�| jd�}| �|dd� | �t j�|�� | �|jd� d S )N�new� Fr! ) r rB rC rD r r0 �existsr rF )r Z mock_lockZnew_dirr
r
r � test_successk s z SetUpCoreDirTest.test_successz certbot.util.make_or_verify_dirc C s" t |_| �tj| j| jdd� d S )NrO F)r r r r �Errorr rD )r Zmock_make_or_verifyr
r
r r" r s z SetUpCoreDirTest.test_failureN) r# r$ r% r& r r( r) rQ r" r
r
r
r rL d s
rL c sH e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z � Z
S )�MakeOrVerifyDirTestz�Tests for certbot.util.make_or_verify_dir.
Note that it is not possible to test for a wrong directory owner,
as this testing script would have to be run as root.
c s2 t t| ��� tj�| jd�| _t�| jd� d S )N�foo� ) r: rS r; r rB rC rD r rE r1 r> r
r r; � s zMakeOrVerifyDirTest.setUpc C s ddl m} |||dd�S )Nr )�make_or_verify_dirT)�strict)r rV )r Z directory�moderV r
r
r r � s zMakeOrVerifyDirTest._callc C sD t j�| jd�}| �|d� | �t j�|�� | �t�|d�� d S )N�bari� ) r rB rC rD r r0 �isdirr �
check_mode)r rB r
r
r � test_creates_dir_when_missing� s z1MakeOrVerifyDirTest.test_creates_dir_when_missingc C s& | � | jd� | �t�| jd�� d S )NrU )r rB r0 r r[ r1 r
r
r �(test_existing_correct_mode_does_not_fail� s z<MakeOrVerifyDirTest.test_existing_correct_mode_does_not_failc C s | � tj| j| jd� d S )N� )r r rR r rB r1 r
r
r � test_existing_wrong_mode_fails� s z2MakeOrVerifyDirTest.test_existing_wrong_mode_failsc C s8 t j�td�� }t� |_| �t| jdd� W 5 Q R X d S )N�makedirsrY i� )r( r) �objectr r r r r )r r` r
r
r �test_reraises_os_error� s z*MakeOrVerifyDirTest.test_reraises_os_error) r# r$ r% r& r; r r\ r] r_ rb rK r
r
r> r rS x s rS c sB e Zd ZdZ� fdd�Zd
dd�Zdd� Zd d
� Zd d � Z� Z S )�UniqueFileTestz#Tests for certbot.util.unique_file.c s$ t t| ��� tj�| jd�| _d S �Nzfoo.txt)r: rc r; r rB rC rD � default_namer1 r> r
r r; � s zUniqueFileTest.setUprU c C s ddl m} || j|�S )Nr )� unique_file)r rf re )r rX rf r
r
r r � s zUniqueFileTest._callc C sF | � � \}}|�d� |�� t|��}| �|�� d� W 5 Q R X d S )NrY )r �write�close�openr �read)r �fd�name�fr
r
r �test_returns_fd_for_writing� s
z*UniqueFileTest.test_returns_fd_for_writingc C sT | � d�\}}| � d�\}}| �t�|d�� | �t�|d�� |�� |�� d S )NrO rU )r r0 r r[ rh )r �fd1�name1�fd2�name2r
r
r �test_right_mode� s z UniqueFileTest.test_right_modec
C s� | � � \}}| � � \}}| � � \}}| �||� | �||� | �||� | �tj�|�| j� | �tj�|�| j� | �tj�|�| j� tj�|�}| �|� d�� tj�|�}| �|� d�� tj�|�} | �| � d�� |�
� |�
� |�
� d S rd ) r ZassertNotEqualr r rB �dirnamerD �basenamer0 �endswithrh )
r ro rp rq rr Zfd3Zname3Z basename1Z basename2Z basename3r
r
r �test_default_exists� s$ z"UniqueFileTest.test_default_exists)rU )
r# r$ r% r& r; r rn rs rw rK r
r
r> r rc � s
rc c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� Zd S )
�UniqueLineageNameTestz+Tests for certbot.util.unique_lineage_name.� c C s ddl m} || j||�S )Nr )�unique_lineage_name)r rz rD )r �filenamerX rz r
r
r r � s zUniqueLineageNameTest._callc C sB | � d�\}}| �t|t�� | �tj�| jd�|� |� � d S )N�wowzwow.conf)
r r0 �
isinstance� file_typer r rB rC rD rh )r rm rB r
r
r �
test_basic� s z UniqueLineageNameTest.test_basicc C s| g }t j�d�D ]}|�| �d�� q|d \}}| �t|t�� | �t|t j�� | �d|k� |D ]\}}|� � qfd S )N�
r| ���z
wow-0009.conf)
�sixZmoves�range�appendr r0 r} r~ Z string_typesrh )r �items�_rm rl r
r
r �
test_multiple� s z#UniqueLineageNameTest.test_multiplec C s4 t jdttj�d�� | �t| jd� W 5 Q R X d S )Nz certbot.compat.filesystem.open)r r| )r( r) r �errnoZEIOr r r1 r
r
r r" � s z"UniqueLineageNameTest.test_failureN)ry )r# r$ r% r& r r r� r" r
r
r
r rx � s
rx c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Z� Z S ) �SafelyRemoveTestz%Tests for certbot.util.safely_remove.c s$ t t| ��� tj�| jd�| _d S )NrT )r: r� r; r rB rC rD r1 r> r
r r; � s zSafelyRemoveTest.setUpc C s ddl m} || j�S )Nr )�
safely_remove)r r� rB )r r� r
r
r r � s zSafelyRemoveTest._callc C s8 t | jd�� W 5 Q R X | �� | �tj�| j�� d S )N�w)ri rB r r3 r rP r1 r
r
r � test_exists s
z SafelyRemoveTest.test_existsc C s | � � | �tj�| j�� d S �N)r r3 r rB rP r1 r
r
r � test_missing s z SafelyRemoveTest.test_missingc C s. t �d��}t|_| �t| j� W 5 Q R X d S )Nzcertbot.util.os.remove)r( r) r r r r )r Z mock_remover
r
r � test_other_error_passthrough s z-SafelyRemoveTest.test_other_error_passthrough)
r# r$ r% r& r; r r� r� r� rK r
r
r> r r� � s r� c @ s, e Zd ZdZedd� �Zdd� Zdd� ZdS ) �
SafeEmailTestzTest safe_email.c C s ddl m} ||�S )Nr )�
safe_email)r r� )r �addrr� r
r
r r s zSafeEmailTest._callc C s. dddg}|D ]}| � | �|�d| � qd S )Nzcertbot@certbot.orgztbd.ade@gmail.comzabc_def.jdk@hotmail.museum�
%s failed.)r0 r �r Zaddrsr� r
r
r �test_valid_emails s �z SafeEmailTest.test_valid_emailsc C s. dddg}|D ]}| � | �|�d| � qd S )Nzcertbot@certbot..orgz.tbd.ade@gmail.comz ~/abc_def.jdk@hotmail.museumr� )r3 r r� r
r
r �test_invalid_emails" s �z!SafeEmailTest.test_invalid_emailsN)r# r$ r% r& r' r r� r� r
r
r
r r� s
r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�AddDeprecatedArgumentTestz Test add_deprecated_argument.c C s t �� | _d S r� )�argparse�ArgumentParser�parserr1 r
r
r r; . s z AddDeprecatedArgumentTest.setUpc C s ddl m} || jj||� d S )Nr )�add_deprecated_argument)r r� r� � add_argument)r Z
argument_name�nargsr� r
r
r r 1 s z AddDeprecatedArgumentTest._callc C sp | � dd� t�d��}| j�dg� W 5 Q R X | �|jd� | �d|jd d k� | �d|jd d � d S )N� --old-optionr �certbot.util.logger.warningr! �
is deprecated� r r( r) r� �
parse_argsr rF r0 rG �r � mock_warnr
r
r �test_warning_no_arg5 s z-AddDeprecatedArgumentTest.test_warning_no_argc C sr | � dd� t�d��}| j�ddg� W 5 Q R X | �|jd� | �d|jd d k� | �d|jd d � d S )Nr� r! r� �42r� r r� r� r
r
r �test_warning_with_arg= s z/AddDeprecatedArgumentTest.test_warning_with_argc C sl | � dd� t�� }tjd|d��. z| j�dg� W n tk
rJ Y nX W 5 Q R X | �d|� � k� d S )Nr� rA z
sys.stdout)rN z-h)
r r� �StringIOr( r) r� r� �
SystemExitr0 �getvalue)r r r
r
r � test_helpE s z#AddDeprecatedArgumentTest.test_helpc C s | � t� dS )z�Test when ACTION_TYPES_THAT_DONT_NEED_A_VALUE is a set.
This variable is a set in configargparse versions < 0.12.0.
N)�_test_constant_common�setr1 r
r
r �test_set_constantO s z+AddDeprecatedArgumentTest.test_set_constantc C s | � t� dS )z�Test when ACTION_TYPES_THAT_DONT_NEED_A_VALUE is a tuple.
This variable is a tuple in configargparse versions >= 0.12.0.
N)r� �tupler1 r
r
r �test_tuple_constantW s z-AddDeprecatedArgumentTest.test_tuple_constantc C sL t �d��&}|� |_| �dd� | �dd� W 5 Q R X | �t|j�d� d S )Nzcertbot.util.configargparser� r! z
--old-option2rA )r( r) Z#ACTION_TYPES_THAT_DONT_NEED_A_VALUEr r rH )r �typZmock_configargparser
r
r r� _ s �z/AddDeprecatedArgumentTest._test_constant_commonN) r# r$ r% r& r; r r� r� r� r� r� r� r
r
r
r r� , s
r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�EnforceLeValidityzTest enforce_le_validity.c C s ddl m} ||�S )Nr )�enforce_le_validity)r r� )r �domainr� r
r
r r j s zEnforceLeValidity._callc C s | � tj| jd� d S )Nz..�r r �ConfigurationErrorr r1 r
r
r � test_sanityn s z EnforceLeValidity.test_sanityc C s | � tj| jd� d S )Nzhello_world.example.comr� r1 r
r
r �test_invalid_charsq s
�z$EnforceLeValidity.test_invalid_charsc C s | � tj| jd� d S )Nz-a.example.comr� r1 r
r
r �test_leading_hyphenu s
�z%EnforceLeValidity.test_leading_hyphenc C s | � tj| jd� d S )Nza-.example.comr� r1 r
r
r �test_trailing_hypheny s
�z&EnforceLeValidity.test_trailing_hyphenc C s | � tj| jd� d S )NZcomr� r1 r
r
r �test_one_label} s z EnforceLeValidity.test_one_labelc C s | � | �d�d� d S )Nz example.com�r r r1 r
r
r �test_valid_domain� s z#EnforceLeValidity.test_valid_domainc C s( | � tj| jd� | � tj| jd� d S )Nzhttp://example.comzhttps://example.comr� r1 r
r
r �test_input_with_scheme� s z(EnforceLeValidity.test_input_with_schemec C s | � | �d�d� d S )Nzhttp.example.comr� r1 r
r
r �!test_valid_input_with_scheme_name� s z3EnforceLeValidity.test_valid_input_with_scheme_nameN)
r# r$ r% r& r r� r� r� r� r� r� r� r� r
r
r
r r� h s r� c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�EnforceDomainSanityTestzTest enforce_domain_sanity.c C s ddl m} ||�S )Nr )�enforce_domain_sanity)r r� )r r� r� r
r
r r � s z EnforceDomainSanityTest._callc C s | � tj| jd�d�� d S )N� eichhörnchen.example.comzutf-8)r r r� r �encoder1 r
r
r �test_nonascii_str� s �z)EnforceDomainSanityTest.test_nonascii_strc C s | � tj| jd� d S )Nr� r� r1 r
r
r �test_nonascii_unicode� s �z-EnforceDomainSanityTest.test_nonascii_unicodec C s d}| � tj| j|� d S )NA aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar� )r Z long_domainr
r
r �
test_too_long� s �z%EnforceDomainSanityTest.test_too_longc C s d� dddd�}| �|� d S )Nz{0}.{1}.{2}.{3}�?aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�?bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbZ?cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccZ?ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd��formatr )r Znot_too_long_domainr
r
r �test_not_too_long� s z)EnforceDomainSanityTest.test_not_too_longc C s d}| � tj| j|� d S )Nzfizz..example.comr� )r Zempty_label_domainr
r
r �test_empty_label� s �z(EnforceDomainSanityTest.test_empty_labelc C s d}| � tj| j|� d S )Nz
example.com..r� )r Zempty_trailing_label_domainr
r
r �test_empty_trailing_label� s �z1EnforceDomainSanityTest.test_empty_trailing_labelc C s d}| � tj| j|� d S )N�@aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar� �r Zlong_label_domainr
r
r �test_long_label_1� s �z)EnforceDomainSanityTest.test_long_label_1c C s" d� dd�}| �tj| j|� d S )N� {0}.{1}.comr� r� )r� r r r� r r� r
r
r �test_long_label_2� s �z)EnforceDomainSanityTest.test_long_label_2c C s d� dd�}| �|� d S )Nr� r� r� r� )r Znot_too_long_label_domainr
r
r �test_not_long_label� s z+EnforceDomainSanityTest.test_not_long_labelc C s d}| � tj| j|� d S )Nr r� )r Z empty_domainr
r
r �test_empty_domain� s �z)EnforceDomainSanityTest.test_empty_domainc C s | � d� d S )Nzthis.is.xn--ls8h.tld)r r1 r
r
r �test_punycode_ok� s z(EnforceDomainSanityTest.test_punycode_okN)r# r$ r% r& r r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �IsWildcardDomainTestz Tests for is_wildcard_domain.c C s d| _ d| _d S )Nz
*.example.orgz example.org)�wildcard� no_wildcardr1 r
r
r r; � s zIsWildcardDomainTest.setUpc C s ddl m} ||�S )Nr )�is_wildcard_domain)r r� )r r� r� r
r
r r � s zIsWildcardDomainTest._callc C s, | � | �| j�� | � | �| j�� �� d S r� )r3 r r� r� r1 r
r
r �test_no_wildcard� s z%IsWildcardDomainTest.test_no_wildcardc C s, | � | �| j�� | � | �| j�� �� d S r� )r0 r r� r� r1 r
r
r �
test_wildcard� s z"IsWildcardDomainTest.test_wildcardN)r# r$ r% r& r; r r� r� r
r
r
r r� � s
r� c @ s e Zd ZdZe�d�e�ej �
d�d�dd� ��Z e�d�e�ej �
d�d�dd� ��Z e�d�e�ej �
d�d�d d
� ��Z
e�d �e�d�e�ej �
d�d�d d
� ���Ze�d�dd� �Ze�d�e�ej �
d�d�dd� ��Ze�d�e�ej �
d�d�dd� ��ZdS )�
OsInfoTestz Test OS / distribution detectionzcertbot.util.distro�linuxzrequires Linuxc C s> dd l m} d|j_|�� }| �t|�d� | �d|k� d S )Nr zfirst debian third� Zdebian)r r< Zliker Zget_systemd_os_liker rH r0 )r �m_distro�cbutilZid_likesr
r
r � test_systemd_os_release_like� s
z'OsInfoTest.test_systemd_os_release_likec C s| dd l m} tjddd��@ d|j_d|j_|jdd� | �|� � d �
|jdd��� W 5 Q R X d
|j_| �|� � d
� d S ) Nr �platform.system_alias)r� r� r� r. r )� somethingz1.0ZcodenameT)Zpretty� Zwhatever) r r< r( r) rl r �linux_distribution�get_python_os_infor Zget_os_info_uarC �r r� r� r
r
r �test_get_os_info_ua� s �
�z OsInfoTest.test_get_os_info_uac C s\ dd l m} t�d��<}d|j_d|_| �|�� d� d|j_| �|�� d� W 5 Q R X d S )Nr zplatform.system)rl �version�xr� )rl r� )r� �else)r r< r( r) r� r r Z get_os_info)r r� r� Z
mock_platformr
r
r �test_get_os_info� s zOsInfoTest.test_get_os_infoz
warnings.warnc C s$ dd l m} |�� | �|j� d S r9 )r r< Zget_systemd_os_infor0 �called)r r� r� r� r
r
r �$test_get_systemd_os_info_deprecation s z/OsInfoTest.test_get_systemd_os_info_deprecationr
c C s* dd l m} t�dd��� tjddd�� | �|�� d d� W 5 Q R X tjddd��P t�� }d d
i}|jf |� ||_| �|�� d d � | �|�� d d
� W 5 Q R X tjddd�� | �|�� d� W 5 Q R X tjddd��0 tjddd�� | �|�� d� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr zcertbot.util._USE_DISTROFr� )Z
NonSystemDr� r� r. Z
nonsystemd)�darwinr r zcommunicate.return_value)�42.42.42�errorr� r! r� )�freebsdz
9.3-RC3-p1r )r� �9)�windowsr r zplatform.win32_ver)Z4242�95�2r )r� r� ) r r< r( r) r r� ZMockZconfigure_mockr )r Z
popen_mockr� Z comm_mockZ
comm_attrsr
r
r �test_non_systemd_os_info s< � �� ���
�z#OsInfoTest.test_non_systemd_os_infoc C s, dd l m} d|j_| �|�� d d� d S )Nr )r r r r� �r r< r� r r r� r� r
r
r � test_python_os_info_notfound& s z'OsInfoTest.test_python_os_info_notfoundc C s( dd l m} d|j_| �|�� d� d S )Nr )�testdistr� r )r� r� r� r� r
r
r �test_python_os_info_custom- s z%OsInfoTest.test_python_os_info_customN)r# r$ r% r& r( r) �unittestZ
skipUnless�sys�platform�
startswithr� r� r� r� r� r� r� r
r
r
r r� � s,
r� c @ s< e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
d � Z d S )
�AtexitRegisterTestz'Tests for certbot.util.atexit_register.c C s t �� | _d| _ddi| _d S )N)�hiZanswer�* )r( Z MagicMock�funcr7 r8 r1 r
r
r r; 7 s
zAtexitRegisterTest.setUpc O s ddl m} |||�S )Nr )�atexit_register)r r )r r7 r8 r r
r
r r < s zAtexitRegisterTest._callc C s$ | � t�� � | jj| j| j� d S r� )� _test_commonr �getpidr Zassert_called_withr7 r8 r1 r
r
r � test_calledA s z AtexitRegisterTest.test_calledc C s | j dd� | �| jj� d S )Nr� )� initial_pid)r r3 r r� r1 r
r
r �test_not_calledE s z"AtexitRegisterTest.test_not_calledc
C s~ t �d|��h t �d�� }| j| jf| j�| j� W 5 Q R X | �|jj� |jj \}}|d }||dd � |� W 5 Q R X d S )Nzcertbot.util._INITIAL_PIDzcertbot.util.atexitr r! )
r( r) r r r7 r8 r0 �registerr� rG )r r Z mock_atexitr7 r8 Z atexit_funcr
r
r r I s " z AtexitRegisterTest._test_commonN)
r# r$ r% r& r; r' r r r r r
r
r
r r 5 s
r �__main__)*r& r� r� r� r� r( r� Z six.movesr Zcertbot.tests.utilZtestsr< Z test_utilr= r Zcertbot.compatr r ZTestCaser r* ZTempDirTestCaser5 rL rS rc �filer~ � NameError�io�
TextIOWrapperrx r� r� r� r� r� r� r� r r# �mainr
r
r
r �<module> sB #%4 <#;[