U 2ܟjp/@sJdZddlmZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZmZddl m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1e2Z3GdddeZ4GdddeZ5dS)z Tests for L{twisted.web.tap}. ) annotationsN)cast)skipIf) endpointsreactor) IReactorCore IReactorUNIX)FilePath) requireModule) ThreadPool) UsageError)PBServerFactory)TestCase)demo)ResourcePublisher UserDirectory) PythonScript)Site)DataFile)Options_AddHeadersResourcemakePersonalServerFactory makeService) DummyRequest) CGIScript) WSGIResourcec@seZdZdZddddZddddZeee  d ddd d Z ddd d Z ddddZ ddddZ ddddZeee  d ddddZeee  d ddddZddddZddddZddddZddddZeed d!k d"ddd#d$Zeed d!kd%ddd&d'Zddd(d)Zddd*d+Zddd,d-Zddd.d/Zddd0d1Zd!S)2 ServiceTestszD Tests for the service creation APIs in L{twisted.web.tap}. ztuple[FilePath[str], File]returncCs:t|}|t}|d|jg|d}||fS)a Helper for the I{--path} tests which creates a directory and creates an L{Options} object which uses that directory as its static filesystem root. @return: A two-tuple of a L{FilePath} referring to the directory and the value associated with the C{'root'} key in the L{Options} instance after parsing a I{--path} option. --pathroot)r mktempmakedirsr parseOptionspath)selfr%optionsr!r(=/tmp/pip-unpacked-wheel-4b529toj/twisted/web/test/test_tap.py _pathOption-s zServiceTests._pathOptionNonecCs,|\}}||t||j|jdS)z The I{--path} option causes L{Options} to create a root resource which serves responses from the specified path. N)r*assertIsInstancer assertEqualr%r&r%r!r(r(r) test_path>s  zServiceTests.test_pathz0The reactor does not support UNIX domain socketscCst|}||}t}|dd|d|jgt|}|||j | |j dj j t||j dj j j|j|tj||tt|jdS)z The I{--path} option to L{makeService} causes it to return a service which will listen on the server address given by the I{--port} option. --portunix:r rN)r r"r#rr$r%r startService addCleanup stopServicer,servicesfactoryresourcerr- assertTrueosexistsstatS_ISSOCKst_mode)r&r%portr'servicer(r(r)test_pathServerGs  zServiceTests.test_pathServercCs4|\}}|dd||ddtdS)z The I{--path} option creates a root resource which serves a L{CGIScript} instance for any child with the C{".cgi"} extension. zfoo.cgiN)r*child setContentr,getChildrr.r(r(r)test_cgiProcessor]s zServiceTests.test_cgiProcessorcCs4|\}}|dd||ddtdS)z The I{--path} option creates a root resource which serves a L{PythonScript} instance for any child with the C{".epy"} extension. zfoo.epyrAN)r*rBrCr,rDrr.r(r(r)test_epyProcessorfs zServiceTests.test_epyProcessorcCsT|\}}|dd|dd}||t||jd||jddS)z The I{--path} option creates a root resource which serves the C{resource} global defined by the Python source in any child with the C{".rpy"} extension. zfoo.rpysNfrom twisted.web.static import Data resource = Data('content', 'major/minor') Ncontentz major/minor) r*rBrCrDr,rr-datatype)r&r%r!rBr(r(r)test_rpyProcessoros    zServiceTests.test_rpyProcessorcCsDttdd}t|}||t||jt||jj|dS)z L{makePersonalServerFactory} returns a PB server factory which has as its root object a L{ResourcePublisher}. sfoo barz text/plainN) rrrr,r r!rassertIdenticalsite)r&rLZ serverFactoryr(r(r)test_makePersonalServerFactorys  z+ServiceTests.test_makePersonalServerFactorycCsl|}t}|dd|dgt|}|||j|tj ||t t |j dS)z The I{--personal} option to L{makeService} causes it to return a service which will listen on the server address given by the I{--port} option. r0r1 --personalN)r"rr$rr2r3r4r8r9r%r:r;r<r=)r&r>r'r?r(r(r)test_personalServers  z ServiceTests.test_personalServercCsHt}|dgtjtjdtj}||ddd|dS)z If the I{--port} option not specified but the I{--personal} option is, L{Options} defaults the port to C{UserDirectory.userSocketName} in the user's home directory. rN~portsrr1N) rr$r9r% expanduserjoinrZuserSocketNamer-)r&r'r%r(r(r)test_defaultPersonalPaths  z%ServiceTests.test_defaultPersonalPathcCs8t}|g|t|dddddddS)zl If the I{--port} option is not specified, L{Options} defaults the port to C{8080}. rQrN)ZTCP)iN)rr$r-rZ _parseServerr&r'r(r(r)test_defaultPorts  zServiceTests.test_defaultPortcCsDt}|ddddg|d|dd|d|ddd S) zQ If the I{--http} option is given twice, there are two listeners z--listenztcp:8001ztcp:8002Z8001rQrZ8002Nrr$ZassertInrVr(r(r) test_twoPortsszServiceTests.test_twoPortscCst}|dtdg|d}||t||jt|t|j t ||j t | |j jtttd||j j| |j jtttd||j jdS)z The I{--wsgi} option takes the fully-qualifed Python name of a WSGI application object and creates a L{WSGIResource} at the root which serves that application. --wsgiz .applicationr!ZstartupshutdownN)rr$__name__r8rrKZ_reactorr isinstanceZ _threadpoolr Z _application application assertFalsestartedrrZfireSystemEventZjoined)r&r'r!r(r(r) test_wsgis zServiceTests.test_wsgicCsFt}tddfD].}|t|jd|g}|t|d|qdS)zn If I{--wsgi} is given an invalid name, L{Options.parseOptions} raises L{UsageError}. z .nosuchthingzfoo.r[zNo such WSGI application: N)rr] assertRaisesr r$r-str)r&r'nameexcr(r(r)test_invalidApplicationsz$ServiceTests.test_invalidApplicationz OpenSSL.SSLNzSSL module is available.cCs.t}|t|jdg}|d|jddS)zp An L{UsageError} is raised when C{https} is requested but there is no support for SSL. --https=443zSSL support not installedrN)rrcr r$r-args)r&r' exceptionr(r(r)test_HTTPSFailureOnMissingSSLsz*ServiceTests.test_HTTPSFailureOnMissingSSLzSSL module is not available.cCs>t}|dg|d|dd|d|dddS)zM When SSL support is present, it accepts the --https option. rhsslrQr443NrYrVr(r(r) test_HTTPSAcceptedOnAvailableSSLs z-ServiceTests.test_HTTPSAcceptedOnAvailableSSLcCs0t}|ddddg||dddgdS)zE When --add-header is specific, the value is parsed. --add-headerK1: V1K2: V2Z extraHeadersK1V1K2V2N)rr$r-rVr(r(r)test_add_header_parsingsz$ServiceTests.test_add_header_parsingcCs`t}|ddddgt|}|jdjj}||t||j ddg||j t j dS)zj When --add-header is specified, the resource is a composition that adds headers. rorprqrrrruN) rr$rr5r6r7r,rr-_headersZ_originalResourcerTest)r&r'r?r7r(r(r)test_add_header_resources z%ServiceTests.test_add_header_resourcecCsdt}|dgt|||jg}||ddt||ddd|t|ddS)z7 Passing --notracebacks is deprecated. z--notracebacksrcategorymessagez/--notracebacks was deprecated in Twisted 19.7.0rXN)rr$rZ flushWarningstest_noTracebacksDeprecationr-DeprecationWarninglen)r&r'warningsr(r(r)r~s  z)ServiceTests.test_noTracebacksDeprecationcCs2t}|dgt|}||jdjjdS)zm Passing --display-tracebacks will enable traceback rendering on the generated Site. z--display-tracebacksrN)rr$rr8r5r6displayTracebacksr&r'r?r(r(r)test_displayTracebackss z#ServiceTests.test_displayTracebackscCs0t}|gt|}||jdjjdS)zt Not passing --display-tracebacks will leave traceback rendering on the generated Site off. rN)rr$rr`r5r6rrr(r(r)test_displayTracebacksNotGiven(s z+ServiceTests.test_displayTracebacksNotGiven)r] __module__ __qualname____doc__r*r/rrZ providedByrr@rErFrJrMrOrTrWrZrbrgr rkrnrxr{r~rrr(r(r(r)r(sF               rc@seZdZddddZdS)AddHeadersResourceTestsr+rcCsZttdddg}tg}|d|||jdddg||jdd gd S) zc When getChildWithDefault is invoked, it adds the headers to the response. rrru)rsV3rsrtrrvrwN)rrrzrZgetChildWithDefaultr-ZresponseHeadersZ getRawHeaders)r&r7requestr(r(r)test_getChildWithDefault4s z0AddHeadersResourceTests.test_getChildWithDefaultN)r]rrrr(r(r(r)r3sr)6r __future__rr9r;typingrZunittestrZtwisted.internetrrZtwisted.internet.interfacesrrZtwisted.python.filepathr Ztwisted.python.reflectr Ztwisted.python.threadpoolr Ztwisted.python.usager Ztwisted.spread.pbr Ztwisted.trial.unittestrZ twisted.webrZtwisted.web.distribrrZtwisted.web.scriptrZtwisted.web.serverrZtwisted.web.staticrrZtwisted.web.taprrrrZtwisted.web.test.requesthelperrZtwisted.web.twcgirZtwisted.web.wsgirobjectr_rrr(r(r(r)s6