3 (hx@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZyddlmZWn ek rddlmZYnXddlmZdd lmZdd lmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%ddl&m'Z'ej(e)Z*e+e,ds,ddZ-ddZ.ddZ/ddZ0ddZ1ddZ2d(ddZ3eddGd d!d!e'eZ4Gd"d#d#e5Z6d$d%Z7d&d'Z8dS)))absolute_importN)settings)site)User)StaticLiveServerTestCase)reverse)override_settings)six)ugettext) ActionChains)By)visibility_of_element_locatedelement_to_be_clickable)SeleniumTestCasecmpcCs||k||kS)N)abrrH/tmp/pip-install-q3hcpn_q/django-nested-admin/nested_admin/tests/base.pyr'scCs t|tjS)N) isinstance collectionsSequence)orrr is_sequence+srcCs t|tjS)N)rr integer_types)rrrr is_integer/srcCs t|tjS)N)rr string_types)rrrris_str3srcCsdj|jj|jjgS)N-)join_meta app_label model_name)mrrr7sr%cCsd|S)Nz*contains(concat(" ", @class, " "), " %s ")r) classnamerrr xpath_cls:sr'cCs(d}dtd|f}|r$|d|7}|S)NzAnot(contains(@class, "-drag")) and not(contains(@class, "thead"))z %s and %szdjn-itemz and contains(@class, "-%s"))r')r#Zxpath_item_predicateexprrrr xpath_item>s  r)znested_admin.tests.urls)Z ROOT_URLCONFcseZdZedejkZedejkZdZdZ e ddZ e fddZ fdd Zd d ZdKd dZddZdLddZdMddZdNddZdOddZdPddZdQddZdRddZejdSdd ZejdTd!d"ZejdUd#d$ZejdVd%d&ZejdWd'd(Zejd)d*Z dXd+d,Z!d-d.Z"d/d0Z#dYd3d4Z$dZd5d6Z%d[d7d8Z&d\d9d:Z'd;d<Z(d]d=d>Z)d?d@Z*dAdBZ+dCdDZ,d^dEdFZ-d_dGdHZ.d`dIdJZ/Z0S)aBaseNestedAdminTestCase grappellisuitNcCsHddddddddg}|jr&|jd d t|jjd d }|j|dS) Nzdjango.contrib.authzdjango.contrib.contenttypeszdjango.contrib.messageszdjango.contrib.sessionszdjango.contrib.siteszdjango.contrib.staticfileszdjango.contrib.admin nested_adminrr+.) has_grappelliinserttype __module__ rpartitionappend)selfZappsZ current_apprrravailable_appsOs z&BaseNestedAdminTestCase.available_appscsTtt|jddl}tj|j}fdd||_fdd|j|_dS)Nrcs:|jgg}x*t|ddpgD]}|dj|qW|S)NZinlinesr)modelgetattrr4)Zadmindatainline)descend_admin_inlinesrrr;hs zABaseNestedAdminTestCase.setUpClass..descend_admin_inlinescs*t|trddt|DSt|SdS)NcSsg|]}|qSrr).0r$rrr rszYBaseNestedAdminTestCase.setUpClass..recursive_map_model_names..)rlistmapget_model_name)r9)recursive_map_model_namesrrrAps zEBaseNestedAdminTestCase.setUpClass..recursive_map_model_names) superr* setUpClassZnested_admin.tests.urls admin_site _registry root_modelmodels model_names)clsr-Z root_admin) __class__)r;rArrCas    z"BaseNestedAdminTestCase.setUpClasscsDtt|j|jjdd|jjdtjjddddd|_ dS) Ni`i rBsecretzsuper@example.com)usernamepasswordemailF) rBr*setUpseleniumset_window_sizeset_page_load_timeoutrZobjectsZcreate_superuser logged_in)r5)rJrrrPxs  zBaseNestedAdminTestCase.setUpc CsPy|jjd}Wn Yn0X|jjj||jj|jjj|jjddS)Nrr)rQwindow_handles switch_towindowclose)r5Z popup_windowrrrtearDowns z BaseNestedAdminTestCase.tearDownrKcCs4ddlm}ddlm}|j|j|j|f|dS)zX Helper function that blocks until a CSS selector is found on the page. r)r )expected_conditionsN)selenium.webdriver.common.byr Zselenium.webdriver.supportrZ wait_untilZpresence_of_element_located CSS_SELECTOR)r5Z css_selectortimeoutr Zecrrrwait_fors   z BaseNestedAdminTestCase.wait_forc Cs4ddlm}y|jdWn|k r.YnXdS)z7 Block until page has started to load. r)TimeoutExceptionbodyN)Zselenium.common.exceptionsr`r_)r5r`rrrwait_page_loadeds  z(BaseNestedAdminTestCase.wait_page_loadedcCs|jj||d|jjd|jdf|jx.|jjjD]\}}|jj||j ddq.z6Timeout waiting for available element at selector='%s')r^rr)r\)r5rsr^r)rsrwait_until_available_selectors z5BaseNestedAdminTestCase.wait_until_available_selectorcs |jfdd|dddS)Ncs |jS)N)find_element_by_xpath)ry)rurrr%szDBaseNestedAdminTestCase.wait_until_available_xpath..z3Timeout waiting for available element at xpath='%s')r^rr)r\)r5rur^r)rurwait_until_available_xpaths z2BaseNestedAdminTestCase.wait_until_available_xpathccs|j|||jj|VdS)N)rtrQrx)r5rsr^rrrvisible_selectors z(BaseNestedAdminTestCase.visible_selectorccs|j|||jj|VdS)N)rwrQrx)r5rsr^rrrclickable_selectors z*BaseNestedAdminTestCase.clickable_selectorccs|j|||jj|VdS)N)rvrQr{)r5rur^rrrclickable_xpaths z'BaseNestedAdminTestCase.clickable_xpathccs|j|||jj|VdS)N)rzrQrx)r5rsr^rrravailable_selectors z*BaseNestedAdminTestCase.available_selectorccs|j|||jj|VdS)N)r|rQr{)r5rur^rrravailable_xpaths z'BaseNestedAdminTestCase.available_xpathccsv|jdd|jjj|jjddVz.y|jddWn|jYnXWd|jjj|jjdXdS)NcSst|jdkS)N)lenrU)drrrr%sz@BaseNestedAdminTestCase.switch_to_popup_window..rcSst|jdkS)Nr)rrU)rrrrr%sr)r\rQrVrWrUrX)r5rrrswitch_to_popup_windows z.BaseNestedAdminTestCase.switch_to_popup_windowcCs|jjj|jjjjf}|r2td||jgd}n td|}|jsT|jdd|n|j j d|j |f|j |j j dd|j jd |j jd |jdS) Nzadmin:%s_%s_change)argszadmin:%s_%s_addrBrLz%s%si`irKzwindow.$ = django.jQuery)rFr!r"Z object_namelowerrpkrTrorQrirjrbrRrSexecute_script"make_header_footer_position_static)r5objinfourlrrr load_admins   z"BaseNestedAdminTestCase.load_admincCsdd|jjdD}t|dkr:tjdtj|ddt|jjd}|rRd nd }|jj d |j |r|j |jj d d |jj d|jjd|jdS)NcSsg|]}d|dkr|qS)Zfaviconrrr)r<errrr=sz5BaseNestedAdminTestCase.save_form..browserrzFound browser errors: %s)indentz0return django.jQuery("[name=_continue]").length;Z _continueZ_savez//*[@name="%s"]i`irKzwindow.$ = django.jQuery)rQZget_logrloggererrorjsondumpsboolrr{clickrbrRrSr)r5Zbrowser_errorsZ has_continueZ name_attrrrr save_forms    z!BaseNestedAdminTestCase.save_formcCs|jjddS)zBMake grappelli header and footer element styles 'position: static'z$('footer').attr('class', 'grp-module grp-submit-row');$('#content-inner').css('bottom', '0');$('#grp-header').css('position', 'static');$('#grp-content').css('top', '0');N)rQr)r5rrrr!sz:BaseNestedAdminTestCase.make_header_footer_position_staticFTc sg}t|j\}}t|pg}|r&tnt|rxt|r|dry&tfdd|ddDrf|SWqYqXn,ytfdd|Dr|SWn YnXd}|rt|rt|drt|ddkr|jd}q|jdn|s|Sx\t |D]N\}} t|dkrt d| dkrr|s6t dt|dkrLt d |rf|j|ddn |jdPt|  rt|  rt d t | j t| rt|dkrt d |d| g} t| rt| d krt d |t| ftdd| Dst d|| \} } || \} }|r>|j| | gn|j| | gqW|dk r|rv|j||dn |j||S)aZ To allow for the case where there are multiple inlines defined on a given ModelAdmin or InlineModelAdmin, we need indexes to be a list of lists (or at any rate an iterable of iterables, but lets call the outermost iterable a list, and the elements themselves tuples). The elements of the top-level list should be seen as directions for how to descend the hierarchy of the inlines (an empty list thus referring to the root model, which is not an inline). Descending down the hierarchy, each element is a 2-tuple (a, b) where ``a`` is the index on the ModelAdmin.inlines attribute, and b is a 0-indexed number corresponding to the actual item in that inline currently on the page. Some operations, such as ``add_inline`` and ``get_num_inlines`` (along with the helper method ``get_group``) refer not to a specific item within an inline, but to the group of inlines as a whole (at least at the last step of descending the hierarchy: for instance when clicking add on a nested inline; the add button is at the group-level, but the parents of that group are all specific inline items). In the case of such operations that operate at the group level, the last item is a 1-tuple to indicate which group of the available inlines at that level is the target. In the trivial case, there is only one inline at every level. Most of the unit tests are trivial in this sense. It would be very burdensome, not to mention difficult to parse, if the operation:: self.add_inline(indexes=[1, 3, 2]) had to be written out as:: self.add_inline(indexes=[(0, 1), (0, 3), (0, 2), (0, ))]) The first example above is hard enough to follow without adding a number of zeros to each element (and at the end!) to accommodate the possibility of there being multiple inline classes when we already now that there is only one. To ease the burden of writing test cases, it is possible to use an integer at a given level of the indexes if there is only one possible inline at that level. It is similarly alright to omit the terminal 1-tuple of (0, ) to ``add_inline`` and ``get_num_inlines`` if there is only one possible inline group that can be added to. The purposes of this function is to normalize the indexes parameter so that it can be converted from the shorter notation to a fully qualified one of 2-tuples (and the occasional terminal 1-tuple). rcs g|]\}}|ot|qSr)r)r<rr)inline_type_checkrrr=dsz>BaseNestedAdminTestCase._normalize_indexes..Ncs g|]\}}|ot|qSr)r)r<rr)rrrr=jsrz'Indexes depth greater than inline depthz Unexpected None found in indexeszdTerminal index to inline class omitted in group-level operation, but parent has more than one inlinez%Unexpected type %s in list of indexeszRindexes[%d] using shorthand integer value, but more than one inline to choose fromrz*Expected indexes[%d] to have len 2, got %dcSsg|] }t|qSr)r)r<irrrr=sz#indexes[%d] must have only integersrrr) r>rHrrrallrpopr4 enumerate ValueErrorr1__name__) r5indexesis_group named_modelsZ norm_indexesZroot_model_nameZinline_model_namesZ group_indexlevelZ level_indexes inline_indexZ inline_itemZinline_model_namer)rr_normalize_indexes)sz1  $         z*BaseNestedAdminTestCase._normalize_indexescCst|||d}|j|ddS)N) from_indexes to_indexes)screenshot_hack)DragAndDropActionmove_to_target)r5rrractionrrrdrag_and_drop_itemsz*BaseNestedAdminTestCase.drag_and_drop_itemcCsH|j|dd}|d}|j|d}|jd}d||f}|jjd|S) NT)rr)ridz#%s .djn-dynamic-form-%szreturn $('%s').lengthr)r get_group get_attributerQr)r5rr#groupZgroup_idrsrrrget_num_inliness    z'BaseNestedAdminTestCase.get_num_inlinescCsn|j|dd}|j}g}x*|D]"\}}|dt||dfg7}q W|d|g7}ddj|}|jj|S)NT)rz /*[%s][%d]rz/*[@data-inline-model='%s']z/%src)rrr)r rQr{)r5rr#Z expr_partsZ parent_modelZparent_item_indexr(rrrrsz!BaseNestedAdminTestCase.get_groupcCsF|j|}|j\}}|j||j|d}|jdt||dfS)N)rz .//*[%s][%d]r)rrr4rr{r))r5rr# item_indexrrrrget_items     z BaseNestedAdminTestCase.get_itemc Cs|j|dd}|j|}|d }d|j|jd|f}|j|}|jWdQRX||g|d <|dk rz|jd||d|dk r|jd||ddS) NT)rrz0#%s .djn-add-item a.djn-add-handler.djn-model-%srrd)rslugrr)rrrrr~r set_field)r5rrdrZ new_indexr#Z add_selectorelrrr add_inlines   z"BaseNestedAdminTestCase.add_inlinec CsP|j|}|dd}d|j|jd|f}|j|}|jWdQRXdS)Nrrz$#%s .djn-remove-handler.djn-model-%srr)rrrr~r)r5rr#Zremove_selectorrrrr remove_inlines    z%BaseNestedAdminTestCase.remove_inlinec Csp|j|}|dd}|j|jd}d||f}|j|}|jWdQRX|jrld||f}|j|dS)Nrrrz$#%s .djn-delete-handler.djn-model-%sz2#%s.grp-predelete .grp-delete-handler.djn-model-%sr)rrrr~rr/rw)r5rr#item_iddelete_selectorrundelete_selectorrrr delete_inlines     z%BaseNestedAdminTestCase.delete_inlinec Csp|j|}|dd}|j|jd}d||f}|j|}|jWdQRX|jrld||f}|j|dS)Nrrrz$#%s .djn-delete-handler.djn-model-%sz8#%s:not(.grp-predelete) .grp-delete-handler.djn-model-%sr)rrrr~rr/rw)r5rr#rrrrrrrundelete_inlines     z'BaseNestedAdminTestCase.undelete_inlinecCs4|j|}|sd|S|j|djd}d||fS)Nz#id_%s)rrz #id_%s-%s)rrr)r5attnamerZ field_prefixrrrget_form_field_selectors  z/BaseNestedAdminTestCase.get_form_field_selectorcCs$|j|}|j||d}|jj|S)N)r)rrrQrx)r5rrfield_selectorrrr get_fields z!BaseNestedAdminTestCase.get_fieldc CsD|j|}|j||d}|j|}|j|j|WdQRXdS)N)r)rrr~clearZ send_keys)r5rrerrrrrrrs   z!BaseNestedAdminTestCase.set_field)rK)N)rKN)rK)rK)rK)rK)rK)rK)rK)rK)rK)rK)N)FT)F)N)N)NNN)N)N)N)1rr2 __qualname__rrINSTALLED_APPSr/has_suitrFZ nested_modelspropertyr6 classmethodrCrPrYr_rbror\rtrvrwrzr| contextlibcontextmanagerr}r~rrrrrrrrrrrrrrrrrrr __classcell__rr)rJrr*FsV                |       r*c@seZdZddZeddZeddZddZd d Zd d Z d dZ ddZ ddZ ddZ eddZddZdddZdS)rcCsr||_|j|_t|t|kr&d|_nd|_|jj|dd|_|jj||jdd|_|jj||jdd}t|d s|d d|d <|jj||_ t|jd r|jddg|jd<dd|jD|_|jj d}x&|jddD]\}}||d}qWt||_ |jdd|jddkrV|jdd|jddkrV|jddd7<|jj t|t|d dS)NTF)r)rrrrcSsg|] }t|qSr)tuple)r<rrrrr=sz.DragAndDropAction.__init__..z+Depth of source and target must be the samerrrrrrrrrrrr) test_caserQrZtarget_is_emptyrrrrrtarget_num_itemsrGtarget_num_inlinesZ assertEqual)r5rrrZnum_inlines_indexesZ inline_modelsrrrrr__init__s2   zDragAndDropAction.__init__cCst|ds|jj|jd}|jdkr*d}nd|jjrfdjdtddtddtd d td g}n(djdtddtd dtd g}|j||_ |j S) N_source)rdivh3rcz*[%s]zdjn-trzdjn-tdZtoolsz/*[%s]zdjn-drag-handlerz is-sortable) hasattrrrrtag_namer/r r'r{r)r5Z source_itemZdrag_handler_xpathrrrsource*s         zDragAndDropAction.sourcecCst|dst|jdkr0|jj|jdd }n|j}dtd|jd ddtdf}|j|jd dkr|dt|jd ddd7}|j ||_ |j S) N_targetrz.//*[%s][%d]//*[%s]z djn-grouprz djn-itemsz/*[%(item_pred)s][%(item_pos)d])Z item_predZitem_posrrrr) rrrrrrQr'rr)r{r)r5Ztarget_inline_parentZ target_xpathrrrtarget@s " zDragAndDropAction.targetc Cs|j}|jjr.|jjdkr.|jdkr.|jd}t|jj|ddj j t j dt|jj ddj t j dt|jj ddj |jjd}|SQRXdS) NZchromerrrg?z.ui-sortable-helperi)rrrrrrxr rQZmove_to_element_with_offsetZclick_and_holdperformtimesleepmove_by_offsetr})r5rrrrrinitialize_dragQs    z!DragAndDropAction.initialize_dragcCsb|dkr dnd}x*td||D]}t|jjd|jqW||kr^t|jjd||jdS)Nrri)ranger rQrr)r5Zx_offsety_offset incrementoffsetrrrr`s z DragAndDropAction.move_by_offsetcCst|jjjdS)N)r rQreleaser)r5rrrrgszDragAndDropAction.releasec Cs~d}d}|jd}|jddt|j}td|jd dkoL|jdkno|jd d|jdkot|dd |jdd d k}x|j d}|j d|} |r| dkr| |jd7} n| |jd|7} t | dkrP||krPt t | d} |jd| |dkrR||j dkrR|j tjd|j}tjd|jd| |j} t|| } | dkrnP|d7}qW|S)Nrheightryg?g?rrrrrr)sizerrrrrrrcurrent_positionlocationabsintroundrrrr) r5helperrlimitcount helper_height desired_posZ target_bottomZhelper_yrZ scaled_offsetcurr_pospos_diffrrr_match_helper_with_targetjsB   *"         z+DragAndDropAction._match_helper_with_targetcCsD|jdjd}d}x*|D]"}|j|jkr,P||r|d7}qW|S)aS For an unknown reason, evaluating XPath expressions of the form preceding-sibling::*[not(contains(@attr, 'value'))] Where 'value' is contained in at least one of the preceding siblings, is extraordinarily slow. So we just grab all siblings and iterate through the elements in python. z parent::**rr)r{Zfind_elements_by_xpathr)r5ctx conditionZsiblingsrrrrr_num_preceding_siblingss    z)DragAndDropAction._num_preceding_siblingscCsdd}|j||dS)NcSs0ttjd|jd}|dh@o.|dddh@ S)Nz\s+classzdjn-itemz djn-no-dragz djn-theadzdjn-item-dragging)setresplitr)rclassesrrr is_djn_items z?DragAndDropAction._num_preceding_djn_items..is_djn_item)r)r)r5rrrrr_num_preceding_djn_itemssz*DragAndDropAction._num_preceding_djn_itemscCsdd}|j||dS)NcSsdtjd|jdkS)Nz djn-groupz\s+r)rrr)rrrr is_djn_groupszADragAndDropAction._num_preceding_djn_groups..is_djn_group)r)r)r5rrrrr_num_preceding_djn_groupssz+DragAndDropAction._num_preceding_djn_groupscCs|jjd}g}d}dtd}xftdt|jD]R}|dkrD|}n |j|}|j|}|jdtd}|j|}|j d||fq2Wt |S)Nz.ui-sortable-placeholderzancestor::*[%s][1]zdjn-itemrz djn-group) rQrxr'rrrr{rrr0r)r5 placeholderposrZancestor_xpathrrrrrrrs    z"DragAndDropAction.current_positionc Csd}d}|jd}|jdkr4tdtt|d}ntt|d}t|j}x@||krZP|j}t||} | dkrtP|j d| ||d7}qPWdS)NrrrgC@g$@) rrmaxrrrrrrr) r5rrrrrZdyrrrrrr_finesse_positions     z#DragAndDropAction._finesse_positionFcCsN|j}|j}|r*d|jjkr*|jjd|j||}|j|||jdS)NZ phantomjsz /dev/null) rrrrrQZsave_screenshotrrr)r5rrrrrrrs   z DragAndDropAction.move_to_targetN)F)rr2rrrrrrrrrrrrrrrrrrrrs#  ( rcCsdtjkrtj|S|S)Nr+)rrunittestexpectedFailure)funcrrrexpected_failure_if_grappellis  r cCsdtjkrtj|S|S)Nr,)rrrr )r rrrexpected_failure_if_suits  r )N)9 __future__rrrcopyrloggingrrrZdjangoZ django.confrZdjango.contrib.admin.sitesrrDZdjango.contrib.auth.modelsrZ"django.contrib.staticfiles.testingrZ django.urlsr ImportErrorZdjango.core.urlresolversZdjango.test.utilsrZ django.utilsr Zdjango.utils.translationr _Z'selenium.webdriver.common.action_chainsr r[r Z.selenium.webdriver.support.expected_conditionsr rrQr getLoggerrrr __builtins__rrrrr@r'r)r*objectrr r rrrrsT               A`