Edit File: dns_test_common_lexicon.cpython-36.pyc
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 er�ddlm Z ne Z dZ ej j ej!d��Z"G dd� dee �Z#G dd� de �Z$G dd� dej%�Z&G dd� d�Z'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 edd�dd�ZdS )�_LexiconAwareTestCasezX Protocol describing a TestCase suitable to test a real LexiconClient instance. N)� unused_args�returnc 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_mock�str� record_prefix� record_name�record_content� Exception�DOMAIN_NOT_FOUND� GENERIC_ERROR� LOGIN_ERROR�UNKNOWN_LOGIN_ERRORr r r r r r r ' s r c @ s6 e Zd Zej� eedd�dd��Zedd�dd�ZdS )� BaseLexiconAuthenticatorTestN)r �unused_mock_get_utilityr c C s< | j j| jg� tjjtdt tj�g}| j|| j j � d S )Nz_acme-challenge.) �authZperformr r �call�add_txt_record�DOMAIN�ANY� 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 tj�g}| 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 Zed�ZeZed�Zed�Z dZ e d e Z dZ edd�d d �Zedd�d d �Zedd�d d�Zedd�dd�Zedd�dd�Zedd�dd�Zedd�dd�Zedd�dd�Zedd�dd�Zedd�dd �Zedd�d d �Zedd�d d �Zedd�d!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 )N�TXT)�rtype�name�content)r r, r- r! r" r � create_record�assert_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- ZJWKRSA�loadZ load_vectorZKEYr r ZBaseAuthenticatorTestr( r6 r r r r �<module> s2
Back to File Manager