BhVcddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZddlm Z m!Z!ddl"m#Z#m$Z$ddl%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;mZ>m?Z?ddl@mAZAddlBmCZCmDZDmEZEes r5r=Nr>r4r5r6r@r@=s   r5r@c6eZdZdZdZdZefdZxZS)WizardTestMixinNc8t|}t|}tt||}t|t|kst|t|kr|jdt |D]"\}\}}||k7s|jd|$y)Nz"Sequence lengths are not the same.zSequences differ at index )listziplenfail enumerate)selfseq_aseq_bzippedidxabs r6assertSequencesEqualz$WizardTestMixin.assertSequencesEqualFsU U c%'( v;U #s6{SZ'? II: ;$V, >KC!QAv 6se<= >r5cNt|dt_Gddt}|t ddt td|_Gdd t}|t d d t |_ Gd dt}|t ddtd|_ y)NTc eZdZy).WizardTestMixin.setUpClass..PageWizardNr0r4r5r6 PageWizardrUW r5rVrdmy_template.html)titleweightformr; template_namec eZdZy)2WizardTestMixin.setUpClass..SettingsWizardNr0r4r5r6SettingsWizardr_crWr5r`r)rZr[r\c eZdZy)5WizardTestMixin.setUpClass..PageContentWizardNr0r4r5r6PageContentWizardrcmrWr5rd)rZr[r\r]) super setUpClassr+ _discoveredr(_r/r page_wizardr8user_settings_wizardr@ title_wizard)clsrVr`rd __class__s r6rfzWizardTestMixin.setUpClassPs #'   %F),   V $2N# $     -F),  r5) r1r2r3rirkrR classmethodrf __classcell__)rms@r6rDrDBs&KL>& & r5rDc0eZdZdZdZdZdZdZdZy)TestWizardBasec|j}|j|jj||j }|j |jj|yN)"get_staff_user_with_no_permissions assertFalseriuser_has_add_permission get_superuser assertTrue)rKusers r6test_user_has_add_permissionz+TestWizardBase.test_user_has_add_permission|s\668 ))AA$GH!!# ((@@FGr5cT|j}tdtdt|dd}|j d}|j |j j|d||j }|j |j j||y)N Sample PageenTrZtemplatelanguage created_byparent in_navigationr)rwrrrget_absolute_url assertEqualriget_success_url)rKrypageurls r6test_get_success_urlz#TestWizardBase.test_get_success_urls!!#/   ##T#2 ))99 4:!"% '##% ))99$?Er5cv|j}tdtdt|dd}t j dj }tj|dt|i5|jj|d}|j|t|ddddy#1swYyxYw)Nr|r}Tr~cmstoolbar_enabled_modelsr)rwrrrrget_app_config cms_extensionrobjectrrirrr)rKryr extensionrs r6test_get_edit_urlz TestWizardBase.test_get_edit_urls!!#/   ''.<< \\)%=d| L ?""22t3%C   (= ? ? ? ?s +;B//B8cttddttdd}|j }t dt dt|dd }tjd j}tj|d t|i5|j|d }|j|t|d dddy#1swYyxYw) NrrXrYF)rZr[r\r;r]edit_mode_on_successr|r}Tr~rrr)r(rhr/rrwrrrrrrrrrrr)rKwizard_preview_moderyrrrs r6test_get_preview_urlz#TestWizardBase.test_get_preview_urls$F),!&  !!#/   ''.<< \\)%=d| L B%55t6%C   +D4@ B B B Bs 1CCcD|j|jjt|j|jjt |j t5|jjdddy#1swYyxYwrs) rri get_modelrrjr assertRaisesr rkrKs r6test_get_modelzTestWizardBase.test_get_modelsv ))335t< 22<<> M   3 4 *    ' ' ) * * *s 2BBcltd}|jdddd}|jj|}|j |j d|j |5|jj|}|j |j ddddy#1swYyxYw)Nzadmin:cms_wizard_createz staff-activeTF)is_staff is_superuser is_activeira)r _create_userclientgetr status_codelogin_user_context)rKendpoint staff_activeresponses r6test_endpoint_auth_requiredz*TestWizardBase.test_endpoint_auth_requireds45(($UZfj(k ;;??8, --s3  $ $\ 2 8{{x0H   X113 7 8 8 8s )8B**B3N) r1r2r3rzrrrrrr4r5r6rqrqzs#HF$?&B8* 8r5rqcXeZdZdZdZdZdZdZdZdZ e dd Z d Z y ) TestWizardPoolctjdj}t|dr|`t j Dcgc]$}t|jdr |j&}}|D]}|j|t jjt j jycc}w)Nrwizards cms_wizards) rrrhasattrrrget_cms_config_apps cms_configconfigure_wizardsget_cms_extension_apps cache_clear)rKrappconfigs_with_wizardsconfigs r6tearDownzTestWizardPool.tearDowns''.<< 9i (!&6&J&J&L "s~~}5 NN  + 0F  ' ' / 0 //;;=,,88: s)Ccddlm}|j|jdjj t jt}|j|y)zn Test for backwards compatibility of is_registered when checking a registered wizard. rrrN) django.appsrrxrrrr+ is_registeredr)rKrrs r6(test_is_registered_for_registered_wizardz7TestWizardPool.test_is_registered_for_registered_wizardsI % +++E2@@HHI#11/B   &r5cdtj|j}|j|y)zq Test for backwards compatibility of is_registered when checking an unregistered wizard. N)r+rriru)rKrs r6*test_is_registered_for_unregistered_wizardz9TestWizardPool.test_is_registered_for_unregistered_wizards( $11$2B2BC  'r5ctjt}tjdj j }|jtj||j|y)z| Test for backwards compatibility of the unregister method. Removes a wizard from the wizards dict. rN) r+ unregisterrrrrr assertNotInidrx)rKwas_unregisteredregistered_wizardss r6!test_unregister_registered_wizardz0TestWizardPool.test_unregister_registered_wizardsV '11/B!007EEMM ++-?@ ()r5cdtj|j}|j|y)zw Test for backwards compatibility of the unregister method. Returns False if wizard not found. N)r+rriru)rKrs r6#test_unregister_unregistered_wizardz2TestWizardPool.test_unregister_unregistered_wizards* '11$2B2BC )*r5c|jt5tjtdddy#1swYyxYw)z Test for backwards compatibility of the register method. Raises AlreadyRegisteredException if the wizard is already registered. N)rr)r+registerrrs r6'test_register_already_registered_wizardz6TestWizardPool.test_register_already_registered_wizards6   9 : 2   1 2 2 2s 9Actj|jtjdj j }|j|jj|y)zw Test for backwards compatibility of the register method. Adds the wizard to the wizards dict. rN) r+rrirrrrassertInr)rKrs r6!test_register_unregistered_wizardz0TestWizardPool.test_register_unregistered_wizardsP T--.!007EEMM d&&))+=>r5z!cms.wizards.wizard_pool.get_entrycVtjt|jy)z Test for backwards compatibility of wizard_pool.get_entry. Checking we use the new get_entry under the hood. N)r+r'rassert_called_once)rKmocked_get_entrys r6test_get_entryzTestWizardPool.test_get_entry"s o.++-r5cgd}t|}tjjtjjt d|5t d|5t dddddd|jd}tjttjttjti}|j|jj|y#1swYxYw#1swYxYw)z Integration-like test checking that if you register your wizard by adding wizard_pool.register(wizard) to cms_wizards.py it will correctly register the wizard. This ensures backwards compatibility. )r treebeardz(cms.test_utils.project.backwards_wizards)installed_appszcms.app_registration.appszcms.wizards.wizard_pool.appsNr)rrrrrrr"rrrrrassertDictEqualrr)rKINSTALLED_APPS mocked_appscms_appexpected_wizardss r6!test_old_registration_still_worksz0TestWizardPool.test_old_registration_still_works+s  .9  //;;=,,88: . < !5{C !  ! !,,U3     ! !#5 IIv  W22::sub-content content.

