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()