ó Y[c@sùdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl Z ddl ZddlZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZd „Zd „Zd „Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#dd„Z$ddd„Z&de'fd„ƒYZ(d„Z)d„Z*d„Z+dej,fd„ƒYZ-de-fd „ƒYZ.d!„Z/d"„Z0dS(#sJTest utilities. .. warning:: This module is not part of the public API. iÿÿÿÿN(tdefault_backend(t serialization(t reload_module(t constants(t interfaces(tstorage(tutil(t configurationcGstjttjjd|ŒƒS(sPath to a test vector.ttestdata(t pkg_resourcestresource_filenamet__name__tostpathtjoin(tnames((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt vector_pathscGstjttjjd|ŒƒS(sLoad contents of a test vector.R(R tresource_stringR R R R(R((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt load_vector$scCsTtjj|ƒ\}}|jƒdkr.|S|jƒdkrD|Stdƒ‚dS(Ns.pems.ders1Loader could not be recognized based on extension(R R tsplitexttlowert ValueError(tfilenamet loader_pemt loader_dert_text((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt _guess_loader+s cGs;t|dtjjtjjƒ}tjj|t|ŒƒS(sLoad certificate.iÿÿÿÿ(RtOpenSSLtcryptot FILETYPE_PEMt FILETYPE_ASN1tload_certificateR(Rtloader((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt load_cert5scGs;t|dtjjtjjƒ}tjj|t|ŒƒS(sLoad certificate request.iÿÿÿÿ(RRRRRtload_certificate_requestR(RR!((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytload_csr<scGstjt|ŒƒS(s(Load ComparableX509 certificate request.(tjosetComparableX509R$(R((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytload_comparable_csrCscGsDt|dtjtjƒ}tj|t|ŒdddtƒƒƒS(sLoad RSA private key.iÿÿÿÿtpasswordtbackendN( RRtload_pem_private_keytload_der_private_keyR%tComparableRSAKeyRtNoneR(RR!((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytload_rsa_private_keyHs  cGs;t|dtjjtjjƒ}tjj|t|ŒƒS(sLoad pyOpenSSL private key.iÿÿÿÿ(RRRRRtload_privatekeyR(RR!((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytload_pyopenssl_private_keyPscCs7ttdƒrtj||ƒS|r,d„Sd„SdS(sqSkip tests unless a condition holds. This implements the basic functionality of unittest.skipUnless which is only available on Python 2.7+. :param bool condition: If ``False``, the test will be skipped :param str reason: the reason for skipping the test :rtype: callable :returns: decorator that hides tests unless condition is ``True`` t skipUnlesscSs|S(N((tcls((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytgscSsdS(N(R-(R2((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyR3isN(thasattrtunittestR1(t conditiontreason((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt skip_unlessWs c s¯|tdƒ }tjjˆtjƒ}tjjˆtj|ƒ}tjjˆtj|ƒ}x9|||fD](}tjj|ƒsotj |ƒqoqoWt dƒ}xEtj |ƒD]4}t j tjj||ƒtjj||ƒƒq·WxQtjD]F}tjtjj|dj|ƒƒtjj|dj|ƒƒƒqùWtjjˆ||ƒ} tt |ƒƒ<} t| dƒ$} | j‡fd†| DƒƒWdQXWdQX| S(spCreates 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-archives{0}1.pems{0}.pemtwc3s!|]}|jdˆƒVqdS(tMAGICDIRN(treplace(t.0tline(t config_dir(s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pys “sN(tlenR R RRtRENEWAL_CONFIGS_DIRt ARCHIVE_DIRtLIVE_DIRtexiststmakedirsRtlistdirtshutiltcopyfileRtALL_FOURtsymlinktformattopent writelines( R>ttestfilet lineage_nametconf_dirt archive_dirtlive_dirt directorytsample_archivetkindt conf_pathtsrctdst((R>s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt make_lineagels.     !##szope.component.getUtilitycCstj|dtƒ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(tmocktpatcht_create_get_utility_mock(ttarget((s6/usr/lib/python2.7/dist-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_stdoutRZR[(R]tstdouttfreezable_mock((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytpatch_get_utility_with_stdout§s t FreezableMockcBsJeZdZedejjd„Zd„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. cCsa|rtƒn tdƒ|_||_tjƒ|_|tjjkrT||_n||_ dS(Ntfreeze(sfreeze( tsett _frozen_sett_funcRZt MagicMockt_mocktsentineltDEFAULTt return_valuet_frozen(tselftfrozentfuncRo((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt__init__Ìs   cCs t|_dS(s)Freeze object preventing further changes.N(tTrueRp(Rq((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyRgÔscOs2|jdk r"|j||Žn|j||ŽS(N(RjR-Rl(Rqtargstkwargs((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt__call__ØscCs¥|dkr8ytj||ƒSWq¡tk r4tSXni|dkr]ttj|dƒ|ƒS|dksx||jkrˆtj||ƒSttj|dƒ|ƒSdS(NRpRot side_effectRlRi(s return_valueRy(tobjectt__getattribute__tAttributeErrortFalsetgetattrRi(Rqtname((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyR{Ýs   cCs–|jrA||jkr+td|ƒ‚qAt|j||ƒSn|dkr`|jj|ƒn|dkrt|j||ƒStj|||ƒSdS(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 RiRoRyN(s return_values side_effect(RpRiR|tsetattrRltaddRzt __setattr__(RqRtvalue((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyR‚ês   N( R t __module__t__doc__R}R-RZRmRnRtRgRxR{R‚(((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyRf¿s     cCsttƒ}xKtjjƒD]:}|dkrtdtdtƒ}t|||ƒqqW|jƒtdtd|ƒS(Nt notificationRrRsRo(RfRtIDisplayRRut_assert_valid_callR€Rg(tdisplayRt frozen_mock((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyR\s   cs·‡fd†‰‡fd†}tƒ}xptjjƒD]_}|dkrqtdtdˆƒ}t|||ƒq7tdtd|ƒ}t|||ƒq7W|jƒtdtd|ƒS(Ncs|rˆj|ƒndS(s$Write to message to stdout. N(twrite(tmessaget unused_argst unused_kwargs(Rc(s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt _write_msgscst||ƒˆ||ŽdS(s5 Mock function for IDisplay methods. N(Rˆ(RvRw(R(s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt mock_methods R†RrRsRo(RfRR‡RRuR€Rg(RcRR‰RRŠ((RRcs6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyRbs       cOsy|r|dn|dg}i}|jddƒ|d<|jddƒ|d<|jdtƒ|dtwork_dirtlogs_dirt cert_pathtfullchain_patht chain_pathtserver(Rq((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyRFs (R R„R…R(((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyR BscCs–ttƒtjƒ}|jƒ||f}tjdtd|ƒ}|jƒ|jƒ|ƒ|j ƒ|j ƒ|j ƒ|j dks’t ‚dS(s­Grab a lock for lock_path and call func. :param callable func: object to call after acquiring the lock :param str lock_path: path to file or directory to lock R]RviN(RRtmultiprocessingt ConditiontacquiretProcesst hold_locktstarttwaittnotifytreleaseRtexitcodetAssertionError(Rst lock_pathtcvt child_argstchild((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyt lock_and_callTs         cCsoddlm}tjj|ƒr4|j|ƒ}n|j|ƒ}|jƒ|jƒ|j ƒ|j ƒdS(s¶Acquire a file lock at lock_path and wait to release it. :param multiprocessing.Condition cv: condition for synchronization :param str lock_path: path to the file lock iÿÿÿÿ(tlockN( tcertbotRÁR R tisdirtlock_dirtLockFileR³R¸R·R¹(R½R¼RÁtmy_lock((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pyRµns   (1R…R±R R RFRšR5tcryptography.hazmat.backendsRtcryptography.hazmat.primitivesRRZRtjosepyR%R`t six.movesRRÂRRRRRtcertbot.displayR•RRRR"R$R'R.R0R8RXR^R-ReRzRfR\RbRˆtTestCaseR™R RÀRµ(((s6/usr/lib/python2.7/dist-packages/certbot/tests/util.pytsP                   ,  F