ó §‰]c@sPdZddlmZddlmZddlmZdefd„ƒYZd S( s'Tests for the HTMLSanitize preprocessori(tPreprocessorTestsBasei(t SanitizeHTMLiÿÿÿÿ(tv4t TestSanitizercBsweZdZd Zd„Zd„Zd„Zd„Zd„Z d„Z d„Z d„Z d „Z d „Zd „ZRS( s'Contains test functions for sanitize.pycCstƒ}t|_|S(s"Make an instance of a preprocessor(RtTruetenabled(tselft preprocessor((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pytbuild_preprocessor s  cCs[|jƒ}|jƒ}||jd_||jd_|||ƒ\}}|jdjS(Ni(tbuild_notebooktbuild_resourcestcellst cell_typetsource(RR R Rtnbtres((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pytpreprocess_sources   cCs|jƒdS(s"Can a SanitizeHTML be constructed?N(R(R((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_constructorscCsD|jƒ}t|_|j|jdd|ƒjƒdjƒƒdS(sÚ Test to make sure that svgs are handled 'properly' We only allow tags (via markdown syntax) and not all the other ways to embed svg: , , sI ![some image](http://example.com/something.svg) N(RRtstript assertEqualR(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_svg_handling"s   cCs8|jƒ}t|_|j|jdd|ƒdƒdS(s4Test tag whitelisting + stripping out offending tagsRs&_A_ few s_A_ few tagsN(RRRRR(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_tag_whitelist_strippingDs  cCsW|jƒ}|j|jdd|ƒdƒt|_|j|jdd|ƒdƒdS(sTest HTML comment strippingRs_A_ few s_A_ few N(RRRtFalsetstrip_comments(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_comment_strippingRs  cCsB|jƒ}ddg|jd<|j|jdd|ƒdƒdS(s Test stylethrefttitletaRs$HisHiN(Rt attributesRR(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_attributes_whitelistis cCsq|jƒ}d|jkr2|jdjdƒndg|jdfew sG_A_ few <script>tags</script>N(RRtappendtstylesRR(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_style_whitelistxs  cCs/|jƒ}|j|jdd|ƒdƒdS(sTest passing through raw outputtraws&_A_ few s2_A_ few <script>tags</script>N(RRR(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_tag_passthroughs cCs |jƒ}|jƒ}tjddidd6dd6dd6ƒtjd d d d d ƒtjd d d d dƒg}||jd_|jƒ}|||ƒ\}}iidd6dd6d6id6dd6id d 6d d6d d 6id d 6d d6dd 6g}|j|jdj|ƒdS(s-Test that outputs are also sanitized properlyt display_datatdatatbs text/plainss text/htmls stext/csststreamtnametstdoutttexttwatsis&<script>more evil</script>tmetadatat output_typeN(RR tnbformatt new_outputR toutputsR R(RRRR3Rtexpected_output((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_output_sanitizingœs2     cCs/|jƒ}|j|jdd|ƒdƒdS(sTest tag whitelistingRs&_A_ few s2_A_ few <script>tags</script>N(RRR(RR((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyttest_tag_whitelistÅs N(t__name__t __module__t__doc__tNonetmaxDiffRRRRRRRR$R&R5R6(((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyRs   "     )N(R9tbaseRtsanitizeRR1RR(((sJlib/python2.7/site-packages/nbconvert/preprocessors/tests/test_sanitize.pyts