o i@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)NFrrraC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/web/test/test_vhost.py-sz1HostResourceTests.test_getChild..N)rZputChildr r defaultaddHostr prepathrsiteisSecurehostgetChildassertIsInstancer)selfZbazrootZbazuriZbaztesthrrootrequeststeprrr 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 resultrrN)r rrr assertEqualrenderrvirtualHostResourcer!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+Z responseCoderignoredr!rrr cbRenderedRzHNameVirtualHostTests.test_renderWithoutHostNoDefault..cbRenderedN)r r r addCallbackrr.dr6rr5rtest_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)r4r!rrrr6ez.cbRenderedsexample.org:8000cr>r?r@)r4requestWithPortrCrrr6orDN) r rrr requestHeaders addRawHeaderrr8r)rr.r!r:r6rEZ dWithPortrrCrtest_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 datarrr= example.comcsdjddS)NrrIr@r3r5rrr6rDzCNameVirtualHostTests.test_renderWithUnknownHost..cbRenderedN)r rrr rFrGrr8r9rr5rtest_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. rr=rJcr0r1r2r3r5rrr6r7zLNameVirtualHostTests.test_renderWithUnknownHostNoDefault..cbRenderedN)r r rFrGrr8r9rr5r#test_renderWithUnknownHostNoDefaults   z8NameVirtualHostTests.test_renderWithUnknownHostNoDefaultcsHt}tdg}|jddt||IdH|dd|jdS)zs L{NameVirtualHost.render} doesn't echo unescaped HTML when present in the I{Host} header. rr=sexample.comNs)r r rFrGrZ assertNotInrArBr-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.orgr=rN) r rZisLeafrr rFrGrrrrr+Zpostpath)rr.Z leafResourceZ normResourcer!rrrr#s$    z"NameVirtualHostTests.test_getChildN) r%r&r'r(r/r;rHrKrLrMr#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 Z assertFalser assertTrue)rZvhmr!rrrr#sz'VHostMonsterResourceTests.test_getChildNr$rrrrrNr)rNN)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 rr*rNrrrrs        $