Have the Filtering API return Deferreds, so we can do the Datastore implementation nicely

1 files changed, 3 insertions, 2 deletions

diff --git a/tests/api/test_filtering.py b/tests/api/test_filtering.py

index c6c5317696..fecadd1056 100644
--- a/tests/api/test_filtering.py
+++ b/tests/api/test_filtering.py
@@ -53,14 +53,15 @@ class FilteringTestCase(unittest.TestCase):
 
         self.filtering = hs.get_filtering()
 
+    @defer.inlineCallbacks
     def test_filter(self):
-        filter_id = self.filtering.add_user_filter(
+        filter_id = yield self.filtering.add_user_filter(
             user_localpart=user_localpart,
             definition={"type": ["m.*"]},
         )
         self.assertEquals(filter_id, 0)
 
-        filter = self.filtering.get_user_filter(
+        filter = yield self.filtering.get_user_filter(
             user_localpart=user_localpart,
             filter_id=filter_id,
         )