U |h$@sddlZddlmZddlmZddlmZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZmZdd lmZGd d d eZGd ddeeZGdddeZGdddeZGdddeZGdddeZdS)N)TestCase)timezone) StreamValue)migrate_operation)RenameStreamChildrenOperationRenameStructChildrenOperation)InvalidBlockDefErrorapply_changes_to_raw_data)#disable_reference_index_auto_update) factoriesmodels)MigrationTestMixinc@s(eZdZdZddZddZddZdS) TestExceptionRaisedInRawDataaDirectly test whether an exception is raised by apply_changes_to_raw_data for invalid defs. This would happen in a situation where the user gives a block path which contains a block name which is not present in the block definition in the project state at which the migration is applied. (There should also be a block in the stream data with the said name for this to happen) cCs`tjdddjj}|dddddddd ddddgd dd d g|d dd<||_dS)N Char Block 1 nestedstructZcontent__0__char1__valueZ content__1 invalid_name10001foochar1Zchar2typeidvalue0002rZ0003)rrrrZ invalid_name2)r ZSampleModelFactorycontentraw_dataextend)selfrr!z/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/tests/streamfield_migrations/test_bad_data.pysetUps$  z"TestExceptionRaisedInRawData.setUpc Cs:|td$t|jdtdddtjjdW5QRXdS)zLTest whether Exception is raised in when recursing through stream block dataz(No current block def named invalid_name1rrrenamed1Zold_namenew_namerZblock_path_strZ operationZ streamfieldN)assertRaisesMessagerr rrr SampleModelrr r!r!r" test_rename_invalid_stream_child5sz=TestExceptionRaisedInRawData.test_rename_invalid_stream_childc Cs:|td$t|jdtdddtjjdW5QRXdS)zLTest whether Exception is raised in when recursing through struct block dataz(No current block def named invalid_name2znestedstruct.invalid_name2rr$r%r'N)r(rr rrr r)rr*r!r!r" test_rename_invalid_struct_childDsz=TestExceptionRaisedInRawData.test_rename_invalid_struct_childN)__name__ __module__ __qualname____doc__r#r+r,r!r!r!r"rsrc@sHeZdZejZeddddfgZdZddZ dd Z d d Z d d Z dS)BadDataMigrationTestCaserr$r%rZstreamfield_migration_testscCstjddd}||_dS)Nrrr)r ZSamplePageFactoryinstance)r r2r!r!r"create_instance^s z(BadDataMigrationTestCase.create_instancecCs^|jjj}|ddddddddddddg|jjj}t||dd|j_|jdS) NrrrrrrT stream_blockZ stream_dataZis_lazy)r2rrrr5rsave)r rr5r!r!r"append_invalid_instance_dataes&   z5BadDataMigrationTestCase.append_invalid_instance_datacCsZ|||}|jjj}|dd}|jjj}t||dd|j_|j|j|j fS)NTr4) r7create_revisionr2rrr5rr6r created_at)r deltaZinvalid_revisionrr5r!r!r"create_invalid_revision{s      z0BadDataMigrationTestCase.create_invalid_revisioncCs,|j}ttj|d|_||S)Ndays)r2Z save_revisionrnowdatetime timedeltar:r6)r r;revisionr!r!r"r9s z(BadDataMigrationTestCase.create_revisionN) r-r.r/r SamplePagemodelrZ default_operation_and_block_pathZapp_namer3r7r<r9r!r!r!r"r1Ts r1c@s eZdZdZddZddZdS)TestExceptionRaisedForInstancezlException should always be raised when applying migration if it occurs while migrating the instance datac Cs&t||W5QRXdS)N)r r3r7r*r!r!r"r#sz$TestExceptionRaisedForInstance.setUpc CsJ|td|jjj|jj"|jt t j dddW5QRXdS)Nz#Invalid block def in {} object ({})r8r=Zrevisions_from) r(rformatr2 __class__r-rapply_migrationrr?r@rAr*r!r!r" test_migratesz+TestExceptionRaisedForInstance.test_migrateNr-r.r/r0r#rJr!r!r!r"rEsrEc@s eZdZdZddZddZdS)$TestExceptionRaisedForLatestRevisionzsException should always be raised when applying migration if it occurs while migrating the latest revision datac CsLt<|tdD]}|d|q|d\|_|_W5QRXdS)Nr)r r3ranger9r<invalid_revision_idinvalid_revision_created_atr ir!r!r"r#s z*TestExceptionRaisedForLatestRevision.setUpc CsB|td|jjj|jj|j|j|j ddW5QRXdSNzFInvalid block def in {} object ({}) for revision id ({}) created at {}rF r(rrGr2rHr-rrPrQrIr*r!r!r"rJs z1TestExceptionRaisedForLatestRevision.test_migrateNrKr!r!r!r"rLs rLc@s eZdZdZddZddZdS)"TestExceptionRaisedForLiveRevisionzqException should always be raised when applying migration if it occurs while migrating the live revision datac CsbtR||d\|_|_|j|j_|jtddD]}| d|q@W5QRXdSNrNr) r r3r<rPrQr2Zlive_revision_idr6rOr9rRr!r!r"r#s  z(TestExceptionRaisedForLiveRevision.setUpc CsB|td|jjj|jj|j|j|j ddW5QRXdSrTrUr*r!r!r"rJs z/TestExceptionRaisedForLiveRevision.test_migrateNrKr!r!r!r"rVsrVc@s&eZdZdZejZddZddZdS)%TestExceptionIgnoredForOtherRevisionszException should not be be raised when applying migration if it occurs while migrating revision data which is not of a live or latest revision. Instead an exception should be loggedc CsNt>||d\|_|_tddD]}|d|q,W5QRXdSrW)r r3r<rPrQrOr9rRr!r!r"r#sz+TestExceptionIgnoredForOtherRevisions.setUpc Cs|jddv}|jdd||jdddtj|jj j|jj |j |j ||jdddt jdt jW5QRXdS) NERROR)levelrFrzOERROR:{}:Invalid block def in {} object ({}) for revision id ({}) created at {}z,{}: No current block def named invalid_name1.)Z assertLogsrIZ assertEqualoutput splitlinesrGrr-r2rHrrPrQrr.)r cmr!r!r"rJs,  z2TestExceptionIgnoredForOtherRevisions.test_migrateN) r-r.r/r0r rCrDr#rJr!r!r!r"rXs rX)r@Z django.testrZ django.utilsrZwagtail.blocksrZwagtail.blocks.migrationsrZ$wagtail.blocks.migrations.operationsrrZwagtail.blocks.migrations.utilsrr Zwagtail.signal_handlersr Z#wagtail.test.streamfield_migrationsr r Z-wagtail.test.streamfield_migrations.testutilsr rr1rErLrVrXr!r!r!r"s      ?=