î
‚Ü Xþ  ã               @   s5   d  d l  Z  d  d l Z Gd d „  d e  j ƒ Z d S)é    Nc               @   s@   e  Z d  Z d d „  Z d d „  Z d d „  Z d d „  Z d	 S)
ÚTestEncodeForHTMLc             C   s"   t  j ƒ  |  _ t  j ƒ  |  _ d  S)N)ÚjsonZJSONDecoderÚdecoderZJSONEncoderForHTMLÚencoder)Úself© r   úb/home/ubuntu/projects/ifolica/lib/python3.4/site-packages/simplejson/tests/test_encode_for_html.pyÚsetUp   s    zTestEncodeForHTML.setUpc             C   sX   |  j  d |  j j d ƒ ƒ |  j  d |  j j d ƒ ƒ |  j  d |  j j d ƒ ƒ d  S)Nz"\u0026"ú&z"\u003c"ú<z"\u003e"ú>)ÚassertEqualr   Úencode)r   r   r   r   Útest_basic_encode   s    z#TestEncodeForHTML.test_basic_encodec             C   s=   x6 d D]. } |  j  | |  j j |  j j | ƒ ƒ ƒ q Wd  S)Nz&<>)r   r   Údecoder   r   )r   Úcharr   r   r   Útest_basic_roundtrip   s    z&TestEncodeForHTML.test_basic_roundtripc             C   sN   d } |  j  d |  j j | ƒ ƒ |  j  | |  j j |  j j | ƒ ƒ ƒ d  S)Nz)</script><script>alert("gotcha")</script>zK"\u003c/script\u003e\u003cscript\u003ealert(\"gotcha\")\u003c/script\u003e")r   r   r   r   r   )r   Z
bad_stringr   r   r   Útest_prevent_script_breakout   s    z.TestEncodeForHTML.test_prevent_script_breakoutN)Ú__name__Ú
__module__Ú__qualname__r	   r   r   r   r   r   r   r   r      s   r   )ZunittestÚ
simplejsonr   ZTestCaser   r   r   r   r   Ú<module>   s   