a
D øgá ã @ s´ d dl Z d dlmZmZmZ d dlmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZmZmZ d d
lm Z d dl m Z d dl m!Z! d dl"m#Z# G dd„ de ƒZ$G dd„ de ƒZ%G dd„ de#e!e ƒZ&G dd„ de#e!e ƒZ'G dd„ de#e!e ƒZ(G dd „ d e!e ƒZ)G d d „ d e ƒZ*G d d „ d e!e ƒZ+G d!d"„ d"e ƒZ,G d#d$„ d$e ƒZ-G d%d&„ d&e ƒZ.G d'd(„ d(eƒZ/G d)d*„ d*eƒZ0dS )+é N)ÚdateÚdatetimeÚ timedelta)ÚBytesIOÚStringIO)Ú
Permission)Ú
management)ÚTestCase)Úreverse)Útimezone)Ú
load_workbook)ÚAdminURLFinder)ÚBaseReportViewTestCase)ÚQueryÚQueryDailyHitsÚSearchPromotion)Úget_search_promotions)Úregistry)ÚWagtailTestUtils)ÚAdminTemplateTestUtilsc @ 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( úf/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/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_ordering: s4 üüüþÿ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é r0 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_popularX 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 )Nr0 ©Údaysr5 r4 ©r Zsurpriser r6 r zSurprising search pick)Z
date_sincer r1 )r Útodayr r r! r7 r8 r r r r9 r" r$ r# r% r r r: )
r' ZTODAYZ TWO_DAYS_AGOZ
FIVE_DAYS_AGOr; r< Z
surpriseQueryZmost_popular_queriesr# Zsurprise_picksr( r( r) Útest_get_most_popular_sincer s>
üü þz0TestSearchPromotions.test_get_most_popular_sinceN)Ú__name__Ú
__module__Ú __qualname__r* r. r3 r= rB 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' ZpickÚ 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)rH r r" )r' rI r( r( r) Ú1test_get_search_promotions_with_none_query_string® s zTTestGetSearchPromotionsTemplateTag.test_get_search_promotions_with_none_query_stringN)rC rD rE rJ rL r( r( r( r) rF ˜ s rF 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 rK ©ÚloginÚuserr& r( r( r) ÚsetUp´ s z#TestSearchPromotionsIndexView.setUpc C sD | j tdƒ¡}| |jd¡ | |d¡ | dddœg|j¡ d S )Nú wagtailsearchpromotions:indexéÈ ú"wagtailsearchpromotions/index.htmlÚ úPromoted search results©ÚurlÚlabel©Úclientr! r
r" Ú status_codeÚassertTemplateUsedÚ assertBreadcrumbsItemsRenderedÚcontent©r' Úresponser( r( r) Ú test_simple· s
þz)TestSearchPromotionsIndexView.test_simplec C sF | j tdƒddi¡}| |jd¡ | |jd d¡ | |d¡ d S )NrR ÚqÚHellorS Ú query_stringz1Sorry, no promoted results match "Hello")r[ r! r
r" r\ ÚcontextÚassertContainsr` 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- rR rc rS re z.zN) r r r r r! r[ r
r" r\ rf rg r` 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 )r7 r r r r r! Ústr©r' r< r( r( r) Úmake_search_picksà s üz/TestSearchPromotionsIndexView.make_search_picksc C sP | ¡ | j tdƒddi¡}| |jd¡ | |d¡ | |jd jd¡ d S )NrR Úpr0 rS rT Úpage_obj© rn r[ r! r
r" r\ r] rf Únumberr` r( r( r) Útest_paginationé s
z-TestSearchPromotionsIndexView.test_paginationc C sP | ¡ | j tdƒddi¡}| |jd¡ | |d¡ | |jd jd¡ d S )NrR ro z Hello World!rS rT rp r rq r` r( r( r) Útest_pagination_invalidõ s ÿ z5TestSearchPromotionsIndexView.test_pagination_invalidc C sX | ¡ | j tdƒddi¡}| |jd¡ | |d¡ | |jd j|jd j ¡ d S )NrR ro iŸ† rS rT rp Z paginator)
rn r[ r! r
r" r\ r] rf rr Z num_pagesr` r( r( r) Ú test_pagination_out_of_range s ÿ
þz:TestSearchPromotionsIndexView.test_pagination_out_of_rangec sú t dƒ}| ¡ | j |¡ | d¡ | j |¡ W d ƒ n1 sH0 Y t ¡ ‰tdƒD ]\}t dt |ƒ ¡‰ ‡ fdd„tdƒD ƒ}‡ ‡fdd„tdƒD ƒ}t
j |¡ t
j |¡ qb| d¡ | j |¡ W d ƒ n1 sì0 Y d S ) NrR é é rk c s* g | ]"}t ˆ |d d |d|› d‘qS )r0 r z Search pick r )r ©Ú.0Új)r r( r) Ú
" s ú
üzBTestSearchPromotionsIndexView.test_num_queries..r5 c s$ g | ] }t ˆ ˆt|d |d‘qS )r> )r r r: )r r rx ©r rA r( r) r{ + s ÿ)r
rn r[ r! ZassertNumQueriesr rA r7 r rl r r Z bulk_creater )r' rX r< Zpromosr: r( r| 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 ssSSSSrR rS rT Úqueries)
rn r r r r r! r[ r
r" r\ r] rf re r` r( r( r) Ú'test_results_are_ordered_alphabetically7 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 r> r@ r r r/ r r0 rG rR z | 10 | T)Úhtmlzwagtailadmin_pages:edit)r ©Úargs)r0 úa[href="ú"])r rA r7 r r! r8 r r r r r[ r
rg r" r: Úget_soupr_ Ú
select_oneZassertIsNotNoner# r$ ) r' rA r< ra ÚsoupZroot_page_edit_urlZhomepage_edit_urlZroot_page_edit_linkZhomepage_edit_linkr( r( r) Ú test_multiple_searchpromotionsQ s4 üü
z ðÿ ÿÿ ÿÿz9TestSearchPromotionsEditView.test_post_with_external_linkc C s’ dddd| j jdddd| jjddddd œ}| j td
| jjfd |¡}| |td ƒ¡ | t j
j | jjd
¡ ¡ |
t j
j | j jd
¡ ¡ d S )Nrd r0 r¤ rU r r rÅ rÇ rð rí r‚ rR ró )rà rÕ rÌ r[ r® r
r r› Ú assertFalser r r° r± r¯ r× r( r( r) Ú test_post_delete_recommendation_ s. òÿÿzTestSearchPromotionsEditView.test_post_without_recommendationsc C sž dddd| j jdddd| jjddddd d
d œ}| j td | jjfd
|¡}| |jd¡ | |j
d dd d¡ | |j
d d d g ¡ | |j
d dd g ¡ d S )Nrd r0 r¤ rU r r rï r+ rº rÇ )re r¦ r§ r¨ rñ r© rª r« r¬ rò r» r¼ r½ r¾ r¿ rÀ rí r‚ rS rÐ rà rô r× r( r( r) rá Ÿ s: ðÿüzBTestSearchPromotionsEditView.test_post_with_page_and_external_linkc C sš dddd| j jddd| jjdddd d
d œ}| j td | jjfd
|¡}| |jd¡ | |j
d dd d¡ | |j
d d d g ¡ | |j
d dd g ¡ d S )Nrd r0 r¤ rU r rï r r+ rº rÇ )re r¦ r§ r¨ rñ r© rª r¬ rò r» r¼ r¾ r¿ rÀ rí r‚ rS rÐ râ rô r× r( r( r) rã  s6 òÿüz=TestSearchPromotionsEditView.test_post_missing_recommendationc C sœ dddd| j jdddd| jjdddd d
d œ}| j td | jjfd
|¡}| |jd¡ | |j
d ddd¡ | |j
d dd g ¡ | |j
d d d g ¡ d S )Nrd r0 r¤ rU r r rï rä rº rÇ )re r¦ r§ r¨ rñ r© rª r« r¬ rò r» r¼ r¾ r¿ rÀ rí r‚ rS rÐ r- rå rô r× r( r( r) ræ ã s8 ñÿüz r r€ zHello r@ zWorld zFoo bar r ZTestr Z searchpromotions_garbage_collect)Ústdout)Zid__in)r Z query_id__in)r Únowr r r7 r r! r8 ÚappendrÕ r r r r Z call_commandr r÷ r° r± r r" r$ ) r' ZnowdtZ old_hit_dateZrecent_hit_dateZquery_ids_to_be_deletedr< rc Zrecent_query_idsZpromoted_query_idsr( r( r) Ú test_garbage_collect_command sT ÿÿÿÿüÿüz@TestGarbageCollectManagementCommand.test_garbage_collect_commandN)rC rD rE 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 rK )rO r& r( r( r) rQ Ö s zTestQueryChooserView.setUpc C s | j d|¡S )Nz#/admin/searchpicks/queries/chooser/)r[ r! )r' Úparamsr( r( r) r! Ù s zTestQueryChooserView.getc C sF | ¡ }| |jd¡ | |d¡ t |j ¡ ¡}| |d d¡ d S )NrS z4wagtailsearchpromotions/queries/chooser/chooser.htmlÚstepZchooser)r! r" r\ r] ÚjsonÚloadsr_ Údecode)r' ra Z
response_jsonr( r( r) rb Ü s ÿz TestQueryChooserView.test_simplec C s | ddi¡}| |jd¡ d S )Nrc rd rS ©r! r" r\ r` r( r( r) rh å s z TestQueryChooserView.test_searchc C s< | ddi¡}| |jd¡ | ddi¡}| |jd¡ d S )Nro r rS i' i” r
r` r( r( r) rs é s z$TestQueryChooserView.test_paginationN)rC rD rE rQ r! rb rh rs r( r( r( r) r Õ s
r c @ s$ e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestHitCounterc C s | t d¡jd¡ d S )Nrd r )r" r r! r: r& r( r( r) Ú test_no_hitsó s zTestHitCounter.test_no_hitsc C s& t d¡ ¡ | t d¡jd¡ d S )Nrd r )r r! r8 r" r: r& r( r( r) Útest_hitö s zTestHitCounter.test_hitc C s4 t dƒD ]}t d¡ ¡ q| t d¡jd¡ d S )Nr€ rd )r7 r r! r8 r" r: rm r( r( r) Ú test_10_hitsý s zTestHitCounter.test_10_hitsN)rC rD rE r r
r r( r( r( r) r ò 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) rQ s z"TestQueryStringNormalisation.setUpc C s | t| jƒd¡ d S )Nz hello world!)r" rl r r& r( r( r) Útest_normalisation
s z/TestQueryStringNormalisation.test_normalisationc C s* g d¢}|D ]}| | jt |¡¡ q d S )N)z Hello World!zHello World! z
hello world!z Hello world! )r" r r r! ©r' r~ r r( r( r) Útest_equivalent_queries
s z4TestQueryStringNormalisation.test_equivalent_queriesc C s* g d¢}|D ]}| | jt |¡¡ q d S )N)Z
HelloWorldz HelloWorld! Hello World! rd )ZassertNotEqualr r r! r r( r( r) Útest_different_queries s z3TestQueryStringNormalisation.test_different_queriesN)rC rD rE rQ 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 ) Nr1 zunpopular queryr€ z
popular queryr0 r r r5 zlittle popular queryrw )r7 r r! r8 r9 r" r$ )r' r< Zpopular_queriesr( r( r) Útest_query_popularity$ s( z)TestQueryPopularity.test_query_popularityN)rC rD rE r r( r( r( r) r # s r c @ sD e Zd ZdZedd„ ƒZdd„ Zdd„ Zdd „ Zd
d „ Z d d
„ Z
dS )ÚTestQueryHitsReportViewú$wagtailsearchpromotions:search_termsc C s` t d¡| _| j ¡ | j ¡ | j ¡ t d¡ t d¡ ¡ t d¡}| ¡ | ¡ d S )NzA query with three hitsza query with no hitszA query with one hitzA query with two hits)r r! r r8 )r' r r( r( r) Ú
setUpTestDataM s
z%TestQueryHitsReportView.setUpTestDatac C s® | ¡ }| |jd¡ | |d¡ | |d¡ | dddœg|j¡ | |j¡}| d¡}| dd „ |D ƒd
d gd d
gddgddgg¡ | |d¡ | |¡ |
|d¡ d S )NrS z%wagtailadmin/reports/base_report.htmlz-wagtailadmin/reports/base_report_results.htmlrU z Search termsrW zmain trc S s g | ]}d d„ | d¡D ƒ‘qS )c S s g | ]}|j ¡ ‘qS r( ©ÚtextÚstrip©ry Úcellr( r( r) r{ k ó zBTestQueryHitsReportView.test_simple...zth,td©Úselect©ry Útrr( r( r) r{ k r z7TestQueryHitsReportView.test_simple..úSearch term(s)ÚViewsúa query with three hitsÚ3úa query with two hitsÚ2úa query with one hitÚ1úThere are no results.zSearch terms - Wagtail) r! r" r\ r] ZassertBreadcrumbsr_ r† r ÚassertNotContainsÚ assertActiveFilterNotRenderedZassertPageTitle)r' ra rˆ Útrsr( r( r) rb Y s0 þ
þ
üþ
z#TestQueryHitsReportView.test_simplec C sH d| j _| j ¡ | j j tjjddd¡ | ¡ }| |t dƒ¡ d S )NFr r‘ r’ r• )
rP r— r˜ r™ rš r r r! r› r
r` r( r( r) Ú test_get_with_no_permissionsx s
ÿÿz4TestQueryHitsReportView.test_get_with_no_permissionsc C sB | j ddid}| |jd¡ | ¡ ¡ ¡ }| |g d¢¡ d S )NÚexportÚcsv©r rS )zSearch term(s),Viewsza query with three hits,3za query with two hits,2za query with one hit,1)r! r" r\ Úgetvaluer Ú
splitlines)r' ra Z
data_linesr( r( r) Útest_csv_export… s þz'TestQueryHitsReportView.test_csv_exportc C sp | j ddid}| |jd¡ | ¡ }tt|ƒdd }dd„ |jD ƒ}| |d d
gd d gd
dgddgg¡ d S )Nr0 Zxlsxr2 rS )ÚfilenameZSheet1c S s g | ]}d d„ |D ƒ‘qS )c S s g | ]
}|j ‘qS r( )Úvaluer r( r( r) r{ ™ r zGTestQueryHitsReportView.test_xlsx_export...r( )ry Úrowr( r( r) r{ ™ r z.r# r$ r% r1 r' r0 r) r )r! r" r\ r3 r r Úrows)r' ra Z
workbook_dataZ worksheetZ
cell_arrayr( r( r) Útest_xlsx_export” s üþz(TestQueryHitsReportView.test_xlsx_exportc C sâ ddgddgddggddgddgddggddgddgddggddgddgddggdœ}| ¡ D ]~\}}| j|dZ | jd |id
}| |jd ¡ | |j¡}| d ¡}| d
d„ |D ƒ|¡ W d ƒ q^1 sÒ0 Y q^d S )Nr) r* r% r& r' r( )re z
-query_stringZ_hitsz-_hits)Úorderingr; r2 rS z
main tbody trc S s g | ]}d d„ | d¡D ƒ‘qS )c S s g | ]}|j ¡ ‘qS r( r r r( r( r) r{  r zDTestQueryHitsReportView.test_ordering...Útdr r! r( r( r) r{  r z9TestQueryHitsReportView.test_ordering..)ÚitemsZsubTestr! r" r\ r† r_ r )r' Zcasesr; Úresultsra rˆ r. r( r( r) Ú
test_ordering¤ s6 ýýýýð
þz%TestQueryHitsReportView.test_orderingN) rC rD rE Úurl_nameÚ classmethodr rb r/ r5 r: r? r( r( r( r) r J s
r c @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) ÚTestFilteredQueryHitsViewr c C s8 | ¡ | _t d¡| _t ¡ ¡ | _| jj| jd d S )NzThis will be foundr@ ) rO rP r r! Z query_hitr r r r8 r& r( r( r) rQ Ê s
z TestFilteredQueryHitsView.setUpc C s” | j ddid}| |jd¡ | |d¡ | |d¡ | | |j¡¡ | j ddid}| |jd¡ | |d¡ | |d¡ | | |j¡¡ d S )Nrc ÚFoundr2 rS úthis will be foundr+ z Not found)r! r" r\ rg r, r- r† r_ r` r( r( r) Útest_search_by_query_stringÐ s z5TestFilteredQueryHitsView.test_search_by_query_stringc C sÊ d| j jdddi}| j|d}| |jd¡ | |d¡ | |d¡ | | |j ¡d|d ¡ | j j| j j
d d|d< | j j| j j
d d|d
< | j|d}| |jd¡ | |d¡ | |d¡ d S ) NZ
hit_date_fromr )ÚdayÚmonthr2 rS rD r+ )Úyearr0 Z hit_date_to) r Úreplacer! r" r\ rg r, ZassertActiveFilterr† r_ rH )r' r ra r( r( r) Útest_filter_by_dateÝ s ÿ ÿ z-TestFilteredQueryHitsView.test_filter_by_dateN)rC rD rE r@ rQ rE rJ r( r( r( r) rB Ç s
rB )1r r r r Úior r Zdjango.contrib.auth.modelsr Z django.corer Z django.testr Z django.urlsr
Z django.utilsr Zopenpyxlr Z wagtail.admin.admin_url_finderr
Z&wagtail.admin.tests.test_reports_viewsr Z(wagtail.contrib.search_promotions.modelsr r r ZKwagtail.contrib.search_promotions.templatetags.wagtailsearchpromotions_tagsr Zwagtail.log_actionsr r´ Zwagtail.test.utilsr Z!wagtail.test.utils.template_testsr r rF rM r¡ rë rû rþ r r r r r rB r( r( r( r) Ú sH } & v N9 '}