o
hM @ 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 ddf 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 sow 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 sluglocationaudiencecost date_from
feed_image1234567 )captionimage
sort_order7654321 abcdefg 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_childr2 strpkexpected_references)self image_model rK b/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/tests/test_reference_index.pysetUp st
+
z!TestCreateOrUpdateForObject.setUpc C s* | tt| jdddd| j d S )Nto_content_type to_object_id
model_path content_path)assertSetEqualsetr get_references_for_objectrA values_listrH rI rK rK rL 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_typerQ
hero_image)rX rY object_idrN rO rP rQ content_path_hashr1 rN rO rP rQ r5 r6 r) r7 r8 r. r0 )r r: rC _get_base_content_typerA r r; r= rG r< r? _get_content_path_hashcreate_or_update_for_object
assertTruefilterr2 exists assertFalserR rS rT rU rF r> rB r@ )rI reference_to_keepreference_to_removerK rK rL test_updatem s`
z'TestCreateOrUpdateForObject.test_updatec C sp | j dd tjj| jjd}| W d n1 s w Y | tt
| j dddd| j d S )NTr )rG rN rO rP rQ )
r9 r
r: rC rA rG saverR rS r rT rU rH )rI pagerK rK rL 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 w Y t| j}| | d d S )NTr z0event page
)contentr ) r9 r rA r2 rg r get_references_to assertEqualcount)rI objrefsrK rK rL test_null_parental_key s
z2TestCreateOrUpdateForObject.test_null_parental_keyc st j dd t }t fdd|_ j|_| W d n1 s%w Y t j} |
d d S )NTr c s j S )N)rA rK rV rK rL s zDTestCreateOrUpdateForObject.test_lazy_parental_key..r) ) r9 r r rh rD link_pagerg r rk rl rm )rI event_page_related_linkro rK rV rL test_lazy_parental_key s
z2TestCreateOrUpdateForObject.test_lazy_parental_keyc C | j dd# td| jd}| jj|d tdd d}| jj|d W d n1 s,w Y t| j}| | d d S )NTr generic snippet pager snippet_content_objectr3 r) ) r9 r rA rD rE r rk rl rm rI page1page2ro rK rK rL test_generic_foreign_key s
z4TestCreateOrUpdateForObject.test_generic_foreign_keyc C ru )NTr rv rw r3 r ) r9 r rA rD rE r rk rl rm ry rK rK rL +test_model_index_ignore_generic_foreign_key s zGTestCreateOrUpdateForObject.test_model_index_ignore_generic_foreign_keyc C s | j dd) tj| j}td|d}| jj|d tdd d}| jj|d W d n1 s2w Y t |}|
| d d S )NTr rv )r snippet_content_type_nonindexedr3 r ) r9 r r: r; rA r rD rE r rk rl rm )rI rY rz r{ ro rK rK rL 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 s(w Y t | j}| | d d S )Nr Tr An advertisement)textrh r) ) r rk rA rl rm r9 r r: r= )rI ro rK rK rL 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 )Nr r )r r r )clsrK rK rL
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)r on_delete_cascadezOn delete cascadez0the various on delete model will also be deletedzon_delete=PROTECT)r on_delete_protectzOn delete protectzprevents deletionzon_delete=RESTRICT)r on_delete_restrictzOn delete restrictzmay prevent deletionzon_delete=SET_NULL)r on_delete_set_nullzOn delete set nullzwill unset the referencezon_delete=SET_DEFAULT)r on_delete_set_defaultzOn delete set defaultz2will be set to the default various on delete modelz
on_delete=SET)r
on_delete_setz
On delete setz@will be set to a various on delete model specified by the systemzon_delete=DO_NOTHING)r on_delete_do_nothingzOn delete do nothingzwill do nothingGenericForeignKey)r content_objectzContent objectzRichTextField model field)r rich_textz Rich textzdeep RichTextBlockr )r r u4 Stream field → Advertisement content → Rich textzdeep SnippetChooserBlockadvert)r* z Deep textu; Stream field → Advertisement content → Captioned advertImageChooserBlockr+ u Stream field → ImageDocumentChooserBlockdocumentu Stream field → Document)r _meta get_field stream_block child_blocksr r: r= r r advert_uuidr firstrh r2 r r r+ r r
r r r r cases)rI fieldr r page_linkrK rK rL rM $ s
!
zTestDescribeOnDelete.setUpc
C s | j D ]\}}}}| j|d ds d|vrd |d< | jdd tjjd i |}W d n1 s4w Y t| }|d \}}|d } | |j
d|v | | d | || | t
|d | | | | | | | W d n1 sw Y qd S )
Nr )rW r Tr r r r) rK )r subTestr9 r r: r= r rk group_by_source_objectassertIs is_protectedrl rm lendescribe_source_fielddescribe_on_deletedelete)
rI init_kwargsreferred_objectfield_descriptionon_delete_descriptionrn usagereferrer
references referencerK rK rL (test_describe_source_field_and_on_delete s:
z=TestDescribeOnDelete.test_describe_source_field_and_on_deletec C sb t j }t| }| | d |D ]\}}|d }| | d | | d qd S )Nr. r r z)the advert placement will also be deleted)
r r: r r rk r rl rm r r )rI r r _r r rK rK rL 5test_describe_source_field_and_on_delete_parental_key s
zJTestDescribeOnDelete.test_describe_source_field_and_on_delete_parental_keyN) r r r fixtures classmethodr rM r r rK rK rK rL r s
) r N)'ior "django.contrib.contenttypes.modelsr django.corer django.testr django.utils.functionalr wagtail.blocksr r wagtail.documentsr wagtail.documents.tests.utilsr
wagtail.imagesr wagtail.images.tests.utilsr wagtail.modelsr
r wagtail.rich_textr wagtail.test.testapp.modelsr r r r r r r r r r r r rK rK rK rL s" 0