Add mdstore method to extract assurance certifications

Signed-off-by: Ivan Kanakarakis <ivan.kanak@gmail.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/src/saml2/mdstore.py b/src/saml2/mdstore.py
index 3a49d4a4..b7f014aa 100644
--- a/src/saml2/mdstore.py
+++ b/src/saml2/mdstore.py
@@ -66,6 +66,7 @@ classnames = {
 
 ENTITY_CATEGORY = "http://macedir.org/entity-category"
 ENTITY_CATEGORY_SUPPORT = "http://macedir.org/entity-category-support"
+ASSURANCE_CERTIFICATION = "urn:oasis:names:tc:SAML:attribute:assurance-certification"
 
 REQ2SRV = {
     # IDP
@@ -1243,6 +1244,15 @@ class MetadataStore(MetaData):
         attributes = self.entity_attributes(entity_id)
         return attributes.get(ENTITY_CATEGORY_SUPPORT, [])
 
+    def assurance_certifications(self, entity_id):
+        assurance_certifications = (
+            certification
+            for name, values in self.entity_attributes(entity_id).items()
+            if name == ASSURANCE_CERTIFICATION
+            for certification in values
+        )
+        return assurance_certifications
+
     def entity_attributes(self, entity_id):
         """
         Get all entity attributes for an entry in the metadata.