/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/__pycache__
a }|�gJ! � @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddl m Z ddl m Z ddlmZ ddlmZ ddlm Z G d d � d ejej�ZG d d � d ej�ZG d d� dej�ZG dd� dej�Zedk�re�e� ej dd� e g �� dS )z%Tests for certbot.plugins.dns_common.� N)�mock)�errors)�util)�os)� dns_common)�dns_test_commonc s� e Zd ZG dd� dej�ZG dd� d�Z� fdd�Ze� � dd� �Z d d � Z e� � d d � �Z e� � d d� �Z e� � dd� �Ze� � dd� �Zdd� Ze� � dd� �Zdd� 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 �S/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/dns_common_test.py� more_info s z4DNSAuthenticatorTest._FakeDNSAuthenticator.more_infoN) �__name__� __module__� __qualname__r � 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�| jd�| _d S )NZfake)�super�setUpr r �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 �ANY)r Zunused_mock_get_utilityr 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_cleanup2 s z!DNSAuthenticatorTest.test_cleanupc C sB |� }t jdft jdff|j_| j�dd� | jjjdks>J �d S )N� �value� other_key)� display_util�OK�input� side_effectr � _configurer r �r �mock_get_utility� mock_displayr r r � test_prompt9 s �z DNSAuthenticatorTest.test_promptc C sT |� }t jdff|j_t�tj�� | j� dd� W d � n1 sF0 Y d S )N�cr( r& ) r) �CANCELr+ r, �pytest�raisesr � PluginErrorr r- r. r r r �test_prompt_canceledB s z)DNSAuthenticatorTest.test_prompt_canceledc C sr t j�| jd�}t|d��� |� }tjdftjdftj| jftj|ff|j_ | j � dd� | j j j |ksnJ �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< r0 r r r �test_prompt_fileJ s �z%DNSAuthenticatorTest.test_prompt_filec C sT |� }t jdff|j_t�tj�� | j� dd� W d � n1 sF0 Y d S )Nr2 r; r& ) r) r3 rA r, r4 r5 r r6 r rB r. r r r �test_prompt_file_canceledX s z.DNSAuthenticatorTest.test_prompt_file_canceledc C sX t j�| jd�}t�ddi|� t| jd|� | j� ddddi�}|� d�dksTJ �d S )Nr8 � fake_testr' �fake_credentials� credentialsr& �test) r r< r= r> r �write�setattrr r �_configure_credentials�conf)r r< rG 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 �dks�J �d S ) Nz bad-file.iniZ fake_otherZ other_valuer8 rE r' rF r& r: rG rH )r r< r= r> r rI rJ r r) r* rA r, r rK rL )r r/ Zbad_pathr< r0 rG r r r �test_prompt_credentialsi s �z,DNSAuthenticatorTest.test_prompt_credentialsc C s d| j �t�� g�v sJ �d S )Nz@try increasing --fake-propagation-seconds (currently 0 seconds).)r Z auth_hintr r r r r r �test_auth_hint| s �z#DNSAuthenticatorTest.test_auth_hint)r r r r ZDNSAuthenticatorr r r � test_utilZpatch_display_utilr$ r% r1 r7 rC rD rM rN rO � __classcell__r r r r r s$ r c @ s6 e Zd ZG dd� dej�Zdd� Zdd� Zdd� Zd S ) � CredentialsConfigurationTestc s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )z0CredentialsConfigurationTest._MockLoggingHandlerNc s | � � t� j|i |�� d S r )�resetr �__init__)r �args�kwargsr r r rT � 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�listrW r r r r rS � s z6CredentialsConfigurationTest._MockLoggingHandler.reset)r r r rW rT r[ rS rQ r r r r �_MockLoggingHandler� s r_ c C sT t j�| jd�}t�ddd�|� t�|�}d|�d�ks>J �d|�d�ksPJ �d S )N�too-permissive-file.inir' r �rH �otherrH �1rb ) r r< r= r> r rI r �CredentialsConfigurationrL )r r< �credentials_configurationr r r �test_valid_file� s z,CredentialsConfigurationTest.test_valid_filec C sJ t j�| jd�}t�tj�� t� |� W d � n1 s<0 Y d S )Nr: ) r r< r= r> r4 r5 r r6 r rd )r r< r r r �test_nonexistent_file� s z2CredentialsConfigurationTest.test_nonexistent_filec C sd | � � }tj�|� tj�| jd�}t� |dd�� � t� |� dt dd� |j d D ��ks`J �d S )Nr` r9 i� r c S s g | ]}|� d �r|�qS )ZUnsafe)� startswith)�.0�_r r r � <listcomp>� � zXCredentialsConfigurationTest.test_valid_file_with_unsafe_permissions.<locals>.<listcomp>Zwarning)r_ r �loggerZ addHandlerr r< r= r> r Z safe_openr@ rd �lenrW )r �logr< r r r �'test_valid_file_with_unsafe_permissions� s zDCredentialsConfigurationTest.test_valid_file_with_unsafe_permissionsN) r r r �loggingZHandlerr_ rf rg rp r r r r rR � s rR c sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z � Z S )�#CredentialsConfigurationRequireTestc s t � �� tj�| jd�| _d S )Nr8 )r r r r< r= r> r r r r r � s z)CredentialsConfigurationRequireTest.setUpc C s t �|| j� d S r )r rI r< )r �valuesr r r �_write� s z*CredentialsConfigurationRequireTest._writec C s0 | � ddd�� t�| j�}|�ddd�� d S )Nr' r ra r& �rt r rd r< �require�r re r r r � test_valid� s z.CredentialsConfigurationRequireTest.test_validc C s. | � ddd�� t�| j�}|�ddi� d S )Nr' r ra rH r& ru rw r r r �test_valid_but_extra� s z8CredentialsConfigurationRequireTest.test_valid_but_extrac C s$ | � i � t�| j�}|�i � d S r ru rw r r r �test_valid_empty� s z4CredentialsConfigurationRequireTest.test_valid_emptyc C sT | � i � t�| j�}t�tj�� |�ddi� W d � n1 sF0 Y d S �NrH r& � rt r rd r< r4 r5 r r6 rv rw r r r � test_missing� s z0CredentialsConfigurationRequireTest.test_missingc C sX | � ddi� t�| j�}t�tj�� |�ddi� W d � n1 sJ0 Y d S r{ r| rw r r r � test_blank� s z.CredentialsConfigurationRequireTest.test_blankc C sX | � ddi� t�| j�}t�tj�� |�ddi� W d � n1 sJ0 Y d S )NZtetsztypo!rH r& r| rw r r r � test_typo� s z-CredentialsConfigurationRequireTest.test_typo) r r r r rt rx ry rz r} r~ r rQ r r r r rr � s rr c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�DomainNameGuessTestc C s dt �d�v sJ �d S )N� example.com�r Zbase_domain_name_guessesr r r r �test_simple_case� s �z$DomainNameGuessTest.test_simple_casec C s dt �d�v sJ �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�v sJ �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__r ) �__doc__r\ rq �sysZunittestr r4 Zcertbotr r Zcertbot.compatr Zcertbot.displayr) Zcertbot.pluginsr r Z certbot.testsrP ZTempDirTestCaseZBaseAuthenticatorTestr rR rr ZTestCaser� r �exit�main�argv�__file__r r r r �<module> s& m*2
.
Edit
..
Edit
__init__.cpython-39.opt-1.pyc
Edit
__init__.cpython-39.pyc
Edit
common_test.cpython-39.opt-1.pyc
Edit
common_test.cpython-39.pyc
Edit
disco_test.cpython-39.opt-1.pyc
Edit
disco_test.cpython-39.pyc
Edit
dns_common_test.cpython-39.opt-1.pyc
Edit
dns_common_test.cpython-39.pyc
Edit
enhancements_test.cpython-39.opt-1.pyc
Edit
enhancements_test.cpython-39.pyc
Edit
manual_test.cpython-39.opt-1.pyc
Edit
manual_test.cpython-39.pyc
Edit
null_test.cpython-39.opt-1.pyc
Edit
null_test.cpython-39.pyc
Edit
selection_test.cpython-39.opt-1.pyc
Edit
selection_test.cpython-39.pyc
Edit
standalone_test.cpython-39.opt-1.pyc
Edit
standalone_test.cpython-39.pyc
Edit
storage_test.cpython-39.opt-1.pyc
Edit
storage_test.cpython-39.pyc
Edit
util_test.cpython-39.opt-1.pyc
Edit
util_test.cpython-39.pyc
Edit
webroot_test.cpython-39.opt-1.pyc
Edit
webroot_test.cpython-39.pyc
Edit