Skip to content
Snippets Groups Projects
Commit 4640239d authored by Kegan Dougal's avatar Kegan Dougal
Browse files

Mock ratelimiter to make tests pass.

parent 2a5b53bc
No related branches found
No related tags found
No related merge requests found
...@@ -19,7 +19,7 @@ presence and profiles; namely, the displayname and avatar_url.""" ...@@ -19,7 +19,7 @@ presence and profiles; namely, the displayname and avatar_url."""
from tests import unittest from tests import unittest
from twisted.internet import defer from twisted.internet import defer
from mock import Mock, call, ANY from mock import Mock, call, ANY, NonCallableMock
from ..utils import MockClock, MockKey from ..utils import MockClock, MockKey
...@@ -75,8 +75,13 @@ class PresenceProfilelikeDataTestCase(unittest.TestCase): ...@@ -75,8 +75,13 @@ class PresenceProfilelikeDataTestCase(unittest.TestCase):
resource_for_federation=Mock(), resource_for_federation=Mock(),
http_client=None, http_client=None,
replication_layer=MockReplication(), replication_layer=MockReplication(),
config=self.mock_config, ratelimiter=NonCallableMock(spec_set=[
) "send_message",
]),
config=self.mock_config
)
self.ratelimiter = hs.get_ratelimiter()
self.ratelimiter.send_message.return_value = (True, 0)
hs.handlers = PresenceAndProfileHandlers(hs) hs.handlers = PresenceAndProfileHandlers(hs)
self.datastore = hs.get_datastore() self.datastore = hs.get_datastore()
......
...@@ -17,7 +17,7 @@ ...@@ -17,7 +17,7 @@
from tests import unittest from tests import unittest
from twisted.internet import defer from twisted.internet import defer
from mock import Mock from mock import Mock, NonCallableMock
from synapse.api.errors import AuthError from synapse.api.errors import AuthError
from synapse.server import HomeServer from synapse.server import HomeServer
...@@ -59,7 +59,14 @@ class ProfileTestCase(unittest.TestCase): ...@@ -59,7 +59,14 @@ class ProfileTestCase(unittest.TestCase):
resource_for_federation=Mock(), resource_for_federation=Mock(),
replication_layer=self.mock_federation, replication_layer=self.mock_federation,
config=self.mock_config, config=self.mock_config,
ratelimiter=NonCallableMock(spec_set=[
"send_message",
])
) )
self.ratelimiter = hs.get_ratelimiter()
self.ratelimiter.send_message.return_value = (True, 0)
hs.handlers = ProfileHandlers(hs) hs.handlers = ProfileHandlers(hs)
self.store = hs.get_datastore() self.store = hs.get_datastore()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment