o
hV @ 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
colHeaders
rowHeaders contextMenueditorstretchHheightlanguagerendererautoColumnSize)default_table_options)self r" a/var/www/html/ndineBlogger/venv/lib/python3.10/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 htLeft)rowcol classNamehtRightr% r) )r+ r, cellr- a
r' Nr. r2 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/ r0 )r! r4 r5 r6 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! r3 r6 r4 r5 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
FrA r) r* a*
htmlr
table_optionsN)r copyr r/ r0 )r! r3 r6 new_optionsr4 r5 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? rB 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.
FTrJ onetwothreefourfivesixr* aQ
| Foo | Bar | Baz |
|---|
| one | two | three |
|---|
| four | five | six |
|---|
Nr? rB r" r" r# test_column_headers
z"TestTableBlock.test_column_headersc C B ddg dg dg dgd}d}t }||}| || dS )z?
Test row and column headers at the same time.
TrJ rO rS r* a
| Foo | Bar | Baz |
|---|
| one | two | three |
|---|
| four | five | six |
|---|
Nr? rB r" r" r# test_row_and_column_headers s
z*TestTableBlock.test_row_and_column_headersc C rY )z-
Test table without headers.
FrJ rO rS r* a-
| Foo | Bar | Baz |
| one | two | three |
| four | five | six |
Nr? rB r" r" r# test_no_headers rX 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)rK r N)g @rL rM r* zs{"first_row_is_table_header": false, "first_col_is_header": false, "data": [["Foo", 1, null], [3.5, "Bar", "Baz"]]}N)r value_for_formassertJSONEqual assertEqualvalue_from_form)r! r3 r4
expected_json
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.
FrD r rE TN)r assertIsis_html_rendererr rG )r! block1rH block2r" 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% )NrL N)NrK Nr* )r& r' r( rL rK r get_searchable_contentr^ r! r3 r4 contentr" r" r# test_searchable_content> s
z&TestTableBlock.test_searchable_contentc C s$ d }t }||}| |g d S Nrh rj 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/ r1 )r! r4 r3 r5 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.
rp Fr% r) )
table_captionr+ r, r- aK
caption
| Test 1 | Test 2 | Test 3 |
| | |
| | |
r' Nr. r2 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/ r0 assertNotInr2 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)rP NrQ )rR rT rU )rV sevenNr r T)r9 r: hidden )r9 r: rowspancolspan)r+ r, r- r=
mergeCellsaG
| one | two |
| three | four | five |
| six | seven |
Nr? r2 r" r" r# test_merge_cells_render s"
z&TestTableBlock.test_merge_cells_renderc C s t }| |d d S rm )r assertIsNone to_pythonr! r4 r" r" r# test_empty_table_block_to_python s z/TestTableBlock.test_empty_table_block_to_pythonN)__name__
__module__ __qualname__r$ r7 r> r@ rC rI rN rW rZ r[ rb rg rl rn rr rt rx r r 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 ) NTShipTypeStatus Galactica
BattlestarActiveValkyrier DestroyedzCylon BasestarBasestarr BrenikzSmall Military Vesselr r* r )r3 r get_languager rG 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 r^ rF get_table_optionsr r r 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-fritjarE r N)r r1 rF r activater^ assertNotEqual)r! r4 block_frblock_itblock_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 TrE FundoredoN)listr r rF r^ rc )r! default_context_menurF table_options_menu_truetable_options_menu_falsetable_options_menu_listr" r" r# test_table_options_context_menu s6
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.
r{ )r r rE r r r none
allowEmptyFN)r rF r^ rc )r! block_1_opts block_2_opts 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.) descriptiontest_tableblockr r r{ zTest tableblockTtablez0w-field w-field--char_field w-field--table_input ADD_COMMENTz Add Comment) labelr requiredicon
blockDefId
isPreviewable classnameshowAddCommentButtonstrings) r set_namer js_argsr^ assertIsInstancer
definition_prefixis_previewable)r! r4 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.
)r3 r r N)r ri r3 r1 )r! r4 search_contentr" r" r# rl G s z*TestTableBlockForm.test_searchable_contentN)
r r r r$ r r r r r rl r" r" r" r# r s r 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 )NTr r r r r r* r{ )idShipsr )typer3 )titler )instance) r3 r objectsget root_pager
jsondumps add_childtable_block_pageloginuser)r! table_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)argszWr r z#initTable("table\u002D0\u002Dvalue"r r N)clientr r r r assertContains)r! responser" r" r# test_page_edit_page_viewh s z/TestTableBlockPageEdit.test_page_edit_page_viewN)r r r r$ unittestexpectedFailurer r" r" r" r# r R s r )r r django.testr r django.urlsr django.utilsr wagtail.blocks.field_blockr "wagtail.contrib.table_block.blocksr r wagtail.modelsr wagtail.test.testapp.modelsr
wagtail.test.utilsr blocksr
r r r r" r" r" r# s$ 3