U
h
@ sZ d Z ddlmZ ddlmZ ddlmZ ddlmZ G dd dejZ e
dkrVe d S )
zUnittest for html_utils. ) annotations)absltest)
html_utils) sheets_idc @ s e Zd Zdd Zdd ZdS )
HtmlUtilsTestc C s$ t jtddd}| d| d S )Nz"https://docs.google.com/?a=b#hellozhelloworldurltextzkhello<evil_tag/>world)r get_anchor_tagr SheetsURL assertEqual)selfhtml r P/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/html_utils_test.py#test_get_anchor_tag_text_is_escaped s z1HtmlUtilsTest.test_get_anchor_tag_text_is_escapedc C s. t d}d|_tj|dd}| d| d S )Nzhttps://docs.google.com/z%https://docs.google.com/"evil_string"z hello worldr zhhello world)r r _urlr r
r )r
r r r r r "test_get_anchor_tag_url_is_escaped& s
z0HtmlUtilsTest.test_get_anchor_tag_url_is_escapedN)__name__
__module__ __qualname__r r r r r r r s
r __main__N) __doc__
__future__r Z absl.testingr Z google.generativeai.notebookr r ZTestCaser r mainr r r r s !