Current File : //lib/python3.6/site-packages/certbot/tests/__pycache__/util.cpython-36.pyc
3
گa�M � @ s� d Z ddlmZ ddlZddlZddlZddlmZ ddlZddl Z ddl
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ ddlZddlZdd
lmZ ddlmZ ddlZddl m Z ddl Z ddl m Z ddl m!Z! ddl"m#Z# ddl"m$Z$ ddl"m%Z% ddl&m'Z( ddl)m*Z* ddl)m+Z+ ddl,m!Z- ddl.m/Z/ yddl0Z0ej1de2� W n" e3k
�r� ddlm0Z0 Y nX G dd � d e/j4�Z5e6e6d �d d �Z7e6e8d �d d!�Z9e6e:e:e:d"�d#d$�Z;e6e j<d �d%d&�Z=e6e j>d �d'd(�Z?e6ej@d �d)d*�ZAe6ejBd �d+d,�ZCe6e jDd �d-d.�ZEdZe6e6eFe6d0�d1d2�ZGd[e6e0jHd4�d5d6�ZId\e6ee e0jHd7�d8d9�ZJe0jHd:�d;d<�ZKd]ee e0jHd=�d>d?�ZLG d@dA� dA�ZMeMd:�dBdC�ZNeeMd=�dDdE�ZOe e ddF�dGdH�ZPG dIdJ� dJejQ�ZRG dKdL� dLeR�ZSejTejTe6ddM�dNdO�ZUe
g e f e6ddP�dQdR�ZVe6e
e
dSe f ge
dSe f f dT�dUdV�ZWe6e6dW�dXdY�ZXdS )^zTest utilities.� )�reloadN)� synchronize)�Any)�Callable)�cast)�IO)�Iterable)�List)�Optional)�Union)�default_backend)�
serialization)�crypto)�
configuration)�util)� constants)�lock)�storage)�obj)�
filesystem)�os)�commonz�The external mock module is being used for backwards compatibility since it is available, however, future versions of Certbot's tests will use unittest.mock. Be sure to update your code accordingly.)�mockc @ s� e Zd ZdZee d�dd�Zeeeeedd�dd�Zd eeee e
e ef dd �d
d �Z e
e d�d d
�Z d ee e
dd�dd�Zdd�dd�Zdd�dd�Zeed dd�dd��Zdd�dd�Zed�d d �ZdS )!�DummyInstallerz(Dummy installer plugin for test purpose.)�returnc C s d S )N� )�selfr r �/usr/lib/python3.6/util.py�
get_all_names7 s z DummyInstaller.get_all_namesN)�domain� cert_path�key_path�
chain_path�fullchain_pathr c C s d S )Nr )r r r r! r" r# r r r � deploy_cert: s zDummyInstaller.deploy_cert)r � enhancement�optionsr c C s d S )Nr )r r r% r& r r r �enhance> s zDummyInstaller.enhancec C s d S )Nr )r r r r �supported_enhancementsB s z%DummyInstaller.supported_enhancementsF)�title� temporaryr c C s d S )Nr )r r) r* r r r �saveE s zDummyInstaller.savec C s d S )Nr )r r r r � config_testH s zDummyInstaller.config_testc C s d S )Nr )r r r r �restartK s zDummyInstaller.restart.)�addr c C s d S )Nr )�clsr. r r r �add_parser_argumentsN s z#DummyInstaller.add_parser_argumentsc C s d S )Nr )r r r r �prepareR s zDummyInstaller.preparec C s d S )Nr )r r r r � more_infoU s zDummyInstaller.more_info)N)NF).N)�__name__�
__module__� __qualname__�__doc__r �strr r$ r
r r r' r( �boolr+ r, r- � classmethodr r0 r1 r2 r r r r r 5 s &r )�namesr c G s t jttjjd| �� �S )zPath to a test vector.�testdata)r; )�
pkg_resourcesZresource_filenamer3 r �path�join)r: r r r � vector_pathY s r? c G sF t jttjjd| �� �}y|j� jdd�j� S t k
r@ |S X dS )z Load contents of a test vector.r; z
�
N)r; )
r< Zresource_stringr3 r r= r> �decode�replace�encode�
ValueError)r: �datar r r � load_vector_ s rF )�filename�
loader_pem�
loader_derr c C s< t jj| �\}}|j� dkr |S |j� dkr0|S td��d S )Nz.pemz.derz1Loader could not be recognized based on extension)r r= �splitext�lowerrD )rG rH rI �_Zextr r r �
_guess_loaderm s rM c G s$ t | d tjtj�}tj|t| � �S )zLoad certificate.� ���)rM r � FILETYPE_PEM�
FILETYPE_ASN1Zload_certificaterF )r: �loaderr r r � load_certv s rS c G s$ t | d tjtj�}tj|t| � �S )zLoad certificate request.rN rO )rM r rP rQ Zload_certificate_requestrF )r: rR r r r �load_csr} s rT c G s t jt| � �S )z(Load ComparableX509 certificate request.)�jose�ComparableX509rT )r: r r r �load_comparable_csr� s rW c G sF t | d tjtj�}|tjkr&tj}ntj}tj|t | � dt
� d��S )zLoad RSA private key.rN N)ZpasswordZbackendrO ) rM r rP rQ r
Zload_pem_private_keyZload_der_private_keyrU �ComparableRSAKeyrF r )r: rR Z loader_fnr r r �load_rsa_private_key� s
rY c G s$ t | d tjtj�}tj|t| � �S )zLoad pyOpenSSL private key.rN rO )rM r rP rQ Zload_privatekeyrF )r: rR r r r �load_pyopenssl_private_key� s rZ F)�
config_dir�testfile�ecr c
sP |dt d� � }tjj� tj�}tjj� tj|�}tjj� tj|�}x(|||fD ]}tjj|�sRt j
|� qRW t dj |r~dnd��}x2tj
|�D ]$} tjtjj|| �tjj|| �� q�W x:tjD ]0} tjtjj|dj | ��tjj|dj | ��� q�W tjj� ||�}
tt |���4} t|
d�� } | j� fd d
�| D �� W dQ R X W dQ R X |
S ) a� Creates a lineage defined by testfile.
This creates the archive, live, and renewal directories if
necessary and creates a simple lineage.
:param str config_dir: path to the configuration directory
:param str testfile: configuration file to base the lineage on
:param bool ec: True if we generate the lineage with an ECDSA key
:returns: path to the renewal conf file for the created lineage
:rtype: str
Nz.confzsample-archive{}z-ec� z{0}1.pemz{0}.pem�wc 3 s | ]}|j d � �V qdS )ZMAGICDIRN)rB )�.0�line)r[ r r � <genexpr>� s z make_lineage.<locals>.<genexpr>)�lenr r= r> r ZRENEWAL_CONFIGS_DIRZ ARCHIVE_DIRZLIVE_DIR�existsr �makedirsr? �format�listdir�shutilZcopyfiler ZALL_FOUR�symlink�open�
writelines)
r[ r\ r] Z lineage_nameZconf_dirZ archive_dirZlive_dirZ directoryZsample_archiveZkindZ conf_path�src�dstr )r[ r � make_lineage� s.
(rn �zope.component.getUtility)�targetr c C s t jd� ttjtj| td��S )z�Deprecated, patch certbot.display.util directly or use patch_display_util instead.
:param str target: path to patch
:returns: mock zope.component.getUtility
:rtype: mock.MagicMock
z�Decorator certbot.tests.util.patch_get_utility is deprecated. You should now patch certbot.display.util yourself directly or use certbot.tests.util.patch_display_util as a temporary workaround.)� new_callable)�warnings�warnr r � MagicMock�patch�_create_display_util_mock)rp r r r �patch_get_utility� s
rw )rp �stdoutr c C s8 t jd� |r|ntj� }t|�}ttjtj| |d��S )aE Deprecated, patch certbot.display.util directly
or use patch_display_util_with_stdout instead.
:param str target: path to patch
:param object stdout: object to write standard output to; it is
expected to have a `write` method
:returns: mock zope.component.getUtility
:rtype: mock.MagicMock
z�Decorator certbot.tests.util.patch_get_utility_with_stdout is deprecated. You should now patch certbot.display.util yourself directly or use use certbot.tests.util.patch_display_util_with_stdout as a temporary workaround.)�new) rr rs �io�StringIO�%_create_display_util_mock_with_stdoutr r rt ru )rp rx Zfreezable_mockr r r � patch_get_utility_with_stdout� s
r} )r c C s t tjtjdtd��S )a� Patch certbot.display.util to use a special mock display utility.
The mock display utility works like a regular mock object, except it also
also asserts that methods are called with valid arguments.
The mock created by this patch mocks out Certbot internals so this can be
used like the old patch_get_utility function. That is, the mock object will
be called by the certbot.display.util functions and the mock returned by
that call will be used as the display utility. This was done to simplify
the transition from zope.component and mocking certbot.display.util
functions directly in test code should be preferred over using this
function in the future.
See https://github.com/certbot/certbot/issues/8948
:returns: patch on the function used internally by certbot.display.util to
get a display utility instance
:rtype: mock.MagicMock
z)certbot._internal.display.obj.get_display)rq )r r rt ru rv r r r r �patch_display_util� s r~ )rx r c C s* | r| nt j� } ttjtjdt| �d��S )aW Patch certbot.display.util to use a special mock display utility.
The mock display utility works like a regular mock object, except it also
asserts that methods are called with valid arguments.
The mock created by this patch mocks out Certbot internals so this can be
used like the old patch_get_utility function. That is, the mock object will
be called by the certbot.display.util functions and the mock returned by
that call will be used as the display utility. This was done to simplify
the transition from zope.component and mocking certbot.display.util
functions directly in test code should be preferred over using this
function in the future.
See https://github.com/certbot/certbot/issues/8948
The `message` argument passed to the display utility methods is passed to
stdout's write method.
:param object stdout: object to write standard output to; it is
expected to have a `write` method
:returns: patch on the function used internally by certbot.display.util to
get a display utility instance
:rtype: mock.MagicMock
z)certbot._internal.display.obj.get_display)ry )rz r{ r r rt ru r| )rx r r r � patch_display_util_with_stdout s r c @ s| e Zd ZdZddejjfeede f e dd�dd�Z
dd�d d
�Z e e ej d �d d
�Z
ee d�dd�Zee dd�dd�ZdS )�
FreezableMocka� Mock object with the ability to freeze attributes.
This class works like a regular mock.MagicMock object, except
attributes and behavior set before the object is frozen cannot
be changed during tests.
If a func argument is provided to the constructor, this function
is called first when an instance of FreezableMock is called,
followed by the usual behavior defined by MagicMock. The return
value of func is ignored.
FN.)�frozen�func� return_valuer c C s> |r
t � ndh| _|| _tj� | _|tjjkr4|| _|| _ d S )N�freeze)
�set� _frozen_set�_funcr rt �_mock�sentinel�DEFAULTr� �_frozen)r r� r� r� r r r �__init__5 s
zFreezableMock.__init__)r c C s
d| _ dS )z)Freeze object preventing further changes.TN)r� )r r r r r� > s zFreezableMock.freeze)�args�kwargsr c O s" | j d k r| j ||� | j||�S )N)r� r� )r r� r� r r r �__call__B s
zFreezableMock.__call__)�namer c C sz |dkr,y t j| |�S tk
r( dS X nJ|dkrFtt j| d�|�S |dksX|| jkrdt j| |�S tt j| d�|�S d S )Nr� Fr� � side_effectr� r� )r� r� )�object�__getattribute__�AttributeError�getattrr� )r r� r r r r� G s z FreezableMock.__getattribute__)r� �valuer c C sb | j r*|| jkr td| ��t| j||�S |dkr>| jj|� |dkrTt| j||�S tj| ||�S )a� Before it is frozen, attributes are set on the FreezableMock
instance and added to the _frozen_set. Attributes in the _frozen_set
cannot be changed after the FreezableMock is frozen. In this case,
they are set on the underlying _mock.
In cases of return_value and side_effect, these attributes are always
passed through to the instance's _mock and added to the _frozen_set
before the object is frozen.
z Cannot change frozen attribute r� r� r� )r� r� )r� r� r� �setattrr� r. r� � __setattr__)r r� r� r r r r� T s
zFreezableMock.__setattr__)r3 r4 r5 r6 r r� r� r8 r r r� r� rt r� r7 r� r� r r r r r� ( s $
r� c C s\ t � } dd� ttj�D �}x,|D ]$}|dkr t dtd�}t| ||� q W | j� t d| d�S )Nc S s, g | ]$}t ttj|��r|jd � r|�qS )�__)�callabler� � display_obj� FileDisplay�
startswith)r` r� r r r �
<listcomp>p s z-_create_display_util_mock.<locals>.<listcomp>� notificationT)r� r� )r� r� )r� �dirr� r� �_assert_valid_callr� r� )�display� method_list�method� frozen_mockr r r rv m s
rv c s� t ttd d��fdd� � ttd d�� fdd� }t� }dd� ttj�D �}x:|D ]2}|d krhtd
� d �}n td
|d �}t|||� qNW |j� td
|d �S )
N)�message� unused_args�
unused_kwargsr c s | r� j | � dS )z$Write to message to stdout.
N)�write)r� r� r� )rx r r �
_write_msg| s z9_create_display_util_mock_with_stdout.<locals>._write_msg)r� r� r c s t | |� � | |� dS )z<
Mock function for display utility methods.
N)r� )r� r� )r� r r � mock_method� s
z:_create_display_util_mock_with_stdout.<locals>.mock_methodc S s, g | ]$}t ttj|��r|jd � r|�qS )r� )r� r� r� r� r� )r` r� r r r r� � s z9_create_display_util_mock_with_stdout.<locals>.<listcomp>r� T)r� r� )r� r� )r7 r r� r� r� r� r� r� )rx r� r� r� r� r� r )r� rx r r| { s
r| )r� r� r c O sZ | r | d n|d g}i }|j dd �|d< |j dd �|d< |j dd�|d< tj||� d S )Nr r� �defaultZcli_flagZforce_interactiveF)�get� display_utilZassert_valid_call)r� r� Z assert_argsZ
assert_kwargsr r r r� � s r� c @ s, e Zd ZdZdd�dd�Zdd�dd�ZdS )�TempDirTestCasezBBase test class which sets up and tears down a temporary directoryN)r c C s t j� | _dS )zExecute before testN)�tempfileZmkdtemp�tempdir)r r r r �setUp� s zTempDirTestCase.setUpc C s* t j� g t j� _tj� tj| j� dS )zExecute after testN) �loggingZshutdownZ getLoggerZhandlersr Z_release_locksrh Zrmtreer� )r r r r �tearDown� s
zTempDirTestCase.tearDown)r3 r4 r5 r6 r� r� r r r r r� � s r� c s& e Zd ZdZdd�� fdd� Z� ZS )�ConfigTestCasez2Test class which sets up a NamespaceConfig object.N)r c s� t � j� tjtjf tj��| _d| jj _
t j j
| jd�| jj _t j j
| jd�| jj _t j j
| jd�| jj _tjd | jj _tjd | jj _tjd | jj _d| jj _d S )NZcertonly�configZworkZlogsZauth_cert_pathZauth_chain_pathzhttps://example.com)�superr� r ZNamespaceConfigr rt r Z CLI_DEFAULTSr� � namespaceZverbr r= r> r� r[ Zwork_dirZlogs_dirr r# r" Zserver)r )� __class__r r r� � s
zConfigTestCase.setUp)r3 r4 r5 r6 r� �
__classcell__r r )r� r r� � s r� )�event_in� event_outr= r c
C sR t jj|�rtj|�}n
tj|�}z |j� | jdd�s@td��W d|j � X dS )a�
Acquire a file lock on given path, then wait to release it. This worker is coordinated
using events to signal when the lock should be acquired and released.
:param multiprocessing.Event event_in: event object to signal when to release the lock
:param multiprocessing.Event event_out: event object to signal when the lock is acquired
:param path: the path to lock
� )�timeoutz*Timeout while waiting to release the lock.N)
r r= �isdirr Zlock_dirZLockFiler� �wait�AssertionError�release)r� r� r= Zmy_lockr r r � _handle_lock� s
r� )�callback� path_to_lockr c C st t t� tj� }tj� }tjt|||fd�}|j� |jdd�sHtd��| � |j � |j
dd� |j dkspt�dS )z�
Grab a lock on path_to_lock from a foreign process then execute the callback.
:param callable callback: object to call after acquiring the lock
:param str path_to_lock: path to file or directory to lock
)rp r� �
)r� z*Timeout while waiting to acquire the lock.r N) �
reload_moduler �multiprocessing�EventZProcessr� �startr� r� r� r> Zexitcode)r� r� Z
emit_eventZ
receive_eventZprocessr r r �
lock_and_call� s r� .)�reasonr c s( t dtf t dtf d�� fdd� }|S )zFDecorator to skip permanently a test on Windows. A reason is required..)�functionr c s t jtjdk� �| �S )zWrapped versionZwin32)�unittestZskipIf�sys�platform)r� )r� r r �wrapper� s z skip_on_windows.<locals>.wrapper)r r )r� r� r )r� r �skip_on_windows� s $r� )r= r c C s t jjtj� | �S )z�
Return the given path joined to the tempdir path for the current platform
Eg.: 'cert' => /tmp/cert (Linux) or 'C:\Users\currentuser\AppData\Temp\cert' (Windows)
)r r= r> r� Z
gettempdir)r= r r r � temp_join s r� )F)ro )ro N)N)Yr6 � importlibr r� rz r� r� r rh r� r� Ztypingr r r r r r r
r r� rr Z cryptography.hazmat.backendsr Z cryptography.hazmat.primitivesr
ZjosepyrU ZOpenSSLr r< Zcertbotr r Zcertbot._internalr r r Zcertbot._internal.displayr r� Zcertbot.compatr r Zcertbot.displayr� Zcertbot.pluginsr r rs �PendingDeprecationWarning� ImportErrorZ Installerr r7 r? �bytesrF �intrM ZX509rS ZX509ReqrT rV rW rX rY ZPKeyrZ r8 rn rt rw r} r~ r r� rv r| r� ZTestCaser� r� r� r� r� r� r� r r r r �<module> s�
$ - E *
Mini Shell