fix UTs by telling all the mock stores about the new methods for tracking retries

1 files changed, 7 insertions, 2 deletions

diff --git a/tests/handlers/test_typing.py b/tests/handlers/test_typing.py

index adb5148351..2f170ac3bf 100644
--- a/tests/handlers/test_typing.py
+++ b/tests/handlers/test_typing.py
@@ -25,6 +25,8 @@ from ..utils import MockHttpResource, MockClock, DeferredMockCallable, MockKey
 from synapse.server import HomeServer
 from synapse.handlers.typing import TypingNotificationHandler
 
+from synapse.storage.transactions import DestinationsTable
+
 
 def _expect_edu(destination, edu_type, content, origin="test"):
     return {
@@ -49,7 +51,6 @@ class JustTypingNotificationHandlers(object):
     def __init__(self, hs):
         self.typing_notification_handler = TypingNotificationHandler(hs)
 
-
 class TypingNotificationsTestCase(unittest.TestCase):
     """Tests typing notifications to rooms."""
     def setUp(self):
@@ -72,6 +73,7 @@ class TypingNotificationsTestCase(unittest.TestCase):
                     "delivered_txn",
                     "get_received_txn_response",
                     "set_received_txn_response",
+                    "get_destination_retry_timings",
                 ]),
                 handlers=None,
                 resource_for_client=Mock(),
@@ -89,6 +91,9 @@ class TypingNotificationsTestCase(unittest.TestCase):
         self.handler.push_update_to_clients = self.mock_update_client
 
         self.datastore = hs.get_datastore()
+        self.datastore.get_destination_retry_timings.return_value = (
+            defer.succeed(DestinationsTable.EntryType("", 0, 0))
+        )
 
         def get_received_txn_response(*args):
             return defer.succeed(None)
@@ -162,7 +167,7 @@ class TypingNotificationsTestCase(unittest.TestCase):
     @defer.inlineCallbacks
     def test_started_typing_remote_send(self):
         self.room_members = [self.u_apple, self.u_onion]
-
+        
         put_json = self.mock_http_client.put_json
         put_json.expect_call_and_return(
             call("farm",