Added functionality needed by the saml2test tool.

1 files changed, 44 insertions, 0 deletions

diff --git a/tests/test_06_setarg.py b/tests/test_06_setarg.py
new file mode 100644
index 00000000..171fefb3
--- /dev/null
+++ b/tests/test_06_setarg.py
@@ -0,0 +1,44 @@
+from saml2 import saml
+from saml2.saml import Subject
+from saml2.samlp import Response
+from saml2.argtree import set_arg, add_path
+from saml2.argtree import find_paths
+
+__author__ = 'roland'
+
+
+def test_path():
+    result = find_paths(Subject, 'in_response_to')
+
+    assert result == [
+        ['subject_confirmation', 'subject_confirmation_data', 'in_response_to']]
+
+    result = find_paths(Response, 'in_response_to')
+
+    assert result == [
+        ['assertion', 'subject', 'subject_confirmation',
+         'subject_confirmation_data', 'in_response_to'],
+        ['in_response_to']
+    ]
+
+
+def test_set_arg():
+    r = set_arg(Subject, 'in_response_to', '123456')
+
+    assert r == [{'subject_confirmation': {
+        'subject_confirmation_data': {'in_response_to': '123456'}}}]
+
+
+def test_multi():
+    t = {}
+    t = add_path(t, ['subject_confirmation','method',saml.SCM_BEARER])
+    x = add_path(
+        t['subject_confirmation'],
+        ['subject_confirmation_data','in_response_to','1234'])
+
+    print(t)
+    assert t == {
+        'subject_confirmation': {
+            'subject_confirmation_data': {'in_response_to': '1234'},
+            'method': 'urn:oasis:names:tc:SAML:2.0:cm:bearer'}
+    }
\ No newline at end of file