U s@g@sdZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZdd lmZdd lmZGd d d eZGdddeZGdddeZdS)z! Tests for L{twisted.web.vhost}. )absolute_importdivision) gatherResults)TestCase) NOT_FOUND) NoResource)Data)Site) _HostResourceNameVirtualHostVHostMonsterResource) DummyRequest)_renderc@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=/usr/lib/python3/dist-packages/twisted/web/test/test_vhost.py/z1HostResourceTests.test_getChild..rN)rZputChildr r defaultaddHostr prepathr ZsiteisSecureZhostgetChildassertIsInstancer)selfZbazrootZbazuriZbaztestZhrrootrequeststeprrr test_getChilds&               zHostResourceTests.test_getChildN__name__ __module__ __qualname____doc__r rrrrrsrc@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 assertEqualZrender)rvirtualHostResourcerrrrtest_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 cbRenderedUszHNameVirtualHostTests.test_renderWithoutHostNoDefault..cbRendered)r r r addCallbackrr(dr/rr.rtest_renderWithoutHostNoDefaultLs    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|jddSNrr4r'joinwritten)r-rrrrr/gsz.cbRenderedsexample.org:8000csd|jddSr6r7)r-requestWithPortr:rrr/os) r rrr requestHeaders addRawHeaderrr0r)rr(rr2r/r;Z dWithPortrr:rtest_renderWithHost[s        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 datarrr5 example.comcsdjddS)Nrr?r7r,r.rrr/szCNameVirtualHostTests.test_renderWithUnknownHost..cbRendered)r rrr r<r=rr0r1rr.rtest_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. rr5r@csjtdSr*r+r,r.rrr/szLNameVirtualHostTests.test_renderWithUnknownHostNoDefault..cbRenderedr r r<r=rr0r1rr.r#test_renderWithUnknownHostNoDefaults   z8NameVirtualHostTests.test_renderWithUnknownHostNoDefaultcsDt}tdgjddt|}fdd}|||S)zs L{NameVirtualHost.render} doesn't echo unescaped HTML when present in the I{Host} header. rr5sexample.comcsdjdS)Nz)Z assertNotInr9r,r.rrr/sz@NameVirtualHostTests.test_renderWithHTMLHost..cbRenderedrBr1rr.rtest_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.orgr5rN) r rZisLeafrr r<r=rrrrr'Zpostpath)rr(Z leafResourceZ normResourcerrrrr s,    z"NameVirtualHostTests.test_getChildN) r"r#r$r%r)r3r>rArCrDr rrrrr&;s 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 assertFalserZ assertTrue)rZvhmrrrrr sz'VHostMonsterResourceTests.test_getChildNr!rrrrrEsrEN)r%Z __future__rrZtwisted.internet.deferrZtwisted.trial.unittestrZtwisted.web.httprZtwisted.web.resourcerZtwisted.web.staticrZtwisted.web.serverr Ztwisted.web.vhostr r r Ztwisted.web.test.test_webr Ztwisted.web.test._utilrrr&rErrrrs        $