ó Y[c@s†dZddlZddlZddlmZddlmZddljjZ de j fd„ƒYZ e dkr‚ej ƒndS(s Tests for new style enhancementsiÿÿÿÿN(t enhancements(tnulltEnhancementTestcBsGeZdZd„Zejƒd„ƒZd„Zd„Zd„Z RS(s@Tests for new style enhancements in certbot.plugins.enhancementscCs/tt|ƒjƒtjdtjƒ|_dS(Ntspec(tsuperRtsetUptmockt MagicMockRtAutoHSTSEnhancementt mockinstaller(tself((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyRscCsëidd6dd6idd6dd6g}tjd|ƒ5t|j_t|j_ttj|jƒƒ}WdQX|j t |ƒdƒ|j g|D]}|ddkr”|^q”ƒ|j g|D]}|ddkrÄ|^qădS( Ntautohststnamet auto_hststcli_destt somethingelset somethings#certbot.plugins.enhancements._INDEXi( RtpatchtTruetconfigR RtlistRtenabled_enhancementst assertEqualtlent assertTrue(R t_t FAKEINDEXtenabledti((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyt%test_enhancement_enabled_enhancementss   0cCs¬|jtgtj|jƒD] }|^qƒdƒ|jtj|jƒƒt|j_|jtgtj|jƒD] }|^qvƒdƒ|j tj|jƒƒdS(Nii( t assertEqualsRRRRt assertFalset are_requestedRR R(R R((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyttest_are_requested's/ /cCs`t|j_tj|jdƒ}|jtj|j|jƒƒ|j tj|j|ƒƒdS(NR( RRR Rt InstallerRRt are_supportedR R(R t unsupported((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyttest_are_supported0s cCswt|j_ddg}d}tj|||j|jƒ|j|jjjƒ|j |jjj d||fƒdS(Ns example.comswww.example.comtlineagei( RRR RtenableR Rtenable_autohststcalledRt call_args(R tdomainsR&((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyt test_enable6s  ( t__name__t __module__t__doc__Rt test_utiltpatch_get_utilityRR!R%R,(((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyR s   t__main__(R/tunittestRtcertbot.pluginsRRtcertbot.tests.utiltteststutilR0tConfigTestCaseRR-tmain(((sE/usr/lib/python2.7/dist-packages/certbot/plugins/enhancements_test.pyts  5