def RequestHuntApproval(self, hunt_id, requestor=None, reason=None, email_cc_address=None, approver=u"approver"): """Request hunt approval for a given hunt.""" if not requestor: requestor = self.token.username if not reason: reason = self.token.reason self.CreateUser(requestor) self.CreateUser(approver) args = api_user.ApiCreateHuntApprovalArgs( hunt_id=hunt_id, approval=api_user.ApiHuntApproval( reason=reason, notified_users=[approver], email_cc_addresses=([email_cc_address] if email_cc_address else []))) handler = api_user.ApiCreateHuntApprovalHandler() result = handler.Handle( args, token=access_control.ACLToken(username=requestor)) return result.id
def RequestHuntApproval(self, hunt_id, requestor=None, reason="Running tests", email_cc_address=None, approver=u"approver"): """Request hunt approval for a given hunt.""" if not requestor: requestor = self.test_username self.CreateUser(requestor) self.CreateUser(approver) args = api_user.ApiCreateHuntApprovalArgs( hunt_id=hunt_id, approval=api_user.ApiHuntApproval( reason=reason, notified_users=[approver], email_cc_addresses=([email_cc_address] if email_cc_address else []))) handler = api_user.ApiCreateHuntApprovalHandler() result = handler.Handle( args, context=api_call_context.ApiCallContext(username=requestor)) return result.id
def setUp(self): super(ApiCreateHuntApprovalHandlerTest, self).setUp() self.SetUpApprovalTest() hunt_id = self.StartHunt(description="foo") self.handler = user_plugin.ApiCreateHuntApprovalHandler() self.args = user_plugin.ApiCreateHuntApprovalArgs(hunt_id=hunt_id) self.args.approval.reason = self.token.reason self.args.approval.notified_users = ["approver"] self.args.approval.email_cc_addresses = ["*****@*****.**"]
def CreateHuntApproval(self, args, context=None): return api_user.ApiCreateHuntApprovalHandler()