File: //lib/python3/dist-packages/django/contrib/admin/__pycache__/tests.cpython-310.pyc
o
�a� � @ 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
�</usr/lib/python3/dist-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)�
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 d&dd��Z
d(dd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S ))�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)� selenium.webdriver.support.waitr �selenium�until)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 � switch_to�windowr" �wait_page_ready)r
r% r r
r$ r � wait_for_and_switch_to_popup& s z2AdminSeleniumTestCase.wait_for_and_switch_to_popupc C �4 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.supportr1 r �presence_of_element_located� CSS_SELECTOR�r
� css_selectorr r/ �ecr
r
r �wait_for0 � �z AdminSeleniumTestCase.wait_forc C �6 ddl m} ddlm} | �|�|j|f|�|� dS )zD
Block until the text is found in the CSS selector.
r r. r0 N)r2 r/ r3 r1 r � text_to_be_present_in_elementr5 �r
r7 �textr r/ r8 r
r
r �
wait_for_text; �
��z#AdminSeleniumTestCase.wait_for_textc C r; )zE
Block until the value is found in the CSS selector.
r r. r0 N)r2 r/ r3 r1 r �#text_to_be_present_in_element_valuer5 r= r
r
r �wait_for_valueG r@ z$AdminSeleniumTestCase.wait_for_valuec C r- )zS
Block until the element described by the CSS selector is visible.
r r. r0 N)r2 r/ r3 r1 r � visibility_of_element_locatedr5 r6 r
r
r �wait_until_visibleS r: z(AdminSeleniumTestCase.wait_until_visiblec C r- )zU
Block until the element described by the CSS selector is invisible.
r r. r0 N)r2 r/ r3 r1 r � invisibility_of_element_locatedr5 r6 r
r
r �wait_until_invisible^ r: z*AdminSeleniumTestCase.wait_until_invisiblec C s | � dd� |� dS )z1
Block until the page is ready.
c S s | � d�dkS )Nzreturn document.readyState;�complete)�execute_script��driverr
r
r r& n r' z7AdminSeleniumTestCase.wait_page_ready.<locals>.<lambda>N)r )r
r r
r
r r+ i s �z%AdminSeleniumTestCase.wait_page_readyc c sD � ddl m} | j�d�}dV | j|�|�|d� | j|d� dS )zA
Block until a new page has loaded and is ready.
r r0 �htmlN)r )r3 r1 r �find_element_by_tag_namer � staleness_ofr+ )r
r r8 �old_pager
r
r �wait_page_loadedr s � z&AdminSeleniumTestCase.wait_page_loaded�/admin/c C s� | j �d| j|f � | j �d�}|�|� | j �d�}|�|� td�}| �� � | j �d| ��� W d � dS 1 s?w Y dS )z&
Log in to the admin.
z%s%s�username�passwordzLog inz//input[@value="%s"]N) r �get�live_server_url�find_element_by_name� send_keys�_rO �find_element_by_xpath�click)r
rQ rR � login_url�username_input�password_input�
login_textr
r
r � admin_login~ s
"�z!AdminSeleniumTestCase.admin_loginc C �* ddl m} || j�|��}|�|� dS )z�
Select the <OPTION> with the value `value` inside the <SELECT> widget
identified by the CSS selector `selector`.
r ��SelectN)� selenium.webdriver.support.uira r � find_element_by_css_selector�select_by_value�r
�selector�valuera �selectr
r
r �
select_option� � z#AdminSeleniumTestCase.select_optionc C r_ )z�
Deselect the <OPTION> with the value `value` inside the <SELECT> widget
identified by the CSS selector `selector`.
r r` N)rb ra r rc �deselect_by_valuere r
r
r �deselect_option� rj z%AdminSeleniumTestCase.deselect_optionc s~ |r | j �� �}g }|D ]
}|�|�d�� q | �||� d S | �� � | �� fdd�� W d � d S 1 s8w Y d S )Nrg c s | � � � S r )� find_elements_by_css_selectorrI ��options_selectorr
r r& � s z<AdminSeleniumTestCase._assertOptionsValues.<locals>.<lambda>)r rm r �
get_attribute� assertEqual�disable_implicit_waitr )r
ro �values�options�
actual_values�optionr
rn r �_assertOptionsValues� s
�"�z*AdminSeleniumTestCase._assertOptionsValuesc C � | � d| |� dS )z{
Assert that the <SELECT> widget identified by `selector` has the
options with the given `values`.
z %s > optionN�rw �r
rf rs r
r
r �assertSelectOptions� � z)AdminSeleniumTestCase.assertSelectOptionsc C rx )z�
Assert that the <SELECT> widget identified by `selector` has the
selected options with the given `values`.
z%s > option:checkedNry rz r
r
r �assertSelectedOptions� r| z+AdminSeleniumTestCase.assertSelectedOptionsc C s | j �|��d��|�dkS )zh
Return True if the element identified by `selector` has the CSS class
`klass`.
�classr( )r rc rp �find)r
rf �klassr
r
r �
has_css_class� s
�z#AdminSeleniumTestCase.has_css_classN)r )r r )rP )r r r �available_appsr r, r9 r? rB rD rF r+ r rO r^ ri rl rw r{ r} r� r
r
r
r r s&
r N)�
contextlibr �"django.contrib.staticfiles.testingr � django.testr �django.test.seleniumr �django.utils.deprecationr �django.utils.translationr rW r r r
r
r
r �<module> s