def ObjectReference(self):
at = rdf_objects.ApprovalRequest.ApprovalType.APPROVAL_TYPE_CRON_JOB
return rdf_objects.ObjectReference(
reference_type=rdf_objects.ObjectReference.Type.APPROVAL_REQUEST,
approval_request=rdf_objects.ApprovalRequestReference(
approval_type=at,
approval_id=self.id,
subject_id=str(self.subject.cron_job_id),
requestor_username=self.requestor))
def ObjectReference(self):
at = rdf_objects.ApprovalRequest.ApprovalType.APPROVAL_TYPE_HUNT
return rdf_objects.ObjectReference(
reference_type=rdf_objects.ObjectReference.Type.APPROVAL_REQUEST,
approval_request=rdf_objects.ApprovalRequestReference(
approval_type=at,
approval_id=utils.SmartStr(self.id),
subject_id=utils.SmartStr(self.subject.hunt_id),
requestor_username=self.requestor))
def testHuntApprovalNotificationIsParsedCorrectly(self):
n = self.InitFromObj_(
rdf_objects.UserNotification.Type.TYPE_HUNT_APPROVAL_REQUESTED,
rdf_objects.ObjectReference(
reference_type=rdf_objects.ObjectReference.Type.APPROVAL_REQUEST,
approval_request=rdf_objects.ApprovalRequestReference(
approval_type=rdf_objects.ApprovalRequest.ApprovalType
.APPROVAL_TYPE_HUNT,
approval_id="foo-bar",
subject_id="H:123456",
requestor_username=self.context.username)))
self.assertEqual(n.reference.type, "HUNT_APPROVAL")
self.assertEqual(n.reference.hunt_approval.hunt_id, "H:123456")
self.assertEqual(n.reference.hunt_approval.username, self.context.username)
self.assertEqual(n.reference.hunt_approval.approval_id, "foo-bar")
def testCronJobApprovalNotificationIsParsedCorrectly(self):
n = self.InitFromObj_(
rdf_objects.UserNotification.Type.TYPE_CRON_JOB_APPROVAL_REQUESTED,
rdf_objects.ObjectReference(
reference_type=rdf_objects.ObjectReference.Type.APPROVAL_REQUEST,
approval_request=rdf_objects.ApprovalRequestReference(
approval_type=rdf_objects.ApprovalRequest.ApprovalType
.APPROVAL_TYPE_CRON_JOB,
approval_id="foo-bar",
subject_id="FooBar",
requestor_username=self.context.username)))
self.assertEqual(n.reference.type, "CRON_JOB_APPROVAL")
self.assertEqual(n.reference.cron_job_approval.cron_job_id, "FooBar")
self.assertEqual(n.reference.cron_job_approval.username,
self.context.username)
self.assertEqual(n.reference.cron_job_approval.approval_id, "foo-bar")
def testClientApprovalNotificationIsParsedCorrectly(self):
n = self.InitFromObj_(
rdf_objects.UserNotification.Type.TYPE_CLIENT_APPROVAL_REQUESTED,
rdf_objects.ObjectReference(
reference_type=rdf_objects.ObjectReference.Type.APPROVAL_REQUEST,
approval_request=rdf_objects.ApprovalRequestReference(
approval_type=rdf_objects.ApprovalRequest.ApprovalType
.APPROVAL_TYPE_CLIENT,
approval_id="foo-bar",
subject_id=self.client_id,
requestor_username=self.context.username)))
self.assertEqual(n.reference.type, "CLIENT_APPROVAL")
client_approval = n.reference.client_approval
self.assertEqual(client_approval.client_id.ToString(), self.client_id)
self.assertEqual(client_approval.username, self.context.username)
self.assertEqual(client_approval.approval_id, "foo-bar")
