Y]c@sddlZddlZddlZddlZddlZddlZddlZddlmZej dde Z ej dZ yddl Z Wnek rdZ nXejjeZejjedZejjedZdejfd YZd e jfd YZd ejfd YZdddYZdejfdYZdejfdYZ de fdYZ!de fdYZ"dZ#de fdYZ$dZ%e&dkre%ndS(iN(t test_supportt mimetoolst deprecatedt threadings keycert.pems keycert2.pemtLoopbackHttpServercBs eZdZdZdZRS(s^HTTP server w/ a few modifications that make it useful for loopback testing purposes. cCs*tjj||||jjddS(Ng?(tBaseHTTPServert HTTPServert__init__tsockett settimeout(tselftserver_addresstRequestHandlerClass((s+lib/python2.7/test/test_urllib2_localnet.pyR s cCs,|jj\}}|jd||fS(s"BaseHTTPServer method, overridden.g$@(RtacceptR (R trequesttclient_address((s+lib/python2.7/test/test_urllib2_localnet.pyt get_request)s (t__name__t __module__t__doc__RR(((s+lib/python2.7/test/test_urllib2_localnet.pyRs tLoopbackHttpServerThreadcBs)eZdZdZdZdZRS(s2Stoppable thread that runs a loopback http server.cCsVtjj|t|_tj|_d|_td||_ |j j |_ dS(NsHTTP/1.0s 127.0.0.1i(s 127.0.0.1i( RtThreadRtFalset_stoptEventtreadytprotocol_versionRthttpdt server_porttport(R trequest_handler((s+lib/python2.7/test/test_urllib2_localnet.pyR8s   cCst|_|jdS(s.Stops the webserver if it's currently running.N(tTrueRtjoin(R ((s+lib/python2.7/test/test_urllib2_localnet.pytstopCs cCs.|jjx|js)|jjqWdS(N(RtsetRRthandle_request(R ((s+lib/python2.7/test/test_urllib2_localnet.pytrunKs  (RRRRR!R$(((s+lib/python2.7/test/test_urllib2_localnet.pyR5s tBasicAuthHandlercBsleZdZdZdZdZdeefZejeZ dZ dZ dZ dZ d ZRS( s,Handler for performing Basic Authentication.ttestUserttestPasstTests%s:%scOstjj|||dS(N(RtBaseHTTPRequestHandlerR(R targstkwargs((s+lib/python2.7/test/test_urllib2_localnet.pyR\scGsdS(N((R tformatR*((s+lib/python2.7/test/test_urllib2_localnet.pyt log_message_scCs+|jd|jdd|jdS(Nis Content-types text/html(t send_responset send_headert end_headers(R ((s+lib/python2.7/test/test_urllib2_localnet.pytdo_HEADcs cCsB|jd|jdd|j|jdd|jdS(NisWWW-AuthenticatesBasic realm="%s"s Content-types text/html(R.R/tREALMR0(R ((s+lib/python2.7/test/test_urllib2_localnet.pyt do_AUTHHEADhs cCsu|jjddkr5|j|jjdn<|jjdd|jkrg|jjdn |jdS(Nt AuthorizationsNo Auth Header ReceivedsBasic s It works!(theaderst getheadertNoneR3twfiletwritet ENCODED_AUTH(R ((s+lib/python2.7/test/test_urllib2_localnet.pytdo_GETns  (RRRtUSERtPASSWDR2t USER_PASSWDtbase64t b64encodeR:RR-R1R3R;(((s+lib/python2.7/test/test_urllib2_localnet.pyR%Ss    tDigestAuthHandlercBs_eZdZdZdZdZdZdZdZdZ dZ d Z RS( s-Handler for performing digest authentication.cCs1d|_g|_i|_d|_d|_dS(Nis Test Realmtauth(t _request_numt_noncest_userst _realm_namet_qop(R ((s+lib/python2.7/test/test_urllib2_localnet.pyR}s     cCs ||_dS(N(RG(R tqop((s+lib/python2.7/test/test_urllib2_localnet.pytset_qopscCs"t|tst||_dS(N(t isinstancetdicttAssertionErrorRE(R tusers((s+lib/python2.7/test/test_urllib2_localnet.pyt set_usersscCs ||_dS(N(RF(R trealm((s+lib/python2.7/test/test_urllib2_localnet.pyt set_realmscCsA|jd7_tjt|jj}|jj||S(Ni(RCthashlibtmd5tstrt hexdigestRDtappend(R tnonce((s+lib/python2.7/test/test_urllib2_localnet.pyt_generate_noncescCs|jd}||d}|jd}i}xu|D]m}|jd\}}|j}|ddkr|ddkr|dd!}n |j}||||j |j gD]*}|j |||j |rt}qqW|s|j|StSdS(sPerforms digest authentication on the given HTTP request handler. Returns True if authentication was successful, False otherwise. If no users have been set, then digest auth is effectively disabled and this method will always return True. isProxy-AuthorizationtusernameRVN(tlenRERR5RsRftgetRDtremoveRtpatht short_pathRrtcommand(R RRbRgtauth_validatedRx((s+lib/python2.7/test/test_urllib2_localnet.pyR#s,        ( RRRRRIRNRPRWRfRrRsR#(((s+lib/python2.7/test/test_urllib2_localnet.pyRAzs        tFakeProxyHandlercBs)eZdZdZdZdZRS(sThis is a 'fake proxy' that makes it look like the entire internet has gone down due to a sudden zombie invasion. It main utility is in providing us with authentication support for testing. cOs#||_tjj|||dS(N(tdigest_auth_handlerRR)R(R R}R*R+((s+lib/python2.7/test/test_urllib2_localnet.pyRs cGsdS(N((R R,R*((s+lib/python2.7/test/test_urllib2_localnet.pyR-scCstj|jd\}}}}}}||_|jj|r|jdd|jdd|j|jj d|j|jj dndS(NthttpitOKs Content-Types text/htmlsYou've reached %s!
sFOur apologies, but our server is down due to a sudden zombie invasion.( turlparseRxRyR}R#R.R/R0R8R9(R tscmtnetlocRxtparamstquerytfragment((s+lib/python2.7/test/test_urllib2_localnet.pyR;s!  (RRRRR-R;(((s+lib/python2.7/test/test_urllib2_localnet.pyR|s  t BaseTestCasecBseZdZdZRS(cCstj|_dS(N(Rtthreading_setupt_threads(R ((s+lib/python2.7/test/test_urllib2_localnet.pytsetUpscCs|jtj|jdS(N(t doCleanupsRtthreading_cleanupR(R ((s+lib/python2.7/test/test_urllib2_localnet.pyttearDowns (RRRR(((s+lib/python2.7/test/test_urllib2_localnet.pyRs tBasicAuthTestscBs;eZdZdZdZdZdZdZdZRS(R&R't IncorrectR(cCsrtt|jd}t||_d|jj|_|jj|jjj |j |jj dS(Nc_s t||S(N(R%(R*R+((s+lib/python2.7/test/test_urllib2_localnet.pyt#http_server_with_basic_auth_handler&sshttp://127.0.0.1:%s( tsuperRRRtserverRt server_urltstartRtwaitt addCleanupR!(R R((s+lib/python2.7/test/test_urllib2_localnet.pyR#s  cCstj}|j|j|j|j|jtjtj|y|j tj |jWn@tj k r|j d|jnt k r}|nXdS(NsBasic Auth Failed for url: %s(turllib2tHTTPBasicAuthHandlert add_passwordR2RR<R=tinstall_openert build_openert assertTrueturlopent HTTPErrortfailt Exception(R tahte((s+lib/python2.7/test/test_urllib2_localnet.pyttest_basic_auth_success.s "cCsdtj}|j|j|j|j|jtjtj||j tj tj |jdS(N( RRRR2RR<tINCORRECT_PASSWDRRt assertRaisesRR(R R((s+lib/python2.7/test/test_urllib2_localnet.pyttest_basic_auth_httperror9s   ( RRR<R=RR2RRR(((s+lib/python2.7/test/test_urllib2_localnet.pyRs tProxyAuthTestscBsMeZdZdZdZdZdZdZdZdZ dZ RS( shttp://localhostttesterttest123t TestRealmcs*ttjd}j|tjjdtjdWdS(Ns pycon 2008...ishttp://localhost:%s(RRRRR(R RRRtline((s+lib/python2.7/test/test_urllib2_localnet.pyttest_iterations  c Csddddddg}dj|}|jdg|fg}tjd |j}xNt|D]@\}}|j|||d |t||t|fqgW|j|d t|dS( NsWe sgot shere s verylong i s Rishttp://localhost:%ssQFetched line number %s doesn't match expected: Expected length was %s, got %si(R RRRRt enumerateRRu(R tlinesRRRtindexR((s+lib/python2.7/test/test_urllib2_localnet.pytztest_line_iterations%N(RRRRR7RRRRRRRRRRRRR RRRRR(((s+lib/python2.7/test/test_urllib2_localnet.pyRs(          cCstjtttdS(N(Rt run_unittestRRR(((s+lib/python2.7/test/test_urllib2_localnet.pyt test_mainst__main__(('RR?RRRtunittestRQttestRt import_moduleRRRRt ImportErrorR7Rxtdirnamet__file__thereR RRRRRRR)R%RAR|tTestCaseRRRRRR R(((s+lib/python2.7/test/test_urllib2_localnet.pyts:         'z  $I *