1 files changed, 55 insertions, 0 deletions
diff --git a/Testcases/cfgm_common/ifmap/response.py b/Testcases/cfgm_common/ifmap/response.py
new file mode 100644
index 0000000..179fd01
--- /dev/null
+++ b/Testcases/cfgm_common/ifmap/response.py
@@ -0,0 +1,55 @@
+# Copyright 2011, Infoblox, All Rights Reserved
+#
+# Open Source, see LICENSE
+#
+
+from xml.etree import ElementTree
+
+class Response():
+    """
+    Base class to handle and parse IF-MAP responses
+    """
+    __xml = ""
+
+    def __init__(self, result):
+        """
+        Take a result string and process it
+        """
+        if result: 
+            env = ElementTree.fromstring(result)
+            body = env.find('{http://www.w3.org/2003/05/soap-envelope}Body')
+            response = body.find('{http://www.trustedcomputinggroup.org/2010/IFMAP/2}response')
+            # xml.etree.ElementTree find is broken in python 2.6
+            children = response.findall('*')
+            if len(children):
+                self.__xml = children[0]
+            
+    def element(self):
+        """
+        Returns the raw Element object
+        """
+        return self.__xml
+    
+    def __str__(self):
+        """
+        Print the XML tree as a string
+        """
+        return ElementTree.tostring(self.__xml)
+
+class newSessionResult(Response):
+    """
+    newSessionResult
+    """
+    def __init__(self, result):
+        #import pdb; pdb.set_trace()
+        self.__newSession = Response(result).element()
+        
+    def get_session_id(self):
+        return self.__newSession.attrib['session-id']
+    
+    def get_publisher_id(self):
+        return self.__newSession.attrib['ifmap-publisher-id']
+        
+
+
+