caps-helper: add extract_data_forms function

Stolen right from gabble.

Signed-off-by: Jonny Lamb <jonny.lamb@collabora.co.uk>

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/twisted/caps_helper.py b/tests/twisted/caps_helper.py
index ab5f0c28..3ec62ef4 100644
--- a/tests/twisted/caps_helper.py
+++ b/tests/twisted/caps_helper.py
@@ -223,6 +223,28 @@ def receive_presence_and_ask_caps(q, stream, expect_dbus=True):
 
     return disco_caps(q, stream, presence) + (signaled_caps,)
 
+def extract_data_forms(x_nodes):
+    dataforms = {}
+
+    if not x_nodes:
+        return dataforms
+
+    for form in x_nodes:
+        name = None
+        fields = {}
+        for field in xpath.queryForNodes('/x/field', form):
+            if field['var'] == 'FORM_TYPE':
+                name = str(field.firstChildElement())
+            else:
+                values = [str(x) for x in xpath.queryForNodes('/field/value', field)]
+
+                fields[field['var']] = values
+
+        if name is not None:
+            dataforms[name] = fields
+
+    return dataforms
+
 def disco_caps(q, stream, presence):
     c_nodes = xpath.queryForNodes('/presence/c', presence.stanza)
     assert c_nodes is not None