�
E��c� � �V � d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlm Z ddl
m Z ddl m
Z
dd lmZ dd
lmZ dd lmZ dd lmZ erdd
lmZ neZdZej � ej d� � � � Z G d� de� � Z G d� d� � Z dee ef de ddfd�Z!dS )z'Base test class for DNS authenticators.� )�Any)�Mapping)�
TYPE_CHECKINGN)�mock)�
challenges)� achallenges)�
filesystem)�DNSAuthenticator)� acme_util)�util)�Protocolz example.comzrsa512_key.pemc �> � e Zd ZU dZeed< deddfd�Zdeddfd�ZdS )� _AuthenticatorCallableTestCasezMProtocol describing a TestCase able to call a real DNSAuthenticator instance.�auth� unused_args�returnNc � � dS )zj
See
https://docs.python.org/3/library/unittest.html#unittest.TestCase.assertTrue
N� ��selfr s �A/usr/lib/python3/dist-packages/certbot/plugins/dns_test_common.py�
assertTruez)_AuthenticatorCallableTestCase.assertTrue � � � � � c � � dS )zk
See
https://docs.python.org/3/library/unittest.html#unittest.TestCase.assertEqual
Nr r s r � assertEqualz*_AuthenticatorCallableTestCase.assertEqual% r r ) �__name__�
__module__� __qualname__�__doc__r
�__annotations__r r r r r r r r sl � � � � � � �W�W�
���� �s� �t� � � � � �� �� � � � � � r r c �r � e Zd ZdZ ej ej ee �� � Z
de ddfd�Z de ddfd�Z
de ddfd�ZdS ) �BaseAuthenticatorTestz�
A base test class to reduce duplication between test code for DNS Authenticator Plugins.
Assumes:
* That subclasses also subclass unittest.TestCase
* That the authenticator is stored as self.auth
)�challb�domain� account_keyr r Nc � � | � t | j � � � t � � � � d S )N)r �
isinstancer � more_info�str�r s r �test_more_infoz$BaseAuthenticatorTest.test_more_info8 s2 � � ���
�4�9�#6�#6�#8�#8�#�>�>�?�?�?�?�?r c �x � | � | j � d� � t j g� � d S )Nz example.org)r r �get_chall_prefr �DNS01r+ s r �test_get_chall_prefz)BaseAuthenticatorTest.test_get_chall_pref; s4 � � �����1�1�-�@�@�:�CS�BT�U�U�U�U�Ur c �� � t j � � }| j � |� � |� dt
t j t j �� � d S )Nzpropagation-seconds)�type�default�help)r � MagicMockr �add_parser_arguments�assert_any_call�int�ANY)r �ms r �test_parser_argumentsz+BaseAuthenticatorTest.test_parser_arguments> sO � � �N� � �� � �&�&�q�)�)�)� ���/�c�4�8�RV�RZ��[�[�[�[�[r )r r r r r �"KeyAuthorizationAnnotatedChallenger r/ �DOMAIN�KEY�achallr r, r0 r; r r r r# r# , s� � � � � � �� � <�[�
;���v�3�@� @� @�F�@� ;� @�� @� @� @� @�V�"@� V�T� V� V� V� V�\�$B� \�t� \� \� \� \� \� \r r# �values�pathr c �� � t j � � }| D ]
}| | ||< �t |d� � 5 }|� |�� � ddd� � n # 1 swxY w Y t j |d� � dS )z�Write the specified values to a config file.
:param dict values: A map of values to write.
:param str path: Where to write the values.
�wb)�outfileNi� )� configobj� ConfigObj�open�writer �chmod)r@ rA �config�key�fs r rH rH E s� � � �
�
"�
"�F�� "� "�� �S�k��s� � �
�d�D� � � �Q�� � �Q� � � � � � � � � � � � � � � ���� � � � � ��T�5�!�!�!�!�!s �A�A� A)"r �typingr r r rE �josepy�jose�unittestr �acmer �certbotr �certbot.compatr �certbot.plugins.dns_commonr
�
certbot.testsr r � test_util�typing_extensionsr
�objectr= �JWKRSA�load� load_vectorr> r r# r* rH r r r �<module>r\ s� �� -� -� � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � %� %� %� %� %� %� 7� 7� 7� 7� 7� 7� #� #� #� #� #� #� +� +� +� +� +� +�� �*�*�*�*�*�*�*��H�
��
�k���,�y�,�-=�>�>�?�?�� � � � � �X� � � �"\� \� \� \� \� \� \� \�2"�'�#�s�(�#� "�3� "�4� "� "� "� "� "� "r |