Ignore:
Timestamp:
04/11/10 13:55:40 (10 years ago)
Author:
pjkersha
Message:

Integrated SAML ESGF Group/Role? attribute value type into SAML Attribute Authority client unit tests.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • TI12-security/trunk/NDGSecurity/python/ndg_security_common/ndg/security/common/saml_utils/esgf/xml/etree.py

    r7287 r7698  
    1919                                QName) 
    2020 
    21 from ndg.security.common.saml_utils.esg import XSGroupRoleAttributeValue 
     21from ndg.security.common.saml_utils.esgf import ESGFGroupRoleAttributeValue 
    2222 
    2323 
    24 class XSGroupRoleAttributeValueElementTree(AttributeValueElementTreeBase, 
    25                                            XSGroupRoleAttributeValue): 
     24class ESGFGroupRoleAttributeValueElementTree(AttributeValueElementTreeBase, 
     25                                             ESGFGroupRoleAttributeValue): 
    2626    """ElementTree XML representation of Earth System Grid custom Group/Role  
    2727    Attribute Value"""  
     
    3232        Attribute Value 
    3333         
    34         @type assertion: saml.saml2.core.XSGroupRoleAttributeValue 
    35         @param assertion: XSGroupRoleAttributeValue to be represented as an  
    36         ElementTree Element 
     34        @type attributeValue: ndg.security.common.saml_utils.esgf.ESGFGroupRoleAttributeValue 
     35        @param attributeValue: Group/Role Attribute Value to be represented as  
     36        an ElementTree Element 
    3737        @rtype: ElementTree.Element 
    3838        @return: ElementTree Element 
     
    4040        elem = AttributeValueElementTreeBase.toXML(attributeValue) 
    4141         
    42         if not isinstance(attributeValue, XSGroupRoleAttributeValue): 
     42        if not isinstance(attributeValue, ESGFGroupRoleAttributeValue): 
    4343            raise TypeError("Expecting %r type; got: %r" %  
    44                             (XSGroupRoleAttributeValue, type(attributeValue))) 
     44                            (ESGFGroupRoleAttributeValue, type(attributeValue))) 
    4545             
    4646        ElementTree._namespace_map[attributeValue.namespaceURI 
     
    6262    def fromXML(cls, elem): 
    6363        """Parse ElementTree ESG Group/Role attribute element into a SAML  
    64         XSGroupRoleAttributeValue object 
     64        ESGFGroupRoleAttributeValue object 
    6565         
    6666        @type elem: ElementTree.Element 
    6767        @param elem: Attribute value as ElementTree XML element 
    68         @rtype: saml.saml2.core.XSGroupRoleAttributeValue 
     68        @rtype: saml.saml2.core.ESGFGroupRoleAttributeValue 
    6969        @return: SAML ESG Group/Role Attribute value 
    7070        """ 
     
    9595 
    9696                                       
    97         attributeValue = XSGroupRoleAttributeValue() 
     97        attributeValue = ESGFGroupRoleAttributeValue() 
    9898        groupName = childElem.attrib.get(cls.GROUP_ATTRIB_NAME) 
    9999        if groupName is None: 
     
    105105        roleName = childElem.attrib.get(cls.ROLE_ATTRIB_NAME) 
    106106        if roleName is None: 
    107             raise XMLTypeParseError('No "%s" attribute found in Group/Role ' 
    108                                     'attribute element' %  
    109                                     cls.GROUP_ATTRIB_NAME) 
     107            roleName = cls.DEFAULT_ROLE_NAME 
     108             
    110109        attributeValue.role = roleName 
    111110 
     
    115114    def factoryMatchFunc(cls, elem): 
    116115        """Match function used by AttributeValueElementTreeFactory to 
    117         determine whether the given attribute is XSGroupRole type 
     116        determine whether the given attribute is ESGFGroupRoleAttributeValue 
     117        type 
    118118         
    119119        @type elem: ElementTree.Element 
    120120        @param elem: Attribute value as ElementTree XML element 
    121         @rtype: saml.saml2.core.XSGroupRoleAttributeValue or None 
    122         @return: SAML ESG Group/Role Attribute Value class if elem is an 
     121        @rtype: ndg.security.common.saml_utils.etree.ESGFGroupRoleAttributeValue 
     122        or None 
     123        @return: SAML ESGF Group/Role Attribute Value class if elem is an 
    123124        Group/role type element or None if if doesn't match this type  
    124125        """ 
     
    140141 
    141142 
    142 class EsgResponseElementTree(ResponseElementTree): 
     143class ESGFResponseElementTree(ResponseElementTree): 
    143144    """Extend ResponseElementTree type for Attribute Query Response to include  
    144145    ESG custom Group/Role Attribute support""" 
     
    146147    @classmethod 
    147148    def toXML(cls, response, **kw): 
    148         # Add mapping for ESG Group/Role Attribute Value to enable ElementTree 
    149         # Attribute Value factory to render the XML output 
     149        """Extend base method adding mapping for ESG Group/Role Attribute Value  
     150        to enable ElementTree Attribute Value factory to render the XML output 
     151         
     152        @type response: ndg.security.common.saml_utils.etree.ESGFGroupRoleAttributeValue 
     153        @param response: ESGF Group/Role attribute value  
     154        @rtype: ElementTree.Element 
     155        @return: ESGF Group/Role attribute value as ElementTree.Element 
     156        """ 
    150157        toXMLTypeMap = kw.get('customToXMLTypeMap', {}) 
    151         toXMLTypeMap[XSGroupRoleAttributeValue 
    152                      ] = XSGroupRoleAttributeValueElementTree 
     158        toXMLTypeMap[ESGFGroupRoleAttributeValue 
     159                     ] = ESGFGroupRoleAttributeValueElementTree 
    153160         
    154161        kw['customToXMLTypeMap'] = toXMLTypeMap 
     
    160167    @classmethod 
    161168    def fromXML(cls, elem, **kw): 
     169        """Extend base method adding mapping for ESG Group/Role Attribute Value 
     170          
     171        @type elem: ElementTree.Element 
     172        @param elem: ESGF Group/Role attribute value as ElementTree.Element 
     173        @rtype: ndg.security.common.saml_utils.etree.ESGFGroupRoleAttributeValue 
     174        @return: ESGF Group/Role attribute value  
     175        """ 
    162176        toSAMLTypeMap = kw.get('customToSAMLTypeMap', []) 
    163177        toSAMLTypeMap.append( 
    164                         XSGroupRoleAttributeValueElementTree.factoryMatchFunc) 
     178                        ESGFGroupRoleAttributeValueElementTree.factoryMatchFunc) 
    165179        kw['customToSAMLTypeMap'] = toSAMLTypeMap 
    166180         
Note: See TracChangeset for help on using the changeset viewer.