Add forgotten test helper

nationalarchives · Sep 12, 2023 · 43aed52 · 43aed52
1 parent e0ae338
commit 43aed52
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/client_helpers/test_xml_helpers.py b/tests/client_helpers/test_xml_helpers.py
@@ -0,0 +1,15 @@
+import lxml.etree
+
+from caselawclient.xml_helpers import get_xpath_match_string, get_xpath_match_strings
+
+
+def test_xpath_single():
+    node = lxml.etree.fromstring("<root><cat>1</cat><cat>2</cat></root>")
+    path = "//cat/text()"
+    assert get_xpath_match_string(node, path) == "1"
+
+
+def test_xpath_multiple():
+    node = lxml.etree.fromstring("<root><cat>1</cat><cat>2</cat></root>")
+    path = "//cat/text()"
+    assert get_xpath_match_strings(node, path) == ["1", "2"]