bpo-38243: Update unittest

Author: corona10

Lib/test/test_docxmlrpc.py

@@ -1,5 +1,6 @@
 from xmlrpc.server import DocXMLRPCServer
 import http.client
+import re
 import sys
 import threading
 import unittest
@@ -193,5 +194,27 @@ def test_annotations(self):
             response.read())
 
 
+class XMLRPCDocGeneratorTest(unittest.TestCase):
+    def setUp(self):
+        self.serv = DocXMLRPCServer(("localhost", 0), logRequests=False)
+
+    def tearDown(self):
+        self.serv.server_close()
+
+    def test_server_title_escape(self):
+        self.serv.set_server_title('test_title<script>')
+        self.serv.set_server_documentation('test_documentation<script>')
+        self.assertEqual('test_title<script>', self.serv.server_title)
+        self.assertEqual('test_documentation<script>',
+                self.serv.server_documentation)
+
+        # bpo-38243
+        generated = self.serv.generate_html_documentation()
+        title = re.search(r'<title>(.+?)</title>', generated).group()
+        documentation = re.search(r'<p><tt>(.+?)</tt></p>', generated).group()
+        self.assertEqual('<title>Python: test_title&lt;script&gt;</title>', title)
+        self.assertEqual('<p><tt>test_documentation&lt;script&gt;</tt></p>', documentation)
+
+
 if __name__ == '__main__':
     unittest.main()