a
D gV @ 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 `/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/contrib/table_block/tests.pysetUp s zTestTableBlock.setUpc C sN ddg dg dg dgd}t }||}d}| || | d| dS )z(
Test a generic render.
FTest 1Test 2Test 3NNNfirst_row_is_table_headerfirst_col_is_headerdataa
r% Nr renderassertHTMLEqualassertInr valueblockresultexpectedr r r! test_table_block_render" s
z&TestTableBlock.test_table_block_renderc C sd ddddddddddgg dg d g d gd
}t }||}d }| || | d | d
S )z@
Test a generic render with some cells aligned.
TFr r ZhtLeft)rowcolZ classNameZhtRightr# r' )r) r* cellr+ a
r% Nr, r0 r r r! !test_table_block_alignment_render> s
z0TestTableBlock.test_table_block_alignment_renderc C s> t }|ddg dg dg dgd}d}| || dS )z4
An empty table should render okay.
Fr' r( a
Nr r- r. )r r2 r3 r4 r r r! test_render_empty_table` s z&TestTableBlock.test_render_empty_tablec C sB ddg dg dg dgd}d}t }||}| || dS )zI
Ensure that raw html doesn't render
by default.
Fz Test
NNr' r( aB
| <p><strong>Test</strong></p> | | |
| | |
| | |
Nr: r r1 r4 r2 r3 r r r! "test_do_not_render_html_by_defaultw s
z1TestTableBlock.test_do_not_render_html_by_defaultc C sX ddg dg dg dgd}d}| j }d|d< t|d}||}| || d S )
zQ
Ensure html renders if table_options set renderer to allow html
Fr< r' r( a*
htmlr
table_optionsN)r copyr r- r. )r r1 r4 new_optionsr2 r3 r r r! test_does_render_html_if_allowed s
z/TestTableBlock.test_does_render_html_if_allowedc C sB ddg dg dg dgd}d}t }||}| || dS )z@
Ensure that row headers are properly rendered.
TFFooBarBazr' r( al
Nr: r= r r r! test_row_headers s
z TestTableBlock.test_row_headersc C sB ddg dg dg dgd}d}t }||}| || dS ) zC
Ensure that column headers are properly rendered.
FTrE onetwothreefourfivesixr( aQ
| Foo | Bar | Baz |
|---|
| one | two | three |
|---|
| four | five | six |
|---|
Nr: r= r r r! test_column_headers s
z"TestTableBlock.test_column_headersc C sB ddg dg dg dgd}d}t }||}| || dS )z?
Test row and column headers at the same time.
TrE rJ rN r( a
| Foo | Bar | Baz |
|---|
| one | two | three |
|---|
| four | five | six |
|---|
Nr: r= r r r! test_row_and_column_headers s
z*TestTableBlock.test_row_and_column_headersc C sB ddg dg dg dgd}d}t }||}| || dS )z-
Test table without headers.
FrE rJ rN r( a-
| Foo | Bar | Baz |
| one | two | three |
| four | five | six |
Nr: r= r r r! test_no_headers s
z TestTableBlock.test_no_headersc C sN ddg dg dgd}t }d}||}| || | ||| dS )ze
Make sure we get back good json and make
sure it translates back to python.
F)rF r N)g @rG rH r( zs{"first_row_is_table_header": false, "first_col_is_header": false, "data": [["Foo", 1, null], [3.5, "Bar", "Baz"]]}N)r Zvalue_for_formZassertJSONEqual assertEqualZvalue_from_form)r r1 r2 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 rB )r Zblock1rC Zblock2r r r! test_is_html_renderer/ s
z$TestTableBlock.test_is_html_rendererc C sB ddg dg dg dgd}t }||}| |g d d S )NFr# )NrG N)NrF Nr( )r$ r% r& rG rF r get_searchable_contentrU r r1 r2 contentr r r! test_searchable_content> s
z&TestTableBlock.test_searchable_contentc C s$ d }t }||}| |g d S NrY r[ r r r! &test_searchable_content_for_null_blockU s
z5TestTableBlock.test_searchable_content_for_null_blockc C sV t dd}ddg dg dg dgd}|j|ddid }| d
| | d | d S )
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( captionzA fascinating table.)contextr$ z A fascinating table.
N)r r- r/ )r r2 r1 r3 r r r! test_render_with_extra_context[ s
z-TestTableBlock.test_render_with_extra_contextc C sP dddg dg dg dgd}t }||}d}| || | d| dS ) z5
Test a generic render with caption.
ra Fr# r' )Z
table_captionr) r* r+ aK
caption
| Test 1 | Test 2 | Test 3 |
| | |
| | |
r% Nr, r0 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 assertNotInr0 r r r! &test_empty_table_block_is_not_rendered s
z5TestTableBlock.test_empty_table_block_is_not_renderedc
C sr ddg dg dg dgddddd d ddgdddd d
dd d dd
gd }t }||}d }| || d
S )z<
Test that merged table cells are rendered.
F)rK NrL )rM rO rP )rQ ZsevenNr r T)r6 r7 hidden )r6 r7 ZrowspanZcolspan)r) r* r+ r8 Z
mergeCellsaG
| one | two |
| three | four | five |
| six | seven |
Nr: r0 r r r! test_merge_cells_render s"
z&TestTableBlock.test_merge_cells_renderc C s t }| |d d S r^ )r Z assertIsNoneZ to_pythonr r2 r r r! test_empty_table_block_to_python s z/TestTableBlock.test_empty_table_block_to_pythonN)__name__
__module__ __qualname__r" r5 r9 r; r> rD rI rR rS rT rV rX r] r_ rc rd rg rj rl 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 sL ddg dg dg dg dg dgd| _ t }t | _|| jd< d S ) NTZShipTypeZStatus Galactica
BattlestarActiveZValkyrieru DestroyedzCylon BasestarZBasestarrv BrenikzSmall Military Vesselrx r( r )r1 r Z get_languager rB 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 rU rA Zget_table_optionsr r rk 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/ rA r activaterU ZassertNotEqual)r r2 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 rA rU rW )r Zdefault_context_menurA 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.
ri )r r r@ r r r noneZ
allowEmptyFN)r rA rU rW )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 sp t dd}|d t |}| |d d | |d t | |d dddd |j|jd
dd d id
d S )NzA table to display data.) descriptionZtest_tableblockr r ri zTest tableblockTtablez0w-field w-field--char_field w-field--table_inputZ ADD_COMMENTz Add Comment) labelr requiredZiconZ
blockDefIdZ
isPreviewable classnameZshowAddCommentButtonstrings) r set_namer js_argsrU ZassertIsInstancer
Zdefinition_prefixZis_previewable)r r2 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.
)r1 rt r{ N)r rZ r1 r/ )r r2 Zsearch_contentr r r! r] G s z*TestTableBlockForm.test_searchable_contentN)
rm rn ro r" r| r r r r r] r r r r! rp s rp c @ s" e Zd Zdd Zejdd ZdS )TestTableBlockPageEditc C sv ddg dg dg dg dg dgd| _ tjjdd | _td
td | j d gd
}| jj|d| _ |
| _ d S )NTrq rs rw ry rz r( ri )idZShipsr )typer1 )titler )instance) r1 r ZobjectsgetZ root_pager
jsondumpsZ add_childtable_block_pageloginuser)r Ztable_block_page_instancer r r! r" S 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)argszWru rt z#initTable("table\u002D0\u002Dvalue"r r N)clientr r r r ZassertContains)r responser r r! test_page_edit_page_viewh s z/TestTableBlockPageEdit.test_page_edit_page_viewN)rm rn ro r" unittestZexpectedFailurer r r r r! r R 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 rp r r r r r! s" 3