ó Y[c@s™dZddlZddlZddlZddlmZddlmZdej fd„ƒYZ dej fd„ƒYZ e d kr•ej ƒndS( sModule for enhancement UI.iÿÿÿÿN(terrors(tutiltAskTestcBsPeZdZd„Zd„Zed„ƒZejdƒd„ƒZ d„Z RS(sTest the ask method.cCstjtjƒdS(N(tloggingtdisabletCRITICAL(tself((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pytsetUp scCstjtjƒdS(N(RRtNOTSET(R((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyttearDownscCsddlm}||ƒS(Niÿÿÿÿ(task(tcertbot.display.enhancementsR (tclst enhancementR ((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyt_callss!certbot.display.enhancements.utilcCs2tjdf|ƒj_|j|jdƒƒdS(Nitredirect(t display_utiltOKtmenut return_valuet assertTrueR(Rt mock_util((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyt test_redirectscCs|jtj|jdƒdS(Ntunknown_enhancement(t assertRaisesRtErrorR(R((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyttest_key_errors( t__name__t __module__t__doc__RR t classmethodRtmocktpatchRR(((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyR s   t RedirectTestcBseeZdZed„ƒZejdƒd„ƒZejdƒd„ƒZejdƒd„ƒZ RS(s$Test the redirect_by_default method.cCsddlm}|ƒS(Niÿÿÿÿ(tredirect_by_default(R R"(R R"((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyR#ss!certbot.display.enhancements.utilcCs/tjdf|ƒj_|j|jƒƒdS(Ni(RRRRRR(RR((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyt test_secure(scCs/tjdf|ƒj_|j|jƒƒdS(Ni(RtCANCELRRt assertFalseR(RR((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyt test_cancel-scCs/tjdf|ƒj_|j|jƒƒdS(Ni(RRRRR%R(RR((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyt test_easy2s( RRRRRRR R#R&R'(((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyR!!s t__main__(RRtunittestRtcertbotRtcertbot.displayRRtTestCaseRR!Rtmain(((sK/usr/lib/python2.7/dist-packages/certbot/tests/display/enhancements_test.pyts