20 lines
664 B
Python
20 lines
664 B
Python
import unittest
|
|
|
|
from scripts.wiki_update_page import extract_script_to_script_js
|
|
|
|
|
|
class WikiUpdatePageTests(unittest.TestCase):
|
|
def test_script_js_keeps_script_element_for_wikijs_injection(self):
|
|
content = "<p>before</p>\n<script>window.example = true;</script>\n<p>after</p>"
|
|
|
|
rendered_content, script_js = extract_script_to_script_js(content)
|
|
|
|
self.assertEqual(script_js, "<script>window.example = true;</script>")
|
|
self.assertNotIn("<script>", rendered_content)
|
|
self.assertIn("<p>before</p>", rendered_content)
|
|
self.assertIn("<p>after</p>", rendered_content)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|