B H±N`ã@snddlZddlmmZddlZddlZddlm Z ddl m Z dd„Z dd„Z dd „Zd d „Zd d „ZdS)éN)Úcore)ÚAeStackcCs(t ¡}t|dƒt | ¡ d¡j¡S)NZae)rZApprÚjsonÚdumpsZsynthÚ get_stackÚtemplate)Úapp©r úK/Users/shuaicao/Documents/artifacts/ae-blog-cdk/tests/unit/test_ae_stack.pyÚ get_templates r cCsŽd}tƒ}||k}|s~t d|fd||f¡t |¡dt ¡ksHt t¡rRt t¡ndt |¡dœ}dd|i}tt |¡ƒ‚d}}}dS)NzAWS::IAM::Role)Úin)z)%(py1)s in %(py5)s {%(py5)s = %(py3)s() }r )Úpy1Úpy3Úpy5zassert %(py7)sÚpy7) r Ú @pytest_arÚ_call_reprcompareÚ _safereprÚ @py_builtinsÚlocalsÚ_should_repr_global_nameÚAssertionErrorÚ_format_explanation)Ú @py_assert0Ú @py_assert4Ú @py_assert2Ú @py_format6Ú @py_format8r r r Útest_role_created sN rcCsŽd}tƒ}||k}|s~t d|fd||f¡t |¡dt ¡ksHt t¡rRt t¡ndt |¡dœ}dd|i}tt |¡ƒ‚d}}}dS)Nz AWS::SageMaker::NotebookInstance)r )z)%(py1)s in %(py5)s {%(py5)s = %(py3)s() }r )r rrzassert %(py7)sr) r rrrrrrrr)rrrrrr r r Útest_sagemaker_createdsN rcCsŽd}tƒ}||k}|s~t d|fd||f¡t |¡dt ¡ksHt t¡rRt t¡ndt |¡dœ}dd|i}tt |¡ƒ‚d}}}dS)NzAWS::Lambda::Function)r )z)%(py1)s in %(py5)s {%(py5)s = %(py3)s() }r )r rrzassert %(py7)sr) r rrrrrrrr)rrrrrr r r Útest_lambda_createdsN r cCsŽd}tƒ}||k}|s~t d|fd||f¡t |¡dt ¡ksHt t¡rRt t¡ndt |¡dœ}dd|i}tt |¡ƒ‚d}}}dS)NzAWS::DynamoDB::Table)r )z)%(py1)s in %(py5)s {%(py5)s = %(py3)s() }r )r rrzassert %(py7)sr) r rrrrrrrr)rrrrrr r r Útest_ddb_createdsN r!)ÚbuiltinsrÚ_pytest.assertion.rewriteÚ assertionÚrewriterrÚpytestZaws_cdkrZ ae.ae_stackrr rrr r!r r r r Ús