rrrZrrcontentrz
z
z
rarr4)r rwrrr!appendr rrrxrrrrrassertContainsassertNotContains)rKrrr templatesrrrrr\rexpected unexpectedrs r6'test_wizard_content_placeholder_settingz6TestPageWizard.test_wizard_content_placeholder_settings  !&&(  $ $Y / )&&(G )#K0  CD'0B3@   * * N}.@$GD3G! !" D %  $& D OODMMO ,99;D((3 N++D16wivF4WIVD ;;??3/##HhC#H&&x&M  N' N N ) )B N N' N Ns1E 2A'E$A!E:E$ EE! E$$E-ct}|j}|j|5|j}dddt d}|j d|ddd}t di|5tddd}d }d d d|d }t|||d } |j| j| j}|j|5|jd} |jj| } |j| |dddddddy#1swYxYw#1swYxYw#1swYyxYw)z Tests that the PageWizard won't respect a 'bad' setting such as targeting a static-placeholder. In this case, will just fail to add the content (without error). Nr r r footerr rr}z

footer content.

rrrrrarr4)r rwrrr!rr rrrxrrrrrr) rKrrrrrrrrr\rrs r6+test_wizard_content_placeholder_bad_settingz:TestPageWizard.test_wizard_content_placeholder_bad_settingsd  !&&(  $ $Y / )&&(G )#K0  CD'0B3;   * * K}.@$GD.G! !" D %  $& D OODMMO ,99;D((3 K++D1;;??3/&&xc&J K' K K ) )D K K' K Ks1D+2A'EAD7E+D47E <EE ct}|j}|j|5|j}ddddddd}t |d|d}|j |j y#1swY?xYw)Nrr}r)r rwrrrrurrs r6"test_create_page_with_empty_fieldsz1TestPageWizard.test_create_page_with_empty_fieldss!&&(  $ $Y / )&&(G )  ! "   ) ) )s A;;Bc:t}|j}|j|5|j}ddddddd}t ddddt |d|d}|j |j|j |jjjdt |d|d|}|j |j|j |jjjd d|d <t |d|d|}|j |j|j |jjjd t |d|d|}|j |j|j |jjjd y#1swYxYw) Nrrnav_playground.htmlr})rrrzpage-2)rzpage-3rzpage-2-2zpage-2-3) r rwrrrrrxrrget_urlsrrs r6#test_create_page_with_existing_slugz2TestPageWizard.test_create_page_with_existing_slugs!&&(  $ $Y / )&&(G )    !  ! "    (  ,,.5585DE! "    (  ,,.5585DE V ! "    (  ,,.55:5FG! "    (  ,,.55:5FGw ) )s HHN) r1r2r3rrrrrrr r$r4r5r6rrVs4FG D>qB/Nb.K`*(?Hr5rceZdZdZy)TestPageWizardSubmissioncddlm}tddd}|dj}t d}d|d|j d}d d d d d }|j |j5|jj||d}|j|jd|jj||}dddttjj|j j!j#dd}|j%t'j(|y#1swYxYw)Nr)r&homer"r}cms_wizard_create0)wizard_create_view-current_stepz0-entryz 0-languagez0-page1z my test pagez my-test-pager)r+z1-titlez1-slugz 1-contentT)followra)pk)r)cms.wizards.helpersr&rrr#r.rrwrpostrrrrrr<lastget_content_objrrformat) rKr&rentry_idr data_step1 data_step2r edit_endpoints r6test_page_wizard_submissionz4TestPageWizardSubmission.test_page_wizard_submissionXs,36#8$?=#&& !45/2gg  03%$   $ $T%7%7%9 : >{{''*T'JH   X113 7{{''*=H > , LL DGG , 1 1 3 C CD I   H&9&@&@]&ST > >s *AEE N)r1r2r3r8r4r5r6r&r&WsUr5r&ceZdZdZdZdZy)TestWizardHelpersc|tjjtjjyrs)rrrrrs r6setUpzTestWizardHelpers.setUpys( //;;=,,88:r5c^tttg}t}|j ||y)zd The get_entries function returns the registered wizards ordered by weight. N)rrrr&assertListEqual)rKrentriess r6!test_get_entries_orders_by_weightz3TestWizardHelpers.test_get_entries_orders_by_weights($]4FG- Wh/r5cbttj}|j|ty)za The get_entry function returns the wizard when a wizard id is supplied. N)r'rrr)rKentrys r6#test_get_entry_returns_wizard_by_idz5TestWizardHelpers.test_get_entry_returns_wizard_by_ids$ -**+  .r5N)r1r2r3r<r@rCr4r5r6r:r:ws;0"/r5r:cTeZdZdZej ededdZy)TestEntryChoicesc\|j}tddd}t||Dcgc]}|}}tjtj ft jt j ftjtj fg}|j||ycc}w)zR The entry_choices function returns the wizards ordered by weight r(r"r}N) rwrr*rrrZrrr>rKryroptionwizard_choicesrs r6(test_generates_choices_in_weighted_orderz9TestEntryChoices.test_generates_choices_in_weighted_orders!!#6#8$?/rGs r6:test_doesnt_generate_choice_if_user_doesnt_have_permissionzKTestEntryChoices.test_doesnt_generate_choice_if_user_doesnt_have_permissions!!#6#8$?/ris %%7)//+1 ?IE)CFH'**,E6* '!& *!)   i  9 5 5 p[8_&<[8|{N_k{N|~H_k~HBU{U@!/ !/H!7{!7r5