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)
