def setUp(self):
self.author = self.createUser(username="******")
self.inviter = self.createUser(username="******")
self.invitee = self.createUser(username="******")
self.notification_day = Notification(name="1 day",
duration=timedelta(days=1))
self.notification_day.save()
self.notification_week = Notification(name="7 day",
duration=timedelta(days=7))
self.notification_week.save()
self.event = Event.create("Event name",self.author,
"Some interesting place")
self.event.starts = timezone.now() + timedelta(hours=23)
self.event.save()
self.invitation_context = InvitationContext(inviter=self.inviter,
event=self.event)
self.invitation_context.save()
self.invitation = Invitation(context=self.invitation_context,
invitee=self.invitee)
self.invitation.save()
self.invitation_notification_day = InvitationNotification(
invitation=self.invitation,
notification=self.notification_day)
self.invitation_notification_week = InvitationNotification(
invitation=self.invitation,
notification=self.notification_week)
self.invitation_notification_day.save()
self.invitation_notification_week.save()
self.signal_by_user = None
self.signal_event = None
self.signal_listened = False
class InvitationTest(TestCase):
def createUser(self,username):
user = User(username=username)
user.save()
return user
def setUp(self):
self.author = self.createUser(username="******")
self.inviter = self.createUser(username="******")
self.invitee = self.createUser(username="******")
self.notification_day = Notification(name="1 day",
duration=timedelta(days=1))
self.notification_day.save()
self.notification_week = Notification(name="7 day",
duration=timedelta(days=7))
self.notification_week.save()
self.event = Event.create("Event name",self.author,
"Some interesting place")
self.event.starts = timezone.now() + timedelta(hours=23)
self.event.save()
self.invitation_context = InvitationContext(inviter=self.inviter,
event=self.event)
self.invitation_context.save()
self.invitation = Invitation(context=self.invitation_context,
invitee=self.invitee)
self.invitation.save()
self.invitation_notification_day = InvitationNotification(
invitation=self.invitation,
notification=self.notification_day)
self.invitation_notification_week = InvitationNotification(
invitation=self.invitation,
notification=self.notification_week)
self.invitation_notification_day.save()
self.invitation_notification_week.save()
self.signal_by_user = None
self.signal_event = None
self.signal_listened = False
def test_accept_invitation(self):
"""
Tests the accept method on InvitationModel.
"""
def invitation_accepted_listener(sender, by_user, event, **kwargs):
self.signal_by_user = by_user
self.signal_event = event
# Connects to the signal
invitation_accepted.connect(invitation_accepted_listener, sender=self.invitation)
# Default value of accepted is none then accepts it
self.assertIsNone(self.invitation.accepted)
self.assertTrue(self.invitation.accept())
# The accept action updated it.
self.assertIsNotNone(self.invitation.accepted)
# Tests that the signal has been sent and received.
self.assertEqual(self.invitee, self.signal_by_user)
self.assertEqual(self.event, self.signal_event)
def test_reject_invitation(self):
"""
Tests the reject method on InvitationModel.
"""
def invitation_rejected_listener(sender, by_user, event, **kwargs):
self.signal_by_user = by_user
self.signal_event = event
# Connects to the signal
invitation_rejected.connect(invitation_rejected_listener, sender=self.invitation)
# Default value of rejected is none then rejects it
self.assertIsNone(self.invitation.rejected)
self.assertTrue(self.invitation.reject())
# The reject action updated it.
self.assertIsNotNone(self.invitation.rejected)
# Tests that the signal has been sent and received.
self.assertEqual(self.invitee, self.signal_by_user)
self.assertEqual(self.event, self.signal_event)
def test_mark_viewed_invitation(self):
"""
Tests the mark_viewed method on InvitationModel.
"""
def invitation_viewed_listener(sender, **kwargs):
self.signal_listened = True
# Connects to the signal
invitation_viewed.connect(invitation_viewed_listener, sender=self.invitation)
# Default value of viewed is none
self.assertIsNone(self.invitation.viewed)
self.assertTrue(self.invitation.mark_viewed())
# The view action updated it.
self.assertIsNotNone(self.invitation.viewed)
# Tests that the signal has been sent and received.
self.assertTrue(self.signal_listened)
def test_find_invitations_to_notify(self):
invitations = find_invitations_to_notify()
# Tests that the invitation was returned.
self.assertEqual(len(invitations),1)
# Tests that the invitation was returned.
self.assertEqual(invitations[0][0].id,self.invitation.id)
# This is the shortest delay which has been taken into account.
self.assertEqual(invitations[0][1],self.notification_day.duration.days * 86400 * 1000000)
# Checks hat the notification has been set to sent !
try:
invitation_notification = InvitationNotification.objects.get(id=self.invitation_notification_day.id)
except InvitationNotification.DoesNotExist:
invitation_notification = None
self.assertIsNotNone(invitation_notification)
self.assertTrue(invitation_notification.sent)