o f8@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZmZGd d d eZGd ddeZGdddeZdS)z! Tests for L{twisted.web.vhost}. ) gatherResults)TestCase) NOT_FOUND) NoResource)Site)Data)_render) DummyRequest)NameVirtualHostVHostMonsterResource _HostResourcec@eZdZdZddZdS)HostResourceTestsz% Tests for L{_HostResource}. cCstdd}tdd}tdd}|d||d|t}t}tdd|_|d|tddg}gd |_t||_ d d |_ d |_ | d|}| |ttddg}| d|}| |td S)z L{_HostResource.getChild} returns the proper I{Resource} for the vhost embedded in the URL. Verify that returning the proper I{Resource} required changing the I{Host} in the header. s root datasuri datas test datastestsuris default databaz.com)sbarhttprcSsdS)NFrrr=/usr/lib/python3/dist-packages/twisted/web/test/test_vhost.py-sz1HostResourceTests.test_getChild..N)rputChildr r defaultaddHostr prepathrsiteisSecurehostgetChildassertIsInstancer)selfbazrootbazuribaztesthrrootrequeststeprrr test_getChilds&               zHostResourceTests.test_getChildN__name__ __module__ __qualname____doc__r'rrrrr rc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)NameVirtualHostTestsz' Tests for L{NameVirtualHost}. cCs2t}tdd|_tdg}|||ddS)z L{NameVirtualHost.render} returns the result of rendering the instance's C{default} if it is not L{None} and there is no I{Host} header in the request. scorrect resultrN)r rrr assertEqualrender)rvirtualHostResourcer%rrrtest_renderWithoutHost=s  z+NameVirtualHostTests.test_renderWithoutHostcs6t}tdgt|}fdd}|||S)z L{NameVirtualHost.render} returns a response with a status of I{NOT FOUND} if the instance's C{default} is L{None} and there is no I{Host} header in the request. rcjtdSNr/ responseCoderignoredr%rrr cbRenderedRzHNameVirtualHostTests.test_renderWithoutHostNoDefault..cbRendered)r r r addCallbackrr1dr:rr9rtest_renderWithoutHostNoDefaultHs    z4NameVirtualHostTests.test_renderWithoutHostNoDefaultcst}|dtddtdg}|jddt||}fdd}|||tdg}|jddt||}fd d}|||t||gS) z L{NameVirtualHost.render} returns the result of rendering the resource which is the value in the instance's C{host} dictionary corresponding to the key indicated by the value of the I{Host} header in the request. s example.orgwinnerrrhostcd|jddSNrr@r/joinwritten)r8r%rrrr:ez.cbRenderedsexample.org:8000crBrCrD)r8requestWithPortrGrrr:orH) r rrr requestHeaders addRawHeaderrr<r)rr1r%r>r:rI dWithPortrrGrtest_renderWithHostXs         z(NameVirtualHostTests.test_renderWithHostcsPt}tdd|_tdgjddt|}fdd}|||S)z L{NameVirtualHost.render} returns the result of rendering the instance's C{default} if it is not L{None} and there is no host matching the value of the I{Host} header in the request. correct datarrrA example.comcsdjddS)NrrNrDr7r9rrr:rHzCNameVirtualHostTests.test_renderWithUnknownHost..cbRendered)r rrr rJrKrr<r=rr9rtest_renderWithUnknownHostvs    z/NameVirtualHostTests.test_renderWithUnknownHostcDt}tdgjddt|}fdd}|||S)z L{NameVirtualHost.render} returns a response with a status of I{NOT FOUND} if the instance's C{default} is L{None} and there is no host matching the value of the I{Host} header in the request. rrArOcr3r4r5r7r9rrr:r;zLNameVirtualHostTests.test_renderWithUnknownHostNoDefault..cbRenderedr r rJrKrr<r=rr9r#test_renderWithUnknownHostNoDefaults   z8NameVirtualHostTests.test_renderWithUnknownHostNoDefaultcrQ)zs L{NameVirtualHost.render} doesn't echo unescaped HTML when present in the I{Host} header. rrAsexample.comcsdjdS)Nz) assertNotInrFr7r9rrr:r;z@NameVirtualHostTests.test_renderWithHTMLHost..cbRenderedrRr=rr9rtest_renderWithHTMLHosts   z,NameVirtualHostTests.test_renderWithHTMLHostcCst}tdd}d|_tdd}|d||d|tg}|jdddg|_|| d|t | |jdg| |j gtg}|jdddg|_|| d|t| |jg| |j dgd S) z L{NameVirtualHost.getChild} returns correct I{Resource} based off the header and modifies I{Request} to ensure proper prepath and postpath are set. s leaf datarTs norm datasleaf.example.orgsnorm.example.orgrArN) r risLeafrr rJrKrrrrr/postpath)rr1 leafResource normResourcer%rrrr's$    z"NameVirtualHostTests.test_getChildN) r)r*r+r,r2r?rMrPrSrUr'rrrrr.8s  r.c@r )VHostMonsterResourceTestsz, Tests for L{VHostMonsterResource}. cCs^t}tg}||d|t||tg}||d|t||dS)z L{VHostMonsterResource.getChild} returns I{_HostResource} and modifies I{Request} with correct L{Request.isSecure}. rshttpsN)r r rrr assertFalser assertTrue)rvhmr%rrrr'sz'VHostMonsterResourceTests.test_getChildNr(rrrrrZr-rZN)r,twisted.internet.deferrtwisted.trial.unittestrtwisted.web.httprtwisted.web.resourcertwisted.web.serverrtwisted.web.staticrtwisted.web.test._utilrtwisted.web.test.test_webr twisted.web.vhostr r r rr.rZrrrrs        $