← Back
Editing: test_vhost.cpython-38.pyc
U s�@g� � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z ddlmZ dd lmZmZmZ dd lmZ dd lmZ G d d � d e�ZG dd� de�ZG dd� de�ZdS )z! Tests for L{twisted.web.vhost}. � )�absolute_import�division)� gatherResults)�TestCase)� NOT_FOUND)� NoResource)�Data)�Site)� _HostResource�NameVirtualHost�VHostMonsterResource)� DummyRequest)�_renderc @ s e Zd ZdZdd� ZdS )�HostResourceTestsz% Tests for L{_HostResource}. c C s� t dd�}t dd�}t dd�}|�d|� |�d|� t� }t� }t dd�|_|�d|� tddg�}d d dg|_t|�|_ d d � |_ d |_ |� d|�}| � |t � tddg�}|� d|�}| � |t� dS )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 data� s uri datas test datas tests uris default datas baz.coms bar� httpc S s dS )NF� r r r �=/usr/lib/python3/dist-packages/twisted/web/test/test_vhost.py�<lambda>/ � z1HostResourceTests.test_getChild.<locals>.<lambda>r N)r ZputChildr r �default�addHostr �prepathr Zsite�isSecureZhost�getChild�assertIsInstancer )�selfZbazrootZbazuriZbaztestZhr�root�request�stepr r r � test_getChild s& z HostResourceTests.test_getChildN��__name__� __module__� __qualname__�__doc__r r r r r r s r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�NameVirtualHostTestsz' Tests for L{NameVirtualHost}. c C s2 t � }tdd�|_tdg�}| �|�|�d� dS )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. s correct resultr N)r r r r � assertEqualZrender)r �virtualHostResourcer r r r �test_renderWithoutHost? s �z+NameVirtualHostTests.test_renderWithoutHostc s6 t � }tdg�� t|� �}� �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. r c s �� � jt� d S �N�r' Z responseCoder ��ignored�r r r r � cbRenderedU s zHNameVirtualHostTests.test_renderWithoutHostNoDefault.<locals>.cbRendered)r r r � addCallback�r r( �dr/ r r. r � test_renderWithoutHostNoDefaultL s z4NameVirtualHostTests.test_renderWithoutHostNoDefaultc s� t � }|�dtdd�� tdg�}|j�dd� t||�}� fdd�}|�||� tdg�}|j�dd� t||�}� fd d�}|�||� t||g�S ) 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.org� winnerr r � hostc s � � d�|j�d� d S �Nr r4 �r' �join�written)r- r �r r r r/ g s z<NameVirtualHostTests.test_renderWithHost.<locals>.cbRendereds example.org:8000c s � � d�|j�d� d S r6 r7 )r- �requestWithPortr: r r r/ o s ) r r r r �requestHeaders� addRawHeaderr r0 r )r r( r r2 r/ r; Z dWithPortr r: r �test_renderWithHost[ s z(NameVirtualHostTests.test_renderWithHostc sP t � }tdd�|_tdg�� � j�dd� t|� �}� �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 datar r r5 � example.comc s �� d�� j�d� d S )Nr r? r7 r, r. r r r/ � s zCNameVirtualHostTests.test_renderWithUnknownHost.<locals>.cbRendered)r r r r r< r= r r0 r1 r r. r �test_renderWithUnknownHostv s z/NameVirtualHostTests.test_renderWithUnknownHostc sD t � }tdg�� � j�dd� t|� �}� �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. r r5 r@ c s �� � jt� d S r* r+ r, r. r r r/ � s zLNameVirtualHostTests.test_renderWithUnknownHostNoDefault.<locals>.cbRendered�r r r< r= r r0 r1 r r. r �#test_renderWithUnknownHostNoDefault� s z8NameVirtualHostTests.test_renderWithUnknownHostNoDefaultc sD t � }tdg�� � j�dd� t|� �}� �fdd�}|�|� |S )zs L{NameVirtualHost.render} doesn't echo unescaped HTML when present in the I{Host} header. r r5 s <b>example</b>.comc s �� d� j� d S )Nz<b>)Z assertNotInr9 r, r. r r r/ � s z@NameVirtualHostTests.test_renderWithHTMLHost.<locals>.cbRenderedrB r1 r r. r �test_renderWithHTMLHost� s z,NameVirtualHostTests.test_renderWithHTMLHostc C s� t � }tdd�}d|_tdd�}|�d|� |�d|� tg �}|j�dd� dg|_| �|� d|�t � | � |jdg� | � |j g � tg �}|j�dd� dg|_| �|� d|�t� | � |jg � | � |j dg� d 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 datar Ts norm datas leaf.example.orgs norm.example.orgr5 r N) r r ZisLeafr r r< r= r r r r r' Zpostpath)r r( Z leafResourceZ normResourcer r r r r � s, ��z"NameVirtualHostTests.test_getChildN) r" r# r$ r% r) r3 r> rA rC rD r r r r r r& ; s r& c @ s e Zd ZdZdd� ZdS )�VHostMonsterResourceTestsz, Tests for L{VHostMonsterResource}. c C s^ t � }tg �}| �|�d|�t� | �|�� � tg �}| �|�d|�t� | �|�� � dS )z� L{VHostMonsterResource.getChild} returns I{_HostResource} and modifies I{Request} with correct L{Request.isSecure}. r s httpsN)r r r r r Z assertFalser Z assertTrue)r Zvhmr r r r r � s z'VHostMonsterResourceTests.test_getChildNr! r r r r rE � s rE N)r% Z __future__r r Ztwisted.internet.deferr Ztwisted.trial.unittestr Ztwisted.web.httpr Ztwisted.web.resourcer Ztwisted.web.staticr Ztwisted.web.serverr Ztwisted.web.vhostr r r Ztwisted.web.test.test_webr Ztwisted.web.test._utilr r r& rE r r r r �<module> s $
Save File
Cancel