← Back
Editing: dns_common_test.cpython-38.pyc
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)/
Save File
Cancel