Ignore:
Timestamp:
22/02/10 13:51:08 (11 years ago)
Author:
pjkersha
Message:

Completed pickle support for AuthzDecision?* types.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • TI12-security/trunk/ndg_security_saml/saml/test/test_saml.py

    r6602 r6603  
    2626                             Response, Issuer, Subject, NameID, StatusCode,  
    2727                             StatusMessage, Status, Conditions, DecisionType, 
    28                              XSStringAttributeValue, Action, 
     28                             XSStringAttributeValue, Action,  
    2929                             AuthzDecisionQuery) 
    3030 
     
    385385        print("_"*80) 
    386386         
    387     def test06CreateAuthzDecisionQuery(self): 
     387    def _createAuthzDecisionQuery(self): 
    388388        authzDecisionQuery = AuthzDecisionQuery() 
    389389 
     
    402402         
    403403        authzDecisionQuery.resource = "http://LOCALHOST:80/My Secured URI" 
     404         
     405        return authzDecisionQuery 
     406     
     407    def test06CreateAuthzDecisionQuery(self): 
     408        samlUtil = SAMLUtil() 
     409        authzDecisionQuery = samlUtil.buildAuthzDecisionQuery() 
     410         
    404411        self.assert_(":80" not in authzDecisionQuery.resource) 
    405412        self.assert_("localhost" in authzDecisionQuery.resource) 
     
    492499            authzDecisionQuery2.actions[0].namespace == Action.GHPP_NS_URI) 
    493500        self.assert_(authzDecisionQuery2.evidence is None) 
    494  
    495501 
    496502    def _createAuthzDecisionQueryResponse(self): 
     
    676682                                    ].attributes[1].attributeValues[0].value) 
    677683              
     684    def test15PickleAuthzDecisionQuery(self): 
     685        samlUtil = SAMLUtil() 
     686        query = samlUtil.buildAuthzDecisionQuery() 
    678687              
     688        jar = pickle.dumps(query) 
     689        query2 = pickle.loads(jar) 
     690         
     691        self.assert_(isinstance(query2, AuthzDecisionQuery)) 
     692        self.assert_(query.resource == query2.resource) 
     693        self.assert_(query.version == query2.version) 
     694        self.assert_(len(query2.actions) == 1) 
     695        self.assert_(query2.actions[0].value == Action.HTTP_GET_ACTION) 
     696        self.assert_(query2.actions[0].namespace == Action.GHPP_NS_URI) 
     697 
     698 
     699         
     700    def test16PickleAuthzDecisionResponse(self): 
     701        response = self._createAuthzDecisionQueryResponse() 
     702         
     703        jar = pickle.dumps(response) 
     704        response2 = pickle.loads(jar) 
     705         
     706        self.assert_(isinstance(response2, Response)) 
     707         
     708        self.assert_(len(response.assertions) == 1) 
     709        self.assert_(len(response.assertions[0].authzDecisionStatements) == 1) 
     710          
     711        self.assert_(response.assertions[0].authzDecisionStatements[0 
     712                        ].resource == response2.assertions[0 
     713                                        ].authzDecisionStatements[0].resource) 
     714         
     715        self.assert_(len(response.assertions[0].authzDecisionStatements[0 
     716                        ].actions) == 1) 
     717        self.assert_(response.assertions[0].authzDecisionStatements[0 
     718                        ].actions[0].value == response2.assertions[0 
     719                                        ].authzDecisionStatements[0 
     720                                                ].actions[0].value) 
     721         
     722        self.assert_(response2.assertions[0].authzDecisionStatements[0 
     723                        ].actions[0].namespace == Action.GHPP_NS_URI)         
     724 
     725        self.assert_(response2.assertions[0].authzDecisionStatements[0 
     726                        ].decision == DecisionType.PERMIT)         
     727         
    679728if __name__ == "__main__": 
    680729    unittest.main()         
Note: See TracChangeset for help on using the changeset viewer.