3 (h@spddlmZddlmZddlmZddlmZddlm Z GdddeZ edd id Gd d d eeZ d S))StaticLiveServerTestCase)modify_settings)SeleniumTestCase)MiddlewareMixin)ugettextc@seZdZdZddZdS) CSPMiddlewarez5The admin's JavaScript should be compatible with CSP.cCs d|d<|S)Nzdefault-src 'self'zContent-Security-Policy)selfrequestresponserr>/tmp/pip-install-q3hcpn_q/Django/django/contrib/admin/tests.pyprocess_response szCSPMiddleware.process_responseN)__name__ __module__ __qualname____doc__r rrrr rsrappendz(django.contrib.admin.tests.CSPMiddleware)Z MIDDLEWAREc@seZdZdddddgZd(ddZd)d d Zd*d d Zd+ddZd,ddZd-ddZ d.ddZ ddZ d/ddZ ddZ ddZdd Zd!d"Zd#d$Zd%d&Zd'S)0AdminSeleniumTestCasezdjango.contrib.adminzdjango.contrib.authzdjango.contrib.contenttypeszdjango.contrib.sessionszdjango.contrib.sites cCs"ddlm}||j|j|dS)a= Helper function that blocks the execution of the tests until the specified callback returns a value that is not falsy. This function 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)Zselenium.webdriver.support.waitrseleniumZuntil)r callbacktimeoutrrrr wait_untils z AdminSeleniumTestCase.wait_untilcs|jfdd|dS)z Block until `num_windows` are present (usually 2, but can be overridden in the case of pop-ups opening other pop-ups). cst|jkS)N)lenZwindow_handles)d) num_windowsrr )sz6AdminSeleniumTestCase.wait_for_popup..N)r)r rrr)rr wait_for_popup$sz$AdminSeleniumTestCase.wait_for_popupcCs4ddlm}ddlm}|j|j|j|f|dS)zX Helper function that blocks until a CSS selector is found on the page. r)By)expected_conditionsN)selenium.webdriver.common.byr selenium.webdriver.supportr!rZpresence_of_element_located CSS_SELECTOR)r css_selectorrr ecrrr wait_for+s   zAdminSeleniumTestCase.wait_forcCs6ddlm}ddlm}|j|j|j|f||dS)zZ Helper function that blocks until the text is found in the CSS selector. r)r )r!N)r"r r#r!rZtext_to_be_present_in_elementr$)r r%textrr r&rrr wait_for_text6s    z#AdminSeleniumTestCase.wait_for_textcCs6ddlm}ddlm}|j|j|j|f||dS)z[ Helper function that blocks until the value is found in the CSS selector. r)r )r!N)r"r r#r!rZ#text_to_be_present_in_element_valuer$)r r%r(rr r&rrr wait_for_valueBs    z$AdminSeleniumTestCase.wait_for_valuecCs4ddlm}ddlm}|j|j|j|f|dS)zS Block until the element described by the CSS selector is visible. r)r )r!N)r"r r#r!rZvisibility_of_element_locatedr$)r r%rr r&rrr wait_until_visibleNs   z(AdminSeleniumTestCase.wait_until_visiblecCs4ddlm}ddlm}|j|j|j|f|dS)zU Block until the element described by the CSS selector is invisible. r)r )r!N)r"r r#r!rZinvisibility_of_element_locatedr$)r r%rr r&rrr wait_until_invisibleYs   z*AdminSeleniumTestCase.wait_until_invisiblec Cs4ddlm}y|jdWn|k r.YnXdS)z7 Block until page has started to load. r)TimeoutExceptionbodyN)selenium.common.exceptionsr-r')r r-rrr wait_page_loadedds  z&AdminSeleniumTestCase.wait_page_loaded/admin/cCsj|jjd|j|f|jjd}|j||jjd}|j|td}|jjd|j|jdS)z8 Helper function to log into the admin. z%s%susernamepasswordzLog inz//input[@value="%s"]N) rgetZlive_server_urlZfind_element_by_nameZ send_keys_Zfind_element_by_xpathZclickr0)r r2r3Z login_urlZusername_inputZpassword_inputZ login_textrrr admin_loginrs    z!AdminSeleniumTestCase.admin_logincCs|jjd||fS)z Helper function that returns the value for the CSS attribute of an DOM element specified by the given selector. Uses the jQuery that ships with Django. z$return django.jQuery("%s").css("%s"))rZexecute_script)r selector attributerrr get_css_valuesz#AdminSeleniumTestCase.get_css_valuecCsPddlm}|jjd|}x|D]}|jd|kr"|Sq"W|d||fdS)z Returns the