a
D g5H @ s d dl Zd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZ d d lmZ d d
lmZ d d lmZ d d
lmZmZmZmZ G dd deZ G dd deZ G dd de Z G dd deZ dS ) N)apps)TestCase)
SafeString)compare) StreamValue)StructBlockValidationError)
ImageBlockImageChooserBlock) JSContext)
StreamPage)WagtailTestUtils )Imageget_test_bad_imageget_test_image_fileget_test_image_filenamec @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )TestImageChooserBlockc C s* t jjdt d| _t | _| j d S )N
Test imagetitlefile)r objectscreater imager bad_imagesaveself r _/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/images/tests/test_blocks.pysetUp s zTestImageChooserBlock.setUpc C s4 t }|| j}dt| jd}| || d S )Nz8
original)r renderr formatr assertHTMLEqualr blockhtmlZ
expected_htmlr r r test_render! s
z!TestImageChooserBlock.test_renderc C sb t }tjjjtdddddd || j}W d n1 sH0 Y | |d d S )NZ
wagtailimagesZ
default_attrsasyncZlazy)ZdecodingZloading)newze
)
r unittestZmockpatchobjectr Zget_app_configr" r r$ )r r& r' r r r %test_render_with_custom_default_attrs, s *z;TestImageChooserBlock.test_render_with_custom_default_attrsc C s& t }|| j}d}| || d S )NzE)r r" r r$ r% r r r test_render_missing9 s
z)TestImageChooserBlock.test_render_missingc C sD t dd}| \}}}| |d | |d | |ddi d S )NFrequiredz'wagtail.images.blocks.ImageChooserBlockr r1 )r Z deconstruct assertEqual)r r& pathargskwargsr r r test_deconstructB s
z&TestImageChooserBlock.test_deconstructc C sJ t }| t|| jtt| jjddfg | t|d g d S )N )r ZassertListEquallistZextract_referencesr r strid)r r& r r r test_extract_referencesI s z-TestImageChooserBlock.test_extract_referencesN) __name__
__module__ __qualname__r r( r. r/ r6 r; r r r r r s
r c @ s* e Zd ZejZdd Zdd Zdd ZdS ) TestImageChooserBlockComparisonc C s: t jjdt d| _t jjdt d| _tjd| _ d S Nz Test image 1r z Test image 2body
r r r r image_1image_2r _meta get_fieldfieldr r r r r X s z%TestImageChooserBlockComparison.setUpc
C s t jd}| |t t|jd| jdfgdt t|jd| jdfgd}| |j | |j
| | d |
}| |t | d| | d| | d| | d | | | d S )
NrA r 1rA Body class="comparison__child-object"class="preview-image"deletionaddition)r rE rF comparison_classr stream_blockrC
assertTrueis_field assertFalseis_child_relationr2 field_labelhtmldiffassertIsInstancer assertIn assertNotIn has_changedr rG
comparisonrV r r r test_hasnt_changede s8
z2TestImageChooserBlockComparison.test_hasnt_changedc
C s t jd}| |t t|jd| jdfgdt t|jd| jdfgd}| |j |
|j | |
d | }| |t | d| | d| | d| | | d S ) NrA r rH rI rJ rK class="preview-image deletion" class="preview-image addition")r rE rF rO r rP rC rD rQ rR rS rT r2 rU rV rW r rX rZ r[ r r r test_has_changed s6
z0TestImageChooserBlockComparison.test_has_changedN) r< r= r> r StreamFieldComparisonrO r r] r` r r r r r? U s
"r? c @ 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#S )$TestImageBlockc C sd t }| jjddd}|||}t|}|d}| |d | d |
d|d d S NSample alt textFr alt_text
decorativeimgaltrf /media/images/testsrc) r r r: r" to_pythonr get_soupfindr2 getrX r r& valuer' ZsoupZimg_tagr r r r( s
zTestImageBlock.test_renderc C sd t }| jjddd}|||}t|}|d}| |d | d |
d|d d S rc ) r r r: Z render_basicrl r rm rn r2 ro rX rp r r r test_render_basic s
z TestImageBlock.test_render_basicc C s^ t }| jjddd}|||}t|}|d}| |d d | d|d d S ) Nrd Tre rh ri r7 rj rk )
r r r: r" rl r rm rn r2 rX rp r r r test_render_as_decorative s
z(TestImageBlock.test_render_as_decorativec C sl t }| jjd dd}| t }||| W d n1 sF0 Y | dt|j j
d d S )NFre z@Please add some alt text for your image or mark it as decorativerf ) r r r: assertRaisesr cleanrl rX r9 exception block_errors)r r& rq contextr r r test_no_alt_text s .z TestImageBlock.test_no_alt_textc C sD t }|| jj}| |j| jj | |jd | |j d S )Nr )r rl r r: r2 contextual_alt_textrS rg r r& rq r r r test_to_python_with_int s
z&TestImageBlock.test_to_python_with_intc C sP t }| jjddd}||}| |j| jj | |jd | |j d S )Nz Sample textFre )r r r: rl r2 rz rS rg r r& rq resultr r r test_to_python_with_dict s
z'TestImageBlock.test_to_python_with_dictc C s" t dd}|d }| | d S NFr0 )r rl assertIsNoner{ r r r test_to_python_with_none s
z'TestImageBlock.test_to_python_with_nonec C s$ t dd}|g }| |g d S r r bulk_to_pythonr2 r r& r~ r r r #test_bulk_to_python_with_empty_list s
z2TestImageBlock.test_bulk_to_python_with_empty_listc C s( t dd}|d g}| |d g d S r r r r r r %test_bulk_to_python_with_list_of_none s
z4TestImageBlock.test_bulk_to_python_with_list_of_nonec C sF t dd}|| jj}|d | jj| jjg}| |d ||g d S r )r rl r r: r r2 )r r& Z single_imager~ r r r %test_bulk_to_python_with_list_of_ints s
z4TestImageBlock.test_bulk_to_python_with_list_of_intsc C s t dd}|d d d d| jjddd| jjdddg}| t|d | |d | |d j| jj | |d jd | |d j| jj | |d jd d S )
NFr0 re Custom alt textzDifferent alt text r r
)r r r r: r2 lenr rz r r r r &test_bulk_to_python_with_list_of_dicts s&
z5TestImageBlock.test_bulk_to_python_with_list_of_dictsc C s> t }d| j_d| j_|| j}| || jjddd d S )Nr Fre )r r rz rg get_prep_valuer2 r: r{ r r r test_get_prep_value, s z"TestImageBlock.test_get_prep_valuec C s, t dd}|d }| |d d d d d S )NFr0 re )r r r2 r{ r r r "test_get_prep_value_for_null_value< s
z1TestImageBlock.test_get_prep_value_for_null_valuec C s8 t }| jjddd}|||}| |dg d S )Nrd Fre )r r r: Zget_searchable_contentrl r2 r} r r r test_get_searchable_contentE s z*TestImageBlock.test_get_searchable_contentc C s t }t |}|d d d }| |d d d |d ddd}| t}|| W d n1 st0 Y | d t |j
j d
d S ) N_argsr
r r r1 r7 Fre zThis field is requiredr ) r r
packrQ rl rt r ru rX r9 rv rw )r r& image_block_defimage_chooser_block_defrq rx r r r test_required_trueQ s (z!TestImageBlock.test_required_truec C sb t dd}t |}|d d d }| |d d d |d ddd }| || d S )
NFr0 r r
r r r1 r7 re )r r
r rS rl r ru )r r& r r rq r r r test_required_falseh s
z"TestImageBlock.test_required_falsec C s t jd}t |jdd| jjddddgd}||jdg}| |j
j d | |j
t | |j
j| jj ||jdd g}| |j
j d | |j
d ||jdd
g}| | d S ) NrA Z123image_with_altrd Fre )r: typerq rI rf Zdoes_not_exist)r rE rF rP rl r r: Zget_block_by_content_pathrA r2 r& namerW rq r r )r rG pageZ bound_blockr r r test_get_block_by_content_pathy s0
z-TestImageBlock.test_get_block_by_content_pathN)r< r= r> r( rr rs ry r| r r r r r r r r r r r r r r r r rb s" rb c @ s2 e Zd ZejZdd Zdd Zdd Zdd Z d S )
TestImageBlockComparisonc C s: t jjdt d| _t jjdt d| _tjd| _ d S r@ rB r r r r r s z TestImageBlockComparison.setUpc C s t jd}t }d| jjdddddg|_t }d| jjdddddg|_| |||}| |j | |j
| | d |
}| |t | d | | d
| | d | | d | | | d S )
NrA r F
Some alt textr rg rf rH r rq r: rJ rK rL rM rN )r rE rF rC r: rA rO rQ rR rS rT r2 rU rV rW r rX rY rZ r rG Zpage_1Zpage_2r\ rV r r r r] s< z+TestImageBlockComparison.test_hasnt_changedc C s t jd}t }d| jjdddddg|_t }d| jjdddddg|_| |||}| |j |
|j | |
d | }| |t | d | | d
| | d | | | d S ) NrA r Fr r rH r rJ rK r^ r_ )r rE rF rC r: rA rD rO rQ rR rS rT r2 rU rV rW r rX rZ r r r r r` s: z)TestImageBlockComparison.test_has_changedc C s t jd}t }d| jjdddddg|_t }d| jjdddddg|_| |||}| |j | |j
| | d |
}| |t | d
| | d | | | d S ) NrA r Fz a cat playing with some stringr rH r z!a kitten playing with some stringrJ rK zia catkitten playing with some string)r rE rF rC r: rA rO rQ rR rS rT r2 rU rV rW r rX rZ r r r r test_alt_text_has_changed s> z2TestImageBlockComparison.test_alt_text_has_changedN)
r< r= r> r ra rO r r] r` r r r r r r s
)(r ) Z
unittest.mockr+ Z django.appsr Z django.testr Zdjango.utils.safestringr Z
wagtail.adminr Zwagtail.blocks.stream_blockr Zwagtail.blocks.struct_blockr Zwagtail.images.blocksr r Zwagtail.telepathr
Zwagtail.test.testapp.modelsr Z wagtail.test.utils.wagtail_testsr utilsr r r r r r? rb r r r r r s >T q