def test_authorization_challenge_request_bad_last_message(self): """ Test the AuthorizationChallengeRequestHandler returns an AuthorizationViolation and closes the connection if the last message was not a ConnectionRequst """ auth_challenge_request = AuthorizationChallengeRequest() roles = {"network": AuthorizationType.TRUST} network = MockNetwork(roles, connection_status={"connection_id": "other"}) handler = AuthorizationChallengeRequestHandler(network) handler_status = handler.handle( "connection_id", auth_challenge_request.SerializeToString()) self.assertEqual(handler_status.status, HandlerStatus.RETURN_AND_CLOSE) self.assertEqual(handler_status.message_type, validator_pb2.Message.AUTHORIZATION_VIOLATION)
def do_authorization_challenge_request(): """ Test the AuthorizationChallengeRequestHandler returns an AuthorizationChallengeResponse. """ auth_challenge_request = AuthorizationChallengeRequest() roles = {"network": AuthorizationType.TRUST} network = MockNetwork(roles, connection_status={ "connection_id": ConnectionStatus.CONNECTION_REQUEST }) handler = AuthorizationChallengeRequestHandler(network) handler_status = handler.handle("connection_id", auth_challenge_request.SerializeToString()) return handler_status
def test_authorization_challenge_request(self): """ Test the AuthorizationChallengeRequestHandler returns an AuthorizationChallengeResponse. """ auth_challenge_request = AuthorizationChallengeRequest() roles = {"network": AuthorizationType.TRUST} network = MockNetwork(roles, connection_status={ "connection_id": ConnectionStatus.CONNECTION_REQUEST }) handler = AuthorizationChallengeRequestHandler(network) handler_status = handler.handle( "connection_id", auth_challenge_request.SerializeToString()) self.assertEqual(handler_status.status, HandlerStatus.RETURN) self.assertEqual( handler_status.message_type, validator_pb2.Message.AUTHORIZATION_CHALLENGE_RESPONSE)
def test_authorization_challenge_request_bad_last_message(self): """ Test the AuthorizationChallengeRequestHandler returns an AuthorizationViolation and closes the connection if the last message was not a ConnectionRequst """ auth_challenge_request = AuthorizationChallengeRequest() roles = {"network": AuthorizationType.TRUST} network = MockNetwork( roles, connection_status={"connection_id": "other"}) handler = AuthorizationChallengeRequestHandler(network) handler_status = handler.handle( "connection_id", auth_challenge_request.SerializeToString()) self.assertEqual(handler_status.status, HandlerStatus.RETURN_AND_CLOSE) self.assertEqual( handler_status.message_type, validator_pb2.Message.AUTHORIZATION_VIOLATION)
def test_authorization_challenge_request(self): """ Test the AuthorizationChallengeRequestHandler returns an AuthorizationChallengeResponse. """ auth_challenge_request = AuthorizationChallengeRequest() roles = {"network": AuthorizationType.TRUST} network = MockNetwork( roles, connection_status={ "connection_id": ConnectionStatus.CONNECTION_REQUEST }) handler = AuthorizationChallengeRequestHandler(network) handler_status = handler.handle( "connection_id", auth_challenge_request.SerializeToString()) self.assertEqual(handler_status.status, HandlerStatus.RETURN) self.assertEqual( handler_status.message_type, validator_pb2.Message.AUTHORIZATION_CHALLENGE_RESPONSE)