o bB.@s.dZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZddlmZddlmZddlm Z m!Z!ddl"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,e-Z.GdddeZ/GdddeZ0dS)z Tests for L{twisted.web.tap}. N)skipIf) endpointsreactor) IReactorUNIX)FilePath) requireModule) ThreadPool) UsageError)PBServerFactory)TestCase)demo)ResourcePublisher UserDirectory) PythonScript)Site)DataFile)Options_AddHeadersResourcemakePersonalServerFactory makeService) DummyRequest) CGIScript) WSGIResourcec@s eZdZdZddZddZeee  dddZ d d Z d d Z d dZ ddZeee  dddZeee  dddZddZddZddZddZeeddudd d!Zeeddud"d#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)/ ServiceTestszD Tests for the service creation APIs in L{twisted.web.tap}. cCs: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)rmktempmakedirsr parseOptionspath)selfr optionsrr#;/usr/lib/python3/dist-packages/twisted/web/test/test_tap.py _pathOption,s zServiceTests._pathOptioncCs,|\}}||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 rr#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:rrN)rrrrrr r startService addCleanup stopServicer&servicesfactoryresourcerr' assertTrueosexistsstatS_ISSOCKst_mode)r!r portr"servicer#r#r$test_pathServerFs  zServiceTests.test_pathServercC4|\}}|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\ zServiceTests.test_cgiProcessorcCr;)z The I{--path} option creates a root resource which serves a L{PythonScript} instance for any child with the C{".epy"} extension. zfoo.epyr<N)r%r=r>r&r?rr(r#r#r$test_epyProcessorerAzServiceTests.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%r=r>r?r&rr'datatype)r!r rr=r#r#r$test_rpyProcessorns    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 rr assertIdenticalsite)r!rH serverFactoryr#r#r$test_makePersonalServerFactory~s  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. r*r+ --personalN)rrrrr,r-r.r2r3r r4r5r6r7)r!r8r"r9r#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. rK~portsrr+N) rrr3r expanduserjoinruserSocketNamer')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}. rNrN)TCP)iN)rrr'r _parseServerr!r"r#r#r$test_defaultPorts  zServiceTests.test_defaultPortcCs@t}|gd|d|dd|d|dddS)zQ If the I{--http} option is given twice, there are two listeners )--listenztcp:8001rXztcp:80028001rNr8002NrrassertInrVr#r#r$ test_twoPortsszServiceTests.test_twoPortscCst}|dtdg|d}||t||jt|t|j t ||j t | |j jtd||j j| |j jtd||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 .applicationrstartupshutdownN)rr__name__r2rrG_reactorr isinstance _threadpoolr _application application assertFalsestartedfireSystemEventjoined)r!r"rr#r#r$ test_wsgis   zServiceTests.test_wsgicCsFt}tddfD]}|t|jd|g}|t|d|q dS)zn If I{--wsgi} is given an invalid name, L{Options.parseOptions} raises L{UsageError}. z .nosuchthingzfoo.r_zNo such WSGI application: N)rrb assertRaisesr rr'str)r!r"nameexcr#r#r$test_invalidApplications z$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)rrmr rr'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. rrsslrNr443Nr\rVr#r#r$ test_HTTPSAcceptedOnAvailableSSLs z-ServiceTests.test_HTTPSAcceptedOnAvailableSSLcCs,t}|gd||dddgdS)zE When --add-header is specific, the value is parsed.  --add-headerzK1: V1rzzK2: V2 extraHeadersK1V1K2V2N)rrr'rVr#r#r$test_add_header_parsingsz$ServiceTests.test_add_header_parsingcCs\t}|gdt|}|jdjj}||t||j ddg||j t j dS)zj When --add-header is specified, the resource is a composition that adds headers. ryrr|rN) rrrr/r0r1r&rr'_headers_originalResourcer Test)r!r"r9r1r#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.0r[N)rrr flushWarningstest_noTracebacksDeprecationr'DeprecationWarninglen)r!r"warningsr#r#r$rs  z)ServiceTests.test_noTracebacksDeprecationcCs2t}|dgt|}||jdjjdS)zm Passing --display-tracebacks will enable traceback rendering on the generated Site. z--display-tracebacksrN)rrrr2r/r0displayTracebacksr!r"r9r#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)rrrrhr/r0rrr#r#r$test_displayTracebacksNotGiven's z+ServiceTests.test_displayTracebacksNotGiven)rb __module__ __qualname____doc__r%r)rr providedByrr:r@rBrFrJrLrRrWr^rlrqrrurxrrrrrr#r#r#r$r'sH                rc@seZdZddZdS)AddHeadersResourceTestscCsXttgd}tg}|d|||jdddg||jddgdS) zc When getChildWithDefault is invoked, it adds the headers to the response. )r|r)r}V3r}r~rrrN)rr rrgetChildWithDefaultr'responseHeaders getRawHeaders)r!r1requestr#r#r$test_getChildWithDefault3s  z0AddHeadersResourceTests.test_getChildWithDefaultN)rbrrrr#r#r#r$r2s r)1rr3r5unittestrtwisted.internetrrtwisted.internet.interfacesrtwisted.python.filepathrtwisted.python.reflectrtwisted.python.threadpoolrtwisted.python.usager twisted.spread.pbr twisted.trial.unittestr twisted.webr twisted.web.distribr rtwisted.web.scriptrtwisted.web.serverrtwisted.web.staticrrtwisted.web.taprrrrtwisted.web.test.requesthelperrtwisted.web.twcgirtwisted.web.wsgirobjectrgrrr#r#r#r$s4