Your IP : 216.73.216.85
B
�|NTq � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd
lm Z dd lm Z dd lm Z dd
lm Z y ddl!Z!W n4 e"k
�rH y ddl#Z!W n dZ!Y nX Y nX y
e$� W n e%k
�rn e&Z$Y nX e
�'e(�Z)e)�*e
�+ej,�� e)�-e
j.� G dd� de/�Z0dZ1dd� Z2dd� Z3dd� Z4G dd� de5�Z6G dd� dej7�Z8G d d � d ej9�Z9G d d � d e5�Z:G d d!� d!e5�Z;G d"d#� d#ej<e:�Z<G d$d%� d%e<�Z=e=e<j>d&< G d'd(� d(e<�Z?e?e<j>d)< G d*d+� d+e<�Z@e@e<j>d,< G d-d.� d.e@�ZAeAe<j>d/< G d0d1� d1e<�ZBeBe<j>d2< G d3d4� d4e<�ZCeCe<j>d5< G d6d7� d7eB�ZDeDe<j>d8< G d9d:� d:eBejE�ZEeEe<j>d;< G d<d=� d=e<ejF�ZFeFe<j>d>< eFe<j>d?< eFe<j>d@< G dAdB� dBe5�ZGG dCdD� dDejHe:�ZHG dEdF� dFe5�ZIG dGdH� dHejJ�ZJG dIdJ� dJejK�ZKG dKdL� dLe �ZLdMdN� ZMdOdP� ZNdWdQdR�ZOdSdT� ZPdUdV� ZQdS )Xz�
Routines for testing WSGI applications with selenium.
Most interesting is :class:`~webtest.sel.SeleniumApp` and the
:func:`~webtest.sel.selenium` decorator
� N)�wraps)�app)�
simple_server)�contextmanager)�PY3)� urlencode)� binary_type)�HTTPConnection)�CannotSendRequest)�
HTTPServer)�SimpleHTTPRequestHandlerFc @ s e Zd ZdZdS )�SeleniumWarningzSpecific warning categoryN)�__name__�
__module__� __qualname__�__doc__� r r �</opt/alt/python37/lib/python3.7/site-packages/webtest/sel.pyr
= s r
)z*chromez*firefoxc s t � �� fdd��}|S )z6run test with selenium. create a new session if neededc sz t � rv| rnt| d tj�rn| d }t|jt�r8� |� qv|j}t|jj�|_z � |� W d |j�� ||_X n� | � d S )Nr )� is_available�
isinstance�unittestZTestCaser � SeleniumApp�close)�args�selfZold_app)�funcr r �wrapperI s
z#function_decorator.<locals>.wrapper)r )r r r )r r �function_decoratorG s r c c s� d| _ t� sdV n�| j}t|j�}x2| jj�� D ]"\}}|�� dkr2|j �
||� q2W t j ddd�}|�
| j� |�� |�dt|jd��}z
|V W d|j �� }tr�|�| jp�d �}|| _d| _d
| _ |�� |�� X dS ) z4A context mamanger to create a session inside a testFN)�hostzwebtest-selenium-z.html)�prefix�suffixz /__file__)�__file__zutf-8T)�updatedr �test_appr r �request�headers�items�lower�browser�addCustomRequestHeader�tempfile�NamedTemporaryFile�write�body�flush�get�dict�name�
getHtmlSourcer �encode�charsetZ_forms_indexedr )�respr# r �h�v�fd�responser- r r r �context_manager^ s.
r: c C sR t | t�rt� rN| S n8t | tj�r*t| �S t | tj�rBtt �| �S t
d| ��dS )z_A callable usable as:
- class decorator
- function decorator
- contextmanager
zUnsuported type %rN) r �typer �types� FunctionTyper �testapp� TestResponser r: � RuntimeError)�objr r r �selenium{ s
rB c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�Seleniuma� Selenium RC control aka ``browser``
A object use to manipulate DOM nodes. This object allow to use the
underlying selenium api. See Selenium `api
<http://goo.gl/IecEk>`_
You can use the original method name::
browser.fireEvent('id=#myid", 'focus')
Or a more pythonic name::
browser.fire_event('id=#myid", 'focus')
Both are equal to::
browser.execute('fireEvent', 'id=#myid', 'focus')
c C s. t j�dd�| _tt j�dd��| _d | _d S )N�
SELENIUM_HOSTz 127.0.0.1�
SELENIUM_POSTi\ )�os�environr/ r �int�port�
session_id)r r r r �__init__� s zSelenium.__init__c C s* t j�dd�| _| �| j|ddd�| _d S )NZSELENIUM_DRIVERz
*googlechrome� zcaptureNetworkTraffic=truezaddCustomRequestHeader=true)rF rG r/ �driverZgetNewBrowserSessionrJ )r �urlr r r �start� s zSelenium.startc C s | � � d | _d S )N)Z testCompleterJ )r r r r �stop� s z
Selenium.stopc G s� t dd� t|�D �|d�}| jr*| j|d< t|�}ddi}t| j| j�}z*|�dd||� |�� }|j }|�
� }W d |� � X t r�t
|d �}|�d
�r�tt|���n|�d �s�tt|���|d d � }|d
kr�|dkr�dp�dS |S )Nc S s g | ]\}}|d |f�qS )� r )�.0�i�ar r r �
<listcomp>� s z$Selenium.execute.<locals>.<listcomp>)�cmdZ sessionIdz Content-Typez0application/x-www-form-urlencoded; charset=utf-8ZPOSTz/selenium-server/driver/zutf-8zERROR: Unknown command:ZOK� )�true�falserX TF)r0 � enumeraterJ r r r rI r$ � getresponse�status�readr r �str�
startswith�AttributeError�reprr@ )r rV r �datar% �connr5 r\ r r r �execute� s,
zSelenium.executec s t |�� � �fdd�}||_|S )Nc s � gdd� | D � } �j | � S )Nc S s g | ] }t |��qS r )r^ )rR rT r r r rU � s z9Selenium.__getattr__.<locals>.wrapped.<locals>.<listcomp>)rd )r )rV r r r �wrapped� s z%Selenium.__getattr__.<locals>.wrapped)� _get_commandr )r �attrre r )rV r r � __getattr__� s zSelenium.__getattr__N) r r r r rK rO rP rd rh r r r r rC � s rC c s` e Zd ZdZg Zd� fdd� Zedd� �Zedd � �Zd
d � Z dd d
�Z
dd� Z dd� Z � Z
S )r zPSee :class:`webtest.TestApp`
SeleniumApp only support ``GET`` requests
N�0u c s� d | _ |r2tt| �j||d� | �| j � | j j}t� s<t�d | _t j
� dd�}t t j
� dd��}t
� | _| j�|� |p|i | _|| _| | _d S )N)� relative_torD z 127.0.0.1rE i\ )r �superr rK � _run_serverrN r �AssertionErrorrJ rF rG r/ rH rC �_browserrO �
extra_environ�timeoutr# ) r r rN rp ro rj �kwargsr rI )� __class__r r rK � s
zSeleniumApp.__init__c C s | j S )z*The current :class:`~webtest.sel.Selenium`)rn )r r r r r( � s zSeleniumApp.browserc C s | j jtkS )N)rn rM �HAS_UPLOAD_SUPPORT)r r r r �has_upload_support� s z SeleniumApp.has_upload_supportc C s� |j dkrt�d��| jr(d| jj |_| j�d� x0|j� � D ]"\}}|�
� dkr@| j� ||� q@W | j� |j
� | �� }|s�| �||� |s�|j}|dkr�|dk s�| �|� |S )N�GETzOnly GET are allowedz%s:%s�json)r i, i� )�methodr> ZAppErrorr �bindr r( �captureNetworkTrafficr% r& r' r) �openrN �
_get_response�
_check_status�
status_int�
_check_errors)r �reqr\ �
expect_errorsr6 r7 r5 r r r �
do_request s"
zSeleniumApp.do_requestc C s� |dkr |p| j }| j�|� t�| j�d��}g }g }�x |D �]}|d �d�rVq@tj� |d �}x"|d D ]}|d |j
|d < qpW t � }| j |_
|_ | j j|_||_||_||_t|d �|_x"|d D ]}|d |j
|d < q�W |jd
k�rd |jk�r|j�sd |_|jd
k�r4|�d|jj|f � d|jk�sJ|jd
kr@|�|� q@W |�rl|�d�}|S |dk �rz|S td��dS )z%Get responses responses from seleniumr rv rN z.icoZrequestHeaders�valuer1 Z
statusCodeZresponseHeaders�� ztext/zutf-8i� z%s %r�htmlNzNo response found)rp r( ZwaitForPageToLoadrv �loadsry �endswith�webob�RequestZblankr% r? r# r � responses�errorsr$ r^ r\ r} � content_typer4 �appendrN �pop� LookupError) r r5 rp Ztraficr� r� �dr r6 r r r r{ sD
zSeleniumApp._get_responsec
s� t � \��t� ��f� | _� � ��fdd�}tj|d�� _� j�� t���}t� d� xPt
d�D ]D}y|� dd� |� � }W n$ t
jtfk
r� t� d� Y qdX P qdW d S )
z&Run a wsgi server in a separate threadc s t j��� ttd�} | �� d S )N)Z server_classZ
handler_class)r Z make_server�
WSGIServer�WSGIRequestHandler�
serve_forever)Zhttpd)r �iprI r r �run> s
z$SeleniumApp._run_server.<locals>.run)�targetg �?�d ru z/__application__g333333�?N)�
_free_port�WSGIApplicationr � threading�Thread�threadrO r �time�sleep�ranger$ r[ �socket�errorr
)r r r� rc rS r5 r )r r� rI r rl 9 s
zSeleniumApp._run_serverc C s� | j rht| j j� }xTtd�D ]H}y|�dd� |�� }W n tjk
rX |�� P Y q X t �
d� q W dt j kr|| j
�� dt j kr�t �tt j d �tj� dS )z,Close selenium and the WSGI server if neededr� ru z/__kill_application__g333333�?ZSELENIUM_KEEP_OPEN� SELENIUM_PIDN)r r rx r� r$ r[ r� r� r r� r� rF rG r( rP �killrH �signal�SIGTERM)r rc rS r5 r r r r R s
zSeleniumApp.close)NNri NN)NN)r r r r ZappsrK �propertyr( rt r� r{ rl r �
__classcell__r r )rr r r � s
&r c sd e Zd Zddd�Zddd�Zedd� �Zed d
� �Zd d � Zee�Z � fd
d�Z
edd� �Z � Z S )r? Nc K sx | j dkr| j dk s"td| j ��t| j�rl| jd }|�dd�s`| j�||� |s`| j�|� | j�d�S t d��dS ) z�If this request is a redirect, follow that redirect. It
is an error if this is not a redirect response. Returns
another response object.
i, i� z'You can only follow 301 and 302. Not %sr r� Fz,Responses queue is empty. Nothing to follow.N)
r} �
ValueError�lenr� r/ r r| r~ r� r� )r r\ �kwr5 r r r �followg s
zTestResponse.followFc
C s8 | j j||||d�} | �� |dkr(| S | jj| |d�S )N)� description�linkid�href�indexr )r5 rp )�doc�link�clickr# r{ )
r r� r� r� �anchorr� �verbosero rp r� r r r r� x s
zTestResponse.clickc C s t | �S )N)�Forms)r r r r �forms� s zTestResponse.formsc C s
t | d�S )Nr )�Form)r r r r �form� s zTestResponse.formc C s@ | j �� }tr |�| jpd�S t|t�r8t|| jp4d�S |S d S )Nzutf-8)r( r2 r r3 r4 r r �unicode)r r- r r r �
_body__get� s
zTestResponse._body__getc s" t |t�r|�� S tt| ��|�S )N)r �Element�isElementPresentrk r? � __contains__)r �item)rr r r r� � s
zTestResponse.__contains__c C s t | �S )z+Expose a :class:`~webtest.browser.Document`)�Document)r r r r r� � s zTestResponse.doc)N)NNNNNFNN)
r r r r� r� r� r� r� r� r- r� r� r� r r )rr r r? e s
r? c @ s� e Zd ZdZdd� Zdd� Zdd� Zd&d d
�Zd'd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zeee�Zdd � Zd d � Zd d � ZeZd!d"� Zd#d$� Zd%S )(r� a� A object use to manipulate DOM nodes. This object allow to use the
underlying selenium api for the specified locator. See Selenium `api
<http://goo.gl/IecEk>`_
You can use the original method name::
element.fireEvent('focus')
Or a more pythonic name::
element.fire_event('focus')
Both are equal to::
browser.execute('fireEvent', element.locator, 'focus')
c C s |j | _ || _|| _d S )N)r( r5 �locator)r r5 r� r r r rK � s zElement.__init__c s t |�� � �fdd�}||_|S )Nc s$ � �j gdd� | D � } �jj| � S )Nc S s g | ] }t |��qS r )r^ )rR rT r r r rU � s z8Element.__getattr__.<locals>.wrapped.<locals>.<listcomp>)r� r( rd )r )rV r r r re � s z$Element.__getattr__.<locals>.wrapped)rf r )r rg re r )rV r r rh � s zElement.__getattr__c C s | � � S )z%return true is the element is present)r� )r r r r �exist� s z
Element.exist� c C sH dt | � }y| j�||� W n$ tk
rB td| |f ��Y nX | S )z+Wait for an element and return this elementz%selenium.isElementPresent(%r) || nullzCan't find %s after %sms)r^ r( ZwaitForConditionr@ )r rp �scriptr r r �wait� s z Element.waitc C s | � � �� S )z8Wait for an element, click on it and return this element)r� r� )r rp r r r �wait_and_click� s zElement.wait_and_clickc C s | � d��� }||kS )z True iif the class is present�class)rg �split)r r1 �classesr r r �hasClass� s zElement.hasClassc C s
| � d�S )z#Return the innerHTML of the elementz e.innerHTML)�eval)r r r r r� � s z Element.htmlc C s | � � S )z Return the text of the element)ZgetText)r r r r �text� s z Element.textc C s | � dt|� �S )z)Return the attribute value of the elementze.getAttribute(%r))r� r^ )r rg r r r rg � s z Element.attrc C s
| � |�S )zDrag and drop to element)ZdragAndDropToObject)r �elementr r r �
drag_and_drop� s zElement.drag_and_dropc C s | � � S )N)ZgetValue)r r r r �
value__get� s zElement.value__getc C s" t |�}d||f }| �|� d S )Nz�(function() {
s.doFireEvent(l, "focus");
s.doType(l, %s);
e.setAttribute("value", %s);
s.doFireEvent(l, "keydown");
s.doFireEvent(l, "keypress");
s.doFireEvent(l, "keyup");
}()))�
_get_valuer� )r r� r� r r r �
value__set� s zElement.value__setc G sJ dt | �d�|��d�f }y | j�|�S tk
rD t|��Y nX dS )z�Eval a javascript expression in Selenium RC. You can use the
following variables:
- s: the ``selenium`` object
- b: the ``browserbot`` object
- l: the element locator string
- e: the element itself
zs(function(s) {var l = %r;var b = s.browserbot; var e = b.findElement(l);var res = %s; return res || 'null';}(this))rL �;N)r^ �join�stripr( �getEvalr@ )r �exprr� r r r r� � s
z Element.evalc C s t |t�r|�� }|| �� kS )N)r r� r� )r �sr r r r� s
zElement.__contains__c C s | � � S )N)r� )r r r r � __nonzero__ s zElement.__nonzero__c C s d| j j| jf S )Nz
<%s at %s>)rr r r� )r r r r �__repr__ s zElement.__repr__c C s t | j�dd��S )N�"�')r^ r� �replace)r r r r �__str__ s zElement.__str__N)r� )r� )r r r r rK rh r� r� r� r� r� r� rg r� r� r� r� r� r� r� r� �__bool__r� r� r r r r r� � s(
r� c @ sf e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdd
d�Z ddd�Z
ddd�Z dd� Z dd� Z
d S )r� zQThe browser document. ``resp.doc.myid`` is egual to
``resp.doc.css('#myid')``c C s
|| _ d S )N)r5 )r r5 r r r rK ' s zDocument.__init__c C s t | jd| �S )Nzcss=#%s)r� r5 )r rg r r r rh * s zDocument.__getattr__c K s t |f|�}t| j|�S )z�Return an element matching ``tag``, an ``attribute`` and an
``index``. For example::
resp.doc.get('input', name='go') => xpath=//input[@name="go"]
resp.doc.get('li', description='Item') => xpath=//li[.="Item"]
)� _eval_xpathr� r5 )r �tagrq r� r r r r/ - s z Document.getc C s t | jd| �S )z4Get an :class:`~webtest.browser.Element` using xpathzxpath=%s)r� r5 )r �pathr r r �xpath7 s zDocument.xpathc C s t | jd| �S )z=Get an :class:`~webtest.browser.Element` using a css selectorzcss=%s)r� r5 )r �selectorr r r �css; s z Document.cssNc C s | j d||||d�S )z
Get a linkrT )r� �idr� r� )r/ )r r� r� r� r� r r r r� ? s
z
Document.linkc C s | j d||||d�S )zGet an input field�input)r� r� r1 r� )r/ )r r� r1 Zinputidr� r r r r� D s zDocument.inputc C s | j d|||d�S )z Get a button�button)r� r� r� )r/ )r r� Zbuttonidr� r r r r� I s zDocument.buttonc C s" t |t�r|�� S | j�t|��S )N)r r� r� r( Z
isTextPresentr� )r r� r r r r� N s
zDocument.__contains__c C s t |�S )N)r� )r r� r r r �__call__S s zDocument.__call__)NNNN)NNNN)NNN)r r r r rK rh r/ r� r� r� r� r� r� r� r r r r r� # s
r� c s. e Zd Zi Z� fdd�Zeejej�Z � Z
S )�Fieldc sJ t t| �j||� | jj| _g | _g | _g | _t| j | jj
| j d�| _
d S )N)r� r1 ) rk r� rK r� r( �optionsZselectedIndicesZ_forced_valuesr� r� r� r1 )r r rq )rr r r rK ` s
zField.__init__) r r r r� rK r� r� r� r� r� r� r r )rr r r� \ s
r� c @ s2 e Zd ZdZdd� Zdd� Zdd� Zeee�ZdS ) �Selectz Field representing ``<select>``c C s | � d| � d S )Nzvalue=%s)�select)r r� r r r � force_valuep s zSelect.force_valuec C s | � d| � d S )Nzvalue=%s)r� )r r� r r r r� s s zSelect.value__setc C s | � � S )N)ZgetSelectedValue)r r r r r� v s zSelect.value__getN) r r r r r� r� r� r� r� r r r r r� m s
r� r� c @ s2 e Zd ZdZdd� Zdd� Zdd� Zeee�ZdS ) �MultipleSelectz3Field representing ``<select multiple="multiple">``c C s6 | � � dd� |D �}x|D ]}| �d| � q W d S )Nc S s g | ]}t �|��qS r )r> �
_stringify)rR r� r r r rU � s z.MultipleSelect.force_value.<locals>.<listcomp>zvalue=%s)�removeAllSelections� addSelection)r �values�
str_valuesr7 r r r r� � s
zMultipleSelect.force_valuec C s6 | � � dd� |D �}x|D ]}| �d| � q W d S )Nc S s g | ]}t �|��qS r )r> r� )rR r� r r r rU � s z-MultipleSelect.value__set.<locals>.<listcomp>zvalue=%s)r� r� )r r� r� r7 r r r r� � s
zMultipleSelect.value__setc C s | � � }|�d�S )N�,)ZgetSelectedValuesr� )r r� r r r r� � s zMultipleSelect.value__getN) r r r r r� r� r� r� r� r r r r r� ~ s
r� Zmultiple_selectc @ s* e Zd ZdZdd� Zdd� Zeee�ZdS )�Radioz+Field representing ``<input type="radio">``c C s |r| � � n| �� d S )N)�checkZuncheck)r r� r r r r� � s
zRadio.value__setc C sb d| j | jjf }| j�|�}dd� t�d�D �}|s:d S t|�dkrN|d S td| |f ��d S )Na� (function(obj) {
var name = '%s';
var element = obj.browserbot.findElement('%s');
var elements = element.getElementsByTagName('input');
var values = [];
for (var i = 0, n = elements.length; i < n; ++i) {
element = elements[i];
if (element.name == name && element.checked) {
values.push('name='+element.value);
}
}
return values.join('&');
}(this))c S s g | ] \}}|�qS r r )rR �kr7 r r r rU � s z$Radio.value__get.<locals>.<listcomp>z name=truerQ r z"Got more than one value for %r: %s) r1 r� r� r( r� �cgi� parse_qslr� r� )r r� r� r r r r� � s
zRadio.value__getN)r r r r r� r� r� r� r r r r r� � s r� Zradioc @ s e Zd ZdZdS )�Checkboxz.Field representing ``<input type="checkbox">``N)r r r r r r r r r� � s r� Zcheckboxc @ s e Zd ZdZdS )�Textz*Field representing ``<input type="text">``N)r r r r r r r r r� � s r� r� c @ s0 e Zd ZdZdd� Zdd� ZeZeej e�Z
dS )�Filez*Field representing ``<input type="file">``c s: t � \��� ��fdd�}tj|d�}|�� d��f S )z(Run a simple server in a separate threadc s � t _t��ft �} | �� d S )N)� FileHandler�filenamer �handle_request)�server)r� r� rI r r r� � s z File._run_server.<locals>.run)r� z
http://%s:%s/)r� r� r� rO )r r� r� r� r )r� r� rI r rl � s
zFile._run_serverc C sv t |ttf�r*t|�dkr*| jgt|� }| jjj}|�|�}|\}}}|j rr| �
|�}|t j �
|�7 }| �|� d S )NrQ )r �list�tupler� r1 r� r5 r# Z_get_file_infort rl rF r� �basenameZ
attachFile)r r� r# � file_infor1 r� �contentrN r r r r� � s
zFile.value__setN) r r r r rl r� r� r� r� r� r� r r r r r� � s
r� �filec @ s e Zd ZdZdS )�Textareaz!Field representing ``<textarea>``N)r r r r r r r r r � s r Ztextareac @ s e Zd ZdZdS )�Hiddenz,Field representing ``<input type="hidden">``N)r r r r r r r r r � s r Zhiddenc @ s, e Zd ZdZdZdd� Zee�Zdd� ZdS )�Submitz=Field representing ``<input type="submit">`` and ``<button>``Fc C s d S )Nr )r r r r r� � s zSubmit.value__getc C s | j S )N)�_value)r r r r �value_if_submitted s zSubmit.value_if_submittedN) r r r r Zsettabler� r� r� r r r r r r � s
r �submitr� Zimagec @ s e Zd Zdd� Zdd� ZdS )r� c C s
|| _ d S )N)r5 )r r5 r r r rK s zForms.__init__c C s t | j|�S )N)r� r5 )r �keyr r r � __getitem__ s zForms.__getitem__N)r r r rK r r r r r r�
s r� c s: e Zd ZdZeZ� fdd�Z� fdd�Zd dd�Z� Z S )
r� zSee :class:`~webtest.Form`c st || _ |j| _|j| _t|t�r0td|d�| _ntd|d�| _| sPtd| j ��| �d�}t t
| �� |d| � d S )Nr� )r� )r� zNo form found at %sz e.innerHTMLz<form>%s</form>) r5 r# r( r rH r� r� r� r� rk r� rK )r r5 r� r� )rr r r rK s
z
Form.__init__c sd t t| ��� xP| j�� D ]B\}}t|�dkrx,t|�D ] \}}| jd|d f 7 _q8W qW d S )NrQ z[%s])rk r� �
_parse_fields�fieldsr& r� rZ r� )r r1 r rS �field)rr r r r
% s
zForm._parse_fieldsNc C sd |dkr| j �d� | j�|� |rDtd| j||d�}| j �|� n| j �| j� | jj| j |d�S )z�Submits the form. If ``name`` is given, then also select that
button (using ``index`` to disambiguate)``.
Returns a :class:`webtest.browser.TestResponse` object.
r rv r� )r� r1 r� )r5 rp )
r( ry r# Z
_make_environr� r� r� r
r{ r5 )r r1 r� ro rp r� r r r r
- s
z Form.submit)NNNN)
r r r r r� Z
FieldClassrK r
r
r� r r )rr r r� s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r� z@A WSGI middleware to handle special calls used to run a test appc C s& || _ d| _|| _d| | _d | _d S )NTz
http://%s:%s/)r r� rx rN r� )r r rx r r r rK G s
zWSGIApplication.__init__c C s� d|d kr$d| _ t�� }|||�S d|d kr�t�|�}t�� }d|_|j�d�}t|��� }|� dd|j
� t r~||_ n||_
|||�S d|d kr�t�� }|||�S | �||�S ) NZ__kill_application__Z PATH_INFOFr! ztext/html; charset=UTF-8zhttp://localhost/z
http://%s/Z__application__)r� r� ZResponser� r� �paramsr/ rz r] r� r r r� r- r )r rG Zstart_responser5 r r� r- r r r r� N s(
zWSGIApplication.__call__c C s d| j | jf S )Nz<WSGIApplication %r at %s>)r rN )r r r r r� e s zWSGIApplication.__repr__N)r r r r rK r� r� r r r r r� D s r� c @ s e Zd ZdZdd� ZdS )r� z(A WSGIRequestHandler who log to a loggerc G s$ t �d| �� | �� || f � d S )Nz%s - - [%s] %s)�log�debug�address_string�log_date_time_string)r �formatr r r r � log_messagel s z WSGIRequestHandler.log_messageN)r r r r r r r r r r� i s r� c @ s e Zd ZdZdd� ZdS )r� z A WSGIServerc C s x| j jr| �� qW d S )N)Z applicationr� r� )r r r r r� v s
zWSGIServer.serve_foreverN)r r r r r� r r r r r� s s r� c @ s e Zd ZdZdd� Zdd� ZdS )r� zHandle a simple filec C s | j S )N)r� )r r� r r r �translate_path~ s zFileHandler.translate_pathc G s$ t �d| �� | �� || f � d S )Nz%s - - [%s] %s
)r r r r )r r r r r r r � s zFileHandler.log_messageN)r r r r r r r r r r r� { s r� c C s t rt �| �S tt| ��S d S )N)rv �dumpsra r^ )r� r r r r� � s
r� c C s: d| kr6| � d�} | �d�gdd� | D � } d�| �} | S )N�_r c S s g | ] }|� � �qS r )�title)rR �cr r r rU � s z _get_command.<locals>.<listcomp>rL )r� r� r� )rV r r r rf � s
rf c K s� |sd}|d| 7 }xP|� � D ]D\}}|dkr8|�d�}|r |dkrR|d| 7 }q |d||f 7 }q W |d k r�|d|d f 7 }|S )
Nzxpath=z//%s)Zfor_�class_r r� z[.="%s"]z
[@%s="%s"]z[%s]rQ )r& r� )r� r� r� rq r� r7 r r r r� � s
r� c C sD t � t jt j�} | �d� | �� \}}| �� tj�dd�}||fS )N)rL r Z
SELENIUM_BINDz 127.0.0.1) r� �AF_INET� SOCK_STREAMrx � getsocknamer rF rG r/ )r� r� rI r r r r� � s
r� c C s� t dkrt�dt� tj�dd�} ttj�dd��}yt| |�}|� dd� W n� t
j k
r� d tjkrndS tjd }t �
d
d |g�}t|j�tjd < xPtd
�D ]D}t�d� yt| |�}|� dd� W n t
j k
r� Y q�X dS q�W dS Y nX dS )zNreturn True if the selenium module is available and a RC server is
runningFz^selenium is not available because no json module are available. Consider installing simplejsonrD z 127.0.0.1rE i\ ru �/Z SELENIUM_JAR�javaz-jarr� � g333333�?T)rv �warnings�warnr
rF rG r/ rH r r$ r� r� �
subprocess�Popenr^ �pidr� r� r� )r rI rc Zjar�prS r r r r � s2
r )NN)Rr rF r� �sysr� r� r� r< r� �shutilZloggingr# r* r r� r% � functoolsr Zwebtestr r> Zwsgirefr �
contextlibr Zwebtest.compatr r r r r
r r rv � ImportErrorZ
simplejsonr� � NameErrorr^ Z getLoggerr r Z
addHandlerZ
StreamHandler�stderrZsetLevel�DEBUG�Warningr
rs r r: rB �objectrC ZTestAppr r? r� r� r� r� r� r� r� r� r� r� r r r r� r� r� r� r� r� r� rf r� r� r r r r r �<module>
s�
Q @~9
#
1%