o
h @ 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(eZ/G d)d* d*eZ0dS )+ N)datedatetime timedelta)BytesIOStringIO)
Permission)
management)TestCase)reverse)timezone)
load_workbook)AdminURLFinder)BaseReportViewTestCase)QueryQueryDailyHitsSearchPromotion)get_search_promotions)registry)WagtailTestUtils)AdminTemplateTestUtilsc @ 4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) TestSearchPromotionsc C sP t jjtddddd | tdj d | tdj j d d S )N root page r First search promotionquerypage_id
sort_order description)
r objectscreater get assertEqual
editors_pickscountfirstr self r) g/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/contrib/search_promotions/tests.pytest_search_pick_create s z,TestSearchPromotions.test_search_pick_createc C sP t jjtddddd | tdj d | tdj 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 jjtddddd t jjtddddd t jjtdddd d
| tdj d | tdj j d | tdj
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}tdD ]}| q tjjt ddddd tjjt dddd d | t d t dj
}| |j d | |j
jd d S )
Npopular r1 r Popular search pickr
uninterestingr zUninteresting search pick)r r" rangeadd_hitr r r! r# get_most_popularr% r$ r& r r hits)r( popularQueryi
popular_picksr) r) r* test_get_most_popularX s,
z*TestSearchPromotions.test_get_most_popularc
C s t }|tdd }|tdd }td}tdD ]}|j|d qtd}|j|d |j|d |j|d tjj tddddd tjj tdddd
d tj
|d }| | d tdj
}| } | | jd
| | jjd
d S )Nr1 daysr6 r5 r surpriser r7 r zSurprising search pick)
date_sincer r2 )r todayr r r" r9 r: r r r! r; r# r% r$ r& r r r< )
r( TODAY TWO_DAYS_AGO
FIVE_DAYS_AGOr= r>
surpriseQuerymost_popular_queriesr$ surprise_picksr) r) r* test_get_most_popular_sincer s>
z0TestSearchPromotions.test_get_most_popular_sinceN)__name__
__module__ __qualname__r+ r/ r4 r@ rM r) r) r) r* r s
r c @ s e Zd Zdd Zdd ZdS )"TestGetSearchPromotionsTemplateTagc C sR t jjtddddd}t jjtddddd ttd}| ||g d S )Nr r r r0 r zroot page againSecond 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)rS r r# )r( rU r) r) r* 1test_get_search_promotions_with_none_query_string s zTTestGetSearchPromotionsTemplateTag.test_get_search_promotions_with_none_query_stringN)rN rO rP rV rX r) r) r) r* rQ s rQ 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 | | _d S rW loginuserr' r) r) r* setUp z#TestSearchPromotionsIndexView.setUpc C sD | j td}| |jd | |d | dddg|j d S )N wagtailsearchpromotions:index "wagtailsearchpromotions/index.html Promoted search resultsurllabelclientr" r
r# status_codeassertTemplateUsed assertBreadcrumbsItemsRenderedcontentr( responser) r) r* test_simple s
z)TestSearchPromotionsIndexView.test_simplec C sF | j tdddi}| |jd | |jd d | |d d S )Nr` qHellora query_stringz1Sorry, no promoted results match "Hello")ri r" r
r# rj contextassertContainsrn r) r) r* test_search s z)TestSearchPromotionsIndexView.test_searchc C s~ t jjtddd t jjtddd | jtdddi}| |jd | |j d d |
|d
|
|d d S ) Nzsearch promotion queryr )r r r, )r r. r` rq ra rs z.zN) r r r! r r" ri r
r# rj rt ru rn r) r) r* test_search_with_results s$ z6TestSearchPromotionsIndexView.test_search_with_resultsc C s4 t dD ]}tjjtdt| dddd qd S )N2 query r r r0 r )r9 r r r! r r" strr( r> r) r) r* make_search_picks s z/TestSearchPromotionsIndexView.make_search_picksc C sP | | jtdddi}| |jd | |d | |jd jd d S )Nr` pr1 ra rb page_obj r| ri r" r
r# rj rk rt numberrn r) r) r* test_pagination s
z-TestSearchPromotionsIndexView.test_paginationc C sP | | jtdddi}| |jd | |d | |jd jd d S )Nr` r} z Hello World!ra rb r~ r r rn r) r) r* test_pagination_invalid s z5TestSearchPromotionsIndexView.test_pagination_invalidc C sX | | jtdddi}| |jd | |d | |jd j|jd j d S )Nr` r} i ra rb r~ paginator)
r| ri r" r
r# rj rk rt r num_pagesrn 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 s$w Y t tdD ].}tdt | fddtdD } fddtdD }t
j | t
j | q1| d | j| W d d S 1 sww Y d S ) Nr` ry c s* g | ]}t |d d |d| dqS )r1 r z Search pick r )r .0j)r r) r*
" s
zBTestSearchPromotionsIndexView.test_num_queries..r6 c s$ g | ]}t t|d |dqS )rA )r r r< )r r r r rF r) r* r + s )r
r| ri r" assertNumQueriesr rF r9 r rz r r bulk_creater )r( rf r> promosr< r) r r* test_num_queries s(
"z.TestSearchPromotionsIndexView.test_num_queriesc C s | tjjtddddd tjjtddddd | jtd}| |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 ssSSSSr` ra rb queries)
r| r r r! r r" ri r
r# rj rk rt rs rn r) r) r* 'test_results_are_ordered_alphabetically7 s$ zETestSearchPromotionsIndexView.test_results_are_ordered_alphabeticallyc C s t }tdD ]}tdj|t|d d qtjj tddddd tjj tdd dd
d | j
t d }| j |d d
d |
tdjd | |j}t ddd}t ddd}|d| d}|d| d}| | | | |
tdj d d S )N
r rA rC r r r0 r r1 rR r` z | 10 | T)htmlzwagtailadmin_pages:edit)r args)r1 a[href=""])r rF r9 r r" r: r r r r! ri r
ru r# r< get_souprm
select_oneassertIsNotNoner$ r% ) r( rF r> ro souproot_page_edit_urlhomepage_edit_urlroot_page_edit_linkhomepage_edit_linkr) r) r* test_multiple_searchpromotionsQ s4
z ro r) r) r* test_results_orderingo sb
z3TestSearchPromotionsIndexView.test_results_orderingc C ^ d| j _| j | j jtjjddd | jt d}|
|j d | |t d d S )NF wagtailadmin access_admincontent_type__app_labelcodenamer` . wagtailadmin_home
r] is_superusersaveuser_permissionsaddr r r" ri r
r# rj assertRedirectsrn r) r) r* test_get_with_no_permission
z9TestSearchPromotionsIndexView.test_get_with_no_permissionc C s d| j _| j | j jtjjdddtjjddd | jt d}|
|j d | |d |
|j}t d
}| |d | d d S )
NFr r r wagtailsearchpromotionschange_searchpromotionr` ra rb wagtailsearchpromotions:addr r )r] r r r r r r r" ri r
r# rj rk r rm assertIsNoner )r( ro r add_urlr) r) r* "test_get_with_edit_permission_only s$
z@TestSearchPromotionsIndexView.test_get_with_edit_permission_onlyN)rN rO rP r^ rp rv rw r| r r r r r r r r r r) r) r) r* rY s # @ rY 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 d Zd S ) TestSearchPromotionsAddViewc C rZ rW r[ r' r) r) r* r^ r_ z!TestSearchPromotionsAddView.setUpc C sP | j td}| |jd | |d | tddddddg|j d S ) Nr ra wagtailsearchpromotions/add.htmlr` rd re rc zNew: Promoted search resultrh rn r) r) r* rp s z'TestSearchPromotionsAddView.test_simplec C s ddddddddd}| j td|}| |td | tdjjdd
t
tdj }| t
|d | |d jd t|d }| t
|d | |d jd d S ) Ntestr r rc rr rs editors_picks-TOTAL_FORMSeditors_picks-INITIAL_FORMSeditors_picks-MAX_NUM_FORMSeditors_picks-0-DELETEeditors_picks-0-ORDEReditors_picks-0-pageeditors_picks-0-descriptionr r` )r wagtail.create)ri postr
r
assertTruer r" r$ filterexistsrS allr# lenr log_registryget_logs_for_instanceaction)r( post_dataro rU logsr) r) r* test_post s$
z%TestSearchPromotionsAddView.test_postc C s ddddddddddddd d
d }| j td |}| |td
ttdj d}|
t |d |
|d j d |
|d j
d |
|d jd |
|d j
d
|D ]"}t|}|
t |d |
|d jd |
|d j| j q]d S )Nr r1 r r rc r rr r, WagtailzThe landing page)rs r r r r r r r editors_picks-1-DELETEeditors_picks-1-ORDEReditors_picks-1-page!editors_picks-1-external_link_url"editors_picks-1-external_link_texteditors_picks-1-descriptionr r` r r )ri r r
r rS r r" r$ r order_byr# r r r r. r r r r] r( r ro rU search_pickr r) r) r* test_with_multiple_picks s>
z4TestSearchPromotionsAddView.test_with_multiple_picksc
C s t d}|jjdddd}|jjdddd}ddddd dd
d d d
}| jtd|}| |td | t t dj
dd|j d f|j d fdh d S )Nr r r Root pager r r r1 Homepager rc r, r zA Django-based CMS rs r r r r r !editors_picks-0-external_link_url"editors_picks-0-external_link_textr r r` r r. )Nr, )
r r" r$ r! ri r r
r r# setr values_listr )r( r
search_pick_1
search_pick_2r ro r) r) r* $test_post_with_existing_query_string. s<
z@TestSearchPromotionsAddView.test_post_with_existing_query_stringc C s ddddddddd}| j td|}| |jd | |jd d
d | t|jd jd | |jd jd j d
j
d | |jd jd j d d | |jd dd
g | |jd dd g | |jd d d g d S )Nrc r r r rr r r ra formrs This field is required.searchpicks_formsetpager ) ri r r
r# rj assertFormErrorrt r forms cleaned_dataidassertFormSetErrorr( r ro r) r) r* #test_post_with_invalid_query_stringX s<
z?TestSearchPromotionsAddView.test_post_with_invalid_query_stringc C s| ddddddddd}| j td |}| |jd
| |jd dd d
| |jd dd g | |jd d d g d S )Nr r r r rc l c( rr r r ra r r GSelect a valid choice. That choice is not one of the available choices.ri r r
r# rj r rt r r) r) r* test_post_with_invalid_page} s&
z7TestSearchPromotionsAddView.test_post_with_invalid_pagec
C s\ dddddddddd }| j td
|}| |td | tdjjdd d S )
Nr r r r rc r, r rr r r r` r. )
ri r r
r r r r" r$ 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 )rs r r r r ra r @Please specify at least one recommendation for this search term.r r r) r) r* !test_post_without_recommendations s z=TestSearchPromotionsAddView.test_post_without_recommendationsc C sj ddddddddddd
}| j td
|}| |jd | |jd dd d
| |jd d d g d S )Nr r r r rc r, r rr )
rs r r r r r r r r r r ra r 4Please 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 sd dddddddd}| j td|}| |jd | |jd
dd d | |jd
d d g d S ) Nr r r r rc rr )rs r r r r r r r ra r .You must recommend a page OR an external link.r r r) r) r* test_post_missing_recommendation s" zTestSearchPromotionsEditView.test_post_without_recommendationsc C s i ddddddddd| j jd d
d d d
dddd| jjdd
dddddddddd}| jtd| jjfd |}| |jd | |j
d dd d | |j
d d d g | |j
d d d g d S ) Nrs rr r r1 r r r r r rc r r r r r r r r r r r r, r r r r r r ra r r r r r) r) r* r sZ
zBTestSearchPromotionsEditView.test_post_with_page_and_external_linkc C s dddd| j jddd| jjdddd d
d }| jtd | jjfd
|}| |jd | |j
d dd d | |j
d d d g | |j
d dd g d S )Nrr r1 r rc r r r r, r r )rs r r r r r r r r r r r r r r r ra r r r r r) r) r* r s6 z=TestSearchPromotionsEditView.test_post_missing_recommendationc C s dddd| j jdddd| jjdddd d
d }| jtd | jjfd
|}| |jd | |j
d ddd | |j
d dd g | |j
d d d g d S )Nrr r1 r rc r r r r r r )rs r r r r r r r r r r r r r r r r ra r r. r r r r) r) r* r s8 z rq recent_query_idspromoted_query_idsr) r) r* test_garbage_collect_command sZ z@TestGarbageCollectManagementCommand.test_garbage_collect_commandN)rN rO rP r@ r) r) r) r* r0 r0 c @ s8 e Zd Zdd Zi fddZdd Zdd Zd d
Zd S ) TestQueryChooserViewc C s | d S rW )r\ r' r) r) r* r^ s zTestQueryChooserView.setUpc C s | j d|S )Nz#/admin/searchpicks/queries/chooser/)ri r" )r( paramsr) r) r* r" r_ zTestQueryChooserView.getc C sF | }| |jd | |d t|j }| |d d d S )Nra z4wagtailsearchpromotions/queries/chooser/chooser.htmlstepchooser)r" r# rj rk jsonloadsrm decode)r( ro
response_jsonr) r) r* rp s z TestQueryChooserView.test_simplec C s | ddi}| |jd d S )Nrq rr ra r" r# rj rn r) r) r* rv s z TestQueryChooserView.test_searchc C s< | ddi}| |jd | ddi}| |jd d S )Nr} r ra i' i rJ rn r) r) r* r s z$TestQueryChooserView.test_paginationN)rN rO rP r^ r" rp rv r r) r) r) r* rB s rB c @ s$ e Zd Zdd Zdd Zdd ZdS )TestHitCounterc C s | tdjd d S )Nrr r )r# r r" r< r' r) r) r* test_no_hits s zTestHitCounter.test_no_hitsc C s& t d | t djd d S )Nrr r )r r" r: r# r< r' r) r) r* test_hit s zTestHitCounter.test_hitc C s4 t dD ] }td q| tdjd d S )Nr rr )r9 r r" r: r# r< r{ r) r) r* test_10_hits s zTestHitCounter.test_10_hitsN)rN rO rP rL rM rN r) r) r) r* rK s rK 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* r^ s z"TestQueryStringNormalisation.setUpc C s | t| jd d S )Nz hello world!)r# rz r r' r) r) r* test_normalisation
s z/TestQueryStringNormalisation.test_normalisationc C * g d}|D ] }| | jt| qd 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 rQ )N)
HelloWorldz HelloWorld! Hello World! rr )assertNotEqualr r r" rR r) r) r* test_different_queries s z3TestQueryStringNormalisation.test_different_queriesN)rN rO rP r^ rP rS rV r) r) r) r* rO s
rO c @ r/ )TestQueryPopularityc C s\ t dD ] }td qt dD ] }td qt }| | d | |d td | |d td t dD ] }td qB| | d | |d td | |d td | |d td t d
D ] }td qy| | d | |d td | |d td | |d td d S ) Nr2 zunpopular queryr z
popular queryr1 r r r6 zlittle popular queryr )r9 r r" r: r; r# r% )r( r> popular_queriesr) r) r* test_query_popularity$ s( z)TestQueryPopularity.test_query_popularityN)rN rO rP rY r) r) r) r* rW # rA rW 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 r: )r( r r) r) r*
setUpTestDataM s
z%TestQueryHitsReportView.setUpTestDatac C s | }| |jd | |d | |d | dddg|j | |j}|d}| dd |D d
d gd d
gddgddgg | |d | | |
|d d S )Nra z%wagtailadmin/reports/base_report.htmlz-wagtailadmin/reports/base_report_results.htmlrc z Search termsre zmain trc S g | ] }d d | dD qS )c S g | ]}|j qS r) textstripr cellr) r) r* r k zBTestQueryHitsReportView.test_simple...zth,tdselectr trr) r) r* r k z7TestQueryHitsReportView.test_simple..Search term(s)Viewsa query with three hits3a query with two hits2a query with one hit1There are no results.zSearch terms - Wagtail) r" r# rj rk assertBreadcrumbsrm r rf assertNotContains assertActiveFilterNotRenderedassertPageTitle)r( ro r trsr) r) r* rp Y s0
z#TestQueryHitsReportView.test_simplec C sH d| j _| j | j jtjjddd | }| |t d d S )NFr r r r )
r] r r r r r r r" r r
rn r) r) r* test_get_with_no_permissionsx s
z4TestQueryHitsReportView.test_get_with_no_permissionsc C sB | j ddid}| |jd | }| |g d d S )NexportcsvrC ra )zSearch term(s),Viewsza query with three hits,3za query with two hits,2za query with one hit,1)r" r# rj getvaluerH
splitlines)r( ro
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 )Nry xlsxr{ ra )filenameSheet1c S s g | ] }d d |D qS )c S s g | ]}|j qS r) )valuerb r) r) r* r s zGTestQueryHitsReportView.test_xlsx_export...r) )r rowr) r) r* r s z.rj rk rl r2 rn r1 rp r )r" r# rj r| r r rows)r( ro
workbook_data worksheet
cell_arrayr) r) r* test_xlsx_export s z(TestQueryHitsReportView.test_xlsx_exportc C s ddgddgddggddgddgddggddgddgddggddgddgddggd}| D ]?\}}| j|d- | jd |id
}| |jd | |j}|d }| d
d |D | W d n1 siw Y q/d S )Nrp rq rl rm rn ro )rs z
-query_string_hitsz-_hits)orderingr r{ ra z
main tbody trc S r] )c S r^ r) r_ rb r) r) r* r rd zDTestQueryHitsReportView.test_ordering...tdre rg r) r) r* r ri z9TestQueryHitsReportView.test_ordering..)itemssubTestr" r# rj r rm rf )r( casesr resultsro r rw r) r) r*
test_ordering s<
z%TestQueryHitsReportView.test_orderingN) rN rO rP url_name classmethodr\ rp rx r r r r) r) r) r* rZ J s
rZ c @ s( e Zd ZdZdd Zdd Zdd ZdS ) TestFilteredQueryHitsViewr[ c C s8 | | _td| _t | _| jj| jd d S )NzThis will be foundrC ) r\ r] r r" query_hitr r7 r r: r' r) r) r* r^ s
z TestFilteredQueryHitsView.setUpc C s | j ddid}| |jd | |d | |d | | |j | j ddid}| |jd | |d | |d | | |j d S )Nrq Foundr{ ra this will be foundrr z Not found)r" r# rj ru rt ru r rm rn 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 ) N
hit_date_fromr )daymonthr{ ra r rr )yearr1 hit_date_to) r replacer" r# rj ru rt assertActiveFilterr rm r )r( rC ro r) r) r* test_filter_by_date s z-TestFilteredQueryHitsView.test_filter_by_dateN)rN rO rP r r^ r r r) r) r) r* r s
r )1rF r r r ior r django.contrib.auth.modelsr django.corer django.testr django.urlsr
django.utilsr openpyxlr wagtail.admin.admin_url_finderr
&wagtail.admin.tests.test_reports_viewsr (wagtail.contrib.search_promotions.modelsr r r Kwagtail.contrib.search_promotions.templatetags.wagtailsearchpromotions_tagsr wagtail.log_actionsr r wagtail.test.utilsr !wagtail.test.utils.template_testsr r rQ rY r r r+ r0 rB rK rO rW rZ r r) r) r) r* sJ } & v N9 '}