U
|hH @ 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 @ sd e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dS )TestCreateOrUpdateForObjectc
C s` t }tj|| _|jjdt d| _|jjdt d| _|jjdt d| _ t
ddddd d
| jd | _ t d | jd
dt d| j ddt d| jddg| j _
tjjdd| _| jj| j d | jjt| jjddf| jjt| jjdd| j j
jd
dj df| jjt| j jdd| j j
jddj df| jjt| jjdd| j j
jddj dfh| _d S )NzTest 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)r 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 r@ j/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/tests/test_reference_index.pysetUp s|
z!TestCreateOrUpdateForObject.setUpc C s* | tt| jdddd| j d S )Nto_content_type to_object_id
model_path content_path)assertSetEqualsetr get_references_for_objectr7 values_listr> r? r@ r@ rA testb 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_typerF Z
hero_image)rM rN Z object_idrC rD rE rF Zcontent_path_hashr( rC rD rE rF r, r- r" r. r/ r& r' )r r0 r9 Z_get_base_content_typer7 r r1 r3 r= r2 r5 Z_get_content_path_hashZcreate_or_update_for_objectZ
assertTruefilterr) exists assertFalserG rH rI rJ r< r4 r8 r6 )r? Zreference_to_keepZreference_to_remover@ r@ rA test_updatel sf
z'TestCreateOrUpdateForObject.test_updatec C sD t jj| jjd}| | tt | j
dddd| j d S )N)r= rC rD rE rF ) r
r0 r9 r7 r= saverG rH r rI rJ r> )r? pager@ r@ rA 1test_saving_base_model_does_not_remove_references s zMTestCreateOrUpdateForObject.test_saving_base_model_does_not_remove_referencesc C s: t d| jj d}| t| j}| | d d S )Nz0event page
)contentr )r r7 r) rS r get_references_to assertEqualcount)r? objrefsr@ r@ rA test_null_parental_key s z2TestCreateOrUpdateForObject.test_null_parental_keyc sH t }t fdd|_ j|_| t j} | d d S )Nc s j S )N)r7 r@ rK r@ rA zDTestCreateOrUpdateForObject.test_lazy_parental_key..r" )
r r rT r: Z link_pagerS r rW rX rY )r? Zevent_page_related_linkr[ r@ rK rA test_lazy_parental_key s z2TestCreateOrUpdateForObject.test_lazy_parental_keyc C sV t d| jd}| jj|d t dd d}| jj|d t| j}| | d d S )Ngeneric snippet pager Zsnippet_content_objectr* r" )r r7 r: r; r rW rX rY r? page1page2r[ r@ r@ rA test_generic_foreign_key s z4TestCreateOrUpdateForObject.test_generic_foreign_keyc C sV t d| jd}| jj|d t dd d}| jj|d t| j}| | d d S )Nr` ra r* r )r r7 r: r; r rW rX rY rb r@ r@ rA +test_model_index_ignore_generic_foreign_key s zGTestCreateOrUpdateForObject.test_model_index_ignore_generic_foreign_keyc C s` t j| j}td|d}| jj|d tdd d}| jj|d t|}| |
d d S )Nr` )r Z snippet_content_type_nonindexedr* r ) r r0 r1 r7 r r: r; r rW rX rY )r? rN rc rd r[ r@ r@ rA 1test_model_field_index_ignore_generic_foreign_key s
zMTestCreateOrUpdateForObject.test_model_field_index_ignore_generic_foreign_keyc C s( t }tjdd|d | | d S )Nrebuild_references_indexr ) verbositystdout)r r call_commandrQ getvaluer? rj r@ r@ rA *test_rebuild_references_index_no_verbosity s zFTestCreateOrUpdateForObject.test_rebuild_references_index_no_verbosityc C s8 t }tjd|d | d| | d| d S )NZshow_references_indexrj z 3 wagtail.images.models.Imagez) 4 wagtail.test.testapp.models.EventPage)r r rk ZassertInrl rm r@ r@ rA test_show_references_index s z6TestCreateOrUpdateForObject.test_show_references_indexN)__name__
__module__ __qualname__rB rL rR rU r\ r_ re rf rg rn rp r@ r@ r@ rA r s D
>
r c @ s6 e Zd ZdgZedd Zdd Zdd Zdd Zd
S ) TestDescribeOnDeletez test.jsonc C s t jdt d d S )Nrh ro )r rk r )clsr@ r@ rA
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)rz Zon_delete_cascadezOn delete cascadez0the various on delete model will also be deletedzon_delete=PROTECT)rz on_delete_protectzOn delete protectzprevents deletionzon_delete=RESTRICT)rz Zon_delete_restrictzOn delete restrictzmay prevent deletionzon_delete=SET_NULL)rz Zon_delete_set_nullzOn delete set nullzwill unset the referencezon_delete=SET_DEFAULT)rz on_delete_set_defaultzOn delete set defaultz2will be set to the default various on delete modelz
on_delete=SET)rz Z
on_delete_setz
On delete setz@will be set to a various on delete model specified by the systemzon_delete=DO_NOTHING)rz Zon_delete_do_nothingzOn delete do nothingzwill do nothingZGenericForeignKey)rz Zcontent_objectzContent objectzRichTextField model field)rz rich_textz Rich textzdeep RichTextBlockr} )rz rw 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 _meta get_fieldZ stream_blockZ child_blocksr r0 r3 r~ r Z advert_uuidr firstrT r) r r r$ r r
r r r r cases)r? fieldrx ry Z page_linkr@ r@ rA rB s
!
zTestDescribeOnDelete.setUpc
C s | j D ]\}}}}| j|d d d|kr4d |d< tjjf |}t| }|d \}}|d } | |j d|k |
| d |
|| |
t |d |
|
| |
| | | W 5 Q R X qd S )Nrz )rL r| r r{ r" )r ZsubTestr r0 r3 r rW group_by_source_objectZassertIsZ is_protectedrX rY lendescribe_source_fielddescribe_on_deletedelete)
r? Z init_kwargsZreferred_objectZfield_descriptionZon_delete_descriptionrZ usageZreferrer
references referencer@ r@ rA (test_describe_source_field_and_on_delete s, 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 r0 r r rW r rX rY r r )r? r~ r _r r r@ r@ rA 5test_describe_source_field_and_on_delete_parental_key s
zJTestDescribeOnDelete.test_describe_source_field_and_on_delete_parental_keyN) rq rr rs Zfixtures classmethodrv rB r r r@ r@ r@ rA rt s
) rt 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 rt r@ r@ r@ rA s 0 l