def _createAssertion(self, timeNow=None, validityDuration=60*60*8, issuerName=BaseTestCase.SITEA_SAML_ISSUER_NAME): if timeNow is None: timeNow = datetime.utcnow() timeExpires = timeNow + timedelta(seconds=validityDuration) assertionStr = Template( self.__class__.ASSERTION_STR).substitute( dict( issuerName=issuerName, timeNow=SAMLDateTime.toString(timeNow), timeExpires=SAMLDateTime.toString(timeExpires) ) ) assertionStream = StringIO() assertionStream.write(assertionStr) assertionStream.seek(0) assertionElem = ElementTree.parse(assertionStream).getroot() return AssertionElementTree.fromXML(assertionElem)
def test02ParseAssertion(self): assertion = self._createAttributeAssertionHelper() # Create ElementTree Assertion Element assertionElem = AssertionElementTree.toXML(assertion) self.assert_(ElementTree.iselement(assertionElem)) # Serialise to output xmlOutput = prettyPrint(assertionElem) print("\n"+"_"*80) print(xmlOutput) print("_"*80) assertionStream = StringIO() assertionStream.write(xmlOutput) assertionStream.seek(0) tree = ElementTree.parse(assertionStream) elem2 = tree.getroot() assertion2 = AssertionElementTree.fromXML(elem2) self.assert_(assertion2)