add tests for escapting <element/>s in matrix

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/test_matrix.py b/tests/test_matrix.py
index 01723a9..e1782d1 100644
--- a/tests/test_matrix.py
+++ b/tests/test_matrix.py
@@ -301,3 +301,23 @@ def test_weird_utf8():
     test = "~\u200b~"
     formatted_body = "<strike>\u200b</strike>"
     assert(format_for_matrix(test) == formatted_body)
+
+    test = "<element>"
+    formatted_body = "&lt;element&gt;"
+    assert(format_for_matrix(test) == formatted_body)
+
+    test = "< element >"
+    formatted_body = "&lt; element &gt;"
+    assert(format_for_matrix(test) == formatted_body)
+
+    test = "< element>"
+    formatted_body = "&lt; element&gt;"
+    assert(format_for_matrix(test) == formatted_body)
+
+    test = "<element >"
+    formatted_body = "&lt;element &gt;"
+    assert(format_for_matrix(test) == formatted_body)
+
+    test = "<element> malicious script </element>"
+    formatted_body = "&lt;element&gt; malicious script &lt;/element&gt;"
+    assert(format_for_matrix(test) == formatted_body)