← Back
Editing: util.cpython-38.pyc
U �]7 � @ s� d Z ddlZddlZddlZddlZddlZddlmZmZ ddl Z ddl Z ddl Z ddl Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd l m Z ddl m Z dd l mZ dd� Z!dd� Z"dd� Z#dd� Z$dd� Z%dd� Z&dd � Z'd d � Z(d d � Z)d:d"d#�Z*d;d$d%�Z+G d&d'� d'e,�Z-d(d)� Z.d*d+� Z/d,d-� Z0G d.d/� d/ej1�Z2G d0d1� d1e2�Z3d2d3� Z4d4d5� Z5d6d7� Z6d8d9� Z7dS )<zJTest utilities. .. warning:: This module is not part of the public API. � N)�Process�Event)� reload_module)�default_backend)� serialization)� configuration)� constants)� interfaces)�lock)�storage)�util)�os)� filesystemc G s t �ttjjd| �� �S )zPath to a test vector.�testdata)r )� pkg_resourcesZresource_filename�__name__r �path�join��names� r �4/usr/lib/python3/dist-packages/certbot/tests/util.py� vector_path! s �r c G sL t �ttjjd| �� �}z|�� �dd��� W S t k rF | Y S X dS )z Load contents of a test vector.r z � N)r ) r Zresource_stringr r r r �decode�replace�encode� ValueError)r �datar r r � load_vector' s �r c C s< t j�| �\}}|�� dkr |S |�� dkr0|S td��d S )Nz.pemz.derz1Loader could not be recognized based on extension)r r �splitext�lowerr )�filenameZ loader_pemZ loader_der�_Zextr r r � _guess_loader5 s r$ c G s* t | d tjjtjj�}tj�|t| � �S )zLoad certificate.���)r$ �OpenSSL�crypto� FILETYPE_PEM� FILETYPE_ASN1Zload_certificater �r �loaderr r r � load_cert? s �r, c G s* t | d tjjtjj�}tj�|t| � �S )zLoad certificate request.r% )r$ r& r' r( r) Zload_certificate_requestr r* r r r �load_csrF s �r- c G s t �t| � �S )z(Load ComparableX509 certificate request.)�joseZComparableX509r- r r r r �load_comparable_csrM s r/ c G s. t | d tjtj�}t�|t| � dt� d��S )zLoad RSA private key.r% N)ZpasswordZbackend)r$ r Zload_pem_private_keyZload_der_private_keyr. ZComparableRSAKeyr r r* r r r �load_rsa_private_keyR s � �r0 c G s* t | d tjjtjj�}tj�|t| � �S )zLoad pyOpenSSL private key.r% )r$ r& r' r( r) Zload_privatekeyr r* r r r �load_pyopenssl_private_keyZ s �r1 c s6 |dt d� � }tj�� tj�}tj�� tj|�}tj�� tj|�}|||fD ]}tj�|�sPt � |� qPt d�}t� |�D ]$}t �tj�||�tj�||�� q~tjD ]0}t�tj�|d�|��tj�|d�|��� q�tj�� ||�} tt |���4} t| d�� } | �� fdd�| D �� W 5 Q R X W 5 Q R X | S ) ap 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 :returns: path to the renewal conf file for the created lineage :rtype: str Nz.confzsample-archivez{0}1.pemz{0}.pem�wc 3 s | ]}|� d � �V qdS )ZMAGICDIRN)r )�.0�line�� config_dirr r � <genexpr>� s z make_lineage.<locals>.<genexpr>)�lenr r r r ZRENEWAL_CONFIGS_DIRZ ARCHIVE_DIRZLIVE_DIR�existsr �makedirsr �listdir�shutilZcopyfiler ZALL_FOUR�symlink�format�open� writelines) r6 ZtestfileZ lineage_nameZconf_dirZ archive_dirZlive_dirZ directoryZsample_archiveZkindZ conf_path�srcZdstr r5 r � make_lineagea sD � � � � � � rB �zope.component.getUtilityc C s t j| td�S )a; Patch zope.component.getUtility to use a special mock IDisplay. The mock IDisplay works like a regular mock object, except it also also asserts that methods are called with valid arguments. :param str target: path to patch :returns: mock zope.component.getUtility :rtype: mock.MagicMock )Z new_callable)�mock�patch�_create_get_utility_mock)�targetr r r �patch_get_utility� s rH c C s& |r|nt �� }t|�}tj| |d�S )a Patch zope.component.getUtility to use a special mock IDisplay. The mock IDisplay works like a regular mock object, except it also also asserts that methods are called with valid arguments. The `message` argument passed to the IDisplay methods is passed to stdout's write method. :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 )�new)�six�StringIO�$_create_get_utility_mock_with_stdoutrD rE )rG �stdoutZfreezable_mockr r r � patch_get_utility_with_stdout� s rN c @ sD e Zd ZdZddejjfdd�Zdd� Zdd � Z d d � Z d d � Z dS )� 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. FNc C s> |r t � ndh| _|| _t�� | _|tjjkr4|| _|| _ d S )N�freeze) �set� _frozen_set�_funcrD � MagicMock�_mock�sentinel�DEFAULT� return_value�_frozen)�self�frozen�funcrX r r r �__init__� s zFreezableMock.__init__c C s d| _ dS )z)Freeze object preventing further changes.TN)rY �rZ r r r rP � s zFreezableMock.freezec O s" | j d k r| j ||� | j||�S )N)rS rU )rZ �args�kwargsr r r �__call__� s zFreezableMock.__call__c C s~ |dkr0zt �| |�W S tk r, Y dS X nJ|dkrJtt �| d�|�S |dks\|| jkrht �| |�S tt �| d�|�S d S )NrY F�rX Z side_effectrU rR )�object�__getattribute__�AttributeError�getattrrR )rZ �namer r r rd � s z FreezableMock.__getattribute__c C sd | j r,|| jkr td| ��nt| j||�S |dkr@| j�|� |dkrVt| j||�S t�| ||�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 rR rb )rY rR re �setattrrU �addrc � __setattr__)rZ rg �valuer r r rj � s zFreezableMock.__setattr__) r � __module__� __qualname__�__doc__rD rV rW r] rP ra rd rj r r r r rO � s rO c C sJ t � } tj�� D ]$}|dkrt dtd�}t| ||� q| �� t d| d�S )N� notificationT�r[ r\ �r[ rX )rO r �IDisplayr �_assert_valid_callrh rP )�displayrg � frozen_mockr r r rF � s rF c s| �fdd�� � fdd�}t � }tj�� D ]>}|dkrNt d� d�}t|||� q(t d|d�}t|||� q(|�� t d|d�S ) Nc s | r� � | � dS )z$Write to message to stdout. N)�write)�messageZ unused_argsZ unused_kwargs)rM r r � _write_msg s z8_create_get_utility_mock_with_stdout.<locals>._write_msgc s t | |� � | |� dS )z5 Mock function for IDisplay methods. N)rs )r_ r` )rx r r � mock_method s z9_create_get_utility_mock_with_stdout.<locals>.mock_methodro Trp rq )rO r rr r rh rP )rM ry rt rg ru r )rx rM r rL s ��rL c O sZ | r | d n|d g}i }|� dd �|d< |� dd �|d< |� dd�|d< tj||� d S )Nr rw �defaultZcli_flagZforce_interactiveF)�get� display_utilZassert_valid_call)r_ r` Z assert_argsZ assert_kwargsr r r rs # s rs c @ s e Zd ZdZdd� Zdd� ZdS )�TempDirTestCasezBBase test class which sets up and tears down a temporary directoryc C s t �� | _dS )zExecute before testN)�tempfileZmkdtemp�tempdirr^ r r r �setUp1 s zTempDirTestCase.setUpc C s* t �� g t �� _t�� t�| j� dS )zExecute after testN) �loggingZshutdownZ getLoggerZhandlersr Z_release_locksr< Zrmtreer r^ r r r �tearDown5 s zTempDirTestCase.tearDownN)r rl rm rn r� r� r r r r r} . s r} c s e Zd ZdZ� fdd�Z� ZS )�ConfigTestCasez2Test class which sets up a NamespaceConfig object.c s� t t| ��� t�tjf tj��| _ d| j _ t j � | jd�| j _t j � | jd�| j _t j � | jd�| j _tjd | j _tjd | j _tjd | j _d| j _d S )NZcertonly�configZworkZlogsZauth_cert_pathZauth_chain_pathzhttps://example.com)�superr� r� r ZNamespaceConfigrD rT r Z CLI_DEFAULTSr� Zverbr r r r r6 Zwork_dirZlogs_dirZ cert_pathZfullchain_pathZ chain_pathZserverr^ �� __class__r r r� F s �zConfigTestCase.setUp)r rl rm rn r� � __classcell__r r r� r r� D s r� c C sR t j�|�rt�|�}n t�|�}z |�� | jdd�s@t d��W 5 |�� 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 � �Ztimeoutz*Timeout while waiting to release the lock.N) r r �isdirr Zlock_dirZLockFile�releaserQ �wait�AssertionError)Zevent_inZ event_outr Zmy_lockr r r � _handle_lockU s r� c C sn t t� t� }t� }tt|||fd�}|�� |jdd�sBtd��| � |�� |j dd� |j dksjt�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 )rG r_ � r� z*Timeout while waiting to acquire the lock.r N) r r r r r� �startr� r� rQ r Zexitcode)�callbackZ path_to_lockZ emit_eventZ receive_eventZprocessr r r � lock_and_callh s r� c s � fdd�}|S )zFDecorator to skip permanently a test on Windows. A reason is required.c s t �tjdk� �| �S )zWrapped versionZwin32)�unittestZskipIf�sys�platform)Zfunction��reasonr r �wrapper� s z skip_on_windows.<locals>.wrapperr )r� r� r r� r �skip_on_windows� s r� c C s t j�t�� | �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� )rC )rC N)8rn r� r< r� r~ r� Zmultiprocessingr r r& Zjosepyr. rD r rJ Z six.movesr Z cryptography.hazmat.backendsr Z cryptography.hazmat.primitivesr Zcertbotr r r r r r Zcertbot.compatr r Zcertbot.displayr| r r r$ r, r- r/ r0 r1 rB rH rN rc rO rF rL rs ZTestCaser} r� r� r� r� r� r r r r �<module> s\ , � E
Save File
Cancel