Directory:~$ /proc/self/root/lib/python2.7/site-packages/certbot/tests/

Upload File

[Home] [System Details] [Kill Me]

�
���_c@s}dZd
dl
Z
d
dlmZ
d
dlmZ
d
dlZd
dlZd
dlZd
dlZd
dl	m
Z

d
dlmZ
d
dl
Zyd
dlZWn!ek
r�


d
dlmZ
n
Xd
dlZd
dlZd
dlZd
dlmZ
d
d	lmZ
d
d
lmZ
d
dlmZ
d
dlmZ
d
d
lmZ
d
dlmZ
d
dlmZ
d
dlm Z

d
d
l!mZ"
d�Z#d�Z$d�Z%d�Z&d�Z'd�Z(d�Z)d�Z*e+d�
Z,dd�
Z-ddd�Z/de0f
d��YZ1d�Z2d
�Z3d!�Z4d"ej5f
d#��YZ6d$e6f
d%��YZ7d&�Z8d'�Z9d(�Z:d)�Z;dS(*sTest
utilities.i����N(
tEvent(
tProcess(
tdefault_backend(
t
serialization(
tmock(
t
reload_module(
t
interfaces(
tutil(
t
configuration(
t	constants(
tlock(
tstorage(
t
filesystem(
tosc
Gstj
ttjjd
|�
�S(sPath to a test
vector.ttestdata(t
pkg_resourcestresource_filenamet__name__R
tpathtjoin(
tnames((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytvector_path!s
cGsZtj
ttjjd
|�
�}
y
|
j�jdd�j�SWnt	k
rU


|
SXdS(sLoad contents of a test vector.Rs
s

N(
Rtresource_stringRR
RRtdecodetreplacetencodet
ValueError(Rtdata((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytload_vector's




cCsTtj
j|�
\}}|j�d
kr.|
S|j�dkrD|Std�
�
dS(Ns.pems.ders1Loader
could not be recognized based on
extension(R
RtsplitexttlowerR(tfilenamet
loader_pemt
loader_dert
_text((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt
_guess_loader5s





cGs;t|d
t
jjt
jj�}
t
jj|
t|��S(sLoad
certificate.i����(R$tOpenSSLtcryptotFILETYPE_PEMt
FILETYPE_ASN1tload_certificateR(Rtloader((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt	load_cert>s

cGs;t|d
t
jjt
jj�}
t
jj|
t|��S(sLoad
certificate
request.i����(R$R%R&R'R(tload_certificate_requestR(RR*((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytload_csrEs

c
Gstj
t|��
S(
s(Load
ComparableX509 certificate
request.(tjosetComparableX509R-(
R((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytload_comparable_csrLscGsDt|d
t
jt
j�}
tj|
t|�dddt��
�
S(sLoad
RSA private
key.i����tpasswordtbackendN(	R$Rtload_pem_private_keytload_der_private_keyR.tComparableRSAKeyRtNoneR(RR*((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytload_rsa_private_keyQs

	
cGs;t|d
t
jjt
jj�}
t
jj|
t|��S(sLoad
pyOpenSSL private
key.i����(R$R%R&R'R(tload_privatekeyR(RR*((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytload_pyopenssl_private_keyYs

c
s�
|
td
�

}t
jj�tj�}t
jj�tj|�}t
jj�tj|�}x9|||fD](}t
jj|�
sot	j
|�

qoqoWtdj|r�dnd�
�
}xEt
j
|�
D]4}	tjt
jj||	�t
jj||	��
q�WxQtjD]F}	t
jt
jj|dj|	�
�t
jj|dj|	�
��
q
Wt
jj�||
�}
tt|
�
�
�<}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.confssample-archive{}s-ects{0}1.pems{0}.pemt
wc
3s!|]}
|
jd��V
qd
S(tMAGICDIRN(
R(t.0tline(
t
config_dir(s6/usr/lib/python2.7/site-packages/certbot/tests/util.pys	<genexpr>�sN(tlenR
RRR	tRENEWAL_CONFIGS_DIRtARCHIVE_DIRtLIVE_DIRtexistsRtmakedirsRtformattlistdirtshutiltcopyfileRtALL_FOURtsymlinktopent
writelines(
R?ttestfiletectlineage_nametconf_dirtarchive_dirtlive_dirt	directorytsample_archivetkindt	conf_pathtsrctdst((
R?s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytmake_lineage`s.
	

	

	


!



!
#



#szope.component.getUtilityc

Cstj
|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(Rtpatcht_create_get_utility_mock(
ttarget((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytpatch_get_utility�scCs7|
r|
n	tj
�}
t|
�
}tj|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

   
tnew(tsixtStringIOt$_create_get_utility_mock_with_stdoutRR\(R^tstdouttfreezable_mock((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytpatch_get_utility_with_stdout�s
t
FreezableMockcBsJeZ
dZedejjd
�Zd�Z	d�Z
d�Zd�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.

   
cCs^|
rt�nd
h
|_
||_tj�|_|tjjkrQ||_n|
|_	dS(Ntfreeze(
tsett_frozen_sett_funcRt	MagicMockt_mocktsentineltDEFAULTtreturn_valuet_frozen(tselftfrozentfuncRp((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt__init__�s

	



c

Cs
t|_
d
S(s)Freeze
object preventing further
changes.N(tTrueRq(
Rr((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyRh�sc
Os2|jdk	r"|j|
|�
n|j|
|�S(
N(RkR6Rm(Rrtargstkwargs((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt__call__�s


cCs�|
d
kr8ytj
||
�SWq�tk
r4


tSXni|
dkr]ttj
|d�|
�S|
dksx|
|jkr�tj
||
�Sttj
|d�|
�SdS(NRqRptside_effectRmRj(sreturn_valueRz(tobjectt__getattribute__tAttributeErrortFalsetgetattrRj(Rrtname((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR|�s









cCs�|jr>|
|j
kr+td
|
�
�
nt|j|
|�S|
dkr]|j
j|
�

n|
dkr|t|j|
|�Stj||
|�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.

        sCannot change frozen attribute
RjRpRz(sreturn_valuesside_effect(RqRjR}tsetattrRmtaddR{t__setattr__(RrR�tvalue((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR��s	




N(
Rt
__module__t__doc__R~R6RRnRoRuRhRyR|R�(((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyRg�s
			
cCstt�}xKt
jj�D]:}
|
d
krtdtdt�}t||
|�
qqW|j�
tdtd|�S(NtnotificationRsRtRp(RgRtIDisplayRRvt_assert_valid_callR�Rh(tdisplayR�tfrozen_mock((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR]�s
	





c
s��
f
d
���f
d�}
t�}xpt
jj�D]_}|dkrqtdtd��}t|||�
q7tdtd|
�}t|||�
q7W|j�
tdtd|�S(Nc
s|r�j|�

nd
S(s$Write
to message to stdout.
       
N(
twrite(tmessagetunused_argst
unused_kwargs(
Rd(s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt
_write_msg
s
cst||
�
�||
�
d
S(s5
        Mock function for IDisplay methods.
       
N(
R�(RwRx(
R�(s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytmock_method	
s

R�RsRtRp(RgRR�RRvR�Rh(RdR�R�R�R�((R�Rds6/usr/lib/python2.7/site-packages/certbot/tests/util.pyRc
s
	


	

	


cOsy|r|d
n|
dg
}i}|
jdd�|d<|
jdd�|d<|
jdt�|d<tj||�
dS(NiR�tdefaulttcli_flagtforce_interactive(tgetR6R~tdisplay_utiltassert_valid_call(RwRxtassert_argst
assert_kwargs((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR�!
s



tTempDirTestCasec
Bs
eZ
dZd
�Zd�ZRS(sBBase test class which sets up and tears down a
temporary directoryc

Cstj
�|_d
S(sExecute before
testN(ttempfiletmkdtempttempdir(
Rr((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytsetUp/
sc

Cs7tj
�
gtj�_tj�
tj|j�

d
S(sExecute
after
testN(	tloggingtshutdownt	getLoggerthandlersRt_release_locksRHtrmtreeR�(
Rr((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyttearDown3
s


(RR�R�R�R�(((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR�,
s
	tConfigTestCasec
BseZ
dZd
�ZRS(s2Test
class which sets up a NamespaceConfig
object.c

Cs�tt
|�j�
tjtjtj��
|_	d
|j	_
tjj
|jd�|j	_tjj
|jd�|j	_tjj
|jd�|j	_tjd|j	_tjd|j	_tjd|j	_d|j	_dS(Ntcertonlytconfigtworktlogstauth_cert_pathtauth_chain_pathshttps://example.com(tsuperR�R�RtNamespaceConfigRRlR	tCLI_DEFAULTSR�tverbR
RRR�R?twork_dirtlogs_dirt	cert_pathtfullchain_patht
chain_pathtserver(
Rr((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR�D
s









(RR�R�R�(((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyR�B
s

cCsqtj
j|�
r$tj|�
}ntj|�
}z,|
j�
|jd
d�
s^td�
�
Wd|j	�
XdS(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
    ttimeoutis*Timeout while waiting to release the lock.N(
R
RtisdirR
tlock_dirtLockFileRitwaittAssertionErrortrelease(tevent_int	event_outRtmy_lock((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt_handle_lockS
s




"cCs�tt
�

t�}t�}td
td|||
f�}|j�
|jdd�
sbtd�
�
|�
|j�
|j	dd�

|j
dks�t�
dS(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^RwR�i
s*Timeout while waiting to acquire the
lock.iN(RRRR
R�tstartR�R�RiRtexitcode(tcallbacktpath_to_lockt
emit_eventt
receive_eventtprocess((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt
lock_and_callf
s
	
	




c
s�f
d
�}
|
S(sFDecorator to skip permanently a test on
Windows. A reason is required.c

stj
tjd
k��|�
S(sWrapped
versiontwin32(tunittesttskipIftsystplatform(
tfunction(
treason(s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytwrapper�
s((R�R�((
R�s6/usr/lib/python2.7/site-packages/certbot/tests/util.pytskip_on_windows�
sc

Cstj
jtj�|�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
RRR�t
gettempdir(
R((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt	temp_join�
s(<R�R�tmultiprocessingRR
RHR�R�R�tcryptography.hazmat.backendsRtcryptography.hazmat.primitivesRtjosepyR.RtImportErrorR%RRat	six.movesRtcertbotRRtcertbot._internalRR	R
Rtcertbot.compatRR
tcertbot.displayR�RRR$R+R-R0R7R9R~RZR_R6RfR{RgR]RcR�tTestCaseR�R�R�R�R�R�(((s6/usr/lib/python2.7/site-packages/certbot/tests/util.pyt<module>
sb

























									,
D