U |®ŽhkVć@s˜ddlmZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZmZGdd„deƒZGdd„deƒZGd d „d eƒZGd d „d eƒZd S)é)ŚTestCase) ŚAlterBlockValueOperationŚ"ListChildrenToStructBlockOperationŚRemoveStreamChildrenOperationŚRemoveStructChildrenOperationŚRenameStreamChildrenOperationŚRenameStructChildrenOperationŚ"StreamChildrenToListBlockOperationŚ$StreamChildrenToStreamBlockOperationŚ$StreamChildrenToStructBlockOperation)Śapply_changes_to_raw_data)Ś factoriesŚmodelsc@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dS)ŚFieldChildBlockTestz+Tests involving changes to top level blockscCs tjdddddjj}||_dS)Nś Char Block 1ś Char Block 2)Ścontent__0__char1__valueZcontent__1__char2__valueZcontent__2__char1__valueŚcontent__3__char2__value©r ZSampleModelFactoryŚcontentŚraw_data©Śselfr©rśƒ/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/tests/streamfield_migrations/test_simple_structures.pyŚsetUpsü zFieldChildBlockTest.setUpcCsīt|jdtdddtjjd}| |ddd”| |ddd”| |dd |jdd ”| |dd |jdd ”| |dd |jdd ”| |dd |jdd ”| |d |jd ”| |d |jd ”d S)zėRename `char1` blocks to `renamed1` Check whether all `char1` blocks have been renamed correctly. Check whether ids and values for renamed blocks are intact. Check whether other block types are intact. ŚŚchar1Śrenamed1©Zold_nameŚnew_name©rZblock_path_strZ operationZ streamfieldrŚtypeéŚidŚvalueééN©r rrrŚ SampleModelrŚ assertEqual©rZaltered_raw_datarrrŚ test_renames"’ś zFieldChildBlockTest.test_renamecCs\t|jdtddtjjd}| t|ƒd”| |d|jd”| |d|jd”d S) zƒRemove all `char1` blocks Check whether all `char1` blocks have been removed and whether other blocks are intact. rr©Śnamer!r#rr&r'N©r rrrr)rr*Ślenr+rrrŚ test_remove:süzFieldChildBlockTest.test_removecCsZt|jdtdddtjjd}| t|ƒd”| |d|jd”| |d|jd”| |d d d”| t|d d ƒd ”| |d d dd d ”| |d d dd d ”| |d d dd |jdd ”| |d d dd |jd d ”| |d d dd |jdd ”| |d d dd |jd d ”dS)aŹCombine all `char1` blocks into a new ListBlock named `list1` Check whether no `char1` blocks are present among the stream children and whether other blocks are intact. Check whether a new `list1` block has been added to the stream children and whether it has child blocks corresponding to the previous `char1` blocks. Check whether the ids and values from the `char1` blocks are intact in the list children. rrŚlist1©Ś block_nameZlist_block_namer!r'rr&r#r"r%Śitemr$N©r rr rr)rr*r0r+rrrŚtest_combine_to_listblockJs4 ’ś && ’ ’z-FieldChildBlockTest.test_combine_to_listblockcCsŠt|jdtdddtjjd}| t|ƒd”| |d|jd”| |d|jd”| |d |jd ”| |d |jd ”d S) zŽCombine all `simplestruct` blocks into a new ListBlock named `list1` We have no `simplestruct` blocks in our existing data, so there should be no list1 blocks created and the data should be intact. rŚ simplestructr2r3r!érr&r#r'Nr6r+rrrŚ.test_combine_to_listblock_no_existing_childrenns’ś zBFieldChildBlockTest.test_combine_to_listblock_no_existing_childrencCsČt|jdtdgddtjjd}| t|ƒd”| |d|jd”| |d|jd”| |d d d”| t|d d ƒd ”| |d d d|jd”| |d d d|jd ”d S) a„Combine all `char1` blocks as children of a new StreamBlock named `stream1` Check whether no `char1` blocks are present among the (top) stream children and whether other blocks are intact. Check whether a new `stream1` block has been added to the (top) stream children. Check whether the new `stream1` block has the `char1` blocks as children and whether they are intact. rrŚstream1©Z block_namesZstream_block_namer!r'rr&r#r"r%N©r rr rr)rr*r0r+rrrŚ'test_combine_single_type_to_streamblockƒs ’ś z;FieldChildBlockTest.test_combine_single_type_to_streamblockcCsŚt|jdtddgddtjjd}| t|ƒd”| |dd d”| t|dd ƒd ”| |dd d|jd”| |dd d|jd”| |dd d |jd ”| |dd d |jd ”dS)aņCombine all `char1` and `char2` blocks as children of a new StreamBlock named `stream1` Check whether no `char1` or `char2` blocks are present among the (top) stream children. Check whether a new `stream1` block has been added to the (top) stream children. Check whether the new `stream1` block has the `char1` and `char2` blocks as children and that they are intact. Note: We only have `char1` and `char2` blocks in our existing data. rrŚchar2r;r<r!r&rr"r%r9r#r'Nr=r+rrrŚ*test_combine_multiple_types_to_streamblock s ’ś z>FieldChildBlockTest.test_combine_multiple_types_to_streamblockcCsŒt|jdtdgddtjjd}| t|ƒd”| |d|jd”| |d|jd”| |d |jd ”| |d |jd ”d S) zīCombine all `simplestruct` blocks as children of a new StreamBlock named `stream1` We have no `simplestruct` blocks in our existing data, so there should be no stream1 blocks created and the data should be intact. rr8r;r<r!r9rr&r#r'Nr=r+rrrŚ0test_combine_to_streamblock_no_existing_children¾s’ś zDFieldChildBlockTest.test_combine_to_streamblock_no_existing_childrencCsąt|jdtddƒtjjd}| |ddd”| |ddd”| |d|jd”| |d |jd ”| d|dd ”| d|dd ”| |dd d|jdd ”| |dd d|jdd ”d S) a«Move each `char1` block inside a new StructBlock named `struct1` Check whether each `char1` block has been replaced with a `struct1` block in the stream children. Check whether other blocks are intact. Check whether each `struct1` block has a `char1` child and whether it has the value of the previous `char1` block. Note: Block ids are not preserved here. rrZstruct1r!rr"r#r&r'r%N)r rr rr)rr*ŚassertInr+rrrŚtest_to_structblockÓs( ü ’ ’z'FieldChildBlockTest.test_to_structblockcCs„t|jdtddtjjd}| |ddd”| |dd|jdd”| |ddd”| |d d|jd d”d S) zĀChange the value of each `char1` block to `foo` Check whether the value of each `char1` block has changed to `foo`. Check whether the values of other blocks are intact. rZfoo)Ś new_valuer!rr%r&r#r'N)r rrrr)rr*r+rrrŚtest_alter_valueösüz$FieldChildBlockTest.test_alter_valueN)Ś__name__Ś __module__Ś __qualname__Ś__doc__rr,r1r7r:r>r@rArCrErrrrrs $#rc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ŚFieldStructChildBlockTestz|Tests involving changes to direct children of a StructBlock We use `simplestruct` blocks as the StructBlocks here. cCs tjdddddjj}||_dS)Nrr8r)rŚ content__1Ś content__2rrrrrrrsü zFieldStructChildBlockTest.setUpcCsĘt|jdtdddtjjd}| |d|jd”| |d|jd”| |dd |jdd ”| |d d |jd d ”| |dd |jdd ”| |d d |jd d ”d S) įsTest whether other blocks and data not passed to an operation are intact. We are checking whether the parts of the data which are not passed to an operation are intact. Since the recursion process depends just on the block path and block structure, this check is independent of the operation used. We will use a rename operation for now. r8rrrr!rr'r&r$r#r"N)r rrrr)rr*r+rrrŚ+test_blocks_and_data_not_operated_on_intacts’ś zEFieldStructChildBlockTest.test_blocks_and_data_not_operated_on_intactcCsbt|jdtdddtjjd}| t|ddƒd”| t|ddƒd”| d|dd”| d|dd”|  d|dd”|  d|dd”| |ddd|jddd”| |ddd|jddd”|  d |dd”|  d |dd”| |ddd |jddd ”| |ddd |jddd ”d S) a Rename `simplestruct.char1` blocks to `renamed1` Check whether all `simplestruct.char1` blocks have been renamed correctly. Check whether values for renamed blocks are intact. Check whether other children of `simplestruct` are intact. r8rrrr!r&r%r#r?N) r rrrr)rr*r0Ś assertNotInrBr+rrrr,2sB’ś ’’’’z%FieldStructChildBlockTest.test_renamecCsģt|jdtddtjjd}| t|ddƒd”| t|ddƒd”| d|dd”| d|dd”|  d|dd”|  d|dd”| |ddd|jddd”| |ddd|jddd”d S) zøRemove `simplestruct.char1` blocks Check whether all `simplestruct.char1` blocks have been removed. Check whether other children of `simplestruct` are intact. r8rr-r!r&r%r#r?N) r rrrr)rr*r0rOrBr+rrrr1Zs(ü’’z%FieldStructChildBlockTest.test_removeN©rFrGrHrIrrNr,r1rrrrrJ s  (rJc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ŚFieldStreamChildBlockTestz|Tests involving changes to direct children of a StreamBlock We use `simplestream` blocks as the StreamBlocks here. c Cs(tjdddddddddjj}||_dS)NrŚ simplestreamr)rrKZ)content__1__simplestream__0__char1__valueZ)content__1__simplestream__1__char2__valueZ)content__1__simplestream__2__char1__valuerLZ)content__2__simplestream__0__char1__valuerrrrrrr}sų zFieldStreamChildBlockTest.setUpcCsĘt|jdtdddtjjd}| |d|jd”| |d|jd”| |dd |jdd ”| |d d |jd d ”| |dd |jdd ”| |d d |jd d ”d S) rMrRrrrr!rr'r&r$r#r"Nr(r+rrrrNŠs’śzEFieldStreamChildBlockTest.test_blocks_and_data_not_operated_on_intactcCs°t|jdtdddtjjd}| |dddd d”| |ddd d d”| |d ddd d”| |dddd |jdddd ”| |ddd d |jddd d ”| |d ddd |jd ddd ”| |dddd|jdddd”| |ddd d|jddd d”| |d ddd|jd ddd”| |ddd|jddd”d S) aRename `simplestream.char1` blocks to `renamed1` Check whether all `simplestream.char1` blocks have been renamed correctly. Check whether values and ids for renamed blocks are intact. Check whether other children of `simplestream` are intact. rRrrrr!r&r%rr"r#r$Nr(r+rrrr,¢sJ’ś ’’’žžžz%FieldStreamChildBlockTest.test_renamecCsvt|jdtddtjjd}| t|ddƒd”| t|ddƒd”| |ddd|jddd”d S) zøRemove `simplestream.char1` blocks Check whether all `simplestream.char1` blocks have been removed. Check whether other children of `simplestream` are intact. rRrr-r!r&r%r#rNr/r+rrrr1Ļsüz%FieldStreamChildBlockTest.test_removeNrPrrrrrQws  -rQc@s eZdZdZdd„Zdd„ZdS)ŚFieldListChildBlockTestzvTests involving changes to direct children of a ListBlock We use `simplelist` blocks as the ListBlocks here. cCs tjdddddjj}||_dS)NrzFoo 1zFoo 2zFoo 3)rZcontent__1__simplelist__0Zcontent__1__simplelist__1Zcontent__2__simplelist__0rrrrrrésü zFieldListChildBlockTest.setUpcCsjt|jdtddtjjd}| t|ddddƒt”| t|ddddƒt”| t|ddddƒt”|  d|dddd”|  d|dddd”|  d|dddd”| |ddddd|jdddd”| |ddddd|jdddd”| |ddddd|jdddd”d S) aoTurn each list child into a StructBlock and move value inside as a child named `text` Check whether each list child has been converted to a StructBlock with a child named `text` in it. Check whether the previous value of each list child is now the value that `text` takes. Note: Block ids are not preserved here. Z simplelistŚtext)r4r!r&r%rr#N) r rrrr)rr*r"ŚdictrBr+rrrrCņs0 ü   žžžz+FieldListChildBlockTest.test_to_structblockN)rFrGrHrIrrCrrrrrSćs rSN)Z django.testrZ$wagtail.blocks.migrations.operationsrrrrrrr r r Zwagtail.blocks.migrations.utilsr Z#wagtail.test.streamfield_migrationsr rrrJrQrSrrrrŚs , yml