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ƒZedkre 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 Ú
¨ ó zXCredentialsConfigurationTest.test_valid_file_with_unsafe_permissions..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 Ú s& m*2