def test_get_messages_for_user(self):
messages = get_messages_for_user(self.user)
self.assertEquals(len(messages), 2)
system_messages = SystemMessage.objects.all()
self.assertEquals(len(system_messages), 2)
self.assertTrue(system_messages[0])
self.assertTrue(system_messages[1])
def test_not_all_messages_are_for_specified_user(self):
add_system_message('Body of message not for this user')
messages = get_messages_for_user(self.user)
self.assertEquals(len(messages), 2)
system_messages = SystemMessage.objects.filter(user=self.user)
self.assertEquals(len(system_messages), 2)
for message in system_messages:
self.assertTrue(message.delivered)
def display_messages_for_user(user):
return {'messages': get_messages_for_user(user)}
