U
{hU @ s d dl Z d dlZd dlmZmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
d dlmZ d dlmZ d d lmZ d
d lmZ G d d
d
eZG dd deeZG dd deeZdS ) N)SimpleTestCaseTestCase)reverse) translation)FieldBlockAdapter)DEFAULT_TABLE_OPTIONS
TableBlock)Page)TableBlockStreamPage)WagtailTestUtils )
TableInputc @ 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 d Zd!d" Zd#d$ Zd%d& Zd'S )(TestTableBlockc
C s$ ddddddddddddd | _ d S )
Nr FTtextalll en) minSpareRows startRows startCols
colHeadersZ
rowHeaders contextMenueditorstretchHheightlanguagerendererZautoColumnSize)default_table_options)self r i/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/contrib/table_block/tests.pysetUp s zTestTableBlock.setUpc C sT dddddgdddgdddggd}t }||}d}| || | d| dS )z(
Test a generic render.
FTest 1Test 2Test 3Nfirst_row_is_table_headerfirst_col_is_headerdataa
r renderassertHTMLEqualassertInr valueblockresultexpectedr r r! test_table_block_render" s
z&TestTableBlock.test_table_block_renderc C sj ddddddddddgdd d
gd d d gd d d ggd }t }||}d
}| || | d | d S )z@
Test a generic render with some cells aligned.
TFr r ZhtLeft)rowcolZ classNameZhtRightr# r$ r% N)r' r( cellr) a
r* r. r r r! !test_table_block_alignment_render> s
z0TestTableBlock.test_table_block_alignment_renderc C sD t }|dddddgdddgdddggd}d}| || dS )z4
An empty table should render okay.
FNr& a
r r+ r, )r r0 r1 r2 r r r! test_render_empty_table` s z&TestTableBlock.test_render_empty_tablec C sH dddddgdddgdddggd}d}t }||}| || dS )zI
Ensure that raw html doesn't render
by default.
F Test
Nr& aB
| <p><strong>Test</strong></p> | | |
| | |
| | |
r8 r r/ r2 r0 r1 r r r! "test_do_not_render_html_by_defaultw s
z1TestTableBlock.test_do_not_render_html_by_defaultc C s^ dddddgdddgdddggd}d}| j }d|d< t|d}||}| || dS ) zQ
Ensure html renders if table_options set renderer to allow html
Fr: Nr& a*
htmlr
table_options)r copyr r+ r, )r r/ r2 new_optionsr0 r1 r r r! test_does_render_html_if_allowed s
z/TestTableBlock.test_does_render_html_if_allowedc C sH dddddgdddgdddggd}d}t }||}| || dS ) z@
Ensure that row headers are properly rendered.
TFFooBarBazNr& al
r8 r; r r r! test_row_headers s
z TestTableBlock.test_row_headersc C sH dddddgdddgd d
d ggd }d
}t }||}| || dS )zC
Ensure that column headers are properly rendered.
FTrC rD rE onetwothreefourfivesixr& aQ
| Foo | Bar | Baz |
|---|
| one | two | three |
|---|
| four | five | six |
|---|
Nr8 r; r r r! test_column_headers s
z"TestTableBlock.test_column_headersc C sH dddddgdddgdd d
ggd }d }t }||}| || d
S )z?
Test row and column headers at the same time.
TrC rD rE rG rH rI rJ rK rL r& a
| Foo | Bar | Baz |
|---|
| one | two | three |
|---|
| four | five | six |
|---|
Nr8 r; r r r! test_row_and_column_headers s
z*TestTableBlock.test_row_and_column_headersc C sH dddddgdddgdd d
ggd }d }t }||}| || d
S )z-
Test table without headers.
FrC rD rE rG rH rI rJ rK rL r& a-
| Foo | Bar | Baz |
| one | two | three |
| four | five | six |
Nr8 r; r r r! test_no_headers s
z TestTableBlock.test_no_headersc C sR dddddgdddggd}t }d }||}| || | ||| dS )
ze
Make sure we get back good json and make
sure it translates back to python.
FrC r Ng @rD rE r& zs{"first_row_is_table_header": false, "first_col_is_header": false, "data": [["Foo", 1, null], [3.5, "Bar", "Baz"]]})r Zvalue_for_formZassertJSONEqual assertEqualZvalue_from_form)r r/ r0 Z
expected_jsonZ
returned_jsonr r r! test_value_for_and_from_form s
z+TestTableBlock.test_value_for_and_from_formc C sF t }| | d | j }d|d< t |d}| | d dS )zc
Test that settings flow through correctly to
the is_html_renderer method.
Fr= r r> TN)r assertIsZis_html_rendererr r@ )r Zblock1rA Zblock2r r r! test_is_html_renderer/ s
z$TestTableBlock.test_is_html_rendererc C sN dddddgd dd gd dd ggd}t }||}| |dddddg d S )NFr# r$ r% rD rC r& r get_searchable_contentrP r r/ r0 contentr r r! test_searchable_content> s$
z&TestTableBlock.test_searchable_contentc C s$ d }t }||}| |g d S NrT rV r r r! &test_searchable_content_for_null_blockU s
z5TestTableBlock.test_searchable_content_for_null_blockc C s\ t dd}dddddgdddgdddggd}|j|d d
id }| d| | d | dS )
zv
Test that extra context variables passed in block.render are passed through
to the template.
z*tests/blocks/table_block_with_caption.html)templateFr# r$ r% Nr& captionzA fascinating table.)contextz A fascinating table.
)r r+ r- )r r0 r/ r1 r r r! test_render_with_extra_context[ s
z-TestTableBlock.test_render_with_extra_contextc C sV ddddddgdddgdddggd}t }||}d}| || | d| dS ) z5
Test a generic render with caption.
r\ Fr# r$ r% N)Z
table_captionr' r( r) aK
caption
| Test 1 | Test 2 | Test 3 |
| | |
| | |
r* r. r r r! test_table_block_caption_rendero s
z.TestTableBlock.test_table_block_caption_renderc C s4 d}t }||}d}| || | d| dS )z6
Test an empty table is not rendered.
N None)r r+ r, Z assertNotInr. r r r! &test_empty_table_block_is_not_rendered s
z5TestTableBlock.test_empty_table_block_is_not_renderedc
C sx dddddgdddgdd dggd
d d d
ddd d
gd
d
d ddd ddd dgd}t }||}d}| || dS )z<
Test that merged table cells are rendered.
FrG NrH rI rJ rK rL sevenr r T)r4 r5 Zhidden )r4 r5 ZrowspanZcolspan)r' r( r) r6 Z
mergeCellsaG
| one | two |
| three | four | five |
| six | seven |
r8 r. r r r! test_merge_cells_render s"
z&TestTableBlock.test_merge_cells_renderc C s t }| |d d S rY )r Z assertIsNoneZ to_pythonr r0 r r r! test_empty_table_block_to_python s z/TestTableBlock.test_empty_table_block_to_pythonN)__name__
__module__ __qualname__r" r3 r7 r9 r< rB rF rM rN rO rQ rS rX rZ r^ r_ rb re rg r r r r! r s& " "r c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dS )TestTableBlockFormc C sV dddddgdddgddd gd
d dgd d
d ggd| _ t }t | _|| jd< d S )NTShipTypeStatus Galactica
BattlestarActiveValkyrie DestroyedCylon BasestarBasestarBrenikSmall Military Vesselr& r )r/ r Z get_languager r@ r )r r r r r! r" s
zTestTableBlockForm.setUpc C s t }| |j| | | j|j | td |jd | td |jd | td |jd | td |jd | td |jd dS )zI
Test options without any custom table_options provided.
r r r r r N)r rP r? Zget_table_optionsr r rf r r r! test_default_table_options s( z-TestTableBlockForm.test_default_table_optionsc C s t }| d|j td t }| d|jd td t }| d|jd t ddid}| d|jd | d|jd td dS )zW
Test that the environment's language is used if no language provided.
r zfr-fritjar> r N)r r- r? r activaterP ZassertNotEqual)r r0 Zblock_frZblock_itZblock_with_langr r r! test_table_options_language s
z.TestTableBlockForm.test_table_options_languagec C s t td }t j}| |d | tddidj}| |d | tddidj}| |d d tdddgidj}| |d ddg tdg idj}| |d g dS )z9
Test how contextMenu is set to default.
r Tr> FZundoZredoN)listr r r? rP rR )r Zdefault_context_menur? Ztable_options_menu_trueZtable_options_menu_falseZtable_options_menu_listr r r! test_table_options_context_menu s&
z2TestTableBlockForm.test_table_options_context_menuc C sv t ddddj}| |d d | |d d t ddidj}| |d d t d d
idj}| |d d
d S ) zE
Test simple options overrides get passed correctly.
rd )r r r> r r r noneZ
allowEmptyFN)r r? rP rR )r Z block_1_optsZ block_2_optsZ block_3_optsr r r! test_table_options_others s z,TestTableBlockForm.test_table_options_othersc
C sb t }|d t |}| |d d | |d t | |d dddddd d
id d S ) NZtest_tableblockr r rd zTest tableblockTtablez0w-field w-field--char_field w-field--table_inputZ ADD_COMMENTz Add Comment)labelrequiredicon classnameZshowAddCommentButtonstrings)r set_namer js_argsrP ZassertIsInstancer
)r r0 r r r r!
test_adapt0 s
z TestTableBlockForm.test_adaptc C s0 t }|j| jd}| d| | d| dS )z?
Test searchable content is created correctly.
)r/ ro rv N)r rU r/ r- )r r0 Zsearch_contentr r r! rX D s z*TestTableBlockForm.test_searchable_contentN)
rh ri rj r" rx r| r~ r r rX r r r r! rk s rk c @ s" e Zd Zdd Zejdd ZdS )TestTableBlockPageEditc C s dddddgdddgddd gd
d dgd d
d ggd| _ tjjdd| _tdtd| j dgd}| jj|d| _ |
| _ d S )NTrl rm rn ro rp rq rr rs rt ru rv rw r& rd )idZShipsr )typer/ )titler )instance) r/ r objectsgetZ root_pager
jsondumpsZ add_childtable_block_pageloginuser)r Ztable_block_page_instancer r r! r" P s$ z TestTableBlockPageEdit.setUpc C sf | j td| jjfd}| |d | |d | |d | |d | |d | |d d S )
z\
Test that edit page loads with saved table data and correct init function.
zwagtailadmin_pages:edit)argszWrp ro z#initTable("table\u002D0\u002Dvalue"r r N)clientr r r r ZassertContains)r responser r r! test_page_edit_page_viewe s z/TestTableBlockPageEdit.test_page_edit_page_viewN)rh ri rj r" unittestZexpectedFailurer r r r r! r O s r )r r Z django.testr r Z django.urlsr Z django.utilsr Zwagtail.blocks.field_blockr Z"wagtail.contrib.table_block.blocksr r Zwagtail.modelsr Zwagtail.test.testapp.modelsr
Zwagtail.test.utilsr blocksr
r rk r r r r r! s" 3