+ except:
+ response_msg_div = etree.Element('div')
+ response_msg_div.text = str(response_msg)
+
+
+ # Set the css class of the message
+ response_msg_div.set("class", "response_message")
+
+ return response_msg_div
+
#-----------------------------------------------------------------------------
diff --git a/common/lib/capa/capa/tests/test_html_render.py b/common/lib/capa/capa/tests/test_html_render.py
index 257e63b611..64f031ea59 100644
--- a/common/lib/capa/capa/tests/test_html_render.py
+++ b/common/lib/capa/capa/tests/test_html_render.py
@@ -135,7 +135,8 @@ class CapaHtmlRenderTest(unittest.TestCase):
# CustomResponse script that sets an overall_message
script=textwrap.dedent("""
def check_func(*args):
- return {'overall_message': '
Test message
',
+ msg = '
Test message 1
Test message 2
'
+ return {'overall_message': msg,
'input_list': [ {'ok': True, 'msg': '' } ] }
""")
@@ -160,9 +161,12 @@ class CapaHtmlRenderTest(unittest.TestCase):
self.assertEqual(msg_div_element.get('class'), "response_message")
# Expect that the
contains our message (as part of the XML tree)
- msg_p_element = msg_div_element.find('p')
- self.assertEqual(msg_p_element.tag, "p")
- self.assertEqual(msg_p_element.text, "Test message")
+ msg_p_elements = msg_div_element.findall('p')
+ self.assertEqual(msg_p_elements[0].tag, "p")
+ self.assertEqual(msg_p_elements[0].text, "Test message 1")
+
+ self.assertEqual(msg_p_elements[1].tag, "p")
+ self.assertEqual(msg_p_elements[1].text, "Test message 2")
def test_substitute_python_vars(self):