U 2ܟj@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@seZdZdZddZdS)HostResourceTestsz% Tests for L{_HostResource}. cCstdd}tdd}tdd}|d||d|t}t}tdd|_|d|tddg}d d dg|_t||_ d d |_ d |_ | d|}| |ttddg}| d|}| |tdS)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 datasbaz.comsbarhttpcSsdS)NFrrr?/tmp/pip-unpacked-wheel-4b529toj/twisted/web/test/test_vhost.py-z1HostResourceTests.test_getChild..rN)rZputChildr r defaultaddHostr prepathrsiteisSecurehostgetChildassertIsInstancer)selfZbazrootZbazuriZbaztesthrrootrequeststeprrr test_getChilds&               zHostResourceTests.test_getChildN__name__ __module__ __qualname____doc__r!rrrrr sr c@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 resultrrN)r rrr assertEqualrenderrvirtualHostResourcerrrrtest_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. rcsjtdSNr(Z responseCoderignoredrrrr cbRenderedRszHNameVirtualHostTests.test_renderWithoutHostNoDefault..cbRendered)r r r addCallbackrr+dr2rr1rtest_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.orgwinnerrrhostcsd|jddSNrr7r(joinwritten)r0rrrrr2esz.cbRenderedsexample.org:8000csd|jddSr9r:)r0requestWithPortr=rrr2os) r rrr requestHeaders addRawHeaderrr3r)rr+rr5r2r>Z dWithPortrr=rtest_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 datarrr8 example.comcsdjddS)NrrBr:r/r1rrr2szCNameVirtualHostTests.test_renderWithUnknownHost..cbRendered)r rrr r?r@rr3r4rr1rtest_renderWithUnknownHostvs    z/NameVirtualHostTests.test_renderWithUnknownHostcsDt}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. rr8rCcsjtdSr-r.r/r1rrr2szLNameVirtualHostTests.test_renderWithUnknownHostNoDefault..cbRendered)r r r?r@rr3r4rr1r#test_renderWithUnknownHostNoDefaults   z8NameVirtualHostTests.test_renderWithUnknownHostNoDefaultcsFt}tdg}|jddt||IdH|dd|jdS)zs L{NameVirtualHost.render} doesn't echo unescaped HTML when present in the I{Host} header. rr8sexample.comNs)r r r?r@rZ assertNotInr;r<r*rrrtest_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.orgr8rN) r rZisLeafrr r?r@rrrrr(Zpostpath)rr+Z leafResourceZ normResourcerrrrr!s$    z"NameVirtualHostTests.test_getChildN) r#r$r%r&r,r6rArDrErFr!rrrrr'8s  r'c@seZdZdZddZdS)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 Z assertFalser assertTrue)rZvhmrrrrr!sz'VHostMonsterResourceTests.test_getChildNr"rrrrrGsrGN)r&Ztwisted.internet.deferrZtwisted.trial.unittestrZtwisted.web.httprZtwisted.web.resourcerZtwisted.web.serverrZtwisted.web.staticrZtwisted.web.test._utilrZtwisted.web.test.test_webr Ztwisted.web.vhostr r r r r'rGrrrrs        $