Speed up and tidy up test_new_message() in UnreadCountsTest.

(imported from commit dee4c591de25b70077d6e765c10a9358a5978120)
This commit is contained in:
Steve Howell
2013-07-02 10:08:12 -04:00
parent a6ceb7dd11
commit 0a42d01003

View File

@@ -2499,10 +2499,11 @@ class UnreadCountTests(AuthedTestCase):
"client": "test suite", "client": "test suite",
"content": content, "content": content,
"subject": "Test subject"}) "subject": "Test subject"})
msgs = Message.objects.all().order_by("id") last_msg = Message.objects.all().order_by("-id")[0]
last = msgs[len(msgs) - 1] self.assertEqual(last_msg.content, "Test message for unset read bit")
self.assertEqual(last.content, "Test message for unset read bit") user_messages = list(UserMessage.objects.filter(message=last_msg))
for um in UserMessage.objects.filter(message=last): self.assertEqual(4, len(user_messages))
for um in user_messages:
self.assertEqual(um.message.content, content) self.assertEqual(um.message.content, content)
if um.user_profile.email != "hamlet@humbughq.com": if um.user_profile.email != "hamlet@humbughq.com":
self.assertFalse(um.flags.read) self.assertFalse(um.flags.read)