U
{®Þh¾‰ ã @ sN d dl Z d dlmZmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d d lmZmZmZ d d
lmZ d d lmZ G d d
„ d
e ƒZG dd„ de ƒZG dd„ dee ƒZG dd„ dee ƒZG dd„ dee ƒZ G dd„ dee ƒZ G dd„ de ƒZ G dd„ dee ƒZ G d d „ d e ƒZ G d d „ d e ƒZ!G d d!„ d!e ƒZ"dS )"é N)ÚdateÚdatetimeÚ timedelta)ÚStringIO)Ú
Permission)Ú
management)ÚTestCase)Úreverse)ÚAdminURLFinder)ÚQueryÚQueryDailyHitsÚSearchPromotion)Úget_search_promotions)ÚWagtailTestUtilsc @ s4 e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd S ) ÚTestSearchPromotionsc C sP t jjt d¡dddd | t d¡j ¡ d¡ | t d¡j ¡ j d¡ d S )Nú root pageé r úFirst search promotion©ÚqueryÚpage_idÚ
sort_orderÚ description)
r
ÚobjectsÚcreater ÚgetÚ assertEqualÚ
editors_picksÚcountÚfirstr ©Úself© r" úo/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/contrib/search_promotions/tests.pyÚtest_search_pick_create s üz,TestSearchPromotions.test_search_pick_createc C sP t jjt d¡dddd | t d¡j ¡ d¡ | t d¡j ¡ j d¡ d S )Nr úhttps://wagtail.orgr r ©r Úexternal_link_urlr r r )
r
r r r r r r r r r' r r" r" r# Ú test_search_pick_link_create$ s üþz1TestSearchPromotions.test_search_pick_link_createc C sž t jjt d¡dddd t jjt d¡dddd t jjt d¡ddd d
| t d¡j ¡ d ¡ | t d¡j ¡ j d¡ | t d¡j
¡ j d¡ d S ) Nr r r úFirst search pickr é zLast search pickr% zMiddle search pickr& é ) r
r r r r r r r r r Úlastr r" r" r# Útest_search_pick_ordering5 s6 üüüþ ÿz.TestSearchPromotions.test_search_pick_orderingc C sš t d¡}tdƒD ] }| ¡ qtjjt d¡dddd tjjt d¡ddd d | t ¡ ¡ d¡ t d¡j
¡ }| |j d¡ | |j
jd¡ d S )
NÚpopularé r* r úPopular search pickr Z
uninterestingr zUninteresting search pick)r r ÚrangeÚadd_hitr
r r r Úget_most_popularr r r r r Úhits)r! Ú popularQueryÚiZ
popular_picksr" r" r# Útest_get_most_popularS s,
üüþz*TestSearchPromotions.test_get_most_popularc
C sü t ¡ }|tdd }|tdd }t d¡}tdƒD ]}|j|d q6t d¡}|j|d |j|d |j|d tjj t d¡dddd tjj t d¡ddd
d tj
|d }| | ¡ d ¡ t d¡j
}| ¡ } | | jd
¡ | | jjd
¡ d S )Nr* ©Údaysr/ r. ©r Zsurpriser r0 r zSurprising search pick)Z
date_sincer r+ )r Útodayr r r r1 r2 r
r r r3 r r r r r r r4 )
r! ZTODAYZ TWO_DAYS_AGOZ
FIVE_DAYS_AGOr5 r6 Z
surpriseQueryZmost_popular_queriesr Zsurprise_picksr" r" r# Útest_get_most_popular_sincem s>
üü þz0TestSearchPromotions.test_get_most_popular_sinceN)Ú__name__Ú
__module__Ú __qualname__r$ r( r- r7 r< r" r" r" r# r s
r c @ s e Zd Zdd„ Zdd„ ZdS )Ú"TestGetSearchPromotionsTemplateTagc C sR t jjt d¡dddd}t jjt d¡dddd ttdƒƒ}| ||g¡ d S )Nr r r r) r zroot page againúSecond search pick)r
r r r r Úlistr r )r! ÚpickÚ search_picksr" r" r# Ú'test_get_search_promotions_template_tag” s üü zJTestGetSearchPromotionsTemplateTag.test_get_search_promotions_template_tagc C s t td ƒƒ}| |g ¡ d S ©N)rB r r )r! rD r" r" r# Ú1test_get_search_promotions_with_none_query_string© s zTTestGetSearchPromotionsTemplateTag.test_get_search_promotions_with_none_query_stringN)r= r> r? rE rG r" r" r" r# r@ “ s r@ c @ s| e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
d„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd „ Zd S ) Ú TestSearchPromotionsIndexViewc C s | ¡ | _d S rF )ÚloginÚuserr r" r" r# ÚsetUp¯ s z#TestSearchPromotionsIndexView.setUpc C s. | j tdƒ¡}| |jd¡ | |d¡ d S )Nú wagtailsearchpromotions:indexéÈ ú"wagtailsearchpromotions/index.html©Úclientr r r Ú status_codeÚassertTemplateUsed©r! Úresponser" r" r# Ú test_simple² s z)TestSearchPromotionsIndexView.test_simplec C sF | j tdƒddi¡}| |jd¡ | |jd d¡ | |d¡ d S )NrL ÚqÚHellorM Ú query_stringz1Sorry, no promoted results match "Hello")rP r r r rQ ÚcontextÚassertContainsrS r" r" r# Ú test_search· s ÿþz)TestSearchPromotionsIndexView.test_searchc C s~ t jjt d¡dd t jjt d¡dd | j tdƒddi¡}| |jd¡ | |j d d¡ |
|d
¡ |
|d ¡ d S ) Nzsearch promotion queryr )r r r% )r r' rL rV rM rX z.zN) r
r r r r rP r r rQ rY rZ rS r" r" r# Útest_search_with_results s& þþ ÿ þz6TestSearchPromotionsIndexView.test_search_with_resultsc C s4 t dƒD ]&}tjjt dt|ƒ ¡dddd qd S )Né2 úquery r r r) r )r1 r
r r r r Ústr©r! r6 r" r" r# Úmake_search_picks× s üz/TestSearchPromotionsIndexView.make_search_picksc C sP | ¡ | j tdƒddi¡}| |jd¡ | |d¡ | |jd jd¡ d S )NrL Úpr* rM rN Úpage_obj) ra rP r r r rQ rR rY ÚnumberrS r" r" r# Útest_paginationà s
z-TestSearchPromotionsIndexView.test_paginationc C s0 | ¡ | j tdƒddi¡}| |jd¡ d S )NrL rb z Hello World!é” ©ra rP r r r rQ rS r" r" r# Útest_pagination_invalidì s ÿz5TestSearchPromotionsIndexView.test_pagination_invalidc C s0 | ¡ | j tdƒddi¡}| |jd¡ d S )NrL rb iŸ† rf rg rS r" r" r# Ú test_pagination_out_of_rangeö s ÿz:TestSearchPromotionsIndexView.test_pagination_out_of_rangec sÒ t dƒ}| ¡ | j |¡ | d¡ | j |¡ W 5 Q R X t ¡ ‰tdƒD ]\}t dt |ƒ ¡‰ ‡ fdd„tdƒD ƒ}‡ ‡fdd„tdƒD ƒ}t
j |¡ t
j |¡ qN| d¡ | j |¡ W 5 Q R X d S ) NrL é é r^ c s* g | ]"}t ˆ |d d |d|› d‘qS )r* r z Search pick r )r
©Ú.0Új)r r" r# Ú
s ú
üzBTestSearchPromotionsIndexView.test_num_queries..r/ c s$ g | ] }t ˆ ˆt|d |d‘qS )r8 )r r r4 )r r rl ©r r; r" r# ro s ÿ)r ra rP r ZassertNumQueriesr r; r1 r r_ r
r Z bulk_creater )r! Úurlr6 Zpromosr4 r" rp r# Útest_num_queries s$
ù þ z.TestSearchPromotionsIndexView.test_num_queriesc C sš | ¡ tjjt d¡dddd tjjt d¡dddd | j tdƒ¡}| |j d ¡ |
|d
¡ | |j d d j d¡ | |j d d j d¡ d S ) Nz aaargh snaker r zooh, it's a snaker z
beloved snakezbeloved snake goes ssSSSSrL rM rN Úqueries)
ra r
r r r r rP r r rQ rR rY rX rS r" r" r# Ú'test_results_are_ordered_alphabetically# s$ üü zETestSearchPromotionsIndexView.test_results_are_ordered_alphabeticallyc C s t ¡ }tdƒD ] }t d¡j|t|d d qtjj t d¡dddd tjj t d¡d dd
d | j
t d ƒ¡}| j |d d
d |
t d¡jd¡ | |j¡}t ddd}t ddd}| d|› d¡}| d|› d¡}| |¡ | |¡ |
t d¡j ¡ d ¡ d S )Né
r r8 r: r r r) r r* rA rL z | 10 | T)Úhtmlzwagtailadmin_pages:edit)r ©Úargs)r* úa[href="ú"])r r; r1 r r r2 r r
r r rP r rZ r r4 Úget_soupÚcontentÚ
select_oneZassertIsNotNoner r ) r! r; r6 rT ÚsoupZroot_page_edit_urlZhomepage_edit_urlZroot_page_edit_linkZhomepage_edit_linkr" r" r# Ú test_multiple_searchpromotions= s4 üü
z r? rK rU r[ r\ ra re rh ri rr rt r r… r r‘ r" r" r" r# rH ® s
# @rH c @ sL e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
d„ Z dd„ Z
dS )ÚTestSearchPromotionsAddViewc C s | ¡ d S rF ©rI r r" r" r# rK Ä s z!TestSearchPromotionsAddView.setUpc C s. | j tdƒ¡}| |jd¡ | |d¡ d S )Nr rM z wagtailsearchpromotions/add.htmlrO rS r" r" r# rU Ç s z'TestSearchPromotionsAddView.test_simplec C sZ dddddddddœ}| j tdƒ|¡}| |td ƒ¡ | t d¡jjdd
¡ ¡ d S ) NÚtestr r éè Ú rW )rX úeditors_picks-TOTAL_FORMSúeditors_picks-INITIAL_FORMSúeditors_picks-MAX_NUM_FORMSúeditors_picks-0-DELETEúeditors_picks-0-ORDERúeditors_picks-0-pageúeditors_picks-0-descriptionr rL )r ©
rP Úpostr rŽ Ú
assertTruer r r ÚfilterÚexists©r! Z post_datarT r" r" r# Ú test_postÌ s ø
z%TestSearchPromotionsAddView.test_postc
C s\ dddddddddd œ }| j td
ƒ|¡}| |td ƒ¡ | t d¡jjdd ¡ ¡ d S )
Nr” r r r• r– r% ÚWagtailrW ) rX r— r˜ r™ rš r› ú!editors_picks-0-external_link_urlú"editors_picks-0-external_link_textr r rL )r' rž r£ r" r" r# Ú test_post_with_external_linkà s" ÷ ÿÿz8TestSearchPromotionsAddView.test_post_with_external_linkc C sH dddddœ}| j tdƒ|¡}| |jd¡ | |jd d d d¡ d S ) Nr” r r• )rX r— r˜ r™ r rM Úsearchpicks_formsetú@Please specify at least one recommendation for this search term.©rP rŸ r r rQ ÚassertFormSetErrorrY r£ r" r" r# Ú!test_post_without_recommendationsù s üüz=TestSearchPromotionsAddView.test_post_without_recommendationsc C sT ddddddddddd œ
}| j td
ƒ|¡}| |jd ¡ | |jd d d d
¡ d S )Nr” r r r• r– r% r¥ rW )
rX r— r˜ r™ rš r› rœ r¦ r§ r r rM r© z4Please only select a page OR enter an external link.r« r£ r" r" r# Ú%test_post_with_page_and_external_link s& ö üzATestSearchPromotionsAddView.test_post_with_page_and_external_linkc C sN ddddddddœ}| j tdƒ|¡}| |jd ¡ | |jd
d d d ¡ d S ) Nr” r r r• r– rW )rX r— r˜ r™ rš r› r r rM r© z.You must recommend a page OR an external link.r« r£ r" r" r# Ú test_post_missing_recommendation$ s ù üz r? rK rU r¤ r¨ r r® r¯ r° r" r" r" r# r’ Ã s r’ c @ s< e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
S )Ú TestSearchPromotionsEditViewc C sF | ¡ | _t d¡| _| jjjdddd| _| jjjdddd| _d S )NrW r r ú Root page)r r r r* ÚHomepage) rI rJ r r r r r Ú search_pickÚ
search_pick_2r r" r" r# rK P s
ÿ ÿz"TestSearchPromotionsEditView.setUpc C sb | j td| jjfd¡}| |jd¡ | |d¡ t| j ƒ}d| jj }| |
| j ¡|¡ d S )Nú wagtailsearchpromotions:editrw rM z!wagtailsearchpromotions/edit.htmlz/admin/searchpicks/%d/) rP r r r Úidr rQ rR r
rJ Z get_edit_urlr´ )r! rT Z
url_finderZ expected_urlr" r" r# rU \ s ÿ
z(TestSearchPromotionsEditView.test_simplec C sv dddd| j jdddd| jjddddd œ}| j td
| jjfd |¡}| |td ƒ¡ | t j
j | j jd
j d¡ d S )NrW r* r• r– r r úDescription has changedr³ ©rX r— r˜ r™ zeditors_picks-0-idrš r› rœ r zeditors_picks-1-idzeditors_picks-1-DELETEzeditors_picks-1-ORDERzeditors_picks-1-pagezeditors_picks-1-descriptionr¶ rw rL ©r· )
r´ r· rµ rP rŸ r r rŽ r r
r r r r£ r" r" r# r¤ g s0 ò ÿþz&TestSearchPromotionsEditView.test_postc C s
| t d¡j ¡ d | j¡ | t d¡j ¡ d | j¡ dddd| jjdddd| jjddddd œ}| j t
d
| j jfd |¡}| |t
d ƒ¡ | t
jj| jjd
jd¡ | t
jj| jjd
jd¡ | t d¡j ¡ d | j¡ | t d¡j ¡ d | j¡ d S )NrW r r r* r• r– r² r³ r¹ r¶ rw rL rº )r r r r Úallr´ rµ r· rP rŸ r r rŽ r
r r r£ r" r" r# Útest_post_reorder† s@ ò ÿ ÿ ÿ z.TestSearchPromotionsEditView.test_post_reorderc C s’ dddd| j jdddd| jjddddd œ}| j td
| jjfd |¡}| |td ƒ¡ | t j
j | jjd
¡ ¡ |
t j
j | j jd
¡ ¡ d S )NrW r* r• r– r r r² r³ r¹ r¶ rw rL rº )r´ r· rµ rP rŸ r r rŽ Ú assertFalser
r r¡ r¢ r r£ r" r" r# Ú test_post_delete_recommendation¯ s0 ò ÿÿzTestSearchPromotionsEditView.test_post_without_recommendationsN) r= r> r? rK rU r¤ r¼ r¾ r r" r" r" r# r± O s )!r± c @ s$ e Zd Zdd„ Zdd„ Zdd„ ZdS )Ú TestSearchPromotionsDeleteViewc C s@ | ¡ t d¡| _| jjjddd| _| jjjddd| _d S )NrW r r² )r r r* r³ )rI r r r r r r´ rµ r r" r" r# rK ñ s ÿ ÿz$TestSearchPromotionsDeleteView.setUpc C s8 | j td| jjfd¡}| |jd¡ | |d¡ d S )Nú wagtailsearchpromotions:deleterw rM z+wagtailsearchpromotions/confirm_delete.html)rP r r r r· r rQ rR rS r" r" r# rU ý s
ÿz*TestSearchPromotionsDeleteView.test_simplec C sf | j td| jjfd¡}| |tdƒ¡ | tjj | j
jd ¡ ¡ | tjj | j jd ¡ ¡ d S )NrÀ rw rL rº )
rP rŸ r r r· rŽ r½ r
r r¡ rµ r¢ r´ rS r" r" r# r¤ s ÿÿÿz(TestSearchPromotionsDeleteView.test_postN)r= r> r? rK rU r¤ r" r" r" r# r¿ ð s r¿ c @ s e Zd Zdd„ ZdS )Ú#TestGarbageCollectManagementCommandc C s t ¡ }|tdd ¡ }|tdd ¡ }g }tdƒD ],}t d|› ¡}|j|d | |j ¡ q8g }tdƒD ],}t d|› ¡}|j|d | |j ¡ qrg }tdƒD ]@}t d|› ¡}|j|d t
j j |dd d
d | |j ¡ q¬t
jd tƒ d
| tj j|d ¡ ¡ | tj j||d ¡ ¡ | tj j|d ¡ d¡ | tj j||d ¡ d¡ | tj j|d ¡ d¡ | tj j||d ¡ d ¡ d S )Né r8 r ru zHello r: zWorld zFoo bar r ZTestr Z searchpromotions_garbage_collect)Ústdout)Zid__in)r Z query_id__in)r Únowr r r1 r r r2 Úappendr· r
r r r Z call_commandr r½ r¡ r¢ r r r ) r! ZnowdtZ old_hit_dateZrecent_hit_dateZquery_ids_to_be_deletedr6 rV Zrecent_query_idsZpromoted_query_idsr" r" r# Ú test_garbage_collect_command s` ÿ ÿÿ ÿü ÿüz@TestGarbageCollectManagementCommand.test_garbage_collect_commandN)r= r> r? rÆ r" r" r" r# rÁ s rÁ c @ s8 e Zd Zdd„ Zi fdd„Zdd„ Zdd„ Zd d
„ Zd S ) ÚTestQueryChooserViewc C s | ¡ d S rF r“ r r" r" r# rK R s zTestQueryChooserView.setUpc C s | j d|¡S )Nz#/admin/searchpicks/queries/chooser/)rP r )r! Úparamsr" r" r# r U s zTestQueryChooserView.getc C sF | ¡ }| |jd¡ | |d¡ t |j ¡ ¡}| |d d¡ d S )NrM z4wagtailsearchpromotions/queries/chooser/chooser.htmlÚstepZchooser)r r rQ rR ÚjsonÚloadsr| Údecode)r! rT Z
response_jsonr" r" r# rU X s ÿz TestQueryChooserView.test_simplec C s | ddi¡}| |jd¡ d S )NrV rW rM ©r r rQ rS r" r" r# r[ a s z TestQueryChooserView.test_searchc C s< | ddi¡}| |jd¡ | ddi¡}| |jd¡ d S )Nrb r rM i' rf rÍ rS r" r" r# re e s z$TestQueryChooserView.test_paginationN)r= r> r? rK r rU r[ re r" r" r" r# rÇ Q s
rÇ c @ s$ e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestHitCounterc C s | t d¡jd¡ d S )NrW r )r r r r4 r r" r" r# Ú test_no_hitso s zTestHitCounter.test_no_hitsc C s& t d¡ ¡ | t d¡jd¡ d S )NrW r )r r r2 r r4 r r" r" r# Útest_hitr s zTestHitCounter.test_hitc C s4 t dƒD ]}t d¡ ¡ q| t d¡jd¡ d S )Nru rW )r1 r r r2 r r4 r` r" r" r# Ú test_10_hitsy s zTestHitCounter.test_10_hitsN)r= r> r? rÏ rÐ rÑ r" r" r" r# rÎ n s rÎ c @ s, e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd S )
Ú TestQueryStringNormalisationc C s t d¡| _d S )Nz Hello World! )r r r r r" r" r# rK ƒ s z"TestQueryStringNormalisation.setUpc C s | t| jƒd¡ d S )Nz hello world!)r r_ r r r" r" r# Útest_normalisation† s z/TestQueryStringNormalisation.test_normalisationc C s. ddddg}|D ]}| | jt |¡¡ qd S )Nz Hello World!zHello World! z
hello world!z Hello world! )r r r r ©r! rs r r" r" r# Útest_equivalent_queries‰ s üz4TestQueryStringNormalisation.test_equivalent_queriesc C s, dddg}|D ]}| | jt |¡¡ qd S )NZ
HelloWorldz HelloWorld! Hello World! rW )ZassertNotEqualr r r rÔ r" r" r# Útest_different_queries” s ýz3TestQueryStringNormalisation.test_different_queriesN)r= r> r? rK rÓ rÕ rÖ r" r" r" r# rÒ ‚ s rÒ c @ s e Zd Zdd„ ZdS )ÚTestQueryPopularityc C s\ t dƒD ]}t d¡ ¡ qt dƒD ]}t d¡ ¡ q$t ¡ }| | ¡ d¡ | |d t d¡¡ | |d t d¡¡ t dƒD ]}t d ¡ ¡ q„| | ¡ d¡ | |d t d¡¡ | |d t d ¡¡ | |d t d¡¡ t d
ƒD ]}t d¡ ¡ qò| | ¡ d¡ | |d t d¡¡ | |d t d¡¡ | |d t d ¡¡ d S ) Nr+ zunpopular queryru z
popular queryr* r r r/ zlittle popular queryrk )r1 r r r2 r3 r r )r! r6 Zpopular_queriesr" r" r# Útest_query_popularity s( z)TestQueryPopularity.test_query_popularityN)r= r> r? rØ r" r" r" r# r× Ÿ s r× )#rÊ r r r Úior Zdjango.contrib.auth.modelsr Z django.corer Z django.testr Z django.urlsr Z wagtail.admin.admin_url_finderr
Z(wagtail.contrib.search_promotions.modelsr r r
ZKwagtail.contrib.search_promotions.templatetags.wagtailsearchpromotions_tagsr Zwagtail.test.utilsr r r@ rH r’ r± r¿ rÁ rÇ rÎ rÒ r× r" r" r" r# Ú s2 }
"(9