Ignore:
Timestamp:
25/11/09 17:11:29 (10 years ago)
Author:
pjkersha
Message:

Refactoring MyProxy? Cert Extension app following updates to the SAML Attribute Query client interface

File:
1 edited

Legend:

Unmodified
Added
Removed
  • TI12-security/trunk/python/ndg_security_test/ndg/security/test/unit/myproxy/certificate_extapp/test_saml_attribute_assertion.py

    r6009 r6051  
    3131    OPENID = Template(OPENID_TMPL).substitute( 
    3232                        {'userIdentifier': BaseTestCase.OPENID_IDENTIFIER}) 
    33     OPENID_SQL_QUERY = ("select openid_identifier from users where username " 
     33    OPENID_SQL_QUERY = ("select openid from users where username " 
    3434                        "= '${username}'")  
    35          
     35    SITEA_SSL_ATTRIBUTEAUTHORITY_PORTNUM = 5443 
     36    SITEA_SSL_ATTRIBUTEAUTHORITY_URI = ('https://localhost:%d/AttributeAuthority' 
     37                                        '/saml' %  
     38                                        SITEA_SSL_ATTRIBUTEAUTHORITY_PORTNUM) 
     39    INI_FILEPATH = os.path.join(THIS_DIR, 'config.ini') 
     40     
    3641    def __init__(self, *arg, **kw): 
    3742        super(CertExtAppTestCase, self).__init__(*arg, **kw)             
    38         self.startSiteAAttributeAuthority() 
     43        self.startSiteAAttributeAuthority(withSSL=True,  
     44                    port=CertExtAppTestCase.SITEA_SSL_ATTRIBUTEAUTHORITY_PORTNUM) 
    3945        self.initDb() 
    4046         
     
    4652        myProxyCertExtApp.openIdSqlQuery = CertExtAppTestCase.OPENID_SQL_QUERY 
    4753         
    48         identifier = myProxyCertExtApp.queryOpenId(CertExtAppTestCase.USERNAME) 
    49         self.assert_(identifier == CertExtAppTestCase.OPENID_IDENTIFIER) 
     54        openid = myProxyCertExtApp.queryOpenId(CertExtAppTestCase.USERNAME) 
     55        self.assert_(openid == CertExtAppTestCase.OPENID_URI) 
    5056         
    5157    def test02AttributeQuery(self): 
    5258        myProxyCertExtApp = CertExtApp() 
    53         myProxyCertExtApp.attributeAuthorityURI = ( 
    54                             'http://localhost:%d/AttributeAuthority/saml' %  
    55                             CertExtAppTestCase.SITEA_ATTRIBUTEAUTHORITY_PORTNUM) 
    56         myProxyCertExtApp.issuerDN = "/O=Site A/CN=Authorisation Service" 
    57         myProxyCertExtApp.userOpenID = CertExtAppTestCase.OPENID 
     59        myProxyCertExtApp.attributeQuery.issuerDN = "/O=Site A/CN=Authorisation Service" 
     60        myProxyCertExtApp.attributeQuery.subjectID = CertExtAppTestCase.OPENID 
    5861                                 
    59         assertion = myProxyCertExtApp.attributeQuery() 
    60         print(assertion) 
     62        response = myProxyCertExtApp.attributeQuery.send( 
     63                    uri=CertExtAppTestCase.SITEA_SSL_ATTRIBUTEAUTHORITY_URI) 
     64        print(response) 
    6165         
    6266    def test03End2End(self): 
     
    6670                                        CertExtAppTestCase.DB_CONNECTION_STR 
    6771                     
    68         myProxyCertExtApp.openIdSqlQuery = ("select openid_identifier from " 
    69                                             "users where username = '%s'" % 
     72        myProxyCertExtApp.openIdSqlQuery = ("select openid from users where " 
     73                                            "username = '%s'" % 
    7074                                            CertExtAppTestCase.USERNAME) 
    71          
    72         myProxyCertExtApp.identityUriTemplate = CertExtAppTestCase.OPENID_TMPL 
    73                              
    74         myProxyCertExtApp.attributeAuthorityURI = ( 
    75                         'http://localhost:%d/AttributeAuthority/saml' %  
    76                         CertExtAppTestCase.SITEA_ATTRIBUTEAUTHORITY_PORTNUM) 
    77         myProxyCertExtApp.issuerDN = "/O=Site A/CN=Authorisation Service" 
     75 
     76        myProxyCertExtApp.attributeAuthorityURI = \ 
     77                            CertExtAppTestCase.SITEA_SSL_ATTRIBUTEAUTHORITY_URI 
     78        myProxyCertExtApp.attributeQuery.issuerDN = \ 
     79            "/O=Site A/CN=Authorisation Service" 
    7880         
    7981        assertion = myProxyCertExtApp(CertExtAppTestCase.USERNAME) 
     
    8284 
    8385    def test04FromConfigFile(self): 
    84         configFilePath = os.path.join(CertExtAppTestCase.THIS_DIR, 'config.ini') 
    85         myProxyCertExtApp = CertExtApp.fromConfigFile(configFilePath) 
     86        myProxyCertExtApp = CertExtApp.fromConfigFile( 
     87                                            CertExtAppTestCase.INI_FILEPATH) 
    8688        assertion = myProxyCertExtApp(CertExtAppTestCase.USERNAME) 
    8789        self.assert_(assertion) 
     
    8991 
    9092    def test05ConsoleApp(self): 
    91         configFilePath = os.path.join(CertExtAppTestCase.THIS_DIR, 'config.ini') 
    9293        import sys 
    9394        sys.argv = [ 
    94             None, "-f", configFilePath, "-u", CertExtAppTestCase.USERNAME 
     95            None,  
     96            "-f", CertExtAppTestCase.INI_FILEPATH,  
     97            "-u", CertExtAppTestCase.USERNAME 
    9598        ] 
    9699        try: 
Note: See TracChangeset for help on using the changeset viewer.