^Q\4@s7dZddlmZddlZddlmZddlmZddlZGddde Z Gdd d e Z Gd d d e Z Gd d d e Z Gddde ZGddde ZGddde ZGdddeeZGdddee ZGdddee ZdS)z8User stories, functional tests for AutocompleteTestCase.)unicode_literalsN)six)StaleElementReferenceExceptionc @seZdZdZddddddddddd ZejdejdddZd d Z d d Z d dZ ddZ ejdejdddZ ddZejdejdddZddZddZejdejdddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*ZdS)+ BaseStoryzBase UserStory class.Nc Cs ||_|p|jj|_|p-|jj|_|pB|jj|_|pW|jj|_|pl|jj|_|p|jj|_|p|jj|_| p|jj|_| p|jj |_ d|j|_ d|j|_ d|j |jf|_ d|_ dS)z2If any kwarg is None, get it from case attributes.zfieldset.aligned .field-%sz#id_%sz%s %sFN)caseclear_selectordropdown_selector field_nameinput_selectorlabel_selectorlabels_selectormodeloption_selectorwidget_selectorfield_container_selectorfield_selectorfield_clear_selectorin_popup) selfrrrr r r r r rrrE/var/www/dbchiro/venv/lib/python3.4/site-packages/dal/test/stories.py__init__s$  zBaseStory.__init__stopcCsP|jjj|j}x!|D]}||jkr|SqWtd|dS)z(Incremental sleep until option appeared.zOption %s not foundN)rbrowser find_by_cssrtext Exception)rroptionsoptionrrr find_option3s  zBaseStory.find_optioncCsd|j|jfS)z+Return CSS selector for field option label.z%s %s)rr )rrrrget_field_label_selector>sz"BaseStory.get_field_label_selectorcCs!|jjjd|jdS)z5Clean child nodes before checking (ie. clear option).z$("%s *").remove()N)rrexecute_scriptr!)rrrrclean_label_from_remove_butonBs z'BaseStory.clean_label_from_remove_butoncCs>|jjj|j}|j|jtj|jS)z!Return autocomplete widget label.) rrrr!r# clean_labelr text_typer)rlabelrrr get_labelHs  zBaseStory.get_labelcCs|S)z.Given an option text, return the actual label.r)rr&rrrr$QszBaseStory.clean_labelcCs.tj|tj|jks*tdS)z0Assert that the autocomplete label matches text.N)rr%r'AssertionError)rrrrr assert_labelUszBaseStory.assert_labelcCs |jjj|j}|dS)z$Return the autocomplete field value.value)rrrr)rfieldrrr get_valueZs  zBaseStory.get_valuecCs%|jtj|ks!tdS)z1Assart that the actual field value matches value.N)r,rr%r()rr*rrr assert_valueaszBaseStory.assert_valuecCs|j||j|dS)z1Assert value is selected and has the given label.N)r)r-)rr*r&rrrassert_selectionfs zBaseStory.assert_selectioncCs8|j|||j|j||j|dS)z9Assert value and lebel, submit the form and assert again.N)r.submitr)r-)rr*r&rrrassert_selection_persistsks  z#BaseStory.assert_selection_persistscCs(t|t|jks$tdS)z7Retrying assert that suggestions match expected labels.N)sortedget_suggestions_labelsr()rexpectedrrrassert_suggestion_labels_arersz&BaseStory.assert_suggestion_labels_arecCs,|jjjd|jjj_d|_dS)zSwitch to popup window.TN)rrwindowscurrentr)rrrrswitch_to_popupwszBaseStory.switch_to_popupcCs,|jjjd|jjj_d|_dS)zSwitch back to main window.rFN)rrr6r7r)rrrrswitch_to_main|szBaseStory.switch_to_mainc Csd}|js|d7}n|jjj|j}|jd}x`|rt|jjjdkru|jruPny |jWn PYnX|d8}t j dqGW|js|jj ndS)zSubmit the form.zinput[type=submit]z[name=_continue]dr5g?N) rrrrfirstclicklenr6visibletimesleepZ wait_script)rseleltriesrrrr/s"    $   zBaseStory.submitcCs$|jjd|j|jfdS)zOpen the autocomplete dropdown.z%s %sN)rr<rr)rrrrtoggle_autocompletes zBaseStory.toggle_autocompletecCs.|j|jjj|j|jdS)zRe-open the autocomplete box.N)rDrris_element_not_present_by_cssr)rrrrrefresh_autocompletes zBaseStory.refresh_autocompletecsgfdd}dd}|}x,||dkrUtjd|}q*Wdd|DS) z7Return the list of suggestions in the autocomplete box.csjjjjS)N)rrrrr)rrr get_optionssz.BaseStory.get_suggestions..get_optionsc SsGyd|djkSWn+tk r-dSYntk rBdSYnXdS)NZ SearchingrT)rr IndexError)rrrr is_searchings   z/BaseStory.get_suggestions..is_searchingTg?cSs#g|]}|d|jfqS)r*)r).0orrr s z-BaseStory.get_suggestions..)r?r@)rrGrIrr)rrget_suggestionss    zBaseStory.get_suggestionscCsdd|jDS)z6Return labels for suggestions in the autocomplete box.cSsg|]}|dqS)r5r)rJrKrrrrLs z4BaseStory.get_suggestions_labels..)rM)rrrrr2sz BaseStory.get_suggestions_labels)__name__ __module__ __qualname____doc__rtenacityretryZstop_after_delayr r!r#r'r$r)r,r-r.r0r4r8r9r/rDrFrMr2rrrrrs8 '   ' '  '      rc@s.eZdZdZddZddZdS) SelectOptionzUser selects an option.cCs||jjj|j}t| s/|j r<|jn|jj|j|jj|j ||j |j dS)z7Assert that selecting option "text" sets input's value.N) rrrrr=r>rDZassert_visible enter_textr r r<)rrZdropdownrrr select_options  zSelectOption.select_optioncCs|jj|jdS)zClear current option.N)rr<r)rrrr clear_optionszSelectOption.clear_optionN)rNrOrPrQrVrWrrrrrTs  rTcs+eZdZdZdfddZS)InlineSelectOptionz/Same as UserCanSelectOption, in a given inline.Nc s$||_|p|j|_tt|j||d|j|j|jf|_d|j|j|jf|_|jj j dj }t |jj j d|j}xq||jdkr|jy8|jj j |jjt|jt|j Wn wYnX|d7}qWdS)z Same as UserCanSelectOption, in inline inline_number. Where inline_related_name should be the related_name option for the foreign key used for the InlineModelAdmin. z#%s-%s .field-%sz #id_%s-%s-%sz Add anotherz .dynamic-%sr5N) inline_numberinline_related_namesuperrXrr rrrrZfind_link_by_partial_textr;r=rr<replacestr)rrrYrZkwargsaddnum) __class__rrrs0       zInlineSelectOption.__init__)rNrOrPrQrrr)rarrXs rXc@s"eZdZdZddZdS) RenameOptionz/User story to rename an option in django admin.cCsx|jjd|j|j|jjjd}|jj|d||jjd||j|j dS)z0Click the change button and rename in the popup.z #change_id_%sZid_namer*z#id_nameN) rr<r r8r find_by_id assertEqualsrUr/r9)rZ current_nameZadd_keysZ name_inputrrr rename_option s   zRenameOption.rename_optionN)rNrOrPrQrerrrrrbs rbc@s"eZdZdZddZdS)AddAnotherOptionzAdd-another user story.cCsU|jjjd|jj|j|jjd||j|jdS)z9Click the add button and add another option in the popup.z add_id_%sz#id_nameN) rrrcr r<r8rUr/r9)rnamerrr add_anothers    zAddAnotherOption.add_anotherN)rNrOrPrQrhrrrrrfs rfc@s"eZdZdZddZdS) CreateOptionzCreate an option on the fly.cCs]|j|jj|j||jjj||jj|j|jjjddS)zf Select the only option after typing name and submit. name should be unique. z.select2-results__optionsN) rDrrUr rZis_element_present_by_textr<rrE)rrgrrr create_option+s   zCreateOption.create_optionN)rNrOrPrQrjrrrrri(s ric@sveZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS) MultipleMixinz,Enable multiple choice support with stories.cCsd|j|jfS)z+Return CSS selector for field option label.z%s %s)rr )rrrrget_field_labels_selector>sz'MultipleMixin.get_field_labels_selectorcCs!|jjjd|jdS)z5Clean child nodes before checking (ie. clear option).z$("%s span").remove()N)rrr"rl)rrrrr#Bs z+MultipleMixin.clean_label_from_remove_butoncs<jjjjj}fdd|DS)z!Return autocomplete widget label.cs+g|]!}jtj|jqSr)r$rr%r)rJr&)rrrrLQs z,MultipleMixin.get_labels..)r#rrrrl)rlabelsr)rr get_labelsHs   zMultipleMixin.get_labelscCs3d|j}|jjj||jjjdS)z$Return the autocomplete field value.z window.GET_VALUES = []; $('%s option:selected').each(function() { GET_VALUES.push($(this).attr('value')); }); zwindow.GET_VALUES)rrrr"Zevaluate_script)rscriptrrr get_valuesUs zMultipleMixin.get_valuescCsS|j}x!|D]}|jj||qW|jjt|t|dS)z#Assert that all labels match texts.N)rnrassertInrdr=)rZtextsrmrrrr assert_labels`s  zMultipleMixin.assert_labelscCsfdd|D}|j}x!|D]}|jj||q&W|jjt|t|dS)z3Assart that the actual field values matches values.cSsg|]}tj|qSr)rr%)rJvrrrrLks z/MultipleMixin.assert_values..N)rprrqrdr=)rvaluesZ text_valuesZ actual_valuesZ actual_valuerrr assert_valuesis    zMultipleMixin.assert_valuescCs|j||j|dS)z)Assert selections have values and labels.N)rrru)rrtrmrrrr.vs zMultipleMixin.assert_selectioncCs8|j|||j|j||j|dS)z9Same as above, but also submits the form and check again.N)r.r/rrru)rrtrmrrrr0{s  z'MultipleMixin.assert_selection_persistsN) rNrOrPrQrlr#rnrprrrur.r0rrrrrk;s    rkc@seZdZdZdS)CreateOptionMultiplez"Multiple version of CreateOptions.N)rNrOrPrQrrrrrvs rvc@seZdZdZdS)SelectOptionMultiplez"Multiple version of CreateOptions.N)rNrOrPrQrrrrrws rwcs+eZdZdZdfddZS)InlineSelectOptionMultiplez(Multiple options for InlineSelectOption.Nc s?tt|j||d||d|j|jf|_dS)z1Set input_selector with field_container_selector.rZz%s %sN)r[rxrrr )rrrYrZr^)rarrrsz#InlineSelectOptionMultiple.__init__)rNrOrPrQrrr)rarrxs rx)rQ __future__rr?Z django.utilsrZselenium.common.exceptionsrrRobjectrrTrXrbrfrirkrvrwrxrrrrs  2 H