3
گax
@ s0 d Z ddlmZ ddlmZ ddlmZ ddlZddlm Z ddlm
Z
ddl m Z dd l
mZ dd
lmZ dd lmZ dd lmZ dd
lmZ y ddlZW n ek
r ddlmZ Y nX erddlm Z ne Z dZ ej j ej!dZ"G dd dee Z#G dd de Z$G dd dej%Z&G dd dZ'dS )z8Base test class for DNS authenticators built on Lexicon. )Any)
TYPE_CHECKING) MagicMockN) HTTPError)RequestException) Challenge)errors)dns_test_common)
LexiconClient) _AuthenticatorCallableTestCase)util)mock)Protocolz example.comzrsa512_key.pemc @ s e Zd ZU dZeedS )%_AuthenticatorCallableLexiconTestCasezq
Protocol describing a TestCase suitable to test challenges against
a mocked LexiconClient instance.
N)__name__
__module__ __qualname____doc__r mock_clientr achall r r -/usr/lib/python3.6/dns_test_common_lexicon.pyr s
r c @ sF e Zd ZU dZeee e
e e
e e e eddddZdS )_LexiconAwareTestCasezX
Protocol describing a TestCase suitable to test a real LexiconClient instance.
N) unused_argsreturnc G s dS )zl
See
https://docs.python.org/3/library/unittest.html#unittest.TestCase.assertRaises
Nr )selfr r r r assertRaises7 s z"_LexiconAwareTestCase.assertRaises)r r r r r
clientr
provider_mockstr
record_prefix record_namerecord_content ExceptionDOMAIN_NOT_FOUND
GENERIC_ERROR LOGIN_ERRORUNKNOWN_LOGIN_ERRORr r r r r r r ' s
r c @ s6 e Zd Zej eeddddZeddddZdS ) BaseLexiconAuthenticatorTestN)r unused_mock_get_utilityr c C s< | j j| jg tjjtdt tjg}| j|| j j
d S )Nz_acme-challenge.) authZperformr r
calladd_txt_recordDOMAINANY assertEqualr
mock_calls)r r) expectedr r r test_performD s z)BaseLexiconAuthenticatorTest.test_perform)r r c C sD d| j _| j j| jg tjjtdt tjg}| j || j
j d S )NTz_acme-challenge.) r* Z_attempt_cleanupZcleanupr r
r+ del_txt_recordr- r. r/ r r0 )r r1 r r r test_cleanupL s z)BaseLexiconAuthenticatorTest.test_cleanup) r r r test_utilZpatch_display_utilr r r2 r4 r r r r r( B s r( c @ s e Zd ZedZeZedZedZ dZ
e
d e Z dZ
eddd d
Zeddd d Zeddd
dZeddddZeddddZeddddZeddddZeddddZeddddZedddd Zeddd d Zeddd d Zeddd!d"ZdS )#BaseLexiconClientTestzNo domain foundz400 Client Error: ...z500 Surprise! Error: ...z_acme-challenge.ZbarN)r r c C s0 | j jt| j| j | jjjd| j| jd d S )NTXT)rtypenamecontent)r r, r- r! r" r
create_recordassert_called_with)r r r r test_add_txt_record^ s
z)BaseLexiconClientTest.test_add_txt_recordc C s@ | j dg| jj_| jjt| j| j | jj j
d| j| jd d S )N r8 )r9 r: r; ) r$ r authenticate side_effectr r, r- r! r" r< r= )r r r r ,test_add_txt_record_try_twice_to_find_domaine s
zBBaseLexiconClientTest.test_add_txt_record_try_twice_to_find_domainc C s6 | j | j | j g| jj_| jtj| jjt | j
| j d S )N) r$ r r@ rA r r PluginErrorr r, r- r! r" )r r r r 'test_add_txt_record_fail_to_find_domainn s z=BaseLexiconClientTest.test_add_txt_record_fail_to_find_domainc C s, | j | jj_| jtj| jjt | j
| j d S )N) r& r r@ rA r r rC r r, r- r! r" )r r r r (test_add_txt_record_fail_to_authenticatew s z>BaseLexiconClientTest.test_add_txt_record_fail_to_authenticatec C s, | j | jj_| jtj| jjt | j
| j d S )N) r' r r@ rA r r rC r r, r- r! r" )r r r r ;test_add_txt_record_fail_to_authenticate_with_unknown_error~ s zQBaseLexiconClientTest.test_add_txt_record_fail_to_authenticate_with_unknown_errorc C s, | j | jj_| jtj| jjt | j
| j d S )N) r% r r@ rA r r rC r r, r- r! r" )r r r r (test_add_txt_record_error_finding_domain s z>BaseLexiconClientTest.test_add_txt_record_error_finding_domainc C s, | j | jj_| jtj| jjt | j
| j d S )N) r% r r< rA r r rC r r, r- r! r" )r r r r 'test_add_txt_record_error_adding_record s z=BaseLexiconClientTest.test_add_txt_record_error_adding_recordc C s0 | j jt| j| j | jjjd| j| jd d S )Nr8 )r9 r: r; )r r3 r- r! r" r
delete_recordr= )r r r r test_del_txt_record s
z)BaseLexiconClientTest.test_del_txt_recordc C s. | j | j | j g| jj_| jjt| j| j d S )N) r$ r r@ rA r r3 r- r! r" )r r r r 'test_del_txt_record_fail_to_find_domain s z=BaseLexiconClientTest.test_del_txt_record_fail_to_find_domainc C s$ | j | jj_| jjt| j| j d S )N) r& r r@ rA r r3 r- r! r" )r r r r (test_del_txt_record_fail_to_authenticate s z>BaseLexiconClientTest.test_del_txt_record_fail_to_authenticatec C s$ | j | jj_| jjt| j| j d S )N) r' r r@ rA r r3 r- r! r" )r r r r ;test_del_txt_record_fail_to_authenticate_with_unknown_error s zQBaseLexiconClientTest.test_del_txt_record_fail_to_authenticate_with_unknown_errorc C s$ | j | jj_| jjt| j| j d S )N) r% r r@ rA r r3 r- r! r" )r r r r (test_del_txt_record_error_finding_domain s z>BaseLexiconClientTest.test_del_txt_record_error_finding_domainc C s$ | j | jj_| jjt| j| j d S )N) r% r rI rA r r3 r- r! r" )r r r r )test_del_txt_record_error_deleting_record s z?BaseLexiconClientTest.test_del_txt_record_error_deleting_record) r r r r# r$ r r% r r& r' r r- r! r" r r> rB rD rE rF rG rH rJ rK rL rM rN rO r r r r r6 T s( r6 )(r Ztypingr r Z
unittest.mockr ZjosepyZjoseZrequests.exceptionsr r Zacme.challengesr Zcertbotr Zcertbot.pluginsr Z"certbot.plugins.dns_common_lexiconr
Z certbot.plugins.dns_test_commonr Z
certbot.testsr r5 r
ImportErrorZunittestZtyping_extensionsr objectr- ZJWKRSAloadZ load_vectorZKEYr r ZBaseAuthenticatorTestr( r6 r r r r s2