From 41c8a67a1653f9acefe78a59b32926592d97e353 Mon Sep 17 00:00:00 2001 From: Jeremy Thurgood Date: Fri, 5 Jun 2015 13:51:01 +0200 Subject: [PATCH] Wait for index query in tests to hopefully fix random failures. --- vumi/components/tests/test_message_store_api.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/vumi/components/tests/test_message_store_api.py b/vumi/components/tests/test_message_store_api.py index 0dbc1fde9..413b7811e 100644 --- a/vumi/components/tests/test_message_store_api.py +++ b/vumi/components/tests/test_message_store_api.py @@ -199,7 +199,8 @@ def test_polling_inbound_match_resource(self): @inlineCallbacks def test_empty_inbound_match_resource(self): - expected_token = yield self.do_query('inbound', self.batch_id, '.*') + expected_token = yield self.do_query( + 'inbound', self.batch_id, '.*', wait=True) response = yield self.do_get('batch/%s/inbound/match/?token=%s' % ( self.batch_id, expected_token)) self.assertResultCount(response, 0) @@ -249,7 +250,8 @@ def test_polling_outbound_match_resource(self): @inlineCallbacks def test_empty_outbound_match_resource(self): - expected_token = yield self.do_query('outbound', self.batch_id, '.*') + expected_token = yield self.do_query( + 'outbound', self.batch_id, '.*', wait=True) response = yield self.do_get('batch/%s/outbound/match/?token=%s' % ( self.batch_id, expected_token)) self.assertResultCount(response, 0)