yield batch_start() all for real redis

praekeltfoundation · Jan 10, 2014 · 8e38e61 · 8e38e61
1 parent 2b5f3d2
commit 8e38e61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vumi/components/tests/test_message_store_cache.py b/vumi/components/tests/test_message_store_cache.py
@@ -29,7 +29,7 @@ def setUp(self):
         self.cache = self.store.cache
         self.batch_id = 'a-batch-id'
         if self.start_batch:
-            self.cache.batch_start(self.batch_id)
+            yield self.cache.batch_start(self.batch_id)
         self.msg_helper = self.add_helper(MessageHelper())
 
     @inlineCallbacks