o hJ<@sddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZmZmZdd lmZdd lmZmZmZmZdd lmZmZdd lmZGd ddeeZ GdddeeZ!GdddeeZ"GdddeeZ#GdddeeZ$dS))get_user_model)Group Permission) ContentType)TestCase)reverse)timezone) freeze_time)LockedPagesPanelRecentEditsPanel$UserObjectsInWorkflowModerationPanelWorkflowObjectsToModeratePanel)get_dummy_request)GroupPagePermissionPageWorkflowWorkflowContentType)FullFeaturedSnippet SimplePage)WagtailTestUtilsc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)TestRecentEditsPanelcCsttjjdd|_tdddd}|jj|d||_|jtjj|j d|_ |j dd d |_ |j d d d dS) Nidz Hello world! hello-worldzSome content here)titleslugcontent)instancealicepasswordusernamer bob) robjectsget root_pager add_child save_revisionrevisionpublishr child_pagecreate_superuser user_alice)selfr+r/b/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/admin/tests/test_dashboard.pysetUps  zTestRecentEditsPanel.setUpcCsd|ddd}|jtd|jjfd|}||td|jjfdtjj|jjd}| |j dS)Nz Some contentr)rrrzwagtailadmin_pages:editargsr) clientpostrr+rassertRedirectsrr$r% assertTruehas_unpublished_changes)r.r post_dataresponsechild_page_newr/r/r0change_something(s z%TestRecentEditsPanel.change_somethingcCs"|jtd}||jd|S)Nwagtailadmin_home)r4r%r assertEqual status_coder.r:r/r/r0go_to_dashboard_response7sz-TestRecentEditsPanel.go_to_dashboard_responsecCs|jddd|jtd}|d|jd|j|jddd|d| }| d|jd|jddd|d | }| d|jd|j|jddd| }| d|jddS) Nr#r r!r=Your most recent editszutf-8r Alice's edit Bob's edit) loginr4r%r assertNotInrdecodelogoutr<rBassertInrAr/r/r0test_your_recent_edits<s     z+TestRecentEditsPanel.test_your_recent_editscCsD|jddd|d|j|jtd}||jddS)Nrr r!rDr=r>) rFr<r+deleter4r%rr?r@rAr/r/r0test_missing_page_record_s   z-TestRecentEditsPanel.test_missing_page_recordcCs|jdddt}t| |dWdn1swYtjjdd|j_ t }| d|ji}t jj|j jd j}||d d d |||d d d |dS) z1Test if the panel actually returns expected pagesr#r r!rENzbob@example.com)emailrequest)pk last_editsr)rFrnowr r<rr$r%r4userr get_context_datarr+rspecificr?)r.edit_timestamppanelctxpager/r/r0 test_paneljs  zTestRecentEditsPanel.test_panelcCs|jddd|dddt|jjddd}|jtd |jjfd || t j j dd |j td }| |jd ||d||d||ddS)Nr#r r!z Bob was erezGoodbye world!z goodbye-worldF) new_titlenew_slugnew_parent_page copy_subpagesaliaszwagtailadmin_pages:copyr2)rr=r>rC)rFr<strr&rr4r5rr+r7rr$r%r?r@assertContainsassertNotContains)r.r9r:r/r/r0&test_copying_does_not_count_as_an_edits"    z;TestRecentEditsPanel.test_copying_does_not_count_as_an_editN) __name__ __module__ __qualname__r1r<rBrKrMr[rdr/r/r/r0rs# rc@"eZdZdgZddZddZdS)TestRecentEditsQueryCount test.jsoncCs|jddd|_t|_|j|j_tj}ddh}ddh}ddh}tt jj gdd  d  }|D]9}|j |jd d }|j|vrJ|||j|j|vr_d |_|j|_t|_||j|vrmt|_|q4dS) Nr#r r! )rkrlrmrn id__inrPT)rT log_action)r,r#r dummy_requestrTrr$firstlistrfilterorder_byrVr(rPstartlocked locked_byrrS locked_atsaveapproved_go_live_at)r.workflowworkflow_pages locked_pagesscheduled_pages pages_to_editrZr)r/r/r0r1s0       zTestRecentEditsQueryCount.setUpcCst}d|ji}||}|d ||}Wdn1s"wY|d|||}|t|ddgd}dd|d D}|||dS) NrOrkAmeristralia Daysvg use[href="#icon-lock"]r)z live + draftlive + scheduledr in moderationrcSs$g|]}d|jdddqS)TF)string recursive)joinfind_allstrip.0er/r/r0 szDTestRecentEditsQueryCount.test_panel_query_count..z .w-status) r rt render_htmlassertNumQueriesrJget_soupr?lenselect)r.rXparent_contexthtmlsoupexpected_statusesstatusesr/r/r0test_panel_query_counts      z0TestRecentEditsQueryCount.test_panel_query_countNrerfrgfixturesr1rr/r/r/r0ris ric@rh)TestLockedPagesQueryCountrjcCs||jddd|_t|_|j|j_tjjgddd}t |D]\}}d|_ |j|_ t t j|d|_|q!dS) Nr#r r!)rmrnrp)pk__inrPT)hours)r,r#rrtrTrr$rwrx enumeraterzr{rrS timedeltar|r})r.pagesirZr/r/r0r1s  zTestLockedPagesQueryCount.setUpcCst}|jdd}||}|d ||}Wdn1s#wY||}gd}dd|dD}|||dS)NdummyrO csrf_token)Saint Patrick (single event)Steal underpantsrcSg|]}|jddqST)rget_textrr/r/r0rzDTestLockedPagesQueryCount.test_panel_query_count...title-wrapper a)r rtrrrrr?r.rXrrrexpected_titlestitlesr/r/r0rs     z0TestLockedPagesQueryCount.test_panel_query_countNrr/r/r/r0rs rc@rh))UserObjectsInWorkflowModerationQueryCountrjcCs|jddd|_|jddd|_|jddd|_tjjdd}|j |j|jt j }t jj |tjtdtjj |td d |j tjjd d tjjd dgdj|jdtjjd dgdD]}||||jd|_|j|_t|_| qhtjjddgdD] }||||jqt!ddD]}tjj d|d}||||jqt"|_#|j|j#_$dS)Nadminr r!r# someoneelseEditorsnamer content_typechangegrouprZpermission_typechange_fullfeaturedsnippetcodenamermrnrq)ownerTrorprR Some obj text)%r, superuser create_userr# someone_elserr$r%user_setaddrrurcreater get_for_modelrrrget_first_root_node permissionsrrwupdaterVr(ryrzr{rrSr|r}rangerrtrT)r.editorsrrZrobjr/r/r0r1sF      z/UserObjectsInWorkflowModerationQueryCount.setUpcCst}d|ji}||}|d ||}Wdn1s"wY||}|t|ddgd}dd|dD}|||dS) NrOrorrz Some obj 2z Some obj 1r ChristmasrrcSrrrrr/r/r0r9rzTUserObjectsInWorkflowModerationQueryCount.test_panel_query_count..r)r rtrrrr?rrrr/r/r0r&s     z@UserObjectsInWorkflowModerationQueryCount.test_panel_query_countNrr/r/r/r0rs .rc@rh)#WorkflowObjectsToModerateQueryCountrjcCs|jddd|_|jddd|_|jddd|_tjjdd}tjjdd}|j |j|j |jt }t jj ||d d t jj ||d d t jj ||d d |j tjjd d |jj tjjd dgdtj}tjj |tjtdt jjddgdD]}||||jd|_|j|_t|_|qt jjddgdD] }||||jqt ddD]}tjj d|d}||||jqt!|_"|j|j"_#dS)Nrr r!r# moderatorrr Moderatorsrrr*rrpublish_fullfeaturedsnippet) codename__inrrmrnrqTrorprRrrr)$r,rrr#rrr$r%rrrrrrrrrwrrurrrrrVr(ryrzr{rrSr|r}rrrtrT)r.r moderatorsrootrrZrrr/r/r0r1@s`     z)WorkflowObjectsToModerateQueryCount.setUpcCst}|jdd}||}|d ||}Wdn1s#wY||}|t|ddgd}dd|d D}|||dS) NrrrprrrcSrrrrr/r/r0rrzNWorkflowObjectsToModerateQueryCount.test_panel_query_count..r)r rtrrrr?rrrr/r/r0rs     z:WorkflowObjectsToModerateQueryCount.test_panel_query_countNrr/r/r/r0r=s @rN)%django.contrib.authrdjango.contrib.auth.modelsrr"django.contrib.contenttypes.modelsr django.testr django.urlsr django.utilsr freezegunr wagtail.admin.views.homer r r r wagtail.coreutilsrwagtail.modelsrrrrwagtail.test.testapp.modelsrrwagtail.test.utilsrrrirrrr/r/r/r0s$        9"H