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.token.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.token.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.token.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.token.username)
self.assertEqual(n.reference.cron_job_approval.approval_id, "foo-bar")