diff options
Diffstat (limited to 'group_messaging/tests.py')
| -rw-r--r-- | group_messaging/tests.py | 221 |
1 files changed, 130 insertions, 91 deletions
diff --git a/group_messaging/tests.py b/group_messaging/tests.py index 9cc69fb8..f0a2dc5c 100644 --- a/group_messaging/tests.py +++ b/group_messaging/tests.py @@ -26,23 +26,43 @@ def create_user(name): user.groups.add(group) return user -class ModelTests(TestCase): - """test cases for the `private_messaging` models""" +class GroupMessagingTests(TestCase): + """base class for the test cases in this app""" def setUp(self): self.sender = create_user('sender') self.recipient = create_user('recipient') - def create_thread(self, recipients): + def create_thread(self, sender, recipient_groups): return Message.objects.create_thread( - sender=self.sender, recipients=recipients, + sender=sender, recipients=recipient_groups, text=MESSAGE_TEXT ) - def create_thread_for_user(self, user): - group = get_personal_group(user) - return self.create_thread([group]) + def create_thread_for_user(self, sender, recipient): + group = get_personal_group(recipient) + return self.create_thread(sender, [group]) + def setup_three_message_thread(self, original_poster=None, responder=None): + """talk in this order: sender, recipient, sender""" + original_poster = original_poster or self.sender + responder = responder or self.recipient + + root_message = self.create_thread_for_user(original_poster, responder) + response = Message.objects.create_response( + sender=responder, + text='some response', + parent=root_message + ) + response2 = Message.objects.create_response( + sender=original_poster, + text='some response2', + parent=response + ) + return root_message, response, response2 + + +class ViewsTests(GroupMessagingTests): def get_view_context(self, view_class, data=None, user=None, method='GET'): spec = ['REQUEST', 'user'] @@ -54,25 +74,87 @@ class ModelTests(TestCase): request.user = user return view_class().get_context(request) - def setup_three_message_thread(self): - """talk in this order: sender, recipient, sender""" - root_message = self.create_thread_for_user(self.recipient) + def test_new_response_marks_thread_heading_as_new(self): + root = self.create_thread_for_user(self.sender, self.recipient) + response = Message.objects.create_response( + sender=self.recipient, + text='some response', + parent=root + ) + #response must show as "new" to the self.sender + context = self.get_view_context( + ThreadsList, + data={'sender_id': '-1'}, + user=self.sender + ) + self.assertEqual(context['threads_data'][root.id]['status'], 'new') + #"visit" the thread + last_visit_time = LastVisitTime.objects.create( + user=self.sender, + message=root + ) + time.sleep(1.5) + + #response must show as "seen" + context = self.get_view_context( + ThreadsList, + data={'sender_id': '-1'}, + user=self.sender + ) + self.assertEqual(context['threads_data'][root.id]['status'], 'seen') + #self.recipient makes another response + response = Message.objects.create_response( + sender=self.recipient, + text='some response', + parent=response + ) + #thread must be "new" again + context = self.get_view_context( + ThreadsList, + data={'sender_id': '-1'}, + user=self.sender + ) + self.assertEqual(context['threads_data'][root.id]['status'], 'new') + + def test_answer_to_deleted_thread_undeletes_thread(self): + #setup: message, reply, responder deletes thread + root_message = self.create_thread_for_user(self.sender, self.recipient) response = Message.objects.create_response( sender=self.recipient, text='some response', parent=root_message ) + memo1, created = MessageMemo.objects.get_or_create( + message=root_message, + user=self.recipient, + status=MessageMemo.ARCHIVED + ) + #OP sends reply to reply response2 = Message.objects.create_response( sender=self.sender, text='some response2', parent=response ) - return root_message, response, response2 + + context = self.get_view_context( + ThreadsList, + data={'sender_id': '-1'}, + user=self.recipient + ) + + self.assertEqual(len(context['threads']), 1) + thread_id = context['threads'][0].id + thread_data = context['threads_data'][thread_id] + self.assertEqual(thread_data['status'], 'new') + + +class ModelsTests(GroupMessagingTests): + """test cases for the `private_messaging` models""" def test_create_thread_for_user(self): """the basic create thread with one recipient tests that the recipient is there""" - message = self.create_thread_for_user(self.recipient) + message = self.create_thread_for_user(self.sender, self.recipient) #message type is stored self.assertEqual(message.message_type, Message.STORED) #recipient is in the list of recipients @@ -100,7 +182,7 @@ class ModelTests(TestCase): member of the group has updated the sender list""" group = Group.objects.create(name='somegroup') self.recipient.groups.add(group) - message = self.create_thread([group]) + message = self.create_thread(self.sender, [group]) senders = SenderList.objects.get_senders_for_user(self.recipient) self.assertEqual(set(senders), set([self.sender])) @@ -108,7 +190,7 @@ class ModelTests(TestCase): """create a thread with one response, then load thread for the user test that only the root message is retrieved""" - root_message = self.create_thread_for_user(self.recipient) + root_message = self.create_thread_for_user(self.sender, self.recipient) response = Message.objects.create_response( sender=self.recipient, text='some response', @@ -131,7 +213,7 @@ class ModelTests(TestCase): self.assertEqual(recipients, expected_recipients) def test_get_threads(self): - root_message = self.create_thread_for_user(self.recipient) + root_message = self.create_thread_for_user(self.sender, self.recipient) threads = set(Message.objects.get_threads(recipient=self.sender)) self.assertEqual(threads, set([])) threads = set(Message.objects.get_threads(recipient=self.recipient)) @@ -147,37 +229,6 @@ class ModelTests(TestCase): threads = set(Message.objects.get_threads(recipient=self.recipient)) self.assertEqual(threads, set([root_message])) - def test_answer_to_deleted_thread_undeletes_thread(self): - #setup: message, reply, responder deletes thread - root_message = self.create_thread_for_user(self.recipient) - response = Message.objects.create_response( - sender=self.recipient, - text='some response', - parent=root_message - ) - memo1, created = MessageMemo.objects.get_or_create( - message=root_message, - user=self.recipient, - status=MessageMemo.ARCHIVED - ) - #OP sends reply to reply - response2 = Message.objects.create_response( - sender=self.sender, - text='some response2', - parent=response - ) - - context = self.get_view_context( - ThreadsList, - data={'sender_id': '-1'}, - user=self.recipient - ) - - self.assertEqual(len(context['threads']), 1) - thread_id = context['threads'][0].id - thread_data = context['threads_data'][thread_id] - self.assertEqual(thread_data['status'], 'new') - def test_deleting_thread_is_user_specific(self): """when one user deletes thread, that same thread should not end up deleted by another user @@ -205,7 +256,7 @@ class ModelTests(TestCase): self.assertEquals(threads.count(), 1) def test_user_specific_inboxes(self): - self.create_thread_for_user(self.recipient) + self.create_thread_for_user(self.sender, self.recipient) threads = Message.objects.get_threads( recipient=self.recipient, sender=self.sender @@ -216,50 +267,8 @@ class ModelTests(TestCase): ) self.assertEqual(threads.count(), 0) - def test_new_response_marks_thread_heading_as_new(self): - root = self.create_thread_for_user(self.recipient) - response = Message.objects.create_response( - sender=self.recipient, - text='some response', - parent=root - ) - #response must show as "new" to the self.sender - context = self.get_view_context( - ThreadsList, - data={'sender_id': '-1'}, - user=self.sender - ) - self.assertEqual(context['threads_data'][root.id]['status'], 'new') - #"visit" the thread - last_visit_time = LastVisitTime.objects.create( - user=self.sender, - message=root - ) - time.sleep(1.5) - - #response must show as "seen" - context = self.get_view_context( - ThreadsList, - data={'sender_id': '-1'}, - user=self.sender - ) - self.assertEqual(context['threads_data'][root.id]['status'], 'seen') - #self.recipient makes another response - response = Message.objects.create_response( - sender=self.recipient, - text='some response', - parent=response - ) - #thread must be "new" again - context = self.get_view_context( - ThreadsList, - data={'sender_id': '-1'}, - user=self.sender - ) - self.assertEqual(context['threads_data'][root.id]['status'], 'new') - def test_response_updates_thread_headline(self): - root = self.create_thread_for_user(self.recipient) + root = self.create_thread_for_user(self.sender, self.recipient) response = Message.objects.create_response( sender=self.recipient, text='some response', @@ -268,8 +277,38 @@ class ModelTests(TestCase): self.assertEqual(root.headline, 'some response') def test_email_alert_sent(self): - root = self.create_thread_for_user(self.recipient) + root = self.create_thread_for_user(self.sender, self.recipient) from django.core.mail import outbox self.assertEqual(len(outbox), 1) self.assertEqual(len(outbox[0].recipients()), 1) self.assertEqual(outbox[0].recipients()[0], self.recipient.email) + + def test_get_sent_threads(self): + root1, re11, re12 = self.setup_three_message_thread() + root2, re21, re22 = self.setup_three_message_thread( + original_poster=self.recipient, responder=self.sender + ) + root3, re31, re32 = self.setup_three_message_thread() + + #mark root2 as seen + root2.mark_as_seen(self.sender) + #mark root3 as deleted + root3.archive(self.sender) + + threads = Message.objects.get_sent_threads(sender=self.sender) + self.assertEqual(threads.count(), 2) + self.assertEqual(set(threads), set([root1, root2]))#root3 is deleted + + def test_recipient_lists_are_in_senders_info(self): + thread = self.create_thread_for_user(self.sender, self.recipient) + self.assertTrue(self.recipient.username in thread.senders_info) + + def test_self_response_not_in_senders_inbox(self): + root = self.create_thread_for_user(self.sender, self.recipient) + response = Message.objects.create_response( + sender=self.sender, + text='some response', + parent=root + ) + threads = Message.objects.get_threads(recipient=self.sender) + self.assertEqual(threads.count(), 0) |