File: //lib/python3/dist-packages/twisted/web/test/__pycache__/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 $