�
���_c @ s} d Z d d l Z d d l m Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l m
Z
d d l m Z d d l
Z y d d l Z Wn! e k
r� d d l m Z n Xd d l Z d d l Z d d l Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d l m Z d d
l! m Z" d � Z# d � Z$ d � Z% d � Z& d � Z' d � Z( d � Z) d � Z* e+ d � Z, d d � Z- d d d � Z/ d e0 f d � � YZ1 d � Z2 d � Z3 d! � Z4 d" e j5 f d# � � YZ6 d$ e6 f d% � � YZ7 d& � Z8 d' � Z9 d( � Z: d) � Z; d S(* s Test utilities.i����N( t Event( t Process( t default_backend( t
serialization( t mock( t
reload_module( t
interfaces( t util( t
configuration( t constants( t lock( t storage( t
filesystem( t osc G s t j t t j j d | � � S( s Path to a test vector.t testdata( t
pkg_resourcest resource_filenamet __name__R
t patht join( t names( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt vector_path! s c G sZ t j t t j j d | � � } y | j � j d d � j � SWn t k
rU | SXd S( s Load contents of a test vector.R s
s
N(
R t resource_stringR R
R R t decodet replacet encodet
ValueError( R t data( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt load_vector' s
c C sT t j j | � \ } } | j � d k r. | S| j � d k rD | St d � � d S( Ns .pems .ders1 Loader could not be recognized based on extension( R
R t splitextt lowerR ( t filenamet
loader_pemt
loader_dert _t ext( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt
_guess_loader5 s c G s; t | d t j j t j j � } t j j | t | � � S( s Load certificate.i����( R$ t OpenSSLt cryptot FILETYPE_PEMt
FILETYPE_ASN1t load_certificateR ( R t loader( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt load_cert> s c G s; t | d t j j t j j � } t j j | t | � � S( s Load certificate request.i����( R$ R% R& R' R( t load_certificate_requestR ( R R* ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt load_csrE s c G s t j t | � � S( s( Load ComparableX509 certificate request.( t joset ComparableX509R- ( R ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt load_comparable_csrL s c G sD t | d t j t j � } t j | t | � d d d t � �� S( s Load RSA private key.i����t passwordt backendN( R$ R t load_pem_private_keyt load_der_private_keyR. t ComparableRSAKeyR t NoneR ( R R* ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt load_rsa_private_keyQ s c G s; t | d t j j t j j � } t j j | t | � � S( s Load pyOpenSSL private key.i����( R$ R% R& R' R( t load_privatekeyR ( R R* ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt load_pyopenssl_private_keyY s c
s� | t d � } t j j � t j � } t j j � t j | � } t j j � t j | � } x9 | | | f D]( } t j j | � so t j
| � qo qo Wt d j | r� d n d � � } xE t j
| � D]4 } t j t j j | | � t j j | | � � q� WxQ t j D]F } t j t j j | d j | � � t j j | d j | � � � qWt j j � | | � }
t t | � � �< } t |
d � �$ } | j � f d � | D� � Wd QXWd QX|
S(
sp 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
s .confs sample-archive{}s -ect s {0}1.pems {0}.pemt wc 3 s! | ] } | j d � � Vq d S( t MAGICDIRN( R ( t .0t line( t
config_dir( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pys <genexpr>� s N( t lenR
R R R t RENEWAL_CONFIGS_DIRt ARCHIVE_DIRt LIVE_DIRt existsR t makedirsR t formatt listdirt shutilt copyfileR t ALL_FOURt symlinkt opent
writelines(
R? t testfilet ect lineage_namet conf_dirt archive_dirt live_dirt directoryt sample_archivet kindt conf_patht srct dst( ( R? s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt make_lineage` s.
!!##s zope.component.getUtilityc C s t j | d t �S( s; 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
t new_callable( R t patcht _create_get_utility_mock( t target( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt patch_get_utility� s c C s7 | r | n t j � } t | � } t j | d | �S( s 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
t new( t sixt StringIOt$ _create_get_utility_mock_with_stdoutR R\ ( R^ t stdoutt freezable_mock( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt patch_get_utility_with_stdout� s t
FreezableMockc B sJ e Z d Z e d e j j d � Z d � Z d � Z
d � Z d � Z RS( s� 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.
c C s^ | r t � n d h | _ | | _ t j � | _ | t j j k rQ | | _ n | | _ d S( Nt freeze(
t sett _frozen_sett _funcR t MagicMockt _mockt sentinelt DEFAULTt return_valuet _frozen( t selft frozent funcRp ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt __init__� s c C s
t | _ d S( s) Freeze object preventing further changes.N( t TrueRq ( Rr ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyRh � s c O s2 | j d k r" | j | | � n | j | | � S( N( Rk R6 Rm ( Rr t argst kwargs( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt __call__� s c C s� | d k r8 y t j | | � SWq� t k
r4 t SXni | d k r] t t j | d � | � S| d k sx | | j k r� t j | | � St t j | d � | � Sd S( NRq Rp t side_effectRm Rj ( s return_valueRz ( t objectt __getattribute__t AttributeErrort Falset getattrRj ( Rr t name( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR| � s
c C s� | j r> | | j k r+ t d | � � n t | j | | � S| d k r] | j j | � n | d k r| t | j | | � St j | | | � S( s� 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.
s Cannot change frozen attribute Rj Rp Rz ( s return_values side_effect( Rq Rj R} t setattrRm t addR{ t __setattr__( Rr R� t value( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR� � s N(
R t
__module__t __doc__R~ R6 R Rn Ro Ru Rh Ry R| R� ( ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyRg � s
c C st t � } xK t j j � D]: } | d k r t d t d t � } t | | | � q q W| j � t d t d | � S( Nt notificationRs Rt Rp ( Rg R t IDisplayR Rv t _assert_valid_callR� Rh ( t displayR� t frozen_mock( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR] � s
c s� � f d � � � f d � } t � } xp t j j � D]_ } | d k rq t d t d � � } t | | | � q7 t d t d | � } t | | | � q7 W| j � t d t d | � S( Nc s | r � j | � n d S( s$ Write to message to stdout.
N( t write( t messaget unused_argst
unused_kwargs( Rd ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt
_write_msg s c s t | | � � | | � d S( s5
Mock function for IDisplay methods.
N( R� ( Rw Rx ( R� ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt mock_method s
R� Rs Rt Rp ( Rg R R� R Rv R� Rh ( Rd R� R� R� R� ( ( R� Rd s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyRc s
c O sy | r | d n | d g } i } | j d d � | d <| j d d � | d <| j d t � | d <t j | | � d S( Ni R� t defaultt cli_flagt force_interactive( t getR6 R~ t display_utilt assert_valid_call( Rw Rx t assert_argst
assert_kwargs( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR� ! s t TempDirTestCasec B s e Z d Z d � Z d � Z RS( sB Base test class which sets up and tears down a temporary directoryc C s t j � | _ d S( s Execute before testN( t tempfilet mkdtempt tempdir( Rr ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt setUp/ s c C s7 t j � g t j � _ t j � t j | j � d S( s Execute after testN( t loggingt shutdownt getLoggert handlersR t _release_locksRH t rmtreeR� ( Rr ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt tearDown3 s
( R R� R� R� R� ( ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR� , s t ConfigTestCasec B s e Z d Z d � Z RS( s2 Test class which sets up a NamespaceConfig object.c C s� t t | � j � t j t j t j � � | _ d | j _
t j j
| j d � | j _ t j j
| j d � | j _ t j j
| j d � | j _ t j d | j _ t j d | j _ t j d | j _ d | j _ d S( Nt certonlyt configt workt logst auth_cert_patht auth_chain_paths https://example.com( t superR� R� R t NamespaceConfigR Rl R t CLI_DEFAULTSR� t verbR
R R R� R? t work_dirt logs_dirt cert_patht fullchain_patht
chain_patht server( Rr ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR� D s ( R R� R� R� ( ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyR� B s c C sS t j j | � r$ t j | � } n t j | � } z | j � Wd | j � Xd S( s�
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
N( R
R t isdirR
t lock_dirt LockFileRi t release( t event_int event_outR t my_lock( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt _handle_lockS s
c C si t t � t � } t � } t d t d | | | f � } | j � | � | j � | j d d � d S( s�
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
R^ Rw t timeouti
N( R R R R R� t startRi R ( t callbackt path_to_lockt
emit_eventt
receive_eventt process( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt
lock_and_callf s
c s � f d � } | S( sF Decorator to skip permanently a test on Windows. A reason is required.c s t j t j d k � � | � S( s Wrapped versiont win32( t unittestt skipIft syst platform( t function( t reason( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt wrapper� s ( ( R� R� ( ( R� s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt skip_on_windows� s c C s t j j t j � | � S( s�
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� t
gettempdir( R ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt temp_join� s (< R� R� t multiprocessingR R RH R� R� R� t cryptography.hazmat.backendsR t cryptography.hazmat.primitivesR t josepyR. R t ImportErrorR% R Ra t six.movesR t certbotR R t certbot._internalR R R
R t certbot.compatR R
t certbot.displayR� R R R$ R+ R- R0 R7 R9 R~ RZ R_ R6 Rf R{ Rg R] Rc R� t TestCaseR� R� R� R� R� R� ( ( ( s6 /usr/lib/python2.7/site-packages/certbot/tests/util.pyt <module> sb
, D
Anons79 File Manager Version 1.0, Coded By Anons79
Email: [email protected]
Anons79 Mini Shell