Your IP : 216.73.216.190
U
�]� � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
mZ ddl m
Z
ddl mZ ddlmZ G dd � d ejej�ZG d
d � d ej�ZG d d
� d
ej�ZG dd� dej�Zedkr�e�� dS )z%Tests for certbot.plugins.dns_common.� N)�errors)�util)�os)�
dns_common)�dns_test_commonc s� e Zd ZG dd� dej�ZG dd� de�Z� fdd�Zdd� Z d d
� Z
e � � d d � �Z
e � � d
d� �Ze � � dd� �Ze � � dd� �Zdd� Ze � � dd� �Z� ZS )�DNSAuthenticatorTestc @ s, e Zd Ze�� Ze�� Ze�� Zdd� ZdS )z*DNSAuthenticatorTest._FakeDNSAuthenticatorc C s dS )Nz!A fake authenticator for testing.� ��selfr r �A/usr/lib/python3/dist-packages/certbot/plugins/dns_common_test.py� more_info s z4DNSAuthenticatorTest._FakeDNSAuthenticator.more_infoN) �__name__�
__module__� __qualname__�mockZ MagicMockZ_setup_credentials�_perform�_cleanupr r r r r �_FakeDNSAuthenticator s r c @ s e Zd ZdZdZdZdZdS )z DNSAuthenticatorTest._FakeConfigr � N)r
r r Zfake_propagation_secondsZfake_config_key�fake_other_key�fake_file_pathr r r r � _FakeConfig s r c s, t t| ��� t�� | _t�| jd�| _d S )NZfake)�superr �setUpr �configr �authr �� __class__r r r # s
zDNSAuthenticatorTest.setUpc C s, | j �| jg� | j j�tjtjtj� d S �N) r Zperform�achallr �assert_called_once_withr �DOMAINr �ANYr r r r � test_perform* s z!DNSAuthenticatorTest.test_performc C s4 d| j _| j �| jg� | j j�tjtj tj � d S )NT)
r Z_attempt_cleanupZcleanupr r r r r! r r" r r r r � test_cleanup/ s z!DNSAuthenticatorTest.test_cleanupc C sB |� }t jdft jdff|j_| j�dd� | �| jjjd� d S )N� �value� other_key) � display_util�OK�input� side_effectr �
_configure� assertEqualr r �r
�mock_get_utility� mock_displayr r r � test_prompt6 s �z DNSAuthenticatorTest.test_promptc C s0 |� }t jdff|j_| �tj| jjdd� d S )N�cr' r% ) r( �CANCELr* r+ � assertRaisesr � PluginErrorr r, r. r r r �test_prompt_canceled? s z)DNSAuthenticatorTest.test_prompt_canceledc C sr t j�| jd�}t|d��� |� }tjdftjdftj| jftj|ff|j_ | j
� dd� | � | j
j
j|� d S )N�file.ini�wbr% �not-a-file.ini� file_path)r �path�join�tempdir�open�closer( r) �directory_selectr+ r �_configure_filer- r r )r
r/ r; r0 r r r �test_prompt_fileF s
�z%DNSAuthenticatorTest.test_prompt_filec C s0 |� }t jdff|j_| �tj| jjdd� d S )Nr2 r: r% ) r( r3 r@ r+ r4 r r5 r rA r. r r r �test_prompt_file_canceledT s z.DNSAuthenticatorTest.test_prompt_file_canceledc C sX t j�| jd�}t�ddi|� t| jd|� | j� ddddi�}| �
|� d�d� d S )Nr7 � fake_testr& �fake_credentials� credentialsr% �test) r r; r< r= r �write�setattrr r �_configure_credentialsr- �conf)r
r; rF r r r �test_configure_credentials[ s
z/DNSAuthenticatorTest.test_configure_credentialsc C s� t j�| jd�}t�ddi|� t j�| jd�}t�ddi|� t| jdd� |� }tj dftj d ftj | jftj |ftj |ff|j
_ | j �
d
dd di�}| �|�d �d� d S ) Nz bad-file.iniZ
fake_otherZ other_valuer7 rD r& rE r% r9 rF rG )r r; r< r= r rH rI r r( r) r@ r+ r rJ r- rK )r
r/ Zbad_pathr; r0 rF r r r �test_prompt_credentialsd s
�z,DNSAuthenticatorTest.test_prompt_credentials)r
r r r ZDNSAuthenticatorr �objectr r r# r$ � test_utilZpatch_get_utilityr1 r6 rB rC rL rM �
__classcell__r r r r r s
r c @ s6 e Zd ZG dd� dej�Zdd� Zdd� Zdd� Zd S )
� CredentialsConfigurationTestc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS )z0CredentialsConfigurationTest._MockLoggingHandlerNc O s | � � tjj| f|�|� d S r )�reset�logging�Handler�__init__)r
�args�kwargsr r r rU | s z9CredentialsConfigurationTest._MockLoggingHandler.__init__c C s | j |j�� �|�� � d S r )�messagesZ levelname�lower�appendZ
getMessage)r
�recordr r r �emit� s z5CredentialsConfigurationTest._MockLoggingHandler.emitc C s t �t�| _dS )z-Allows the handler to be reset between tests.N)� collections� defaultdict�listrX r r r r rR � s z6CredentialsConfigurationTest._MockLoggingHandler.reset)r
r r rX rU r\ rR r r r r �_MockLoggingHandlery s r` c C sT t j�| jd�}t�ddd�|� t�|�}| �d|� d�� | �d|� d�� d S )N�too-permissive-file.inir& r �rG �otherrG �1rc )
r r; r< r= r rH r �CredentialsConfigurationr- rK )r
r; �credentials_configurationr r r �test_valid_file� s
z,CredentialsConfigurationTest.test_valid_filec C s& t j�| jd�}| �tjtj|� d S )Nr9 ) r r; r< r= r4 r r5 r re )r
r; r r r �test_nonexistent_file� s z2CredentialsConfigurationTest.test_nonexistent_filec C sd | � � }tj�|� tj�| jd�}t� |dd��
� t� |� | � dt
dd� |jd D ��� d S )Nra r8 i� r c S s g | ]}|� d �r|�qS )ZUnsafe)�
startswith)�.0�_r r r �
<listcomp>� s
zXCredentialsConfigurationTest.test_valid_file_with_unsafe_permissions.<locals>.<listcomp>Zwarning)r` r ZloggerZ
addHandlerr r; r< r= r Z safe_openr? re r- �lenrX )r
�logr; r r r �'test_valid_file_with_unsafe_permissions� s
zDCredentialsConfigurationTest.test_valid_file_with_unsafe_permissionsN) r
r r rS rT r` rg rh ro r r r r rQ x s rQ c sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�#CredentialsConfigurationRequireTestc s$ t t| ��� tj�| jd�| _d S )Nr7 )r rp r r r; r< r= r r r r r � s z)CredentialsConfigurationRequireTest.setUpc C s t �|| j� d S r )r rH r; )r
�valuesr r r �_write� s z*CredentialsConfigurationRequireTest._writec C s0 | � ddd�� t�| j�}|�ddd�� d S )Nr& r rb r% �rr r re r; �require�r
rf r r r �
test_valid� s z.CredentialsConfigurationRequireTest.test_validc C s. | � ddd�� t�| j�}|�ddi� d S )Nr& r rb rG r% rs ru r r r �test_valid_but_extra� s z8CredentialsConfigurationRequireTest.test_valid_but_extrac C s$ | � i � t�| j�}|�i � d S r rs ru r r r �test_valid_empty� s
z4CredentialsConfigurationRequireTest.test_valid_emptyc C s0 | � i � t�| j�}| �tj|jddi� d S �NrG r% �rr r re r; r4 r r5 rt ru r r r � test_missing� s
z0CredentialsConfigurationRequireTest.test_missingc C s4 | � ddi� t�| j�}| �tj|jddi� d S ry rz ru r r r �
test_blank� s z.CredentialsConfigurationRequireTest.test_blankc C s4 | � ddi� t�| j�}| �tj|jddi� d S )NZtetsztypo!rG r% rz ru r r r � test_typo� s z-CredentialsConfigurationRequireTest.test_typo) r
r r r rr rv rw rx r{ r| r} rP r r r r rp � s rp c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�DomainNameGuessTestc C s | � dt�d�k� d S )N� example.com�Z
assertTruer Zbase_domain_name_guessesr r r r �test_simple_case� s
��z$DomainNameGuessTest.test_simple_casec C s | � dt�d�k� d S )Nr zfoo.bar.baz.example.comr� r r r r �test_sub_domain� s
��z#DomainNameGuessTest.test_sub_domainc C s | � dt�d�k� d S )Nz
example.co.ukzfoo.bar.baz.example.co.ukr� r r r r �test_second_level_domain� s
��z,DomainNameGuessTest.test_second_level_domainN)r
r r r� r� r� r r r r r~ � s r~ �__main__)�__doc__r] rS Zunittestr Zcertbotr r Zcertbot.compatr Zcertbot.displayr( Zcertbot.pluginsr r Z
certbot.testsrO ZTempDirTestCaseZBaseAuthenticatorTestr rQ rp ZTestCaser~ r
�mainr r r r �<module> s" f)/