OwlCyberSecurity - MANAGER
Edit File: tests.cpython-39.pyc
a ���eL! � @ s| d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl m Z G dd� de �Z ed d id �G d d � d ee��ZdS )� )�contextmanager)�StaticLiveServerTestCase)�modify_settings)�SeleniumTestCase)�MiddlewareMixin)�gettextc @ s e Zd ZdZdd� ZdS )� CSPMiddlewarez5The admin's JavaScript should be compatible with CSP.c C s d|j d< |S )Nzdefault-src 'self'zContent-Security-Policy)�headers)�self�request�response� r �[/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/contrib/admin/tests.py�process_response s z CSPMiddleware.process_responseN)�__name__� __module__� __qualname__�__doc__r r r r r r s r �appendz(django.contrib.admin.tests.CSPMiddleware)Z MIDDLEWAREc @ s� e Zd Zg d�Zd(dd�Zd)dd�Zd*dd �Zd+d d �Zd,d d �Zd-dd�Z d.dd�Z d/dd�Z e d0dd��Z d1dd�Zdd� Zdd � Zd2d d �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd S )3�AdminSeleniumTestCase)zdjango.contrib.adminzdjango.contrib.authzdjango.contrib.contenttypeszdjango.contrib.sessionszdjango.contrib.sites� c C s" ddl m} || j|��|� dS )a% Block the execution of the tests until the specified callback returns a value that is not falsy. This method can be called, for example, after clicking a link or submitting a form. See the other public methods that call this function for more details. r )� WebDriverWaitN)Z selenium.webdriver.support.waitr �seleniumZuntil)r �callback�timeoutr r r r � wait_until s z AdminSeleniumTestCase.wait_until� c s6 | � � fdd�|� | jj�| jjd � | �� dS )z� Block until `num_windows` are present and are ready (usually 2, but can be overridden in the case of pop-ups opening other pop-ups). Switch the current window to the new pop-up. c s t | j�� kS �N)�len�window_handles)�d�� num_windowsr r �<lambda>- � zDAdminSeleniumTestCase.wait_for_and_switch_to_popup.<locals>.<lambda>���N)r r Z switch_toZwindowr �wait_page_ready)r r" r r r! r � wait_for_and_switch_to_popup' s z2AdminSeleniumTestCase.wait_for_and_switch_to_popupc C s4 ddl m} ddlm} | �|�|j|f�|� dS )zB Block until a CSS selector is found on the page. r ��By��expected_conditionsN)� selenium.webdriver.common.byr) �selenium.webdriver.supportr+ r Zpresence_of_element_located� CSS_SELECTOR�r � css_selectorr r) �ecr r r �wait_for1 s �z AdminSeleniumTestCase.wait_forc C s6 ddl m} ddlm} | �|�|j|f|�|� dS )zD Block until the text is found in the CSS selector. r r( r* N)r, r) r- r+ r Z text_to_be_present_in_elementr. �r r0 �textr r) r1 r r r � wait_for_text< s �z#AdminSeleniumTestCase.wait_for_textc C s6 ddl m} ddlm} | �|�|j|f|�|� dS )zE Block until the value is found in the CSS selector. r r( r* N)r, r) r- r+ r Z#text_to_be_present_in_element_valuer. r3 r r r �wait_for_valueH s ��z$AdminSeleniumTestCase.wait_for_valuec C s4 ddl m} ddlm} | �|�|j|f�|� dS )zS Block until the element described by the CSS selector is visible. r r( r* N)r, r) r- r+ r Z visibility_of_element_locatedr. r/ r r r �wait_until_visibleV s �z(AdminSeleniumTestCase.wait_until_visiblec C s4 ddl m} ddlm} | �|�|j|f�|� dS )zU Block until the element described by the CSS selector is invisible. r r( r* N)r, r) r- r+ r Z invisibility_of_element_locatedr. r/ r r r �wait_until_invisiblea s �z*AdminSeleniumTestCase.wait_until_invisiblec C s | � dd� |� dS )z1 Block until the page is ready. c S s | � d�dkS )Nzreturn document.readyState;Zcomplete)Zexecute_script�Zdriverr r r r# q s �z7AdminSeleniumTestCase.wait_page_ready.<locals>.<lambda>N)r )r r r r r r& l s �z%AdminSeleniumTestCase.wait_page_readyc c sR ddl m} ddlm} | j�|jd�}dV | j|�|�|d� | j |d� dS )zA Block until a new page has loaded and is ready. r r( r* �htmlN)r ) r, r) r- r+ r � find_elementZTAG_NAMEr Z staleness_ofr& )r r r) r1 Zold_pager r r �wait_page_loadedv s z&AdminSeleniumTestCase.wait_page_loaded�/admin/c C s� ddl m} | j�d| j|f � | j�|jd�}|�|� | j�|jd�}|�|� td�}| � � �( | j�|j d| �� � W d � n1 s�0 Y dS ) z& Log in to the admin. r r( z%s%s�username�passwordzLog inz//input[@value="%s"]N) r, r) r �getZlive_server_urlr; �NAMEZ send_keys�_r< ZXPATHZclick)r r> r? Z login_urlr) Zusername_inputZpassword_inputZ login_textr r r � admin_login� s �z!AdminSeleniumTestCase.admin_loginc C s: ddl m} ddlm} || j�|j|��}|�|� dS )z� Select the <OPTION> with the value `value` inside the <SELECT> widget identified by the CSS selector `selector`. r r( ��SelectN)r, r) � selenium.webdriver.support.uirE r r; r. Zselect_by_value�r �selector�valuer) rE �selectr r r � select_option� s z#AdminSeleniumTestCase.select_optionc C s: ddl m} ddlm} || j�|j|��}|�|� dS )z� Deselect the <OPTION> with the value `value` inside the <SELECT> widget identified by the CSS selector `selector`. r r( rD N)r, r) rF rE r r; r. Zdeselect_by_valuerG r r r �deselect_option� s z%AdminSeleniumTestCase.deselect_optionNc C s4 ddl m} |p| j}| �t|�|j|��|� dS )z� Assert number of matches for a CSS selector. `root_element` allow restriction to a pre-selected node. r r( N)r, r) r � assertEqualr � find_elementsr. )r rH �countZ root_elementr) r r r �assertCountSeleniumElements� s �z1AdminSeleniumTestCase.assertCountSeleniumElementsc s� ddl m� |rL| j�� j��}g }|D ]}|�|�d�� q(| �||� n<| �� �$ | � � �fdd�� W d � n1 s~0 Y d S )Nr r( rI c s | � � j�� S r )rN r. r9 �r) �options_selectorr r r# � s �z<AdminSeleniumTestCase._assertOptionsValues.<locals>.<lambda>) r, r) r rN r. r � get_attributerM Zdisable_implicit_waitr )r rR �values�optionsZ actual_values�optionr rQ r �_assertOptionsValues� s �z*AdminSeleniumTestCase._assertOptionsValuesc C s | � d| |� dS )z{ Assert that the <SELECT> widget identified by `selector` has the options with the given `values`. z %s > optionN�rW �r rH rT r r r �assertSelectOptions� s z)AdminSeleniumTestCase.assertSelectOptionsc C s | � d| |� dS )z� Assert that the <SELECT> widget identified by `selector` has the selected options with the given `values`. z%s > option:checkedNrX rY r r r �assertSelectedOptions� s z+AdminSeleniumTestCase.assertSelectedOptionsc C s, ddl m} | j�|j|��d��|�dkS )zh Return True if the element identified by `selector` has the CSS class `klass`. r r( �classr% )r, r) r r; r. rS �find)r rH �klassr) r r r � has_css_class� s �����z#AdminSeleniumTestCase.has_css_class)r )r r )r )r )r )r )r )r )r )r= )N)r r r Zavailable_appsr r' r2 r5 r6 r7 r8 r& r r< rC rK rL rP rW rZ r[ r_ r r r r r s& r N)� contextlibr Z"django.contrib.staticfiles.testingr Z django.testr Zdjango.test.seleniumr Zdjango.utils.deprecationr Zdjango.utils.translationr rB r r r r r r �<module> s