o
h @ s d Z ddlmZ ddlmZmZ ddlmZmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ G d d
d
ee ZG d d d eZG d
d deZG dd deZdS )zO
Tests for the search box in the admin side menu, and the custom search hooks.
)
Permission)ContextTemplate)RequestFactorySimpleTestCaseTestCase)reverse) user_has_any_page_permission)
SearchArea)WagtailTestUtilsc @ s e Zd Ze ZdddZdS )BaseSearchAreaTestCase/admin/Nc C s2 | j j||d}| j|_td}|td|iS )N)dataz.{% load wagtailadmin_tags %}{% search_other %}request)rfgetuserr renderr )self current_urlr r template r e/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/admin/tests/test_admin_search.py search_other s z#BaseSearchAreaTestCase.search_other)r
N)__name__
__module__ __qualname__r r r r r r r r s r c s, e Zd Z fddZdd Zdd Z ZS )TestSearchAreasc s t | | _d S N)supersetUploginr r __class__r r r s
zTestSearchAreas.setUpc C s t d}d}d}d}d}| j|d|i}| |jd | |d | |d | |d
| j|||||f d d | j||d
d}| j|||||f dd d | j||d
d}| j||||d |f dd d d S )Nwagtailadmin_pages:searchHellozsearch--custom-classzkzI%sMy Searchq z wagtailadmin/pages/search.htmlz$wagtailadmin/shared/search_area.htmlz%wagtailadmin/shared/search_other.htmlT)htmltrue)r' z hide-option) status_coder) )r' z
active-optionz nolink)r clientr assertEqualr+ assertTemplateUsedassertContainsassertNotContains)r
search_urlquerybase_cssicon test_stringresponser r r test_other_searches s.
z#TestSearchAreas.test_other_searchesc C s@ | }| td| | d| | d| | d| d S )Nr% /customsearch/Pages My Search)r assertInr r renderedr r r test_search_other< s
z!TestSearchAreas.test_search_other)r r r r r7 r>
__classcell__r r r# r r s r c s8 e Zd ZdZdd Z fddZdd Zdd Z ZS )
TestSearchAreaNoPagePermissionsz
Test the admin search when the user does not have permission to manage
pages. The search bar should show the first available search area instead.
c C s | | _| t| j d S r )r! r assertFalser r" r r r r K s
z%TestSearchAreaNoPagePermissions.setUpc s4 t }d|_|jtjjddd | |S )NF wagtailadmin access_admin)content_type__app_labelcodename) r create_test_user is_superuseruser_permissionsaddr objectsr save)r r r# r r rF O s
z0TestSearchAreaNoPagePermissions.create_test_userc C s, | j d}| |td | |d dS )z{
Check that the menu search area on the dashboard is not searching
pages, as they are not allowed.
r
r% zF{"_type": "wagtail.sidebar.SearchModule", "_args": ["/customsearch/"]}N)r, r r0 r r/ )r r6 r r r test_dashboardZ s z.TestSearchAreaNoPagePermissions.test_dashboardc C s@ | }| td| | d| | d| | d| dS )zHThe pages search link should be hidden, custom search should be visible.r% r8 r9 r: N)r assertNotInr r; r< r r r r> g s
z1TestSearchAreaNoPagePermissions.test_search_other) r r r __doc__r rF rL r> r? r r r# r r@ E s
r@ c @ s@ e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dS ) SearchAreaComparisonTestCasez Tests the comparison functions.c C sD t dddd| _t dddd| _t ddd d| _t dddd| _d S )
NzLabel 1z/url1d )orderzLabel 2z/url2r( z/url3i, )r
search_area1 search_area2 search_area3 search_area4r" r r r r t s z"SearchAreaComparisonTestCase.setUpc C s8 | | j| jk | | j| jk | | jdk d S N Something)
assertTruerR rU rA rS r" r r r test_eqz s z$SearchAreaComparisonTestCase.test_eqc C sp | | j| jk | | j| jk | | j| jk | t | jdk W d d S 1 s1w Y d S rV )rX rR rS rT rA assertRaises TypeErrorr" r r r test_lt "z$SearchAreaComparisonTestCase.test_ltc C s | | j| jk | | j| jk | | j| jk | | j| jk | | j| jk | t | jdk W d d S 1 sCw Y d S rV )rX rR rS rT rU rA rZ r[ r" r r r test_le "z$SearchAreaComparisonTestCase.test_lec C sp | | j| jk | | j| jk | | j| jk | t | jdk W d d S 1 s1w Y d S rV )rX rS rR rT rA rZ r[ r" r r r test_gt r] z$SearchAreaComparisonTestCase.test_gtc C s | | j| jk | | j| jk | | j| jk | | j| jk | | j| jk | t | jdk W d d S 1 sCw Y d S rV )rX rS rR rT rU rA rZ r[ r" r r r test_ge r_ z$SearchAreaComparisonTestCase.test_geN)
r r r rN r rY r\ r^ r` ra r r r r rO q s
rO N)rN django.contrib.auth.modelsr django.templater r django.testr r r django.urlsr wagtail.admin.authr wagtail.admin.searchr
wagtail.test.utilsr r r r@ rO r r r r s
-,