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)}