U
{®Þ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 eƒZG d
d„ deƒZG dd„ deƒZdS )zO
Tests for the search box in the admin side menu, and the custom search hooks.
é )Ú
Permission)ÚContextÚTemplate)ÚRequestFactoryÚSimpleTestCaseÚTestCase)Úreverse)Ú user_has_any_page_permission)Ú
SearchArea)ÚWagtailTestUtilsc @ s e Zd Zeƒ Zddd„ZdS )ÚBaseSearchAreaTestCaseú/admin/Nc C s2 | j j||d}| j|_tdƒ}| td|iƒ¡S )N)Údataz.{% load wagtailadmin_tags %}{% search_other %}Úrequest)ÚrfÚgetÚuserr Úrenderr )ÚselfZ current_urlr r Útemplate© r úm/var/www/dating/data/www/fatepal.com/env/lib/python3.8/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‡ fdd„Zdd„ Zdd„ Z‡ ZS )ÚTestSearchAreasc s t ƒ ¡ | ¡ | _d S ©N)ÚsuperÚsetUpÚloginr ©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 )Núwagtailadmin_pages:searchZHellozsearch--custom-classzkzI%sMy SearchÚqéÈ z wagtailadmin/pages/search.htmlz$wagtailadmin/shared/search_area.htmlz%wagtailadmin/shared/search_other.htmlT)ÚhtmlÚtrue)r% z hide-option)Ú status_coder' )r% z
active-optionz nolink)r Úclientr Z assertEqualr) ZassertTemplateUsedÚassertContainsÚassertNotContains)r Z
search_urlÚqueryZbase_cssÚiconZ test_stringÚresponser r r Útest_other_searches s4 ÿ ÿü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 r0 r7 Ú
__classcell__r r r" r r s r c s8 e Zd ZdZdd„ Z‡ fdd„Zdd„ 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|_|j tjjddd¡ | ¡ |S )NFZ wagtailadminZ access_admin)Zcontent_type__app_labelÚcodename) r Úcreate_test_userZ is_superuserZuser_permissionsÚaddr Úobjectsr Úsave)r r r" r r r< 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 r, r r+ )r r/ 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$ r1 r2 r3 N)r Z assertNotInr r4 r5 r r r r7 g s
z1TestSearchAreaNoPagePermissions.test_search_other) r r r Ú__doc__r r< r@ r7 r8 r r r" r r9 E s
r9 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/url1éd )Ú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 ©NZ Something)Ú
assertTruerE rH r: rF r! r r r Útest_eqz s z$SearchAreaComparisonTestCase.test_eqc C sZ | | j| jk ¡ | | j| jk ¡ | | j| jk ¡ | t¡ | jdk W 5 Q R X d S rI )rJ rE rF rG r: Ú assertRaisesÚ TypeErrorr! r r r Útest_lt„ s
z$SearchAreaComparisonTestCase.test_ltc C s~ | | j| jk¡ | | j| jk¡ | | j| jk¡ | | j| jk¡ | | j| jk¡ | t¡ | jdk W 5 Q R X d S rI )rJ rE rF rG rH r: rL rM r! r r r Útest_le’ s z$SearchAreaComparisonTestCase.test_lec C sZ | | j| jk¡ | | j| jk¡ | | j| jk¡ | t¡ | jdk W 5 Q R X d S rI )rJ rF rE rG r: rL rM r! r r r Útest_gt¦ s
z$SearchAreaComparisonTestCase.test_gtc C s~ | | j| jk¡ | | j| jk¡ | | j| jk¡ | | j| jk¡ | | j| jk¡ | t¡ | jdk W 5 Q R X d S rI )rJ rF rE rG rH r: rL rM r! r r r Útest_ge´ s z$SearchAreaComparisonTestCase.test_geN)
r r r rA r rK rN rO rP rQ r r r r rB q s
rB N)rA Zdjango.contrib.auth.modelsr Zdjango.templater r Z django.testr r r Z django.urlsr Zwagtail.admin.authr Zwagtail.admin.searchr
Zwagtail.test.utilsr r r r9 rB r r r r Ú s
-,