From 33338f78e927de8eced4bb345917a5dc52dda7ae Mon Sep 17 00:00:00 2001 From: Steve Howell Date: Sun, 15 Oct 2017 06:26:15 -0700 Subject: [PATCH] tests: Make test_bulk_message_fetching() more thorough. We include the post-processing step when counting queries now. --- zerver/tests/test_messages.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/zerver/tests/test_messages.py b/zerver/tests/test_messages.py index e55b3038ac..a8d62c9cd8 100644 --- a/zerver/tests/test_messages.py +++ b/zerver/tests/test_messages.py @@ -696,8 +696,11 @@ class MessageDictTest(ZulipTestCase): with queries_captured() as queries: rows = list(MessageDict.get_raw_db_rows(ids)) - for row in rows: + objs = [ MessageDict.build_dict_from_raw_db_row(row, False) + for row in rows + ] + MessageDict.post_process_dicts(objs) delay = time.time() - t # Make sure we don't take longer than 1.5ms per message to @@ -706,7 +709,7 @@ class MessageDictTest(ZulipTestCase): # slower. error_msg = "Number of ids: {}. Time delay: {}".format(num_ids, delay) self.assertTrue(delay < 0.0015 * num_ids, error_msg) - self.assert_length(queries, 5) + self.assert_length(queries, 6) self.assertEqual(len(rows), num_ids) def test_applying_markdown(self):