a
D gM @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z d dl
mZ d dlmZ d d lmZ d d
lmZ d d lmZmZ d d lmZ d d
lmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$ G dd deZ%G dd deZ&dS ) )StringIO) ContentType)
management)TestCase)SimpleLazyObject) StreamValue StructValue)get_document_model)get_test_document_file)get_image_model)get_test_image_file)PageReferenceIndex)RichText)
Advert AdvertWithCustomUUIDPrimaryKey EventPageEventPageCarouselItemEventPageRelatedLink GenericSnippetNoFieldIndexPageGenericSnippetNoIndexPageGenericSnippetPage ModelWithNullableParentalKeyVariousOnDeleteModelc @ sl e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd ZdS )TestCreateOrUpdateForObjectc
C s | j dd t }tj|| _|jjdt d| _|jjdt d| _ |jjdt d| _
t ddd d
d d | jd
| _ t
d| j ddt
d| j
ddt
d| j ddg| j _tjjdd| _| jj| j d W d n1 s0 Y | jjt| jjddf| jjt| j jdd| j jjddj df| jjt| j
jdd| j jjddj df| jjt| j jdd| j jjddj dfh| _d S ) NTexecutezTest feed imagetitlefilez Test image 1z Test image 2z
Event pagez
event-pagezthe moonpublicfreez
2001-01-01)r ZsluglocationZaudienceZcostZ date_from
feed_imageZ1234567 )captionimage
sort_orderZ7654321 Zabcdefg idinstancer# carousel_items.item.imagecarousel_items.r' .image)captureOnCommitCallbacksr r objects
get_for_modelimage_content_typecreater test_feed_image test_image_1 test_image_2r
event_pager carousel_itemsr
get root_page add_childr+ strpkexpected_references)selfZ image_model rC a/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/tests/test_reference_index.pysetUp sr
.
z!TestCreateOrUpdateForObject.setUpc C s* | tt| jdddd| j d S )Nto_content_type to_object_id
model_path content_path)assertSetEqualsetr get_references_for_objectr: values_listrA rB rC rC rD testc s z TestCreateOrUpdateForObject.testc
C sd t jjt | jtj| jdd}t jjt | jtj| j| jj| j | j
jddt dd}t | j |
t jj|jd | t jj|jd | tt | jdddd | j jt| jjddf| j jt| j
jd
d | jjjd d
j df| j jt| jjd
d | jjjdd
j df| j jt| j
jd
d | jjjdd
j dfh d S )Nr# )base_content_type content_typerI Z
hero_image)rP rQ Z object_idrF rG rH rI Zcontent_path_hashr* rF rG rH rI r. r/ r$ r0 r1 r( r) )r r3 r< Z_get_base_content_typer: r r4 r6 r@ r5 r8 Z_get_content_path_hashZcreate_or_update_for_object
assertTruefilterr+ exists assertFalserJ rK rL rM r? r7 r; r9 )rB Zreference_to_keepZreference_to_removerC rC rD test_updatem s`
z'TestCreateOrUpdateForObject.test_updatec C sp | j dd* tjj| jjd}| W d n1 s<0 Y | tt
| j dddd| j d S )NTr )r@ rF rG rH rI )
r2 r
r3 r< r: r@ saverJ rK r rL rM rA )rB pagerC rC rD 1test_saving_base_model_does_not_remove_references s & zMTestCreateOrUpdateForObject.test_saving_base_model_does_not_remove_referencesc C sf | j dd* td| jj d}| W d n1 s<0 Y t| j}| | d d S )NTr z0event page
)contentr ) r2 r r: r+ rW r get_references_to assertEqualcount)rB objrefsrC rC rD test_null_parental_key s & z2TestCreateOrUpdateForObject.test_null_parental_keyc st j dd8 t }t fdd|_ j|_| W d n1 sJ0 Y t j} |
d d S )NTr c s j S )N)r: rC rN rC rD zDTestCreateOrUpdateForObject.test_lazy_parental_key..r$ ) r2 r r rX r= Z link_pagerW r r[ r\ r] )rB Zevent_page_related_linkr_ rC rN rD test_lazy_parental_key s & z2TestCreateOrUpdateForObject.test_lazy_parental_keyc C s | j ddF td| jd}| jj|d tdd d}| jj|d W d n1 sX0 Y t| j}| | d d S )NTr generic snippet pager Zsnippet_content_objectr, r$ ) r2 r r: r= r> r r[ r\ r] rB page1page2r_ rC rC rD test_generic_foreign_key s , z4TestCreateOrUpdateForObject.test_generic_foreign_keyc C s | j ddF td| jd}| jj|d tdd d}| jj|d W d n1 sX0 Y t| j}| | d d S )NTr rd re r, r ) r2 r r: r= r> r r[ r\ r] rf rC rC rD +test_model_index_ignore_generic_foreign_key s , zGTestCreateOrUpdateForObject.test_model_index_ignore_generic_foreign_keyc C s | j ddR tj| j}td|d}| jj|d tdd d}| jj|d W d n1 sd0 Y t |}|
| d d S )NTr rd )r Z snippet_content_type_nonindexedr, r ) r2 r r3 r4 r: r r= r> r r[ r\ r] )rB rQ rg rh r_ rC rC rD 1test_model_field_index_ignore_generic_foreign_key s ,
zMTestCreateOrUpdateForObject.test_model_field_index_ignore_generic_foreign_keyc C sz t | j}| | d | jdd" tjjd| jd W d n1 sP0 Y t | j}| | d d S )Nr Tr An advertisement)textrX r$ ) r r[ r: r\ r] r2 r r3 r6 )rB r_ rC rC rD test_model_with_uuid_primary_key s $ z
r c @ s6 e Zd ZdgZedd Zdd Zdd Zdd Zd
S ) TestDescribeOnDeletez test.jsonc C s t jdt d d S )Nro rv )r rr r )clsrC rC rD
setUpTestData s z"TestDescribeOnDelete.setUpTestDatac C s t jd}|jjd }|jd }tjjdd| _t jjdd| _
t j | _
d| j
j d}t jjd t d
| _t jjd t d
| _d | jd
| jddfd| jd| jddfd| jd| jddfd| jd| jddfd | jd | jd d fd | jd!| jd"d#fd$| jd%| jd&d'fd(| j
d)| j
d*dfd+|d,| j
d-dfd.t|jdt|d/t|fgfgd0| j
d1dfd2t|jdt|dt|d3| jfd4gfgfgd0| jd5dfd6t|jd7| jfgd0| jd8dfd9t|jd:| jfgd0| jd;dfg
| _d S )Link to a pagezMy imager z My documentzon_delete=CASCADE)rm Zon_delete_cascadezOn delete cascadez0the various on delete model will also be deletedzon_delete=PROTECT)rm on_delete_protectzOn delete protectzprevents deletionzon_delete=RESTRICT)rm Zon_delete_restrictzOn delete restrictzmay prevent deletionzon_delete=SET_NULL)rm Zon_delete_set_nullzOn delete set nullzwill unset the referencezon_delete=SET_DEFAULT)rm on_delete_set_defaultzOn delete set defaultz2will be set to the default various on delete modelz
on_delete=SET)rm Z
on_delete_setz
On delete setz@will be set to a various on delete model specified by the systemzon_delete=DO_NOTHING)rm Zon_delete_do_nothingzOn delete do nothingzwill do nothingZGenericForeignKey)rm Zcontent_objectzContent objectzRichTextField model field)rm rich_textz Rich textzdeep RichTextBlockr )rm r~ u4 Stream field → Advertisement content → Rich textzdeep SnippetChooserBlockadvert)r% z Deep textu; Stream field → Advertisement content → Captioned advertZImageChooserBlockr& u Stream field → ImageZDocumentChooserBlockdocumentu Stream field → Document)r Z_meta get_fieldZ stream_blockZ child_blocksr r3 r6 r r Z advert_uuidr firstrX r+ r r r& r r
r r r r cases)rB fieldr r Z page_linkrC rC rD rE $ s
!
zTestDescribeOnDelete.setUpc
C s" | j D ]\}}}}| j|d d d|vr6d |d< | jdd" tjjf i |}W d n1 sj0 Y t| }|d \}}|d } | |j
d|v | | d | || | t
|d | | | | | | | W d q1 s0 Y qd S ) Nrm )rO r Tr r r r$ )r ZsubTestr2 r r3 r6 r r[ group_by_source_objectZassertIsZ is_protectedr\ r] lendescribe_source_fielddescribe_on_deletedelete)
rB Z init_kwargsZreferred_objectZfield_descriptionZon_delete_descriptionr^ usageZreferrer
references referencerC rC rD (test_describe_source_field_and_on_delete s. 0 z=TestDescribeOnDelete.test_describe_source_field_and_on_deletec C sb t j }t| }| | d |D ]0\}}|d }| | d | | d q,d S )Nr( r r z)the advert placement will also be deleted)
r r3 r r r[ r r\ r] r r )rB r r _r r rC rC rD 5test_describe_source_field_and_on_delete_parental_key s
zJTestDescribeOnDelete.test_describe_source_field_and_on_delete_parental_keyN) rx ry rz Zfixtures classmethodr} rE r r rC rC rC rD r{ s
) r{ N)'ior Z"django.contrib.contenttypes.modelsr Z django.corer Z django.testr Zdjango.utils.functionalr Zwagtail.blocksr r Zwagtail.documentsr Z wagtail.documents.tests.utilsr
Zwagtail.imagesr Zwagtail.images.tests.utilsr Zwagtail.modelsr
r Zwagtail.rich_textr Zwagtail.test.testapp.modelsr r r r r r r r r r r r{ rC rC rC rD s 0