Ignore:
Timestamp:
09/12/08 14:08:49 (11 years ago)
Author:
pjkersha
Message:

Fixed sessionmanager unit tests- updated SessionManager?.getAttCert calls to allow for refactored error handling.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • TI12-security/trunk/python/ndg.security.test/ndg/security/test/sessionmanager/test_sessionmanager.py

    r4573 r4576  
    2121 
    2222from ndg.security.common.utils.configfileparsers import \ 
    23                                                     CaseSensitiveConfigParser 
     23    CaseSensitiveConfigParser 
    2424from ndg.security.common.X509 import X509CertParse 
    25 from ndg.security.server.sessionmanager import SessionManager 
     25from ndg.security.server.sessionmanager import SessionManager, \ 
     26    CredentialWalletAttributeRequestDenied 
    2627from ndg.security.server.attributeauthority import AttributeAuthority 
    2728 
     
    275276        section = 'test07GetAttCertWithSessID' 
    276277        aaURI = self.cfg.get(section, 'aaURI') 
    277         attCert, errMsg, extAttCertList=self.sm.getAttCert(sessID=self.sessID,  
    278                                                 attributeAuthorityURI=aaURI) 
    279         if errMsg: 
    280             self.fail(errMsg) 
    281              
     278        attCert = self.sm.getAttCert(sessID=self.sessID,  
     279                                     attributeAuthorityURI=aaURI) 
    282280        print("Attribute Certificate:\n%s" % attCert)  
    283281        attCert.filePath = xpdVars(self.cfg.get(section, 'acOutputFilePath'))  
     
    294292        aaURI = self.cfg.get('test08GetAttCertRefusedWithSessID', 'aaURI') 
    295293         
    296         attCert, errMsg, extAttCertList=self.sm.getAttCert(sessID=self.sessID,  
    297                                                  attributeAuthorityURI=aaURI, 
    298                                                  mapFromTrustedHosts=False) 
    299         if errMsg: 
    300             print("SUCCESS - obtained expected result: %s" % errMsg) 
     294        try: 
     295            attCert = self.sm.getAttCert(sessID=self.sessID,  
     296                                         attributeAuthorityURI=aaURI, 
     297                                         mapFromTrustedHosts=False) 
     298        except CredentialWalletAttributeRequestDenied, e: 
     299            print("SUCCESS - obtained expected result: %s" % e) 
    301300            return 
    302301         
     
    316315        aaURI = self.cfg.get('test09GetMappedAttCertWithSessID', 'aaURI') 
    317316         
    318         attCert, errMsg, extAttCertList=self.sm.getAttCert(sessID=self.sessID, 
    319                                                    attributeAuthorityURI=aaURI, 
    320                                                    mapFromTrustedHosts=True) 
    321         if errMsg: 
    322             self.fail(errMsg) 
     317        attCert = self.sm.getAttCert(sessID=self.sessID, 
     318                                     attributeAuthorityURI=aaURI, 
     319                                     mapFromTrustedHosts=True) 
    323320             
    324321        print("Attribute Certificate:\n%s" % attCert)   
     
    337334        extAttCert = open(extACFilePath).read() 
    338335         
    339         attCert, errMsg, extAttCertList = self.sm.getAttCert( 
    340                                                    sessID=self.sessID,  
    341                                                    attributeAuthorityURI=aaURI, 
    342                                                    extAttCertList=[extAttCert]) 
    343         if errMsg: 
    344             self.fail(errMsg) 
    345            
     336        attCert = self.sm.getAttCert(sessID=self.sessID,  
     337                                     attributeAuthorityURI=aaURI, 
     338                                     extAttCertList=[extAttCert]) 
    346339        print("Attribute Certificate:\n%s" % attCert)   
    347340 
     
    356349         
    357350        aaURI = self.cfg.get('test11GetAttCertWithUserX509Cert', 'aaURI') 
    358         attCert, errMsg, extAttCertList = self.sm.getAttCert( 
    359                                      userX509Cert=self.userX509Cert,  
     351        attCert = self.sm.getAttCert(userX509Cert=self.userX509Cert,  
    360352                                     attributeAuthorityURI=aaURI) 
    361         if errMsg: 
    362             self.fail(errMsg) 
    363            
    364353        print("Attribute Certificate:\n%s" % attCert)   
    365354 
     
    376365                                              propPrefix='attributeAuthority') 
    377366         
    378         attCert, errMsg, extAttCertList=self.sm.getAttCert(sessID=self.sessID,  
    379                                         attributeAuthority=attributeAuthority) 
    380         if errMsg: 
    381             self.fail(errMsg) 
    382              
     367        attCert = self.sm.getAttCert(sessID=self.sessID,  
     368                                     attributeAuthority=attributeAuthority)             
    383369        print("Attribute Certificate:\n%s" % attCert)  
    384370        attCert.filePath = xpdVars(self.cfg.get(section, 'acOutputFilePath'))  
Note: See TracChangeset for help on using the changeset viewer.